82 lines
2.4 KiB
Go
82 lines
2.4 KiB
Go
package web
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httputil"
|
|
"time"
|
|
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"goauthentik.io/internal/utils/sentry"
|
|
)
|
|
|
|
func (ws *WebServer) configureProxy() {
|
|
// Reverse proxy to the application server
|
|
director := func(req *http.Request) {
|
|
req.URL.Scheme = ws.ul.Scheme
|
|
req.URL.Host = ws.ul.Host
|
|
if _, ok := req.Header["User-Agent"]; !ok {
|
|
// explicitly disable User-Agent so it's not set to default value
|
|
req.Header.Set("User-Agent", "")
|
|
}
|
|
if req.TLS != nil {
|
|
req.Header.Set("X-Forwarded-Proto", "https")
|
|
}
|
|
ws.log.WithField("url", req.URL.String()).WithField("headers", req.Header).Trace("tracing request to backend")
|
|
}
|
|
rp := &httputil.ReverseProxy{
|
|
Director: director,
|
|
Transport: ws.upstreamHttpClient().Transport,
|
|
}
|
|
rp.ErrorHandler = ws.proxyErrorHandler
|
|
rp.ModifyResponse = ws.proxyModifyResponse
|
|
ws.m.Path("/-/health/live/").HandlerFunc(sentry.SentryNoSample(func(rw http.ResponseWriter, r *http.Request) {
|
|
rw.WriteHeader(204)
|
|
}))
|
|
ws.m.PathPrefix("/").HandlerFunc(sentry.SentryNoSample(func(rw http.ResponseWriter, r *http.Request) {
|
|
if !ws.g.IsRunning() {
|
|
ws.proxyErrorHandler(rw, r, errors.New("authentik starting"))
|
|
return
|
|
}
|
|
before := time.Now()
|
|
if ws.ProxyServer != nil && ws.ProxyServer.HandleHost(rw, r) {
|
|
elapsed := time.Since(before)
|
|
Requests.With(prometheus.Labels{
|
|
"dest": "embedded_outpost",
|
|
}).Observe(float64(elapsed) / float64(time.Second))
|
|
return
|
|
}
|
|
elapsed := time.Since(before)
|
|
Requests.With(prometheus.Labels{
|
|
"dest": "core",
|
|
}).Observe(float64(elapsed) / float64(time.Second))
|
|
r.Body = http.MaxBytesReader(rw, r.Body, 32*1024*1024)
|
|
rp.ServeHTTP(rw, r)
|
|
}))
|
|
}
|
|
|
|
func (ws *WebServer) proxyErrorHandler(rw http.ResponseWriter, req *http.Request, err error) {
|
|
ws.log.WithError(err).Warning("failed to proxy to backend")
|
|
rw.WriteHeader(http.StatusBadGateway)
|
|
em := fmt.Sprintf("failed to connect to authentik backend: %v", err)
|
|
// return json if the client asks for json
|
|
if req.Header.Get("Accept") == "application/json" {
|
|
err = json.NewEncoder(rw).Encode(map[string]string{
|
|
"error": em,
|
|
})
|
|
} else {
|
|
_, err = rw.Write([]byte(em))
|
|
}
|
|
if err != nil {
|
|
ws.log.WithError(err).Warning("failed to write error message")
|
|
}
|
|
}
|
|
|
|
func (ws *WebServer) proxyModifyResponse(r *http.Response) error {
|
|
r.Header.Set("X-Powered-By", "authentik")
|
|
r.Header.Del("Server")
|
|
return nil
|
|
}
|