Skip to content

Commit

Permalink
Merge pull request geosolutions-it#266 from geosolutions-it/265_17x_l…
Browse files Browse the repository at this point in the history 
…dap_roles
  • Loading branch information
@taba90
Nuno Oliveira authored and taba90 committed May 9, 2022
1 parent 71eac5f commit 81c0f6d
Show file tree
Hide file tree
Showing 3 changed files with 132 additions and 56 deletions.
66 changes: 48 additions & 18 deletions .../src/main/java/it/geosolutions/geostore/core/security/SimpleGrantedAuthoritiesMapper.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -32,40 +32,70 @@
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import org.springframework.security.core.GrantedAuthority;
import org.springframework.security.core.authority.SimpleGrantedAuthority;

/**
* Map based implementation of GrantedAuthoritiesMapper.
* *
* *
* @author Mauro Bartolomeoli
*
*/
public class SimpleGrantedAuthoritiesMapper implements GrantedAuthoritiesMapper {

private Map<String, String> mappings = new HashMap<String, String>();

private Map<String, String> mappings = new HashMap<>();

/**
* Do not consider authorities that do not exist in the mapping
**/
private boolean dropUnmapped = false;

private static final Log logger = LogFactory.getLog(SimpleGrantedAuthoritiesMapper.class);

public SimpleGrantedAuthoritiesMapper(Map<String, String> mappings) {
super();
this.mappings = mappings;
}

@Override
public Collection<? extends GrantedAuthority> mapAuthorities(
Collection<? extends GrantedAuthority> authorities) {
if(mappings.isEmpty()) {
return authorities;
}
List<GrantedAuthority> result = new ArrayList<GrantedAuthority>();
for(GrantedAuthority authority : authorities) {
if(mappings.containsKey(authority.getAuthority())) {
result.add(new SimpleGrantedAuthority(mappings.get(authority.getAuthority())));
} else {
result.add(authority);

@Override
public Collection<? extends GrantedAuthority> mapAuthorities(
Collection<? extends GrantedAuthority> authorities) {
if (mappings.isEmpty()) {
return authorities;
}

List<GrantedAuthority> result = new ArrayList<>();

for (GrantedAuthority authority : authorities) {
String src = authority.getAuthority();
if (mappings.containsKey(src)) {
String dst = mappings.get(authority.getAuthority());
result.add(new SimpleGrantedAuthority(dst));
if (logger.isDebugEnabled()) {
logger.debug("Mapping authority: " + src + " --> " + dst );
}
}
return result;
}
} else if(dropUnmapped) {
if (logger.isDebugEnabled()) {
logger.debug("Dropping unmapped authority: " + src);
}
}
else
{
result.add(authority);
if (logger.isDebugEnabled()) {
logger.debug("Adding unmapped authority: " + src );
}
}
}
return result;
}

public void setDropUnmapped(boolean dropUnmapped) {
this.dropUnmapped = dropUnmapped;
}

}
11 changes: 11 additions & 0 deletions .../test/java/it/geosolutions/geostore/core/security/SimpleGrantedAuthoritiesMapperTest.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -35,4 +35,15 @@ public void testMapping() {
assertEquals("B", mapped.iterator().next().getAuthority());
}

@Test
public void testDropUnmappedAuthiorities() {
mapper.setDropUnmapped(true);
roleMappings.put("A", "B");
authorities.add(new GrantedAuthorityImpl("A"));
authorities.add(new GrantedAuthorityImpl("C"));
Collection<? extends GrantedAuthority> mapped = mapper.mapAuthorities(authorities);
assertEquals(1, mapped.size());
assertEquals("B", mapped.iterator().next().getAuthority());
}

}
111 changes: 73 additions & 38 deletions ...ava/it/geosolutions/geostore/services/rest/security/GeoStoreLdapAuthoritiesPopulator.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -124,7 +124,8 @@ public Authority(String name, String dn) {
*/
private boolean convertToUpperCase = true;

private GrantedAuthoritiesMapper authoritiesMapper = null;
private GrantedAuthoritiesMapper roleMapper = null;
private GrantedAuthoritiesMapper groupMapper = null;

/**
* @param contextSource
Expand Down Expand Up @@ -156,64 +157,89 @@ public GeoStoreLdapAuthoritiesPopulator(ContextSource contextSource, String grou
}
}

@Deprecated
public void setAuthoritiesMapper(GrantedAuthoritiesMapper authoritiesMapper) {
this.authoritiesMapper = authoritiesMapper;
logger.error("AuthoritiesMapper is deprecated, please set roleMapper and groupMapper separately");
this.roleMapper = authoritiesMapper;
this.groupMapper = authoritiesMapper;
}

public void setRoleMapper(GrantedAuthoritiesMapper roleMapper) {
this.roleMapper = roleMapper;
}

public void setGroupMapper(GrantedAuthoritiesMapper groupMapper) {
this.groupMapper = groupMapper;
}

@Override
public Set<GrantedAuthority> getGroupMembershipRoles(String userDn, String username) {
return getGroupsOrRoles(userDn, username, true, true);
// TODO: double check if we really want to return groups+roles
Set<GrantedAuthority> ret = new HashSet<>();
ret.addAll(getGroups(userDn, username));
ret.addAll(getRoles(userDn, username));
return ret;
}

private Set<GrantedAuthority> getGroupsOrRoles(String userDn, String username, boolean groups, boolean roles) {
if (roleSearchBase == null && groupSearchBase == null) {
return new HashSet<GrantedAuthority>();
private Set<GrantedAuthority> getRoles(String userDn, String username) {
if (roleSearchBase == null) {
return new HashSet<>();
}

Set<GrantedAuthority> authorities = new HashSet<GrantedAuthority>();
Set<GrantedAuthority> authorities = new HashSet<>();

String[] searchParams = username == null ? new String[] {} : new String[] {userDn, username};
if(roles) {
// Searching for ROLES
if (logger.isDebugEnabled()) {
logger.debug("Searching for roles for user '" + username + "', DN = " + "'" + userDn + "', with filter "
+ roleSearchFilter + " in search base '" + roleSearchBase + "'");
}

String[] rolesRoots = roleSearchBase.split(";");
String filter = username == null ? allRolesSearchFilter : roleSearchFilter;

for(String rolesRoot : rolesRoots) {
addAuthorities(searchParams, authorities, rolesRoot, filter, rolePrefix, false);
}

// Searching for ROLES
if (logger.isDebugEnabled()) {
logger.debug("Searching for roles for user '" + username + "', DN = " + "'" + userDn + "', with filter "
+ roleSearchFilter + " in search base '" + roleSearchBase + "'");
}

String[] rolesRoots = roleSearchBase.split(";");
String filter = username == null ? allRolesSearchFilter : roleSearchFilter;

for(String rolesRoot : rolesRoots) {
addAuthorities(searchParams, authorities, rolesRoot, filter, rolePrefix, false);
}

if(roleMapper != null) {
authorities = new HashSet<>(roleMapper.mapAuthorities(authorities));
}

if(groups) {
// Searching for Groups
if (logger.isDebugEnabled()) {
logger.debug("Searching for groups for user '" + username + "', DN = " + "'" + userDn + "', with filter "
+ groupSearchFilter + " in search base '" + groupSearchBase + "'");
}
String[] groupsRoots = groupSearchBase.split(";");
String filter = username == null ? allGroupsSearchFilter : groupSearchFilter;
for(String groupsRoot : groupsRoots) {
addAuthorities(searchParams, authorities, groupsRoot, filter, null, enableHierarchicalGroups);
}
return authorities;
}

private Set<GrantedAuthority> getGroups(String userDn, String username) {
if (groupSearchBase == null) {
return new HashSet<>();
}

Set<GrantedAuthority> authorities = new HashSet<>();
String[] searchParams = username == null ? new String[] {} : new String[] {userDn, username};

// Searching for Groups
if (logger.isDebugEnabled()) {
logger.debug("Searching for groups for user '" + username + "', DN = " + "'" + userDn + "', with filter "
+ groupSearchFilter + " in search base '" + groupSearchBase + "'");
}
String[] groupsRoots = groupSearchBase.split(";");
String filter = username == null ? allGroupsSearchFilter : groupSearchFilter;
for(String groupsRoot : groupsRoots) {
addAuthorities(searchParams, authorities, groupsRoot, filter, null, enableHierarchicalGroups);
}

if(authoritiesMapper != null) {
return new HashSet<GrantedAuthority>(authoritiesMapper.mapAuthorities(authorities));
if(groupMapper != null) {
authorities = new HashSet<>(groupMapper.mapAuthorities(authorities));
}
return authorities;
}

public Set<GrantedAuthority> getAllGroups() {
return getGroupsOrRoles(null, null, true, false);
return getGroups(null, null);
}

public Set<GrantedAuthority> getAllRoles() {
return getGroupsOrRoles(null, null, false, true);
return getRoles(null, null);
}

private void addAuthorities(String[] params, Set<GrantedAuthority> authorities,
Expand All @@ -233,7 +259,7 @@ protected Object doMapFromContext(DirContextOperations ctx) {
});

if (logger.isDebugEnabled()) {
logger.debug("Authorities from search: " + ldapAuthorities);
logger.debug("Found " + ldapAuthorities.size() + " authorities from search");
}
for (Object authority : ldapAuthorities) {
Authority ldapAuthority = (Authority)authority;
Expand All @@ -248,17 +274,26 @@ protected Object doMapFromContext(DirContextOperations ctx) {

private boolean addAuthority(Set<GrantedAuthority> authorities, String authorityPrefix,
String authority) {

if (logger.isDebugEnabled()) {
logger.debug("Adding authority: " + authorityPrefix + "::" + authority);
}

if (convertToUpperCase) {
authority = authority.toUpperCase();
}

String prefix = (authorityPrefix != null && !authority.startsWith(authorityPrefix) ? authorityPrefix : "");

String rolename = prefix + authority;
SimpleGrantedAuthority role = new SimpleGrantedAuthority(prefix + authority);
if (!authorities.contains(role)) {
authorities.add(role);
if (logger.isDebugEnabled()) {
logger.debug("Authority added: " + rolename);
}
return true;
}
logger.debug("Authority not added: " + rolename);
return false;
}

Expand Down

0 comments on commit 81c0f6d

Please sign in to comment.