Skip to content
LdapGroupDAO.java 41.2 KiB
Newer Older
/*
 ************************************************************************
 *******************  CANADIAN ASTRONOMY DATA CENTRE  *******************
 **************  CENTRE CANADIEN DE DONNÉES ASTRONOMIQUES  **************
 *
 *  (c) 2014.                            (c) 2014.
 *  Government of Canada                 Gouvernement du Canada
 *  National Research Council            Conseil national de recherches
 *  Ottawa, Canada, K1A 0R6              Ottawa, Canada, K1A 0R6
 *  All rights reserved                  Tous droits réservés
 *
 *  NRC disclaims any warranties,        Le CNRC dénie toute garantie
 *  expressed, implied, or               énoncée, implicite ou légale,
 *  statutory, of any kind with          de quelque nature que ce
 *  respect to the software,             soit, concernant le logiciel,
 *  including without limitation         y compris sans restriction
 *  any warranty of merchantability      toute garantie de valeur
 *  or fitness for a particular          marchande ou de pertinence
 *  purpose. NRC shall not be            pour un usage particulier.
 *  liable in any event for any          Le CNRC ne pourra en aucun cas
 *  damages, whether direct or           être tenu responsable de tout
 *  indirect, special or general,        dommage, direct ou indirect,
 *  consequential or incidental,         particulier ou général,
 *  arising from the use of the          accessoire ou fortuit, résultant
 *  software.  Neither the name          de l'utilisation du logiciel. Ni
 *  of the National Research             le nom du Conseil National de
 *  Council of Canada nor the            Recherches du Canada ni les noms
 *  names of its contributors may        de ses  participants ne peuvent
 *  be used to endorse or promote        être utilisés pour approuver ou
 *  products derived from this           promouvoir les produits dérivés
 *  software without specific prior      de ce logiciel sans autorisation
 *  written permission.                  préalable et particulière
 *                                       par écrit.
 *
 *  This file is part of the             Ce fichier fait partie du projet
 *  OpenCADC project.                    OpenCADC.
 *
 *  OpenCADC is free software:           OpenCADC est un logiciel libre ;
 *  you can redistribute it and/or       vous pouvez le redistribuer ou le
 *  modify it under the terms of         modifier suivant les termes de
 *  the GNU Affero General Public        la “GNU Affero General Public
 *  License as published by the          License” telle que publiée
 *  Free Software Foundation,            par la Free Software Foundation
 *  either version 3 of the              : soit la version 3 de cette
 *  License, or (at your option)         licence, soit (à votre gré)
 *  any later version.                   toute version ultérieure.
 *
 *  OpenCADC is distributed in the       OpenCADC est distribué
 *  hope that it will be useful,         dans l’espoir qu’il vous
 *  but WITHOUT ANY WARRANTY;            sera utile, mais SANS AUCUNE
 *  without even the implied             GARANTIE : sans même la garantie
 *  warranty of MERCHANTABILITY          implicite de COMMERCIALISABILITÉ
 *  or FITNESS FOR A PARTICULAR          ni d’ADÉQUATION À UN OBJECTIF
 *  PURPOSE.  See the GNU Affero         PARTICULIER. Consultez la Licence
 *  General Public License for           Générale Publique GNU Affero
 *  more details.                        pour plus de détails.
 *
 *  You should have received             Vous devriez avoir reçu une
 *  a copy of the GNU Affero             copie de la Licence Générale
 *  General Public License along         Publique GNU Affero avec
 *  with OpenCADC.  If not, see          OpenCADC ; si ce n’est
 *  <http://www.gnu.org/licenses/>.      pas le cas, consultez :
 *                                       <http://www.gnu.org/licenses/>.
 *
 *  $Revision: 4 $
 *
 ************************************************************************
 */
package ca.nrc.cadc.ac.server.ldap;

Jeff Burke's avatar
Jeff Burke committed
import ca.nrc.cadc.ac.ActivatedGroup;
import ca.nrc.cadc.ac.Group;
import ca.nrc.cadc.ac.GroupAlreadyExistsException;
import ca.nrc.cadc.ac.GroupNotFoundException;
Jeff Burke's avatar
Jeff Burke committed
import ca.nrc.cadc.ac.Role;
import ca.nrc.cadc.ac.User;
import ca.nrc.cadc.ac.UserNotFoundException;
import ca.nrc.cadc.net.TransientException;
import com.unboundid.ldap.sdk.AddRequest;
import com.unboundid.ldap.sdk.Attribute;
import com.unboundid.ldap.sdk.DN;
Jeff Burke's avatar
Jeff Burke committed
import com.unboundid.ldap.sdk.Filter;
import com.unboundid.ldap.sdk.LDAPException;
import com.unboundid.ldap.sdk.LDAPResult;
import com.unboundid.ldap.sdk.Modification;
import com.unboundid.ldap.sdk.ModificationType;
import com.unboundid.ldap.sdk.ModifyRequest;
import com.unboundid.ldap.sdk.SearchRequest;
import com.unboundid.ldap.sdk.SearchResult;
import com.unboundid.ldap.sdk.SearchResultEntry;
import com.unboundid.ldap.sdk.SearchScope;
import com.unboundid.ldap.sdk.controls.ProxiedAuthorizationV2RequestControl;
import java.security.AccessControlException;
import java.security.Principal;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Date;
import java.util.List;
Jeff Burke's avatar
Jeff Burke committed
import javax.security.auth.x500.X500Principal;
import org.apache.log4j.Logger;

