mirror of
https://github.com/caddyserver/caddy.git
synced 2024-11-26 02:09:47 +08:00
de7bf4f241
* Fix for stripping of 'Content-Disposition' and other headers from 'X-Accel-Redirect' redirect scripts. * Added test case for header manipulation of redirect response.
97 lines
2.8 KiB
Go
97 lines
2.8 KiB
Go
// Package internalsrv provides a simple middleware that (a) prevents access
|
|
// to internal locations and (b) allows to return files from internal location
|
|
// by setting a special header, e.g. in a proxy response.
|
|
//
|
|
// The package is named internalsrv so as not to conflict with Go tooling
|
|
// convention which treats folders called "internal" differently.
|
|
package internalsrv
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/mholt/caddy/caddyhttp/httpserver"
|
|
)
|
|
|
|
// Internal middleware protects internal locations from external requests -
|
|
// but allows access from the inside by using a special HTTP header.
|
|
type Internal struct {
|
|
Next httpserver.Handler
|
|
Paths []string
|
|
}
|
|
|
|
const (
|
|
redirectHeader string = "X-Accel-Redirect"
|
|
contentLengthHeader string = "Content-Length"
|
|
contentEncodingHeader string = "Content-Encoding"
|
|
maxRedirectCount int = 10
|
|
)
|
|
|
|
func isInternalRedirect(w http.ResponseWriter) bool {
|
|
return w.Header().Get(redirectHeader) != ""
|
|
}
|
|
|
|
// ServeHTTP implements the httpserver.Handler interface.
|
|
func (i Internal) ServeHTTP(w http.ResponseWriter, r *http.Request) (int, error) {
|
|
|
|
// Internal location requested? -> Not found.
|
|
for _, prefix := range i.Paths {
|
|
if httpserver.Path(r.URL.Path).Matches(prefix) {
|
|
return http.StatusNotFound, nil
|
|
}
|
|
}
|
|
|
|
// Use internal response writer to ignore responses that will be
|
|
// redirected to internal locations
|
|
iw := internalResponseWriter{ResponseWriter: w}
|
|
status, err := i.Next.ServeHTTP(iw, r)
|
|
|
|
for c := 0; c < maxRedirectCount && isInternalRedirect(iw); c++ {
|
|
// Redirect - adapt request URL path and send it again
|
|
// "down the chain"
|
|
r.URL.Path = iw.Header().Get(redirectHeader)
|
|
iw.ClearHeader()
|
|
|
|
status, err = i.Next.ServeHTTP(iw, r)
|
|
}
|
|
|
|
if isInternalRedirect(iw) {
|
|
// Too many redirect cycles
|
|
iw.ClearHeader()
|
|
return http.StatusInternalServerError, nil
|
|
}
|
|
|
|
return status, err
|
|
}
|
|
|
|
// internalResponseWriter wraps the underlying http.ResponseWriter and ignores
|
|
// calls to Write and WriteHeader if the response should be redirected to an
|
|
// internal location.
|
|
type internalResponseWriter struct {
|
|
http.ResponseWriter
|
|
}
|
|
|
|
// ClearHeader removes script headers that would interfere with follow up
|
|
// redirect requests.
|
|
func (w internalResponseWriter) ClearHeader() {
|
|
w.Header().Del(redirectHeader)
|
|
w.Header().Del(contentLengthHeader)
|
|
w.Header().Del(contentEncodingHeader)
|
|
}
|
|
|
|
// WriteHeader ignores the call if the response should be redirected to an
|
|
// internal location.
|
|
func (w internalResponseWriter) WriteHeader(code int) {
|
|
if !isInternalRedirect(w) {
|
|
w.ResponseWriter.WriteHeader(code)
|
|
}
|
|
}
|
|
|
|
// Write ignores the call if the response should be redirected to an internal
|
|
// location.
|
|
func (w internalResponseWriter) Write(b []byte) (int, error) {
|
|
if isInternalRedirect(w) {
|
|
return 0, nil
|
|
}
|
|
return w.ResponseWriter.Write(b)
|
|
}
|