--- /dev/null
+/*
+ * The Shibboleth License, Version 1. Copyright (c) 2002 University Corporation for Advanced Internet Development, Inc.
+ * All rights reserved Redistribution and use in source and binary forms, with or without modification, are permitted
+ * provided that the following conditions are met: Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer. Redistributions in binary form must reproduce the above
+ * copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials
+ * provided with the distribution, if any, must include the following acknowledgment: "This product includes software
+ * developed by the University Corporation for Advanced Internet Development <http://www.ucaid.edu> Internet2 Project.
+ * Alternately, this acknowledegement may appear in the software itself, if and wherever such third-party
+ * acknowledgments normally appear. Neither the name of Shibboleth nor the names of its contributors, nor Internet2, nor
+ * the University Corporation for Advanced Internet Development, Inc., nor UCAID may be used to endorse or promote
+ * products derived from this software without specific prior written permission. For written permission, please contact
+ * shibboleth@shibboleth.org Products derived from this software may not be called Shibboleth, Internet2, UCAID, or the
+ * University Corporation for Advanced Internet Development, nor may Shibboleth appear in their name, without prior
+ * written permission of the University Corporation for Advanced Internet Development. THIS SOFTWARE IS PROVIDED BY THE
+ * COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND WITH ALL FAULTS. ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, AND NON-INFRINGEMENT ARE
+ * DISCLAIMED AND THE ENTIRE RISK OF SATISFACTORY QUALITY, PERFORMANCE, ACCURACY, AND EFFORT IS WITH LICENSEE. IN NO
+ * EVENT SHALL THE COPYRIGHT OWNER, CONTRIBUTORS OR THE UNIVERSITY CORPORATION FOR ADVANCED INTERNET DEVELOPMENT, INC.
+ * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+ * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
+ * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
+ * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+package edu.internet2.middleware.shibboleth.common;
+
+import java.security.GeneralSecurityException;
+import java.security.cert.CertPath;
+import java.security.cert.CertPathValidator;
+import java.security.cert.CertPathValidatorException;
+import java.security.cert.CertificateFactory;
+import java.security.cert.CertificateParsingException;
+import java.security.cert.PKIXCertPathValidatorResult;
+import java.security.cert.PKIXParameters;
+import java.security.cert.X509Certificate;
+import java.util.Arrays;
+import java.util.Collection;
+import java.util.HashSet;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Set;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+import javax.security.auth.x500.X500Principal;
+
+import org.apache.log4j.Logger;
+import org.apache.xml.security.exceptions.XMLSecurityException;
+import org.apache.xml.security.keys.KeyInfo;
+import org.apache.xml.security.keys.content.KeyName;
+import org.apache.xml.security.keys.keyresolver.KeyResolverException;
+
+import edu.internet2.middleware.shibboleth.metadata.EntitiesDescriptor;
+import edu.internet2.middleware.shibboleth.metadata.EntityDescriptor;
+import edu.internet2.middleware.shibboleth.metadata.ExtendedEntitiesDescriptor;
+import edu.internet2.middleware.shibboleth.metadata.ExtendedEntityDescriptor;
+import edu.internet2.middleware.shibboleth.metadata.KeyAuthority;
+import edu.internet2.middleware.shibboleth.metadata.KeyDescriptor;
+import edu.internet2.middleware.shibboleth.metadata.RoleDescriptor;
+
+/**
+ * @author Walter Hoehn
+ */
+public class ShibbolethTrust extends Trust {
+
+ private static Logger log = Logger.getLogger(ShibbolethTrust.class.getName());
+ private static Pattern regex = Pattern.compile(".*?CN=([^,/]+).*");
+
+ public boolean validate(RoleDescriptor descriptor, X509Certificate[] certificateChain, int keyUse) {
+
+ // If we can successfully validate with an inline key, that's fine
+ boolean defaultValidation = super.validate(descriptor, certificateChain, keyUse);
+ if (defaultValidation == true) { return true; }
+
+ // Make sure we have the data we need
+ if (descriptor == null || certificateChain == null || certificateChain.length < 1) {
+ log.error("Appropriate data was not supplied for trust evaluation.");
+ return false;
+ }
+
+ // If not, try PKIX validation against the shib-custom metadata extensions
+
+ // First, we want to see if we can match a keyName from the metadata against the cert
+ // Iterator through all the keys in the metadata
+ Iterator keyDescriptors = descriptor.getKeyDescriptors();
+ while (keyDescriptors.hasNext()) {
+ // Look for a key descriptor with the right usage bits
+ KeyDescriptor keyDescriptor = (KeyDescriptor) keyDescriptors.next();
+ if (keyDescriptor.getUse() != KeyDescriptor.UNSPECIFIED && keyDescriptor.getUse() != keyUse) {
+ log.debug("Role contains a key descriptor, but the usage specification is not valid for this action.");
+ continue;
+ }
+
+ // We found one, see if we can match the metadata's keyName against the cert
+ KeyInfo keyInfo = keyDescriptor.getKeyInfo();
+ if (keyInfo.containsKeyName()) {
+ for (int i = 0; i < keyInfo.lengthKeyName(); i++) {
+ try {
+ if (matchKeyName(certificateChain[0], keyInfo.itemKeyName(i))) {
+ // If we find a match, try to do path validation against any key authorities we might have
+ // in the metadata
+ if (pkixValidate(certificateChain, descriptor.getEntityDescriptor())) { return true; }
+ }
+ } catch (XMLSecurityException e) {
+ log.error("Problem retrieving key name from metadata: " + e);
+ }
+ }
+ }
+ }
+ return false;
+ }
+
+ private boolean pkixValidate(X509Certificate[] certChain, EntityDescriptor entity) {
+
+ if (entity instanceof ExtendedEntityDescriptor) {
+ Iterator keyAuthorities = ((ExtendedEntityDescriptor) entity).getKeyAuthorities();
+ // if we have any key authorities, construct a flat list of trust anchors representing each and attempt to
+ // validate against them in turn
+ while (keyAuthorities.hasNext()) {
+ if (pkixValidate(certChain, (KeyAuthority) keyAuthorities.next())) { return true; }
+ }
+ }
+
+ // We couldn't do path validation based on metadata attached to the entity, we now need to walk up the chain of
+ // nested entities and attempt to validate at each group level
+ EntitiesDescriptor group = entity.getEntitiesDescriptor();
+ if (group != null) {
+ if (pkixValidate(certChain, group)) { return true; }
+ }
+
+ // We've walked the entire metadata chain with no success, so fail
+ return false;
+ }
+
+ private boolean pkixValidate(X509Certificate[] certChain, EntitiesDescriptor group) {
+
+ if (group instanceof ExtendedEntitiesDescriptor) {
+ Iterator keyAuthorities = ((ExtendedEntitiesDescriptor) group).getKeyAuthorities();
+ // if we have any key authorities, construct a flat list of trust anchors representing each and attempt to
+ // validate against them in turn
+ while (keyAuthorities.hasNext()) {
+ if (pkixValidate(certChain, (KeyAuthority) keyAuthorities.next())) { return true; }
+ }
+ }
+
+ // If not, attempt to walk up the chain for validation
+ EntitiesDescriptor parent = group.getEntitiesDescriptor();
+ if (group != null) {
+ if (pkixValidate(certChain, group)) { return true; }
+ }
+
+ return false;
+ }
+
+ private boolean pkixValidate(X509Certificate[] certChain, KeyAuthority authority) {
+
+ Set anchors = new HashSet();
+ Iterator keyInfos = authority.getKeyInfos();
+ while (keyInfos.hasNext()) {
+ KeyInfo keyInfo = (KeyInfo) keyInfos.next();
+ if (keyInfo.containsX509Data()) {
+ try {
+ anchors.add(keyInfo.getX509Certificate());
+ } catch (KeyResolverException e) {
+ log.error("Encountered an error constructing trust list from shibboleth metadata: " + e);
+ }
+ }
+ }
+
+ // alright, if we were able to create a trust list, attempt a pkix validation against the list
+ if (anchors.size() > 0) {
+ log.debug("Constructed a trust list from key authority. Attempting path validation...");
+ try {
+ CertPath path = CertificateFactory.getInstance("X.509").generateCertPath(Arrays.asList(certChain));
+ CertPathValidator validator = CertPathValidator.getInstance("PKIX");
+ PKIXParameters params = new PKIXParameters(anchors);
+ PKIXCertPathValidatorResult result = (PKIXCertPathValidatorResult) validator.validate(path, params);
+
+ System.err.println(result.getPolicyTree().getDepth());
+ // TODO honor verify depth
+ log.debug("Path successfully validated.");
+ return true;
+
+ } catch (CertPathValidatorException e) {
+ log.debug("Path failed to validate.");
+ } catch (GeneralSecurityException e) {
+ log.error("Encountered an error during validation: " + e);
+ }
+ }
+ return false;
+ }
+
+ private static boolean matchKeyName(X509Certificate certificate, KeyName keyName) {
+
+ // First, try to match DN against metadata
+ try {
+ if (certificate.getSubjectX500Principal().getName(X500Principal.RFC2253).equals(
+ new X500Principal(keyName.getKeyName()).getName(X500Principal.RFC2253))) {
+ log.debug("Matched against DN.");
+ return true;
+ }
+ } catch (IllegalArgumentException iae) {
+ // squelch this runtime exception, since
+ // this might be a valid case
+ }
+
+ // If that doesn't work, we try matching against
+ // some Subject Alt Names
+ try {
+ Collection altNames = certificate.getSubjectAlternativeNames();
+ if (altNames != null) {
+ for (Iterator nameIterator = altNames.iterator(); nameIterator.hasNext();) {
+ List altName = (List) nameIterator.next();
+ if (altName.get(0).equals(new Integer(2)) || altName.get(0).equals(new Integer(6))) {
+ // 2 is DNS, 6 is URI
+ if (altName.get(0).equals(keyName.getKeyName())) {
+ log.debug("Matched against SubjectAltName.");
+ return true;
+ }
+ }
+ }
+ }
+ } catch (CertificateParsingException e1) {
+ log.error("Encountered an problem trying to extract Subject Alternate "
+ + "Name from supplied certificate: " + e1);
+ }
+
+ // If that doesn't work, try to match using
+ // SSL-style hostname matching
+ if (getHostNameFromDN(certificate.getSubjectX500Principal()).equals(keyName.getKeyName())) {
+ log.debug("Matched against hostname.");
+ return true;
+ }
+
+ return false;
+ }
+
+ private static String getHostNameFromDN(X500Principal dn) {
+
+ Matcher matches = regex.matcher(dn.getName(X500Principal.RFC2253));
+ if (!matches.find() || matches.groupCount() > 1) {
+ log.error("Unable to extract host name name from certificate subject DN.");
+ return null;
+ }
+ return matches.group(1);
+ }
+
+}