public class LdapGroupDAO<T extends Principal> extends LdapDAO
{
    private static final Logger logger = Logger.getLogger(LdapGroupDAO.class);
    private static final String ACTUAL_GROUP_TOKEN = "<ACTUAL_GROUP>";
Jeff Burke's avatar
Jeff Burke committed
    private static final String GROUP_READ_ACI = "(targetattr = \"*\") " + 
            "(version 3.0;acl \"Group Read\";allow (read,compare,search)" + 
            "(groupdn = \"ldap:///<ACTUAL_GROUP>\");)";
    private static final String GROUP_WRITE_ACI = "(targetattr = \"*\") " + 
            "(version 3.0;acl \"Group Write\";allow " + 
            "(read,compare,search,selfwrite,write,add)" + 
            "(groupdn = \"ldap:///<ACTUAL_GROUP>\");)";
    private static final String PUB_GROUP_ACI = "(targetattr = \"*\") " + 
            "(version 3.0;acl \"Group Public\";" + 
            "allow (read,compare,search)userdn=\"ldap:///all\";)";
    private LdapUserDAO<T> userPersist;

    public LdapGroupDAO(LdapConfig config, LdapUserDAO<T> userPersist)
    {
        super(config);
        if (userPersist == null)
        {
Jeff Burke's avatar
Jeff Burke committed
            throw new IllegalArgumentException(
                    "User persistence instance required");
        }
        this.userPersist = userPersist;
    }

Jeff Burke's avatar
Jeff Burke committed
    /**
     * Creates the group.
     * 
     * @param group The group to create
     * 
     * @return created group
     * 
     * @throws GroupAlreadyExistsException If a group with the same ID already 
     *                                     exists.
     * @throws TransientException If an temporary, unexpected problem occurred.
     * @throws UserNotFoundException If owner or a member not valid user.
     */
    public Group addGroup(Group group)
Jeff Burke's avatar
Jeff Burke committed
        throws GroupAlreadyExistsException, TransientException,
               UserNotFoundException, AccessControlException
        if (group.getOwner() == null)
        {
            throw new IllegalArgumentException("Group owner must be specified");
        }
        
