56 lines
1.6 KiB
Go
56 lines
1.6 KiB
Go
package web
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"net"
|
|
|
|
"github.com/pires/go-proxyproto"
|
|
|
|
"goauthentik.io/internal/config"
|
|
"goauthentik.io/internal/crypto"
|
|
"goauthentik.io/internal/utils"
|
|
"goauthentik.io/internal/utils/web"
|
|
)
|
|
|
|
func (ws *WebServer) GetCertificate() func(ch *tls.ClientHelloInfo) (*tls.Certificate, error) {
|
|
cert, err := crypto.GenerateSelfSignedCert()
|
|
if err != nil {
|
|
ws.log.WithError(err).Error("failed to generate default cert")
|
|
}
|
|
return func(ch *tls.ClientHelloInfo) (*tls.Certificate, error) {
|
|
if ch.ServerName == "" {
|
|
return &cert, nil
|
|
}
|
|
if ws.ProxyServer != nil {
|
|
appCert := ws.ProxyServer.GetCertificate(ch.ServerName)
|
|
if appCert != nil {
|
|
return appCert, nil
|
|
}
|
|
}
|
|
if ws.BrandTLS != nil {
|
|
return ws.BrandTLS.GetCertificate(ch)
|
|
}
|
|
ws.log.Trace("using default, self-signed certificate")
|
|
return &cert, nil
|
|
}
|
|
}
|
|
|
|
// ServeHTTPS constructs a net.Listener and starts handling HTTPS requests
|
|
func (ws *WebServer) listenTLS() {
|
|
tlsConfig := utils.GetTLSConfig()
|
|
tlsConfig.GetCertificate = ws.GetCertificate()
|
|
|
|
ln, err := net.Listen("tcp", config.Get().Listen.HTTPS)
|
|
if err != nil {
|
|
ws.log.WithError(err).Warning("failed to listen (TLS)")
|
|
return
|
|
}
|
|
proxyListener := &proxyproto.Listener{Listener: web.TCPKeepAliveListener{TCPListener: ln.(*net.TCPListener)}}
|
|
defer proxyListener.Close()
|
|
|
|
tlsListener := tls.NewListener(proxyListener, tlsConfig)
|
|
ws.log.WithField("listen", config.Get().Listen.HTTPS).Info("Starting HTTPS server")
|
|
ws.serve(tlsListener)
|
|
ws.log.WithField("listen", config.Get().Listen.HTTPS).Info("Stopping HTTPS server")
|
|
}
|