Spring Boot / Spring LDAP 获取用户的memberof列表

我想通过从如下结构的 LDAP 存储库查询用户 ID 来获取用户属性列表

dn: uid=E000001 ,ou=People,o=Company,o=Internal
cn: BOB DOLE
statusid: active
memberof: cn=foo_group, cn=Foos, ou=Groups, o=Company,o=Internal
memberof: cn=bar_group, cn=Foos, ou=Groups, o=Company,o=Internal

dn: uid=E000002 ,ou=People,o=Company,o=Internal
cn: MARK TEST
statusid: active
memberof: cn=foo_group, cn=Foos, ou=Groups, o=Company,o=Internal
memberof: cn=bar_group, cn=Foos, ou=Groups, o=Company,o=Internal

例如，我查询用户 ID“E00001”。我想退货

["cn=foo_group, cn=Foos, ou=Groups, o=Company,o=Internal", "cn=bar_group, cn=Foos, ou=Groups, o=Company,o=Internal"

以下是检索用户组的多种方法：

• 如果您有一个没有嵌套组的简单 LDAP 服务器，memberOf 通常就足够了：

  String userCN = "user1";
  
  //Get the attribute of user's "memberOf"
  ArrayList<?> membersOf = ldapTemplate.search(
          query().where("sAMAccountName").is(userCN),
          (AttributesMapper<ArrayList<?>>) attrs -> Collections.list(attrs.get("memberOf").getAll())
  ).get(0);
  

• 但如果你有嵌套组，事情就会变得更加复杂：

  /*
   * Get user distinguised name, example: "user" -> "CN=User Name,OU=Groups,OU=Domain Users,DC=company,DC=something,DC=org"
   * This will be used for our query later
   */
  String distinguishedName = ldapTemplate.search(
          query().where("sAMAccountName").is(userCN),
          (AttributesMapper<String>) attrs -> attrs.get("distinguishedName").get().toString()
  ).get(0); //.get(0): we assume that search will return a result 
  
  /*
   * This one recursively search for all (nested) group that this user belongs to
   * "member:1.2.840.113556.1.4.1941:" is a magic attribute, Reference: 
   * https://msdn.microsoft.com/en-us/library/aa746475(v=vs.85).aspx
   * However, this filter is usually slow in case your ad directory is large.
   */
  List<String> allGroups = ldapTemplate.search(
          query().searchScope(SearchScope.SUBTREE)
                  .where("member:1.2.840.113556.1.4.1941:").is(distinguishedName),
          (AttributesMapper<String>) attrs -> attrs.get("cn").get().toString()
  );
  

对于谷歌用户：请注意，这依赖于org.springframework.boot:spring-boot-starter-data-ldap，如果有人需要 Bean 初始化代码：

@Component
@EnableConfigurationProperties
public class Ldap {
    @Bean
    @ConfigurationProperties(prefix="ldap.contextSource")
    public LdapContextSource contextSource() {
        return new LdapContextSource();
    }

    @Bean
    public LdapTemplate ldapTemplate(ContextSource contextSource) {
        return new LdapTemplate(contextSource);
    }
}

在 application.yml 中使用以下配置模板：

ldap:
    contextSource:
        url: ldap://your-ldap.server
        base: dc=Company,dc=Domain,dc=Controller
        userDn: username
        password: hunter2
        #you'll want connection polling set to true so ldapTemplate reuse the connection when searching recursively
        pooled: true

当幻数性能不佳时：如果您的 ldap 目录很大，最后一个使用幻数实际上相当慢，在这种情况下递归搜索 ldap 会更快。这是一个帮助程序类，用于彻底搜索用户所属的所有组：

public class LdapSearchRecursive {
    private final LdapTemplate ldapTemplate;
    private Set<String> groups;

    public LdapSearchRecursive(LdapTemplate ldapTemplate) {
        this.ldapTemplate = ldapTemplate;
        this.groups = new HashSet<>();
    }

    /**
     * Retrieve all groups that this user belongs to.
     */
    public Set<String> getAllGroupsForUserRecursively(String userCN) {
        List<String> distinguishedNames = this.ldapTemplate.search(
                query().where("objectCategory").is("user").and(
                        query().where("sAMAccountName").is(userCN)
                                .or(query().where("userPrincipalName").is(userCN))
                ),
                (AttributesMapper<String>) attrs -> attrs.get("distinguishedName").get().toString()
        );

        if (distinguishedNames.isEmpty()) {
            throw new UsernameNotFoundException("User not recognized in LDAP");
        }

        return this.getAllGroupsRecursivelyByUserDistinguishedName(distinguishedNames.get(0), null);
    }

    private Set<String> getAllGroupsRecursivelyByUserDistinguishedName(String dn, @Nullable String parentDN) {
        List<String> results = this.ldapTemplate.search(
                query().where("member").is(dn),
                (AttributesMapper<String>) attrs -> attrs.get("distinguishedName").get().toString()
        );

        for (String result : results) {
            if (!(result.equals(parentDN) //circular, ignore
                    || this.groups.contains(result) //duplicate, ignore
                    )) {
                this.getAllGroupsRecursivelyByUserDistinguishedName(result, dn);
            }
        }
        this.groups.addAll(results);

        return this.groups;
    }
}