        try
        {
            getGroup(group.getID());
            throw new GroupAlreadyExistsException(group.getID());
        }
        catch (GroupNotFoundException ex)
        {
            try
Jeff Burke's avatar
Jeff Burke committed
            {             
                try
                {
                    Group inactiveGroup = getInactiveGroup(group.getID());

                    // Check requestor owns the group.
                    DN ownerDN = userPersist.getUserDN(group.getOwner());
                    if (!ownerDN.equals(getSubjectDN()))
                    {
                       throw new AccessControlException(
                           "Unable to activate group " + group.getID() + 
Jeff Burke's avatar
Jeff Burke committed
                           + " is not the owner"); 
                    }
                    
                    List<Modification> mods = new ArrayList<Modification>();
                    Modification mod = 
                        new Modification(ModificationType.DELETE, 
                                         "nsaccountlock");
                    mods.add(mod);
                    
                    Group modifiedGroup = modifyGroup(group, inactiveGroup, 
                                                      mods);
Jeff Burke's avatar
Jeff Burke committed
                    Group activatedGroup = 
                            new ActivatedGroup(modifiedGroup.getID(),
                                               modifiedGroup.getOwner());
                    activatedGroup.description = modifiedGroup.description;
                    activatedGroup.publicRead = modifiedGroup.publicRead;
                    activatedGroup.groupRead = modifiedGroup.groupRead;
                    activatedGroup.groupWrite = modifiedGroup.groupWrite;
                    activatedGroup.getProperties()
                            .addAll(modifiedGroup.getProperties());
                    activatedGroup.getGroupMembers()
                            .addAll(modifiedGroup.getGroupMembers());
                    activatedGroup.getUserMembers()
                            .addAll(modifiedGroup.getUserMembers());
                    return activatedGroup;
Jeff Burke's avatar
Jeff Burke committed
                }
                catch (GroupNotFoundException ignore) {}
                
Jeff Burke's avatar
Jeff Burke committed
                if (!group.getProperties().isEmpty())
Jeff Burke's avatar
Jeff Burke committed
                    throw new UnsupportedOperationException(
                            "Support for groups properties not available");
Jeff Burke's avatar
Jeff Burke committed
                DN ownerDN = userPersist.getUserDN(group.getOwner());
                String groupWriteAci = null;
                String groupReadAci = null;
                if (group.groupWrite != null)
                {
                    DN groupWrite = getGroupDN(group.groupWrite.getID());
Jeff Burke's avatar
Jeff Burke committed
                    groupWriteAci = GROUP_WRITE_ACI.replace(
                            ACTUAL_GROUP_TOKEN, 
                            groupWrite.toNormalizedString());
                }

                if (group.groupRead != null)
                {
                    DN groupRead = getGroupDN(group.groupRead.getID());
Jeff Burke's avatar
Jeff Burke committed
                    groupReadAci = GROUP_READ_ACI.replace(
                            ACTUAL_GROUP_TOKEN, 
                            groupRead.toNormalizedString());
Jeff Burke's avatar
Jeff Burke committed
                // add new group
                List<Attribute> attributes = new ArrayList<Attribute>();
                attributes.add(new Attribute("objectClass", 
                                             "groupofuniquenames"));

                attributes.add(new Attribute("cn", group.getID()));
                if (group.description != null)
                {
Jeff Burke's avatar
Jeff Burke committed
                    attributes.add(new Attribute("description", 
                                                 group.description));
Jeff Burke's avatar
Jeff Burke committed
                attributes.add(new Attribute("owner", 
                                             ownerDN.toNormalizedString()));
Jeff Burke's avatar
Jeff Burke committed
                // acis
                List<String> acis = new ArrayList<String>();
                if (group.publicRead)
                {
Jeff Burke's avatar
Jeff Burke committed
                    acis.add(PUB_GROUP_ACI);
                }
                if (groupWriteAci != null)
                {
                    acis.add(groupWriteAci);
                }
                if (groupReadAci != null)
                {
                    acis.add(groupReadAci);
                }

Jeff Burke's avatar
Jeff Burke committed
                if (!acis.isEmpty())
Jeff Burke's avatar
Jeff Burke committed
                    attributes.add(new Attribute("aci", 
                            (String[]) acis.toArray(new String[acis.size()])));
Jeff Burke's avatar
Jeff Burke committed
                List<String> members = new ArrayList<String>();
                for (User<?> member : group.getUserMembers())
                {
                    DN memberDN = this.userPersist.getUserDN(member);
                    members.add(memberDN.toNormalizedString());
                }
                for (Group gr : group.getGroupMembers())
                {
                    DN grDN = getGroupDN(gr.getID());
                    members.add(grDN.toNormalizedString());
                }
Jeff Burke's avatar
Jeff Burke committed
                if (!members.isEmpty())
Jeff Burke's avatar
Jeff Burke committed
                    attributes.add(new Attribute("uniquemember", 
                        (String[]) members.toArray(new String[members.size()])));
Jeff Burke's avatar
Jeff Burke committed
                AddRequest addRequest = 
                        new AddRequest(getGroupDN(group.getID()), attributes);
Jeff Burke's avatar
Jeff Burke committed
                addRequest.addControl(
                        new ProxiedAuthorizationV2RequestControl(
                                "dn:" + getSubjectDN().toNormalizedString()));

                LDAPResult result = getConnection().add(addRequest);
                try
                {
                    return getGroup(group.getID());
                }
                catch (GroupNotFoundException e)
                {
                    throw new RuntimeException("BUG: new group not found");
                }
            }
            catch (LDAPException e)
            {
                e.printStackTrace();
                throw new RuntimeException(e);
            }
        }
    }

Jeff Burke's avatar
Jeff Burke committed
    /**
Jeff Burke's avatar
Jeff Burke committed
     * 
     * @param groupID The Group unique ID.
     * 
     * @return A Group instance
Jeff Burke's avatar
Jeff Burke committed
     * 
     * @throws GroupNotFoundException If the group was not found.
     * @throws TransientException  If an temporary, unexpected problem occurred.
Jeff Burke's avatar
Jeff Burke committed
     */
Jeff Burke's avatar
Jeff Burke committed
        throws GroupNotFoundException, TransientException,
               AccessControlException
    private Group getGroup(String groupID, boolean withMembers)
Jeff Burke's avatar
Jeff Burke committed
        throws GroupNotFoundException, TransientException, 
               AccessControlException
Jeff Burke's avatar
Jeff Burke committed
            Filter filter = Filter.createANDFilter(
                    Filter.createEqualityFilter("cn", groupID),
                    Filter.createNOTFilter(
                        Filter.createEqualityFilter("nsaccountlock", "TRUE")));
Jeff Burke's avatar
Jeff Burke committed
            SearchRequest searchRequest =  new SearchRequest(
                    config.getGroupsDN(), SearchScope.SUB, 
Jeff Burke's avatar
Jeff Burke committed
                    filter, new String[] {"entrydn", "cn", "description", 
                                          "owner", "uniquemember", "aci", 
                                          "modifytimestamp"});
Jeff Burke's avatar
Jeff Burke committed

            searchRequest.addControl(
                    new ProxiedAuthorizationV2RequestControl("dn:" + 
                            getSubjectDN().toNormalizedString()));

            SearchResultEntry group = 
                    getConnection().searchForEntry(searchRequest);
            if (group == null)
            {
                String msg = "Group not found " + groupID;
                logger.debug(msg);
                throw new GroupNotFoundException(groupID);
            }
            String groupCN = group.getAttributeValue("cn");
            DN groupOwner = group.getAttributeValueAsDN("owner");
Jeff Burke's avatar
Jeff Burke committed
            Date lastModified = 
                group.getAttributeValueAsDate("modifytimestamp");
            
            User<X500Principal> owner;
Jeff Burke's avatar
Jeff Burke committed
                owner = userPersist.getMember(groupOwner);
            }
            catch (UserNotFoundException e)
            {
                throw new RuntimeException("BUG: group owner not found");
            }
            Group ldapGroup = new Group(groupCN, owner);
            ldapGroup.description = group.getAttributeValue("description");
            ldapGroup.lastModified = lastModified;

