75 lines
1.9 KiB
Go
75 lines
1.9 KiB
Go
package utils
|
|
|
|
import (
|
|
goldap "github.com/go-ldap/ldap/v3"
|
|
ber "github.com/nmcclain/asn1-ber"
|
|
"github.com/nmcclain/ldap"
|
|
"goauthentik.io/api/v3"
|
|
"goauthentik.io/internal/outpost/ldap/constants"
|
|
)
|
|
|
|
func ParseFilterForUser(req api.ApiCoreUsersListRequest, f *ber.Packet, skip bool) (api.ApiCoreUsersListRequest, bool) {
|
|
switch f.Tag {
|
|
case ldap.FilterEqualityMatch:
|
|
return parseFilterForUserSingle(req, f)
|
|
case ldap.FilterAnd:
|
|
for _, child := range f.Children {
|
|
r, s := ParseFilterForUser(req, child, skip)
|
|
skip = skip || s
|
|
req = r
|
|
}
|
|
return req, skip
|
|
}
|
|
return req, skip
|
|
}
|
|
|
|
func parseFilterForUserSingle(req api.ApiCoreUsersListRequest, f *ber.Packet) (api.ApiCoreUsersListRequest, bool) {
|
|
// We can only handle key = value pairs here
|
|
if len(f.Children) < 2 {
|
|
return req, false
|
|
}
|
|
k := f.Children[0].Value
|
|
// Ensure key is string
|
|
if _, ok := k.(string); !ok {
|
|
return req, false
|
|
}
|
|
v := f.Children[1].Value
|
|
// Null values are ignored
|
|
if v == nil {
|
|
return req, false
|
|
}
|
|
// Switch on type of the value, then check the key
|
|
switch vv := v.(type) {
|
|
case string:
|
|
switch k {
|
|
case "cn":
|
|
return req.Username(vv), false
|
|
case "name":
|
|
case "displayName":
|
|
return req.Name(vv), false
|
|
case "mail":
|
|
return req.Email(vv), false
|
|
case "member":
|
|
fallthrough
|
|
case "memberOf":
|
|
groupDN, err := goldap.ParseDN(vv)
|
|
if err != nil {
|
|
return req.GroupsByName([]string{vv}), false
|
|
}
|
|
name := groupDN.RDNs[0].Attributes[0].Value
|
|
// If the DN's first ou is virtual-groups, ignore this filter
|
|
if len(groupDN.RDNs) > 1 {
|
|
if groupDN.RDNs[1].Attributes[0].Value == constants.OUUsers || groupDN.RDNs[1].Attributes[0].Value == constants.OUVirtualGroups {
|
|
// Since we know we're not filtering anything, skip this request
|
|
return req, true
|
|
}
|
|
}
|
|
return req.GroupsByName([]string{name}), false
|
|
}
|
|
//TODO: Support int
|
|
default:
|
|
return req, false
|
|
}
|
|
return req, false
|
|
}
|