package ldap import ( "context" "errors" "strings" "time" goldap "github.com/go-ldap/ldap/v3" "github.com/nmcclain/ldap" log "github.com/sirupsen/logrus" "goauthentik.io/api" "goauthentik.io/internal/outpost" ) const ContextUserKey = "ak_user" func (pi *ProviderInstance) getUsername(dn string) (string, error) { if !strings.HasSuffix(strings.ToLower(dn), strings.ToLower(pi.BaseDN)) { return "", errors.New("invalid base DN") } dns, err := goldap.ParseDN(dn) if err != nil { return "", err } for _, part := range dns.RDNs { for _, attribute := range part.Attributes { if strings.ToLower(attribute.Type) == "cn" { return attribute.Value, nil } } } return "", errors.New("failed to find cn") } func (pi *ProviderInstance) Bind(username string, req BindRequest) (ldap.LDAPResultCode, error) { fe := outpost.NewFlowExecutor(pi.flowSlug, pi.s.ac.Client.GetConfig(), log.Fields{ "bindDN": req.BindDN, "client": req.conn.RemoteAddr().String(), "requestId": req.id, }) fe.DelegateClientIP(req.conn.RemoteAddr()) fe.Params.Add("goauthentik.io/outpost/ldap", "true") fe.Answers[outpost.StageIdentification] = username fe.Answers[outpost.StagePassword] = req.BindPW passed, err := fe.Execute() if !passed { return ldap.LDAPResultInvalidCredentials, nil } if err != nil { req.log.WithError(err).Warning("failed to execute flow") return ldap.LDAPResultOperationsError, nil } access, err := fe.CheckApplicationAccess(pi.appSlug) if !access { req.log.Info("Access denied for user") return ldap.LDAPResultInsufficientAccessRights, nil } if err != nil { req.log.WithError(err).Warning("failed to check access") return ldap.LDAPResultOperationsError, nil } req.log.Info("User has access") // Get user info to store in context userInfo, _, err := fe.ApiClient().CoreApi.CoreUsersMeRetrieve(context.Background()).Execute() if err != nil { req.log.WithError(err).Warning("failed to get user info") return ldap.LDAPResultOperationsError, nil } pi.boundUsersMutex.Lock() cs := pi.SearchAccessCheck(userInfo.User) pi.boundUsers[req.BindDN] = UserFlags{ UserInfo: userInfo.User, CanSearch: cs != nil, } if pi.boundUsers[req.BindDN].CanSearch { req.log.WithField("group", cs).Info("Allowed access to search") } defer pi.boundUsersMutex.Unlock() pi.delayDeleteUserInfo(username) return ldap.LDAPResultSuccess, nil } // SearchAccessCheck Check if the current user is allowed to search func (pi *ProviderInstance) SearchAccessCheck(user api.User) *string { for _, group := range user.Groups { for _, allowedGroup := range pi.searchAllowedGroups { pi.log.WithField("userGroup", group.Pk).WithField("allowedGroup", allowedGroup).Trace("Checking search access") if group.Pk == allowedGroup.String() { return &group.Name } } } return nil } func (pi *ProviderInstance) delayDeleteUserInfo(dn string) { ticker := time.NewTicker(30 * time.Second) quit := make(chan struct{}) go func() { for { select { case <-ticker.C: pi.boundUsersMutex.Lock() delete(pi.boundUsers, dn) pi.boundUsersMutex.Unlock() close(quit) case <-quit: ticker.Stop() return } } }() }