101 lines
3.3 KiB
Go
101 lines
3.3 KiB
Go
package middleware
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/gorilla/sessions"
|
|
)
|
|
|
|
// SecurityHeaders adds security and cache-control headers to all responses
|
|
func SecureHeaders() Middleware {
|
|
return func(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
// Caching headers
|
|
w.Header().Set("Cache-Control", "no-store")
|
|
|
|
// XSSProtection provides protection against cross-site scripting attack (XSS)
|
|
w.Header().Set("X-XSS-Protection", "1; mode=block")
|
|
|
|
// ContentTypeNosniff provides protection against overriding Content-Type
|
|
w.Header().Set("X-Content-Type-Options", "nosniff")
|
|
|
|
// XFrameOptions prevents the page from being displayed in a frame
|
|
w.Header().Set("X-Frame-Options", "DENY")
|
|
|
|
// HSTS (HTTP Strict Transport Security) forces the browser to use HTTPS
|
|
w.Header().Set("Strict-Transport-Security", "max-age=3600; includeSubDomains")
|
|
|
|
// ReferrerPolicy sets the referrer information passed during navigation
|
|
w.Header().Set("Referrer-Policy", "no-referrer")
|
|
|
|
// CSP controls the resources the user agent is allowed to load for a page
|
|
w.Header().Set("Content-Security-Policy",
|
|
"default-src 'self'; "+
|
|
// Allow HTMX to load from unpkg.com
|
|
"script-src 'self' https://unpkg.com/htmx.org@*; "+
|
|
"style-src 'self'; "+
|
|
"img-src 'self' data:; "+
|
|
"connect-src 'self'; "+
|
|
"frame-ancestors 'none'; "+
|
|
"form-action 'self'; "+
|
|
"base-uri 'self';")
|
|
|
|
// Cross-Origin-Embedder-Policy prevents cross-origin resources from being loaded
|
|
w.Header().Set("Cross-Origin-Embedder-Policy", "require-corp")
|
|
|
|
// Cross-Origin-Opener-Policy prevents cross-origin documents from being loaded
|
|
w.Header().Set("Cross-Origin-Opener-Policy", "same-origin")
|
|
|
|
// Cross-Origin-Resource-Policy prevents cross-origin resources from being loaded
|
|
w.Header().Set("Cross-Origin-Resource-Policy", "same-origin")
|
|
|
|
next.ServeHTTP(w, r)
|
|
})
|
|
}
|
|
}
|
|
|
|
// middleware/csrf.go
|
|
func CSRFMiddleware(store sessions.Store) Middleware {
|
|
return func(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method == "POST" {
|
|
session, _ := store.Get(r, "auth-session")
|
|
csrfToken := session.Values["csrf_token"].(string)
|
|
formToken := r.FormValue("_csrf")
|
|
|
|
if csrfToken != formToken {
|
|
http.Error(w, "Invalid CSRF token", http.StatusForbidden)
|
|
return
|
|
}
|
|
}
|
|
next.ServeHTTP(w, r)
|
|
})
|
|
}
|
|
}
|
|
|
|
// MaxBodySize limits the maximum size of request bodies
|
|
// size parameter is in bytes
|
|
func MaxBodySize(maxSize int64) Middleware {
|
|
return func(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
// Skip restricting GET, HEAD, and OPTIONS requests as they shouldn't have bodies
|
|
if r.Method == http.MethodGet || r.Method == http.MethodHead || r.Method == http.MethodOptions {
|
|
next.ServeHTTP(w, r)
|
|
return
|
|
}
|
|
|
|
// Check Content-Length header first for efficiency
|
|
if r.ContentLength > maxSize {
|
|
http.Error(w, "Request body too large", http.StatusRequestEntityTooLarge)
|
|
return
|
|
}
|
|
|
|
// If Content-Length is not set or potentially spoofed, use LimitReader
|
|
r.Body = http.MaxBytesReader(w, r.Body, maxSize)
|
|
|
|
// Continue to next middleware/handler
|
|
next.ServeHTTP(w, r)
|
|
})
|
|
}
|
|
}
|