package server import ( "fmt" "io" "os" "path/filepath" "text/template" "github.com/valyala/fasthttp" "github.com/authelia/authelia/v4/internal/logging" "github.com/authelia/authelia/v4/internal/utils" ) // ServeTemplatedFile serves a templated version of a specified file, // this is utilised to pass information between the backend and frontend // and generate a nonce to support a restrictive CSP while using material-ui. func ServeTemplatedFile(publicDir, file, assetPath, duoSelfEnrollment, rememberMe, resetPassword, session, theme string, https bool) fasthttp.RequestHandler { logger := logging.Logger() a, err := assets.Open(publicDir + file) if err != nil { logger.Fatalf("Unable to open %s: %s", file, err) } b, err := io.ReadAll(a) if err != nil { logger.Fatalf("Unable to read %s: %s", file, err) } tmpl, err := template.New("file").Parse(string(b)) if err != nil { logger.Fatalf("Unable to parse %s template: %s", file, err) } return func(ctx *fasthttp.RequestCtx) { base := "" if baseURL := ctx.UserValue("base_url"); baseURL != nil { base = baseURL.(string) } logoOverride := f if assetPath != "" { if _, err := os.Stat(assetPath + logoFile); err == nil { logoOverride = t } } var scheme = "https" if !https { proto := string(ctx.Request.Header.Peek(fasthttp.HeaderXForwardedProto)) switch proto { case "": scheme = "http" default: scheme = proto } } baseURL := scheme + "://" + string(ctx.Request.Host()) + base + "/" nonce := utils.RandomString(32, utils.AlphaNumericCharacters, true) switch extension := filepath.Ext(file); extension { case ".html": ctx.SetContentType("text/html; charset=utf-8") default: ctx.SetContentType("text/plain; charset=utf-8") } switch { case publicDir == swaggerAssets: ctx.Response.Header.Add("Content-Security-Policy", fmt.Sprintf("base-uri 'self' ; default-src 'self' ; img-src 'self' https://validator.swagger.io data: ; object-src 'none' ; script-src 'self' 'unsafe-inline' 'nonce-%s' ; style-src 'self' 'nonce-%s'", nonce, nonce)) case os.Getenv("ENVIRONMENT") == dev: ctx.Response.Header.Add("Content-Security-Policy", fmt.Sprintf("default-src 'self' 'unsafe-eval'; object-src 'none'; style-src 'self' 'nonce-%s'", nonce)) default: ctx.Response.Header.Add("Content-Security-Policy", fmt.Sprintf("default-src 'self' ; object-src 'none'; style-src 'self' 'nonce-%s'", nonce)) } err := tmpl.Execute(ctx.Response.BodyWriter(), struct{ Base, BaseURL, CSPNonce, DuoSelfEnrollment, LogoOverride, RememberMe, ResetPassword, Session, Theme string }{Base: base, BaseURL: baseURL, CSPNonce: nonce, DuoSelfEnrollment: duoSelfEnrollment, LogoOverride: logoOverride, RememberMe: rememberMe, ResetPassword: resetPassword, Session: session, Theme: theme}) if err != nil { ctx.Error("an error occurred", 503) logger.Errorf("Unable to execute template: %v", err) return } } } func writeHealthCheckEnv(disabled bool, scheme, host, path string, port int) (err error) { if disabled { return nil } _, err = os.Stat("/app/healthcheck.sh") if err != nil { return nil } _, err = os.Stat("/app/.healthcheck.env") if err != nil { return nil } file, err := os.OpenFile("/app/.healthcheck.env", os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0755) if err != nil { return err } defer func() { _ = file.Close() }() if host == "0.0.0.0" { host = "localhost" } _, err = file.WriteString(fmt.Sprintf(healthCheckEnv, scheme, host, port, path)) return err }