All Downloads are FREE. Search and download functionalities are using the official Maven repository.

se.litsec.opensaml.saml2.metadata.build.IdpEntityDescriptorBuilder Maven / Gradle / Ivy

There is a newer version: 1.4.5
Show newest version
/*
 * Copyright 2016-2018 Litsec AB
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package se.litsec.opensaml.saml2.metadata.build;

import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.Optional;

import org.opensaml.core.xml.io.MarshallingException;
import org.opensaml.core.xml.io.UnmarshallingException;
import org.opensaml.core.xml.util.XMLObjectSupport;
import org.opensaml.saml.common.xml.SAMLConstants;
import org.opensaml.saml.ext.saml2mdattr.EntityAttributes;
import org.opensaml.saml.saml2.core.Attribute;
import org.opensaml.saml.saml2.metadata.EntityDescriptor;
import org.opensaml.saml.saml2.metadata.IDPSSODescriptor;
import org.opensaml.saml.saml2.metadata.SSODescriptor;
import org.opensaml.saml.saml2.metadata.SingleSignOnService;

import net.shibboleth.utilities.java.support.xml.XMLParserException;
import se.litsec.opensaml.saml2.attribute.AttributeTemplate;
import se.litsec.opensaml.saml2.metadata.MetadataUtils;
import se.litsec.opensaml.utils.ObjectUtils;

/**
 * A builder for building an {@code md:EntityDescription} (metadata) object for an Identity Provider.
 * 
 * @author Martin Lindström ([email protected])
 */
public class IdpEntityDescriptorBuilder extends AbstractEntityDescriptorBuilder {

  /**
   * The attribute name for the assurance certification attribute stored as an attribute in the entity attributes
   * extension.
   */
  public static final String ASSURANCE_CERTIFICATION_ATTRIBUTE_NAME = "urn:oasis:names:tc:SAML:attribute:assurance-certification";

  /**
   * The attribute template for the assurance certification attribute stored as an attribute in the entity attributes
   * extension.
   */
  public static final AttributeTemplate ASSURANCE_CERTIFICATION_ATTRIBUTE_TEMPLATE = new AttributeTemplate(
    ASSURANCE_CERTIFICATION_ATTRIBUTE_NAME, null);

  /**
   * Constructor setting up the builder with no template. This means that the entire {@code EntityDescriptor} object is
   * created from data assigned using the builder.
   */
  public IdpEntityDescriptorBuilder() {
    super();
  }

  /**
   * Constructor setting up the builder with a template {@code EntityDescriptor} that is read from a resource. Users of
   * the bean may now change, add or delete, the elements and attributes of the template object using the assignment
   * methods of the builder.
   * 
   * @param resource
   *          the template resource
   * @throws IOException
   *           if the resource can not be read
   * @throws UnmarshallingException
   *           for unmarshalling errors
   * @throws XMLParserException
   *           for XML parsing errors
   */
  public IdpEntityDescriptorBuilder(InputStream resource) throws XMLParserException, UnmarshallingException, IOException {
    super(resource);
  }

  /**
   * Constructor setting up the builder with a template {@code EntityDescriptor}. Users of the bean may now change, add
   * or delete, the elements and attributes of the template object using the assignment methods of the builder.
   * 
   * @param template
   *          the template
   * @throws UnmarshallingException
   *           for unmarshalling errors
   * @throws MarshallingException
   *           for marshalling errors
   */
  public IdpEntityDescriptorBuilder(EntityDescriptor template) throws UnmarshallingException, MarshallingException {
    super(template);
  }

  /** {@inheritDoc} */
  @Override
  protected IdpEntityDescriptorBuilder getThis() {
    return this;
  }