            if (withMembers)
            {
                if (group.getAttributeValues("uniquemember") != null)
                {
Jeff Burke's avatar
Jeff Burke committed
                    for (String member : group
                            .getAttributeValues("uniquemember"))
                    {
                        DN memberDN = new DN(member);
Jeff Burke's avatar
Jeff Burke committed
                        if (memberDN.isDescendantOf(config.getUsersDN(), false))
                                user = userPersist.getMember(memberDN, false);
                            }
                            catch (UserNotFoundException e)
                            {
Jeff Burke's avatar
Jeff Burke committed
                                throw new RuntimeException(
                                    "BUG: group member not found");
Jeff Burke's avatar
Jeff Burke committed
                        else if (memberDN.isDescendantOf(config.getGroupsDN(),
                                                         false))
                            ldapGroup.getGroupMembers().add(new Group(memberDN.getRDNString().replace("cn=", "")));
Jeff Burke's avatar
Jeff Burke committed
                            throw new RuntimeException(
                                "BUG: unknown member DN type: " + memberDN);
Jeff Burke's avatar
Jeff Burke committed
                // TODO not sure this is going to fly...
                if (group.getAttributeValues("aci") != null)
                {
                    for (String aci : group.getAttributeValues("aci"))
                    {
                        if (aci.contains("Group Read"))
                        {
Jeff Burke's avatar
Jeff Burke committed
                            // TODO it's gotta be a better way to do this.
                            String grRead = aci.substring(
                                    aci.indexOf("ldap:///"));
                            grRead = grRead.substring(grRead.indexOf("cn=") + 3,
                                                      grRead.indexOf(','));
                            Group groupRead = new Group(grRead.trim());
                            ldapGroup.groupRead = groupRead;
                        }
                        else if (aci.contains("Group Write"))
                        {
Jeff Burke's avatar
Jeff Burke committed
                            // TODO it's gotta be a better way to do this.
                            String grWrite = aci.substring(
                                    aci.indexOf("ldap:///"));
                            grWrite = grWrite.substring(grWrite.indexOf("cn=") + 3, 
                                                    grWrite.indexOf(','));
                            Group groupWrite = getGroup(grWrite.trim());
                            ldapGroup.groupWrite = groupWrite;
                        }
Jeff Burke's avatar
Jeff Burke committed
                        else if (aci.equals(PUB_GROUP_ACI))
                        {
                            ldapGroup.publicRead = true;
                        }
                    }
                }
            }
            return ldapGroup;
        }
        catch (LDAPException e1)
        {
Jeff Burke's avatar
Jeff Burke committed
            // TODO check which LDAP exceptions are transient and which
            // ones are
            // access control
            throw new TransientException("Error getting the group", e1);
        }
    }

Jeff Burke's avatar
Jeff Burke committed
    /**
     * Modify the given group.
     *
     * @param group The group to update.
     * 
     * @return The newly updated group.
     * 
     * @throws GroupNotFoundException If the group was not found.
     * @throws TransientException If an temporary, unexpected problem occurred.
     * @throws AccessControlException If the operation is not permitted.
     * @throws UserNotFoundException If owner or group members not valid users.
     */
    public Group modifyGroup(Group group)
Jeff Burke's avatar
Jeff Burke committed
        throws GroupNotFoundException, TransientException,
               AccessControlException, UserNotFoundException
Jeff Burke's avatar
Jeff Burke committed
        // check if group exists
        Group oldGroup = getGroup(group.getID());
Jeff Burke's avatar
Jeff Burke committed
        
