src/edu/internet2/middleware/shibboleth/aa/attrresolv/provider/SimpleAttributeDefinition.java

   1 /*
   2  * The Shibboleth License, Version 1.
   3  * Copyright (c) 2002
   4  * University Corporation for Advanced Internet Development, Inc.
   5  * All rights reserved
   6  *
   7  *
   8  * Redistribution and use in source and binary forms, with or without
   9  * modification, are permitted provided that the following conditions are met:
  10  *
  11  * Redistributions of source code must retain the above copyright notice, this
  12  * list of conditions and the following disclaimer.
  13  *
  14  * Redistributions in binary form must reproduce the above copyright notice,
  15  * this list of conditions and the following disclaimer in the documentation
  16  * and/or other materials provided with the distribution, if any, must include
  17  * the following acknowledgment: "This product includes software developed by
  18  * the University Corporation for Advanced Internet Development
  19  * <http://www.ucaid.edu>Internet2 Project. Alternately, this acknowledegement
  20  * may appear in the software itself, if and wherever such third-party
  21  * acknowledgments normally appear.
  22  *
  23  * Neither the name of Shibboleth nor the names of its contributors, nor
  24  * Internet2, nor the University Corporation for Advanced Internet Development,
  25  * Inc., nor UCAID may be used to endorse or promote products derived from this
  26  * software without specific prior written permission. For written permission,
  27  * please contact shibboleth@shibboleth.org
  28  *
  29  * Products derived from this software may not be called Shibboleth, Internet2,
  30  * UCAID, or the University Corporation for Advanced Internet Development, nor
  31  * may Shibboleth appear in their name, without prior written permission of the
  32  * University Corporation for Advanced Internet Development.
  33  *
  34  *
  35  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  36  * AND WITH ALL FAULTS. ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  37  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY, FITNESS FOR A
  38  * PARTICULAR PURPOSE, AND NON-INFRINGEMENT ARE DISCLAIMED AND THE ENTIRE RISK
  39  * OF SATISFACTORY QUALITY, PERFORMANCE, ACCURACY, AND EFFORT IS WITH LICENSEE.
  40  * IN NO EVENT SHALL THE COPYRIGHT OWNER, CONTRIBUTORS OR THE UNIVERSITY
  41  * CORPORATION FOR ADVANCED INTERNET DEVELOPMENT, INC. BE LIABLE FOR ANY DIRECT,
  42  * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  43  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  44  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  45  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  46  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  47  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  48  */
  49
  50 package edu.internet2.middleware.shibboleth.aa.attrresolv.provider;
  51
  52 import java.security.Principal;
  53 import java.util.Arrays;
  54 import java.util.HashSet;
  55 import java.util.Iterator;
  56 import java.util.Set;
  57
  58 import javax.naming.NamingEnumeration;
  59 import javax.naming.NamingException;
  60 import javax.naming.directory.Attribute;
  61 import javax.naming.directory.Attributes;
  62
  63 import org.apache.log4j.Logger;
  64 import org.w3c.dom.Element;
  65
  66 import edu.internet2.middleware.shibboleth.aa.attrresolv.AttributeDefinitionPlugIn;
  67 import edu.internet2.middleware.shibboleth.aa.attrresolv.Dependencies;
  68 import edu.internet2.middleware.shibboleth.aa.attrresolv.ResolutionPlugInException;
  69 import edu.internet2.middleware.shibboleth.aa.attrresolv.ResolverAttribute;
  70
  71 /**
  72  * Basic <code>AttributeDefinitionPlugIn</code> implementation.  Operates as a proxy for attributes
  73  * gathered by Connectors.
  74  *
  75  * @author Walter Hoehn (wassa@columbia.edu)
  76  */
  77 public class SimpleAttributeDefinition extends BaseAttributeDefinition implements AttributeDefinitionPlugIn {
  78
  79         private static Logger log = Logger.getLogger(SimpleAttributeDefinition.class.getName());
  80         private String connectorMapping;
  81         private String smartScope;
  82         private String schemaType;
  83         private String schemaNamespace;
  84         private ValueHandler valueHandler;
  85
  86         /**
  87          * Constructor for SimpleAttributeDefinition.  Creates a PlugIn based on configuration
  88          * information presented in a DOM Element.
  89          */
  90         public SimpleAttributeDefinition(Element e) throws ResolutionPlugInException {
  91
  92                 super(e);
  93
  94                 String sourceName = e.getAttribute("sourceName");
  95                 if (sourceName == null || sourceName.equals("")) {
  96                         int index = getId().lastIndexOf("#");
  97                         if (index < 0) {
  98                                 index = getId().lastIndexOf(":");
  99                                 int slashIndex = getId().lastIndexOf("/");
 100                                 if (slashIndex > index) {
 101                                         index = slashIndex;
 102                                 }
 103                         }
 104                         connectorMapping = getId().substring(index + 1);
 105                 } else {
 106                         connectorMapping = sourceName;
 107                 }
 108
 109                 log.debug("Mapping attribute to name (" + connectorMapping + ") in connector.");
 110
 111                 String smartScopingSpec = e.getAttribute("smartScope");
 112                 if (smartScopingSpec != null && !smartScopingSpec.equals("")) {
 113                         smartScope = smartScopingSpec;
 114                 }
 115                 if (smartScope != null) {
 116                         log.debug("Smart Scope (" + smartScope + ") enabled for attribute (" + getId() + ").");
 117                 } else {
 118                         log.debug("Smart Scoping disabled for attribute (" + getId() + ").");
 119                 }
 120
 121                 String valueHandlerSpec = e.getAttribute("valueHandler");
 122
 123                 if (valueHandlerSpec != null && !valueHandlerSpec.equals("")) {
 124                         if (smartScope == null) {
 125                                 try {
 126                                         Class handlerClass = Class.forName(valueHandlerSpec);
 127                                         valueHandler = (ValueHandler) handlerClass.newInstance();
 128                                 } catch (ClassNotFoundException cnfe) {
 129                                         log.error(
 130                                                 "Value Handler implementation specified for attribute ("
 131                                                         + getId()
 132                                                         + ") cannot be found: "
 133                                                         + cnfe);
 134                                         throw new ResolutionPlugInException(
 135                                                 "Value Handler implementation specified for attribute (" + getId() + ") cannot be found.");
 136                                 } catch (Exception oe) {
 137                                         log.error(
 138                                                 "Value Handler implementation specified for attribute ("
 139                                                         + getId()
 140                                                         + ") coudl not be loaded: "
 141                                                         + oe);
 142                                         throw new ResolutionPlugInException(
 143                                                 "Value Handler implementation specified for attribute (" + getId() + ") could not be loaded.");
 144                                 }
 145                         } else {
 146                                 log.error(
 147                                         "Specification of \"valueHandler\' cannot be used in combination with \"smartScope\".  Ignoring Value Handler for attribute ("
 148                                                 + getId()
 149                                                 + ").");
 150                         }
 151                 }
 152
 153                 if (valueHandler != null) {
 154                         log.debug("Custom Value Handler enabled for attribute (" + getId() + ").");
 155                 }
 156
 157         }
 158
 159         /**
 160          * @see edu.internet2.middleware.shibboleth.aa.attrresolv.AttributeDefinitionPlugIn#resolve(edu.internet2.middleware.shibboleth.aa.attrresolv.ArpAttribute, java.security.Principal, java.lang.String, edu.internet2.middleware.shibboleth.aa.attrresolv.Dependencies)
 161          */
 162         public void resolve(ResolverAttribute attribute, Principal principal, String requester, Dependencies depends)
 163                 throws ResolutionPlugInException {
 164                 log.debug("Resolving attribute: (" + getId() + ")");
 165                 Set results = new HashSet();
 166                 if (!connectorDependencyIds.isEmpty()) {
 167                         results.addAll(Arrays.asList(getValuesFromConnectors(depends)));
 168                 }
 169
 170                 if (!attributeDependencyIds.isEmpty()) {
 171                         results.addAll(Arrays.asList(getValuesFromAttributes(depends)));
 172                 }
 173
 174                 if (lifeTime != -1) {
 175                         attribute.setLifetime(lifeTime);
 176                 }
 177
 178                 Iterator resultsIt = results.iterator();
 179                 while (resultsIt.hasNext()) {
 180                         if (smartScope != null) {
 181                                 attribute.registerValueHandler(new ScopedStringValueHandler(smartScope));
 182                         }
 183                         if (smartScope == null && valueHandler != null) {
 184                                 attribute.registerValueHandler(valueHandler);
 185                         }
 186                         attribute.addValue(resultsIt.next());
 187                 }
 188                 attribute.setResolved();
 189         }
 190
 191         protected Object[] getValuesFromAttributes(Dependencies depends) {
 192
 193                 Set results = new HashSet();
 194
 195                 Iterator attrDependIt = attributeDependencyIds.iterator();
 196                 while (attrDependIt.hasNext()) {
 197                         ResolverAttribute attribute = depends.getAttributeResolution((String) attrDependIt.next());
 198                         if (attribute != null) {
 199                                 log.debug("Found value(s) for attribute (" + getId() + ").");
 200                                 for (Iterator iterator = attribute.getValues(); iterator.hasNext();) {
 201                                         results.add(iterator.next());
 202                                 }
 203                         } else {
 204                                 log.error(
 205                                         "An attribute dependency of attribute (" + getId() + ") was not included in the dependency chain.");
 206                         }
 207                 }
 208
 209                 if (results.isEmpty()) {
 210                         log.debug("An attribute dependency of attribute (" + getId() + ") supplied no values.");
 211                 }
 212                 return results.toArray();
 213         }
 214
 215         protected Object[] getValuesFromConnectors(Dependencies depends) {
 216
 217                 Set results = new HashSet();
 218
 219                 Iterator connectorDependIt = connectorDependencyIds.iterator();
 220                 while (connectorDependIt.hasNext()) {
 221                         Attributes attrs = depends.getConnectorResolution((String) connectorDependIt.next());
 222                         if (attrs != null) {
 223                                 Attribute attr = attrs.get(connectorMapping);
 224                                 if (attr != null) {
 225                                         log.debug("Found value(s) for attribute (" + getId() + ").");
 226                                         try {
 227                                                 NamingEnumeration valuesEnum = attr.getAll();
 228                                                 while (valuesEnum.hasMore()) {
 229                                                         results.add(valuesEnum.next());
 230                                                 }
 231                                         } catch (NamingException e) {
 232                                                 log.error(
 233                                                         "An problem was encountered resolving the dependencies of attribute ("
 234                                                                 + getId()
 235                                                                 + "): "
 236                                                                 + e);
 237                                         }
 238                                 }
 239                         }
 240                 }
 241
 242                 if (results.isEmpty()) {
 243                         log.debug("A connector dependency of attribute (" + getId() + ") supplied no values.");
 244                 }
 245                 return results.toArray();
 246         }
 247 }