78 lines
2.2 KiB
Go
78 lines
2.2 KiB
Go
package ldap
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
"strings"
|
|
|
|
"github.com/getsentry/sentry-go"
|
|
"github.com/google/uuid"
|
|
"github.com/nmcclain/ldap"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
log "github.com/sirupsen/logrus"
|
|
"goauthentik.io/internal/outpost/ldap/metrics"
|
|
"goauthentik.io/internal/utils"
|
|
)
|
|
|
|
type BindRequest struct {
|
|
BindDN string
|
|
BindPW string
|
|
id string
|
|
conn net.Conn
|
|
log *log.Entry
|
|
ctx context.Context
|
|
}
|
|
|
|
func (ls *LDAPServer) Bind(bindDN string, bindPW string, conn net.Conn) (ldap.LDAPResultCode, error) {
|
|
span := sentry.StartSpan(context.TODO(), "authentik.providers.ldap.bind",
|
|
sentry.TransactionName("authentik.providers.ldap.bind"))
|
|
rid := uuid.New().String()
|
|
span.SetTag("request_uid", rid)
|
|
span.SetTag("user.username", bindDN)
|
|
|
|
bindDN = strings.ToLower(bindDN)
|
|
req := BindRequest{
|
|
BindDN: bindDN,
|
|
BindPW: bindPW,
|
|
conn: conn,
|
|
log: ls.log.WithField("bindDN", bindDN).WithField("requestId", rid).WithField("client", utils.GetIP(conn.RemoteAddr())),
|
|
id: rid,
|
|
ctx: span.Context(),
|
|
}
|
|
defer func() {
|
|
span.Finish()
|
|
metrics.Requests.With(prometheus.Labels{
|
|
"outpost_name": ls.ac.Outpost.Name,
|
|
"type": "bind",
|
|
"filter": "",
|
|
"dn": req.BindDN,
|
|
"client": utils.GetIP(req.conn.RemoteAddr()),
|
|
}).Observe(float64(span.EndTime.Sub(span.StartTime)))
|
|
req.log.WithField("took-ms", span.EndTime.Sub(span.StartTime).Milliseconds()).Info("Bind request")
|
|
}()
|
|
for _, instance := range ls.providers {
|
|
username, err := instance.getUsername(bindDN)
|
|
if err == nil {
|
|
return instance.Bind(username, req)
|
|
} else {
|
|
ls.log.WithError(err).Debug("Username not for instance")
|
|
}
|
|
}
|
|
req.log.WithField("request", "bind").Warning("No provider found for request")
|
|
metrics.RequestsRejected.With(prometheus.Labels{
|
|
"outpost_name": ls.ac.Outpost.Name,
|
|
"type": "bind",
|
|
"reason": "no_provider",
|
|
"dn": bindDN,
|
|
"client": utils.GetIP(conn.RemoteAddr()),
|
|
}).Inc()
|
|
return ldap.LDAPResultOperationsError, nil
|
|
}
|
|
|
|
func (ls *LDAPServer) TimerFlowCacheExpiry() {
|
|
for _, p := range ls.providers {
|
|
ls.log.WithField("flow", p.flowSlug).Debug("Pre-heating flow cache")
|
|
p.TimerFlowCacheExpiry()
|
|
}
|
|
}
|