  /** {@inheritDoc} */
  @Override
  protected SSODescriptor ssoDescriptor() {
    if (this.object().getIDPSSODescriptor(SAMLConstants.SAML20P_NS) == null) {
      IDPSSODescriptor d = ObjectUtils.createSamlObject(IDPSSODescriptor.class);
      d.addSupportedProtocol(SAMLConstants.SAML20P_NS);
      this.object().getRoleDescriptors().add(d);
    }
    return this.object().getIDPSSODescriptor(SAMLConstants.SAML20P_NS);
  }

  /** {@inheritDoc} */
  @Override
  protected boolean matchingSSODescriptorType(EntityDescriptor descriptor) {
    if (this.object().getRoleDescriptors().isEmpty()) {
      return true;
    }
    return this.object().getIDPSSODescriptor(SAMLConstants.SAML20P_NS) != null;
  }
  
  /**
   * Assigns the {@code WantAuthnRequestsSigned} attribute of the {@code md:IDPSSODescriptor} element.
   * 
   * @param b
   *          boolean (if {@code null}, the attribute is not set)
   * @return the builder
   */
  public IdpEntityDescriptorBuilder wantAuthnRequestsSigned(Boolean b) {
    ((IDPSSODescriptor) this.ssoDescriptor()).setWantAuthnRequestsSigned(b);
    return this;
  }

  /**
   * Adds a set of URIs to the assurance certification attribute
   * ({@code urn:oasis:names:tc:SAML:attribute:assurance-certification}) that is part of the
   * {@code mdattr:EntityAttributes} element that is part of the metadata extension element.
   * 

* The method does not update any of the other attributes that may exist in the entity attributes extension. *

* * @param uris * the assurance URI values that should be added * @return the builder * @see #entityAttributesExtension(List) */ public IdpEntityDescriptorBuilder assuranceCertificationUris(List uris) { Optional entityAttributes = MetadataUtils.getEntityAttributes(this.object()); if (!entityAttributes.isPresent()) { if (uris == null || uris.isEmpty()) { return this; } return this.entityAttributesExtension( Collections.singletonList(ASSURANCE_CERTIFICATION_ATTRIBUTE_TEMPLATE.createBuilder().value(uris).build())); } List attributeList = new ArrayList<>(); entityAttributes.get().getAttributes().stream() .filter(a -> !ASSURANCE_CERTIFICATION_ATTRIBUTE_NAME.equals(a.getName())) .forEach(attributeList::add); if (uris != null) { attributeList.add(ASSURANCE_CERTIFICATION_ATTRIBUTE_TEMPLATE.createBuilder().value(uris).build()); } return this.entityAttributesExtension(attributeList); } /** * @see #assuranceCertificationUris(List) * * @param uris * the assurance URI values that should be added * @return the builder * @see #entityAttributesExtension(List) */ public IdpEntityDescriptorBuilder assuranceCertificationUris(String... uris) { return this.assuranceCertificationUris(uris != null ? Arrays.asList(uris) : null); } /** * Adds {@code md:SingleSignOnService} elements to the {@code IDPSSODescriptor}. * * @param singleSignOnServices * single sign on service objects (cloned before assignment) * @return the builder */ public IdpEntityDescriptorBuilder singleSignOnServices(List singleSignOnServices) { IDPSSODescriptor idpDescriptor = (IDPSSODescriptor) this.ssoDescriptor(); idpDescriptor.getSingleSignOnServices().clear(); if (singleSignOnServices == null) { return this; } for (SingleSignOnService sso : singleSignOnServices) { try { idpDescriptor.getSingleSignOnServices().add(XMLObjectSupport.cloneXMLObject(sso)); } catch (MarshallingException | UnmarshallingException e) { throw new RuntimeException(e); } } return this; } /** * @see #singleLogoutServices(List) * * @param singleSignOnServices * single sign on service objects (cloned before assignment) * @return the builder */ public IdpEntityDescriptorBuilder singleSignOnServices(SingleSignOnService... singleSignOnServices) { return this.singleSignOnServices(singleSignOnServices != null ? Arrays.asList(singleSignOnServices) : null); } }




© 2015 - 2024 Weber Informatics LLC | Privacy Policy