        return modifyGroup(group, oldGroup, null);
    }
    
    private Group modifyGroup(Group newGroup, Group oldGroup,
                             List<Modification> modifications)
        throws UserNotFoundException, TransientException,
               AccessControlException
    {
        if (!newGroup.getProperties().isEmpty())
Jeff Burke's avatar
Jeff Burke committed
            throw new UnsupportedOperationException(
                    "Support for groups properties not available");
Jeff Burke's avatar
Jeff Burke committed
        List<Modification> modifs = new ArrayList<Modification>();
Jeff Burke's avatar
Jeff Burke committed
        if (modifications != null)
        {
            modifs.addAll(modifications);
        }

        if (newGroup.description == null && oldGroup.description != null)
Jeff Burke's avatar
Jeff Burke committed
            modifs.add(new Modification(ModificationType.DELETE, 
                                        "description"));
        else if (newGroup.description != null && oldGroup.description == null)
        {
            modifs.add(new Modification(ModificationType.ADD, "description", 
                                        newGroup.description));
        }
        else if (newGroup.description != null && oldGroup.description != null)
Jeff Burke's avatar
Jeff Burke committed
            modifs.add(new Modification(ModificationType.REPLACE, "description", 
Jeff Burke's avatar
Jeff Burke committed
                                        newGroup.description));
Jeff Burke's avatar
Jeff Burke committed
        List<String> acis = new ArrayList<String>();
Jeff Burke's avatar
Jeff Burke committed
        if (newGroup.groupRead != null)
Jeff Burke's avatar
Jeff Burke committed
            if (newGroup.groupRead.equals(newGroup))
Jeff Burke's avatar
Jeff Burke committed
                throw new IllegalArgumentException(
                        "cyclical reference from groupRead to group");
Jeff Burke's avatar
Jeff Burke committed
            DN readGrDN = getGroupDN(newGroup.groupRead.getID());
Jeff Burke's avatar
Jeff Burke committed
            acis.add(GROUP_READ_ACI.replace(ACTUAL_GROUP_TOKEN, 
                                            readGrDN.toNormalizedString()));
Jeff Burke's avatar
Jeff Burke committed
        if (newGroup.groupWrite != null)
Jeff Burke's avatar
Jeff Burke committed
            if (newGroup.groupWrite.equals(newGroup))
Jeff Burke's avatar
Jeff Burke committed
                throw new IllegalArgumentException(
                        "cyclical reference from groupWrite to group");
Jeff Burke's avatar
Jeff Burke committed
            DN writeGrDN = getGroupDN(newGroup.groupWrite.getID());
Jeff Burke's avatar
Jeff Burke committed
            acis.add(GROUP_WRITE_ACI.replace(ACTUAL_GROUP_TOKEN, 
                                             writeGrDN.toNormalizedString()));
Jeff Burke's avatar
Jeff Burke committed
        if (newGroup.publicRead)
Jeff Burke's avatar
Jeff Burke committed
            acis.add(PUB_GROUP_ACI);
Jeff Burke's avatar
Jeff Burke committed
        modifs.add(new Modification(ModificationType.REPLACE, "aci", (String[]) 
                                    acis.toArray(new String[acis.size()])));
Jeff Burke's avatar
Jeff Burke committed
        List<String> newMembers = new ArrayList<String>();
Jeff Burke's avatar
Jeff Burke committed
        for (User<?> member : newGroup.getUserMembers())
        {
            if (!oldGroup.getUserMembers().remove(member))
            {
                DN memberDN;
                try
                {
Jeff Burke's avatar
Jeff Burke committed
                    memberDN = userPersist.getUserDN(member);
                }
                catch (LDAPException e)
                {
Jeff Burke's avatar
Jeff Burke committed
                    throw new UserNotFoundException(
                            "User not found " + member.getUserID());
                }
                newMembers.add(memberDN.toNormalizedString());
            }
        }
Jeff Burke's avatar
Jeff Burke committed
        for (Group gr : newGroup.getGroupMembers())
Jeff Burke's avatar
Jeff Burke committed
            if (gr.equals(newGroup))
Jeff Burke's avatar
Jeff Burke committed
                throw new IllegalArgumentException(
                        "cyclical reference from group member to group");
            }

            if (!oldGroup.getGroupMembers().remove(gr))
            {
                DN grDN = getGroupDN(gr.getID());
                newMembers.add(grDN.toNormalizedString());
            }
        }
