X-Git-Url: http://git.argeo.org/?a=blobdiff_plain;f=org.argeo.security.core%2Fsrc%2Forg%2Fargeo%2Fosgi%2Fuseradmin%2FLdifUserAdmin.java;h=521ae8bb6e89916c6b88ea490767658e8b3012fa;hb=8260f4470f514ea347ca53f5b4dfc632c4a4de66;hp=dacae7964f1ee74a24415529655fbd7940133a44;hpb=e86e3691db8b4a426a6d5fd3cf255dc82fe05fc1;p=lgpl%2Fargeo-commons.git diff --git a/org.argeo.security.core/src/org/argeo/osgi/useradmin/LdifUserAdmin.java b/org.argeo.security.core/src/org/argeo/osgi/useradmin/LdifUserAdmin.java index dacae7964..521ae8bb6 100644 --- a/org.argeo.security.core/src/org/argeo/osgi/useradmin/LdifUserAdmin.java +++ b/org.argeo.security.core/src/org/argeo/osgi/useradmin/LdifUserAdmin.java @@ -1,136 +1,146 @@ package org.argeo.osgi.useradmin; +import static org.argeo.osgi.useradmin.LdifName.inetOrgPerson; +import static org.argeo.osgi.useradmin.LdifName.objectClass; + +import java.io.File; +import java.io.FileOutputStream; +import java.io.IOException; import java.io.InputStream; -import java.net.URI; -import java.net.URISyntaxException; +import java.io.OutputStream; import java.util.ArrayList; import java.util.Dictionary; -import java.util.LinkedHashMap; +import java.util.HashSet; +import java.util.Hashtable; import java.util.List; -import java.util.Map; +import java.util.Set; import java.util.SortedMap; import java.util.TreeMap; -import javax.naming.InvalidNameException; import javax.naming.NamingEnumeration; import javax.naming.directory.Attributes; -import javax.naming.directory.BasicAttributes; import javax.naming.ldap.LdapName; -import javax.naming.ldap.Rdn; +import javax.transaction.TransactionManager; +import org.argeo.util.naming.LdifParser; +import org.argeo.util.naming.LdifWriter; import org.osgi.framework.Filter; -import org.osgi.framework.FrameworkUtil; -import org.osgi.framework.InvalidSyntaxException; -import org.osgi.service.useradmin.Authorization; import org.osgi.service.useradmin.Role; -import org.osgi.service.useradmin.User; - -/** User admin implementation using LDIF file(s) as backend. */ -public class LdifUserAdmin extends AbstractLdapUserAdmin { - SortedMap users = new TreeMap(); - SortedMap groups = new TreeMap(); - private Map> userIndexes = new LinkedHashMap>(); +/** + * A user admin based on a LDIF files. Requires a {@link TransactionManager} and + * an open transaction for write access. + */ +public class LdifUserAdmin extends AbstractUserDirectory { + private SortedMap users = new TreeMap(); + private SortedMap groups = new TreeMap(); - public LdifUserAdmin(String uri) { - this(uri, true); + public LdifUserAdmin(String uri, String baseDn) { + this(fromUri(uri, baseDn)); } - public LdifUserAdmin(String uri, boolean isReadOnly) { - setReadOnly(isReadOnly); - try { - setUri(new URI(uri)); - } catch (URISyntaxException e) { - throw new ArgeoUserAdminException("Invalid URI " + uri, e); - } - - if (!isReadOnly && !getUri().getScheme().equals("file:")) - throw new UnsupportedOperationException(getUri().getScheme() - + "not supported read-write."); - + public LdifUserAdmin(Dictionary properties) { + super(properties); } public LdifUserAdmin(InputStream in) { + super(new Hashtable()); load(in); - setReadOnly(true); - setUri(null); + } + + private static Dictionary fromUri(String uri, String baseDn) { + Hashtable res = new Hashtable(); + res.put(UserAdminConf.uri.name(), uri); + res.put(UserAdminConf.baseDn.name(), baseDn); + return res; } public void init() { try { + if (getUri().getScheme().equals("file")) { + File file = new File(getUri()); + if (!file.exists()) + return; + } load(getUri().toURL().openStream()); } catch (Exception e) { - throw new ArgeoUserAdminException("Cannot open URL " + getUri(), e); + throw new UserDirectoryException("Cannot open URL " + getUri(), e); + } + } + + public void save() { + if (getUri() == null) + throw new UserDirectoryException("Cannot save LDIF user admin: no URI is set"); + if (isReadOnly()) + throw new UserDirectoryException("Cannot save LDIF user admin: " + getUri() + " is read-only"); + try (FileOutputStream out = new FileOutputStream(new File(getUri()))) { + save(out); + } catch (IOException e) { + throw new UserDirectoryException("Cannot save user admin to " + getUri(), e); + } + } + + public void save(OutputStream out) throws IOException { + try { + LdifWriter ldifWriter = new LdifWriter(out); + for (LdapName name : groups.keySet()) + ldifWriter.writeEntry(name, groups.get(name).getAttributes()); + for (LdapName name : users.keySet()) + ldifWriter.writeEntry(name, users.get(name).getAttributes()); + } finally { + out.close(); } } protected void load(InputStream in) { try { + users.clear(); + groups.clear(); + LdifParser ldifParser = new LdifParser(); SortedMap allEntries = ldifParser.read(in); for (LdapName key : allEntries.keySet()) { Attributes attributes = allEntries.get(key); - NamingEnumeration objectClasses = attributes.get( - "objectClass").getAll(); + // check for inconsistency + Set lowerCase = new HashSet(); + NamingEnumeration ids = attributes.getIDs(); + while (ids.hasMoreElements()) { + String id = ids.nextElement().toLowerCase(); + if (lowerCase.contains(id)) + throw new UserDirectoryException(key + " has duplicate id " + id); + lowerCase.add(id); + } + + // analyse object classes + NamingEnumeration objectClasses = attributes.get(objectClass.name()).getAll(); + // System.out.println(key); objectClasses: while (objectClasses.hasMore()) { String objectClass = objectClasses.next().toString(); - if (objectClass.equals("inetOrgPerson")) { - users.put(key, new LdifUser(key, attributes)); + // System.out.println(" " + objectClass); + if (objectClass.equals(inetOrgPerson.name())) { + users.put(key, new LdifUser(this, key, attributes)); break objectClasses; - } else if (objectClass.equals("groupOfNames")) { - groups.put(key, new LdifGroup(key, attributes)); + } else if (objectClass.equals(getGroupObjectClass())) { + groups.put(key, new LdifGroup(this, key, attributes)); break objectClasses; } } } - - // optimise - for (LdifGroup group : groups.values()) - loadMembers(group); - - // indexes - for (String attr : getIndexedUserProperties()) - userIndexes.put(attr, new TreeMap()); - - for (LdifUser user : users.values()) { - Dictionary properties = user.getProperties(); - for (String attr : getIndexedUserProperties()) { - Object value = properties.get(attr); - if (value != null) { - LdifUser otherUser = userIndexes.get(attr).put( - value.toString(), user); - if (otherUser != null) - throw new ArgeoUserAdminException("User " + user - + " and user " + otherUser - + " both have property " + attr - + " set to " + value); - } - } - } } catch (Exception e) { - throw new ArgeoUserAdminException( - "Cannot load user admin service from LDIF", e); + throw new UserDirectoryException("Cannot load user admin service from LDIF", e); } } public void destroy() { + if (users == null || groups == null) + throw new UserDirectoryException("User directory " + getBaseDn() + " is already destroyed"); users.clear(); users = null; groups.clear(); groups = null; } - @Override - public Role getRole(String name) { - LdapName key; - try { - key = new LdapName(name); - } catch (InvalidNameException e) { - // TODO implements default base DN - throw new IllegalArgumentException("Badly formatted role name: " - + name, e); - } - + protected DirectoryUser daoGetRole(LdapName key) { if (groups.containsKey(key)) return groups.get(key); if (users.containsKey(key)) @@ -138,139 +148,91 @@ public class LdifUserAdmin extends AbstractLdapUserAdmin { return null; } - @Override - public Authorization getAuthorization(User user) { - return new LdifAuthorization((LdifUser) user); + protected Boolean daoHasRole(LdapName dn) { + return users.containsKey(dn) || groups.containsKey(dn); } - @Override - public Role createRole(String name, int type) { - try { - LdapName dn = new LdapName(name); - if (users.containsKey(dn) || groups.containsKey(dn)) - throw new ArgeoUserAdminException("Already a role " + name); - - BasicAttributes attrs = new BasicAttributes(); - attrs.put("dn", dn.toString()); - Rdn nameRdn = dn.getRdn(dn.size() - 1); - // TODO deal with multiple attr RDN - attrs.put(nameRdn.getType(), nameRdn.getValue()); - LdifUser newRole; - if (type == Role.USER) { - newRole = new LdifUser(dn, attrs); - users.put(dn, newRole); - } else if (type == Role.GROUP) { - newRole = new LdifGroup(dn, attrs); - groups.put(dn, (LdifGroup) newRole); - } else - throw new ArgeoUserAdminException("Unsupported type " + type); - return newRole; - } catch (InvalidNameException e) { - throw new ArgeoUserAdminException("Cannot create role " + name, e); - } - } - - @Override - public boolean removeRole(String name) { - try { - LdapName dn = new LdapName(name); - LdifUser role = null; - if (users.containsKey(dn)) - role = users.remove(dn); - else if (groups.containsKey(dn)) - role = groups.remove(dn); - else - throw new ArgeoUserAdminException("There is no role " + name); - if (role == null) - return false; - for (LdifGroup group : role.directMemberOf) { - group.directMembers.remove(role); - group.getAttributes().get(group.getMemberAttrName()) - .remove(dn.toString()); - } - if (role instanceof LdifGroup) { - LdifGroup group = (LdifGroup) role; - for (Role user : group.directMembers) { - if (user instanceof LdifUser) - ((LdifUser) user).directMemberOf.remove(group); - } - } - return true; - } catch (InvalidNameException e) { - throw new ArgeoUserAdminException("Cannot create role " + name, e); - } - } - - @Override - public Role[] getRoles(String filter) throws InvalidSyntaxException { - ArrayList res = new ArrayList(); - if (filter == null) { + @SuppressWarnings("unchecked") + protected List doGetRoles(Filter f) { + ArrayList res = new ArrayList(); + if (f == null) { res.addAll(users.values()); res.addAll(groups.values()); } else { - Filter f = FrameworkUtil.createFilter(filter); - for (LdifUser user : users.values()) + for (DirectoryUser user : users.values()) { + // System.out.println("\n" + user.getName()); + // Dictionary props = user.getProperties(); + // for (Enumeration keys = props.keys(); keys + // .hasMoreElements();) { + // String key = keys.nextElement(); + // System.out.println(" " + key + "=" + props.get(key)); + // } if (f.match(user.getProperties())) res.add(user); - for (LdifUser group : groups.values()) + } + for (DirectoryUser group : groups.values()) if (f.match(group.getProperties())) res.add(group); } - return res.toArray(new Role[res.size()]); + return res; } @Override - public User getUser(String key, String value) { - // TODO check value null or empty - if (key != null) { - if (!userIndexes.containsKey(key)) - return null; - return userIndexes.get(key).get(value); + protected List getDirectGroups(LdapName dn) { + List directGroups = new ArrayList(); + for (LdapName name : groups.keySet()) { + DirectoryGroup group = groups.get(name); + if (group.getMemberNames().contains(dn)) + directGroups.add(group.getDn()); } + return directGroups; + } - // Try all indexes - List collectedUsers = new ArrayList( - getIndexedUserProperties().size()); - // try dn - LdifUser user = null; - try { - user = (LdifUser) getRole(value); - if (user != null) - collectedUsers.add(user); - } catch (Exception e) { - // silent + @Override + protected void prepare(UserDirectoryWorkingCopy wc) { + // delete + for (LdapName dn : wc.getDeletedUsers().keySet()) { + if (users.containsKey(dn)) + users.remove(dn); + else if (groups.containsKey(dn)) + groups.remove(dn); + else + throw new UserDirectoryException("User to delete not found " + dn); } - for (String attr : userIndexes.keySet()) { - user = userIndexes.get(attr).get(value); - if (user != null) - collectedUsers.add(user); + // add + for (LdapName dn : wc.getNewUsers().keySet()) { + DirectoryUser user = wc.getNewUsers().get(dn); + if (users.containsKey(dn) || groups.containsKey(dn)) + throw new UserDirectoryException("User to create found " + dn); + else if (Role.USER == user.getType()) + users.put(dn, user); + else if (Role.GROUP == user.getType()) + groups.put(dn, (DirectoryGroup) user); + else + throw new UserDirectoryException("Unsupported role type " + user.getType() + " for new user " + dn); } + // modify + for (LdapName dn : wc.getModifiedUsers().keySet()) { + Attributes modifiedAttrs = wc.getModifiedUsers().get(dn); + DirectoryUser user; + if (users.containsKey(dn)) + user = users.get(dn); + else if (groups.containsKey(dn)) + user = groups.get(dn); + else + throw new UserDirectoryException("User to modify no found " + dn); + user.publishAttributes(modifiedAttrs); + } + } - if (collectedUsers.size() == 1) - return collectedUsers.get(0); - return null; - // throw new UnsupportedOperationException(); + @Override + protected void commit(UserDirectoryWorkingCopy wc) { + save(); } - protected void loadMembers(LdifGroup group) { - group.directMembers = new ArrayList(); - for (LdapName ldapName : group.getMemberNames()) { - LdifUser role = null; - if (groups.containsKey(ldapName)) - role = groups.get(ldapName); - else if (users.containsKey(ldapName)) - role = users.get(ldapName); - else { - if (getExternalRoles() != null) - role = (LdifUser) getExternalRoles().getRole( - ldapName.toString()); - if (role == null) - throw new ArgeoUserAdminException("No role found for " - + ldapName); - } - role.directMemberOf.add(group); - group.directMembers.add(role); - } + @Override + protected void rollback(UserDirectoryWorkingCopy wc) { + init(); } }