118 lines
3.1 KiB
Go
118 lines
3.1 KiB
Go
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
|
|
}
|
|
}
|
|
}()
|
|
}
|