Jeff Burke's avatar
Jeff Burke committed
        if (!newMembers.isEmpty())
Jeff Burke's avatar
Jeff Burke committed
            modifs.add(new Modification(ModificationType.ADD, "uniquemember", 
                (String[]) newMembers.toArray(new String[newMembers.size()])));
Jeff Burke's avatar
Jeff Burke committed
        List<String> delMembers = new ArrayList<String>();
        for (User<?> member : oldGroup.getUserMembers())
        {
            DN memberDN;
            try
            {
                memberDN = this.userPersist.getUserDN(member);
            }
            catch (LDAPException e)
            {
Jeff Burke's avatar
Jeff Burke committed
                throw new UserNotFoundException(
                        "User not found " + member.getUserID());
            }
            delMembers.add(memberDN.toNormalizedString());
        }
        for (Group gr : oldGroup.getGroupMembers())
        {
            DN grDN = getGroupDN(gr.getID());
            delMembers.add(grDN.toNormalizedString());
        }
Jeff Burke's avatar
Jeff Burke committed
        if (!delMembers.isEmpty())
Jeff Burke's avatar
Jeff Burke committed
            modifs.add(new Modification(ModificationType.DELETE, "uniquemember",
                (String[]) delMembers.toArray(new String[delMembers.size()])));
Jeff Burke's avatar
Jeff Burke committed
        ModifyRequest modifyRequest = 
Jeff Burke's avatar
Jeff Burke committed
                new ModifyRequest(getGroupDN(newGroup.getID()), modifs);
Jeff Burke's avatar
Jeff Burke committed
            modifyRequest.addControl(
                    new ProxiedAuthorizationV2RequestControl(
                            "dn:" + getSubjectDN().toNormalizedString()));
            LDAPResult result = getConnection().modify(modifyRequest);
        }
        catch (LDAPException e1)
        {
            throw new RuntimeException("LDAP problem", e1);
        }
        try
        {
Jeff Burke's avatar
Jeff Burke committed
            return getGroup(newGroup.getID());
        }
        catch (GroupNotFoundException e)
        {
            throw new RuntimeException("BUG: modified group not found");
Jeff Burke's avatar
Jeff Burke committed
    /**
Jeff Burke's avatar
Jeff Burke committed
     * 
     * @param groupID The group to delete
     * 
     * @throws GroupNotFoundException If the group was not found.
     * @throws TransientException If an temporary, unexpected problem occurred.
Jeff Burke's avatar
Jeff Burke committed
     */
    public void deleteGroup(String groupID)
        throws GroupNotFoundException, TransientException,
               AccessControlException
        Group group = getGroup(groupID);
        List<Modification> modifs = new ArrayList<Modification>();
        modifs.add(new Modification(ModificationType.ADD, "nsaccountlock", "true"));
        
        if (group.description != null)
            modifs.add(new Modification(ModificationType.DELETE, "description"));
        }
        
        if (group.groupRead != null || 
            group.groupWrite != null || 
            group.publicRead)
        {
            modifs.add(new Modification(ModificationType.DELETE, "aci"));
        }
        
        if (!group.getGroupMembers().isEmpty() || 
            !group.getUserMembers().isEmpty())
        {
            modifs.add(new Modification(ModificationType.DELETE, "uniquemember"));
        ModifyRequest modifyRequest = 
                new ModifyRequest(getGroupDN(group.getID()), modifs);
        try
        {
            modifyRequest.addControl(
                    new ProxiedAuthorizationV2RequestControl(
                            "dn:" + getSubjectDN().toNormalizedString()));
            LDAPResult result = getConnection().modify(modifyRequest);
        }
        catch (LDAPException e1)
        {
            throw new RuntimeException("LDAP problem", e1);
        }
        
        try
        {
            getGroup(group.getID());
            throw new RuntimeException("BUG: group not deleted " + 
                                       group.getID());
        }
        catch (GroupNotFoundException ignore) {}
    
    /**
     * Obtain a Collection of Groups that fit the given query.
     * 
     * @param userID The userID.
     * @param role Role of the user, either owner, member, or read/write.
     * @param groupID The Group ID.
     * 
     * @return Collection of Groups
     *         matching GROUP_READ_ACI.replace(ACTUAL_GROUP_TOKEN,
     *         readGrDN.toNormalizedString()) the query, or empty
     *         Collection. Never null.
     * @throws TransientException  If an temporary, unexpected problem occurred.
     * @throws UserNotFoundException
     * @throws GroupNotFoundException
     */
    public Collection<Group> getGroups(T userID, Role role, String groupID)
        throws TransientException, AccessControlException,
               GroupNotFoundException, UserNotFoundException
        }
        catch (LDAPException e)
        {
            // TODO check which LDAP exceptions are transient and which
            // ones are
            // access control
            throw new TransientException("Error getting user", e);
        
        if (role == Role.OWNER)
        {
            return getOwnerGroups(user, userDN, groupID);
        }
        else if (role == Role.MEMBER)
        {
            return getMemberGroups(user, userDN, groupID);
        }
        else if (role == Role.RW)
        {
            return getRWGroups(user, userDN, groupID);
        }
        throw new IllegalArgumentException("Unknown role " + role);
Jeff Burke's avatar
Jeff Burke committed
    
    protected Collection<Group> getOwnerGroups(User<T> user, DN userDN,
                                               String groupID)
        throws TransientException, AccessControlException,
               GroupNotFoundException, UserNotFoundException
    {
        try
        {                           
            Filter filter = Filter.createEqualityFilter("owner", 
                                                        userDN.toString());
            if (groupID != null)
            {
                getGroup(groupID);
                filter = Filter.createANDFilter(filter, 
                                Filter.createEqualityFilter("cn", groupID));
            }
            
            SearchRequest searchRequest =  new SearchRequest(
                    config.getGroupsDN(), SearchScope.SUB, filter, 
                    new String[] {"cn", "description", "modifytimestamp"});
            
            searchRequest.addControl(
                    new ProxiedAuthorizationV2RequestControl("dn:" + 
                            getSubjectDN().toNormalizedString()));
            
            Collection<Group> groups = new ArrayList<Group>();
            SearchResult results = getConnection().search(searchRequest);
            for (SearchResultEntry result : results.getSearchEntries())
            {
                String groupName = result.getAttributeValue("cn");
                // Ignore existing illegal group names.
                try
                {
                    Group group = new Group(groupName, user);
                    group.description = result.getAttributeValue("description");
                    group.lastModified = 
                        result.getAttributeValueAsDate("modifytimestamp");
                    groups.add(group);
                }
                catch (IllegalArgumentException ignore) { }   
            }
            
            return groups; 
        }
        catch (LDAPException e1)
        {
            // TODO check which LDAP exceptions are transient and which
            // ones are
            // access control
            throw new TransientException("Error getting groups", e1);
        }
    }
    
    protected Collection<Group> getMemberGroups(User<T> user, DN userDN, 
                                                String groupID)
        throws TransientException, AccessControlException,
               GroupNotFoundException, UserNotFoundException
    {
        if (groupID != null)
        {
            Collection<Group> groups = new ArrayList<Group>();
            if (userPersist.isMember(user.getUserID(), groupID))
            {
                groups.add(getGroup(groupID));
            }
            return groups;
        }
        else
        {
            return userPersist.getUserGroups(user.getUserID());
        }
    }
    
    protected Collection<Group> getRWGroups(User<T> user, DN userDN,
                                            String groupID)
        throws TransientException, AccessControlException,
               GroupNotFoundException, UserNotFoundException
    {
        try
        {
            Collection<Group> queryGroups =  new ArrayList<Group>();
            if (groupID != null)
            {
                queryGroups.add(new Group(groupID, user));
            }
            else
            {
                // List of Groups the user belongs to.
                queryGroups.addAll(getMemberGroups(user, userDN, groupID));
            
                // List of Groups the user owns;
                queryGroups.addAll(getOwnerGroups(user, userDN, groupID));
            }
            
            System.out.println("# groups: " + queryGroups.size());
                    
            List<Filter> filters = new ArrayList<Filter>();
            for (Group member : queryGroups)
            {
                // Require both groupRead and groupWrite
                if (member.groupRead != null && member.groupWrite != null)
                {
                    DN groupRead = getGroupDN(member.groupRead.getID());
                    String groupReadAci = 
                        GROUP_READ_ACI.replace(ACTUAL_GROUP_TOKEN, 
                                           groupRead.toNormalizedString());
                    DN groupWrite = getGroupDN(member.groupRead.getID());
                    String groupWriteAci = 
                        GROUP_WRITE_ACI.replace(ACTUAL_GROUP_TOKEN, 
                                            groupWrite.toNormalizedString());
                    System.out.println(groupReadAci);
                    System.out.println(groupWriteAci);

                    Filter filter = Filter.createANDFilter(
                            Filter.createEqualityFilter("aci", groupReadAci),
                            Filter.createEqualityFilter("aci", groupWriteAci));
                    filters.add(filter);
                }
            }

            Collection<Group> groups = new ArrayList<Group>();
            if (filters.isEmpty())
            {
                return groups;
            }
            
            Filter filter = Filter.createORFilter(filters);
            SearchRequest searchRequest =  new SearchRequest(
                        config.getGroupsDN(), SearchScope.SUB, filter, 
                        new String[] {"cn", "owner", "description", 
                                      "modifytimestamp"});

            searchRequest.addControl(
                    new ProxiedAuthorizationV2RequestControl("dn:" + 
                            getSubjectDN().toNormalizedString()));
            
            SearchResult results = getConnection().search(searchRequest);
            for (SearchResultEntry result : results.getSearchEntries())
            {
                String groupName = result.getAttributeValue("cn");
                DN ownerDN = result.getAttributeValueAsDN("owner");
                User<X500Principal> owner = userPersist.getMember(ownerDN);
                
                // Ignore existing illegal group names.
                try
                {
                    Group group = new Group(groupName, owner);
                    group.description = result.getAttributeValue("description");
                    group.lastModified = 
                            result.getAttributeValueAsDate("modifytimestamp");
                    groups.add(group);
                }
                catch (IllegalArgumentException ignore) { }   
            }
            return groups;
        }
        catch (LDAPException e)
        {
            // TODO check which LDAP exceptions are transient and which
            // ones are
            // access control
            throw new TransientException("Error getting groups", e);
        }
    }
    
    protected Collection<Group> getRWGroups2(User<T> user, DN userDN, 
                                             String groupID)
        throws TransientException, AccessControlException,
               GroupNotFoundException, UserNotFoundException
    {
        try
        {
            Collection<Group> groups = new ArrayList<Group>();
            
            Collection<Group> queryGroups =  new ArrayList<Group>();
            if (groupID != null)
            {
                queryGroups.add(new Group(groupID, user));
            }
            else
            {
                // List of Groups the user belongs to.
                queryGroups.addAll(getMemberGroups(user, userDN, groupID));
            
                // List of Groups the user owns;
                queryGroups.addAll(getOwnerGroups(user, userDN, groupID));
            }
            
            for (Group member : queryGroups)
            {
                // Require both groupRead and groupWrite
                if (member.groupRead != null && member.groupWrite != null)
                {
                    DN groupRead = getGroupDN(member.groupRead.getID());
                    String groupReadAci = 
                            GROUP_READ_ACI.replace(ACTUAL_GROUP_TOKEN, 
                                            groupRead.toNormalizedString());
                    DN groupWrite = getGroupDN(member.groupWrite.getID());
                    String groupWriteAci = 
                            GROUP_WRITE_ACI.replace(ACTUAL_GROUP_TOKEN, 
                                            groupWrite.toNormalizedString());

                    Filter filter = Filter.createANDFilter(
                            Filter.createEqualityFilter("aci", groupReadAci),
                            Filter.createEqualityFilter("aci", groupWriteAci));

                    SearchRequest searchRequest = new SearchRequest(
                            config.getGroupsDN(), SearchScope.SUB, filter, 
                            new String[] {"cn", "owner", "description", 
                                          "modifytimestamp"});

                    searchRequest.addControl(
                            new ProxiedAuthorizationV2RequestControl("dn:" + 
                                    getSubjectDN().toNormalizedString()));

                    SearchResult results = getConnection().search(searchRequest);
                    for (SearchResultEntry result : results.getSearchEntries())
                    {
                        String groupName = result.getAttributeValue("cn");
                        DN ownerDN = result.getAttributeValueAsDN("owner");
                        User<X500Principal> owner = userPersist.getMember(ownerDN);

                        // Ignore existing illegal group names.
                        try
                        {
                            Group group = new Group(groupName, owner);
                            group.description = result.getAttributeValue("description");
                            group.lastModified = 
                                    result.getAttributeValueAsDate("modifytimestamp");
                            groups.add(group);
                        }
                        catch (IllegalArgumentException ignore) { } 
                    }
                }
            }
            return groups;
        }
        catch (LDAPException e1)
        {
            // TODO check which LDAP exceptions are transient and which
            // ones are
            // access control
            throw new TransientException("Error getting groups", e1);
        }
    }
    
    private Group getInactiveGroup(String groupID)
        throws UserNotFoundException, GroupNotFoundException, LDAPException
    {
        Filter filter = Filter.createANDFilter(
                Filter.createEqualityFilter("cn", groupID),
                Filter.createEqualityFilter("nsaccountlock", "true"));

        SearchRequest searchRequest =  new SearchRequest(
                config.getGroupsDN(), SearchScope.SUB, 
                filter, new String[] {"cn", "owner"});

        searchRequest.addControl(
                new ProxiedAuthorizationV2RequestControl("dn:" + 
                        getSubjectDN().toNormalizedString()));

        SearchResultEntry searchResult = 
                getConnection().searchForEntry(searchRequest);
        
        if (searchResult == null)
        {
            String msg = "Inactive Group not found " + groupID;
Jeff Burke's avatar
Jeff Burke committed
            logger.debug(msg);
            throw new GroupNotFoundException(msg);
        }