Пример #1
0
 private boolean isSupportedNameIdFormat(String nameIdFormat) {
   if (nameIdFormat.equals(JBossSAMLURIConstants.NAMEID_FORMAT_EMAIL.get())
       || nameIdFormat.equals(JBossSAMLURIConstants.NAMEID_FORMAT_TRANSIENT.get())
       || nameIdFormat.equals(JBossSAMLURIConstants.NAMEID_FORMAT_PERSISTENT.get())
       || nameIdFormat.equals(JBossSAMLURIConstants.NAMEID_FORMAT_UNSPECIFIED.get())) {
     return true;
   }
   return false;
 }
Пример #2
0
 protected String getNameId(
     String nameIdFormat, ClientSessionModel clientSession, UserSessionModel userSession) {
   if (nameIdFormat.equals(JBossSAMLURIConstants.NAMEID_FORMAT_EMAIL.get())) {
     return userSession.getUser().getEmail();
   } else if (nameIdFormat.equals(JBossSAMLURIConstants.NAMEID_FORMAT_TRANSIENT.get())) {
     // "G-" stands for "generated" Add this for the slight possibility of collisions.
     return "G-" + UUID.randomUUID().toString();
   } else if (nameIdFormat.equals(JBossSAMLURIConstants.NAMEID_FORMAT_PERSISTENT.get())) {
     return getPersistentNameId(clientSession, userSession);
   } else if (nameIdFormat.equals(JBossSAMLURIConstants.NAMEID_FORMAT_UNSPECIFIED.get())) {
     // TODO: Support for persistent NameID (pseudo-random identifier persisted in user object)
     return userSession.getUser().getUsername();
   } else {
     return userSession.getUser().getUsername();
   }
 }
Пример #3
0
/**
 * @author <a href="mailto:[email protected]">Bill Burke</a>
 * @version $Revision: 1 $
 */
public class SamlProtocol implements LoginProtocol {
  protected static final Logger logger = Logger.getLogger(SamlProtocol.class);

  public static final String ATTRIBUTE_TRUE_VALUE = "true";
  public static final String ATTRIBUTE_FALSE_VALUE = "false";
  public static final String SAML_ASSERTION_CONSUMER_URL_POST_ATTRIBUTE =
      "saml_assertion_consumer_url_post";
  public static final String SAML_ASSERTION_CONSUMER_URL_REDIRECT_ATTRIBUTE =
      "saml_assertion_consumer_url_redirect";
  public static final String SAML_SINGLE_LOGOUT_SERVICE_URL_POST_ATTRIBUTE =
      "saml_single_logout_service_url_post";
  public static final String SAML_SINGLE_LOGOUT_SERVICE_URL_REDIRECT_ATTRIBUTE =
      "saml_single_logout_service_url_redirect";
  public static final String LOGIN_PROTOCOL = "saml";
  public static final String SAML_BINDING = "saml_binding";
  public static final String SAML_IDP_INITIATED_LOGIN = "******";
  public static final String SAML_POST_BINDING = "post";
  public static final String SAML_SOAP_BINDING = "soap";
  public static final String SAML_REDIRECT_BINDING = "get";
  public static final String SAML_REQUEST_ID = "SAML_REQUEST_ID";
  public static final String SAML_LOGOUT_BINDING = "saml.logout.binding";
  public static final String SAML_LOGOUT_REQUEST_ID = "SAML_LOGOUT_REQUEST_ID";
  public static final String SAML_LOGOUT_RELAY_STATE = "SAML_LOGOUT_RELAY_STATE";
  public static final String SAML_LOGOUT_CANONICALIZATION = "SAML_LOGOUT_CANONICALIZATION";
  public static final String SAML_LOGOUT_BINDING_URI = "SAML_LOGOUT_BINDING_URI";
  public static final String SAML_LOGOUT_SIGNATURE_ALGORITHM = "saml.logout.signature.algorithm";
  public static final String SAML_NAME_ID = "SAML_NAME_ID";
  public static final String SAML_NAME_ID_FORMAT = "SAML_NAME_ID_FORMAT";
  public static final String SAML_DEFAULT_NAMEID_FORMAT =
      JBossSAMLURIConstants.NAMEID_FORMAT_UNSPECIFIED.get();
  public static final String SAML_PERSISTENT_NAME_ID_FOR = "saml.persistent.name.id.for";
  public static final String SAML_IDP_INITIATED_SSO_RELAY_STATE =
      "saml_idp_initiated_sso_relay_state";
  public static final String SAML_IDP_INITIATED_SSO_URL_NAME = "saml_idp_initiated_sso_url_name";

  protected KeycloakSession session;

  protected RealmModel realm;

  protected UriInfo uriInfo;

  protected HttpHeaders headers;

  protected EventBuilder event;

  @Override
  public SamlProtocol setSession(KeycloakSession session) {
    this.session = session;
    return this;
  }

  @Override
  public SamlProtocol setRealm(RealmModel realm) {
    this.realm = realm;
    return this;
  }

  @Override
  public SamlProtocol setUriInfo(UriInfo uriInfo) {
    this.uriInfo = uriInfo;
    return this;
  }

  @Override
  public SamlProtocol setHttpHeaders(HttpHeaders headers) {
    this.headers = headers;
    return this;
  }

  @Override
  public SamlProtocol setEventBuilder(EventBuilder event) {
    this.event = event;
    return this;
  }

  @Override
  public Response sendError(ClientSessionModel clientSession, Error error) {
    try {
      if ("true".equals(clientSession.getClient().getAttribute(SAML_IDP_INITIATED_LOGIN))) {
        if (error == Error.CANCELLED_BY_USER) {
          UriBuilder builder =
              RealmsResource.protocolUrl(uriInfo).path(SamlService.class, "idpInitiatedSSO");
          Map<String, String> params = new HashMap<>();
          params.put("realm", realm.getName());
          params.put("protocol", LOGIN_PROTOCOL);
          params.put(
              "client", clientSession.getClient().getAttribute(SAML_IDP_INITIATED_SSO_URL_NAME));
          URI redirect = builder.buildFromMap(params);
          return Response.status(302).location(redirect).build();
        } else {
          return ErrorPage.error(session, translateErrorToIdpInitiatedErrorMessage(error));
        }
      } else {
        SAML2ErrorResponseBuilder builder =
            new SAML2ErrorResponseBuilder()
                .destination(clientSession.getRedirectUri())
                .issuer(getResponseIssuer(realm))
                .status(translateErrorToSAMLStatus(error).get());
        try {
          JaxrsSAML2BindingBuilder binding =
              new JaxrsSAML2BindingBuilder()
                  .relayState(clientSession.getNote(GeneralConstants.RELAY_STATE));
          Document document = builder.buildDocument();
          return buildErrorResponse(clientSession, binding, document);
        } catch (Exception e) {
          return ErrorPage.error(session, Messages.FAILED_TO_PROCESS_RESPONSE);
        }
      }
    } finally {
      RestartLoginCookie.expireRestartCookie(realm, session.getContext().getConnection(), uriInfo);
      session.sessions().removeClientSession(realm, clientSession);
    }
  }

  protected Response buildErrorResponse(
      ClientSessionModel clientSession, JaxrsSAML2BindingBuilder binding, Document document)
      throws ConfigurationException, ProcessingException, IOException {
    if (isPostBinding(clientSession)) {
      return binding.postBinding(document).response(clientSession.getRedirectUri());
    } else {
      return binding.redirectBinding(document).response(clientSession.getRedirectUri());
    }
  }

  private JBossSAMLURIConstants translateErrorToSAMLStatus(Error error) {
    switch (error) {
      case CANCELLED_BY_USER:
      case CONSENT_DENIED:
        return JBossSAMLURIConstants.STATUS_REQUEST_DENIED;
      case PASSIVE_INTERACTION_REQUIRED:
      case PASSIVE_LOGIN_REQUIRED:
        return JBossSAMLURIConstants.STATUS_NO_PASSIVE;
      default:
        logger.warn(
            "Untranslated protocol Error: " + error.name() + " so we return default SAML error");
        return JBossSAMLURIConstants.STATUS_REQUEST_DENIED;
    }
  }

  private String translateErrorToIdpInitiatedErrorMessage(Error error) {
    switch (error) {
      case CONSENT_DENIED:
        return Messages.CONSENT_DENIED;
      case PASSIVE_INTERACTION_REQUIRED:
      case PASSIVE_LOGIN_REQUIRED:
        return Messages.UNEXPECTED_ERROR_HANDLING_REQUEST;
      default:
        logger.warn(
            "Untranslated protocol Error: " + error.name() + " so we return default error message");
        return Messages.UNEXPECTED_ERROR_HANDLING_REQUEST;
    }
  }

  protected String getResponseIssuer(RealmModel realm) {
    return RealmsResource.realmBaseUrl(uriInfo).build(realm.getName()).toString();
  }

  protected boolean isPostBinding(ClientSessionModel clientSession) {
    ClientModel client = clientSession.getClient();
    SamlClient samlClient = new SamlClient(client);
    return SamlProtocol.SAML_POST_BINDING.equals(clientSession.getNote(SamlProtocol.SAML_BINDING))
        || samlClient.forcePostBinding();
  }

  public static boolean isLogoutPostBindingForInitiator(UserSessionModel session) {
    String note = session.getNote(SamlProtocol.SAML_LOGOUT_BINDING);
    return SamlProtocol.SAML_POST_BINDING.equals(note);
  }

  protected boolean isLogoutPostBindingForClient(ClientSessionModel clientSession) {
    ClientModel client = clientSession.getClient();
    SamlClient samlClient = new SamlClient(client);
    String logoutPostUrl = client.getAttribute(SAML_SINGLE_LOGOUT_SERVICE_URL_POST_ATTRIBUTE);
    String logoutRedirectUrl =
        client.getAttribute(SAML_SINGLE_LOGOUT_SERVICE_URL_REDIRECT_ATTRIBUTE);

    if (logoutPostUrl == null) {
      // if we don't have a redirect uri either, return true and default to the admin url + POST
      // binding
      if (logoutRedirectUrl == null) return true;
      return false;
    }

    if (samlClient.forcePostBinding()) {
      return true; // configured to force a post binding and post binding logout url is not null
    }

    String bindingType = clientSession.getNote(SAML_BINDING);

    // if the login binding was POST, return true
    if (SAML_POST_BINDING.equals(bindingType)) return true;

    if (logoutRedirectUrl == null)
      return true; // we don't have a redirect binding url, so use post binding

    return false; // redirect binding
  }

  protected String getNameIdFormat(SamlClient samlClient, ClientSessionModel clientSession) {
    String nameIdFormat = clientSession.getNote(GeneralConstants.NAMEID_FORMAT);

    boolean forceFormat = samlClient.forceNameIDFormat();
    String configuredNameIdFormat = samlClient.getNameIDFormat();
    if ((nameIdFormat == null || forceFormat) && configuredNameIdFormat != null) {
      nameIdFormat = configuredNameIdFormat;
    }
    if (nameIdFormat == null) return SAML_DEFAULT_NAMEID_FORMAT;
    return nameIdFormat;
  }

  protected String getNameId(
      String nameIdFormat, ClientSessionModel clientSession, UserSessionModel userSession) {
    if (nameIdFormat.equals(JBossSAMLURIConstants.NAMEID_FORMAT_EMAIL.get())) {
      return userSession.getUser().getEmail();
    } else if (nameIdFormat.equals(JBossSAMLURIConstants.NAMEID_FORMAT_TRANSIENT.get())) {
      // "G-" stands for "generated" Add this for the slight possibility of collisions.
      return "G-" + UUID.randomUUID().toString();
    } else if (nameIdFormat.equals(JBossSAMLURIConstants.NAMEID_FORMAT_PERSISTENT.get())) {
      return getPersistentNameId(clientSession, userSession);
    } else if (nameIdFormat.equals(JBossSAMLURIConstants.NAMEID_FORMAT_UNSPECIFIED.get())) {
      // TODO: Support for persistent NameID (pseudo-random identifier persisted in user object)
      return userSession.getUser().getUsername();
    } else {
      return userSession.getUser().getUsername();
    }
  }

  /**
   * Attempts to retrieve the persistent type NameId as follows:
   *
   * <ol>
   *   <li>saml.persistent.name.id.for.$clientId user attribute
   *   <li>saml.persistent.name.id.for.* user attribute
   *   <li>G-$randomUuid
   * </ol>
   *
   * If a randomUuid is generated, an attribute for the given saml.persistent.name.id.for.$clientId
   * will be generated, otherwise no state change will occur with respect to the user's attributes.
   *
   * @return the user's persistent NameId
   */
  protected String getPersistentNameId(
      final ClientSessionModel clientSession, final UserSessionModel userSession) {
    // attempt to retrieve the UserID for the client-specific attribute
    final UserModel user = userSession.getUser();
    final String clientNameId =
        String.format(
            "%s.%s", SAML_PERSISTENT_NAME_ID_FOR, clientSession.getClient().getClientId());
    String samlPersistentNameId = user.getFirstAttribute(clientNameId);
    if (samlPersistentNameId != null) {
      return samlPersistentNameId;
    }

    // check for a wildcard attribute
    final String wildcardNameId = String.format("%s.*", SAML_PERSISTENT_NAME_ID_FOR);
    samlPersistentNameId = user.getFirstAttribute(wildcardNameId);
    if (samlPersistentNameId != null) {
      return samlPersistentNameId;
    }

    // default to generated.  "G-" stands for "generated"
    samlPersistentNameId = "G-" + UUID.randomUUID().toString();
    user.setSingleAttribute(clientNameId, samlPersistentNameId);
    return samlPersistentNameId;
  }

  @Override
  public Response authenticated(UserSessionModel userSession, ClientSessionCode accessCode) {
    ClientSessionModel clientSession = accessCode.getClientSession();
    ClientModel client = clientSession.getClient();
    SamlClient samlClient = new SamlClient(client);
    String requestID = clientSession.getNote(SAML_REQUEST_ID);
    String relayState = clientSession.getNote(GeneralConstants.RELAY_STATE);
    String redirectUri = clientSession.getRedirectUri();
    String responseIssuer = getResponseIssuer(realm);
    String nameIdFormat = getNameIdFormat(samlClient, clientSession);
    String nameId = getNameId(nameIdFormat, clientSession, userSession);

    // save NAME_ID and format in clientSession as they may be persistent or transient or email and
    // not username
    // we'll need to send this back on a logout
    clientSession.setNote(SAML_NAME_ID, nameId);
    clientSession.setNote(SAML_NAME_ID_FORMAT, nameIdFormat);

    SAML2LoginResponseBuilder builder = new SAML2LoginResponseBuilder();
    builder
        .requestID(requestID)
        .destination(redirectUri)
        .issuer(responseIssuer)
        .assertionExpiration(realm.getAccessCodeLifespan())
        .subjectExpiration(realm.getAccessTokenLifespan())
        .sessionIndex(clientSession.getId())
        .requestIssuer(clientSession.getClient().getClientId())
        .nameIdentifier(nameIdFormat, nameId)
        .authMethod(JBossSAMLURIConstants.AC_UNSPECIFIED.get());
    if (!samlClient.includeAuthnStatement()) {
      builder.disableAuthnStatement(true);
    }

    List<ProtocolMapperProcessor<SAMLAttributeStatementMapper>> attributeStatementMappers =
        new LinkedList<>();
    List<ProtocolMapperProcessor<SAMLLoginResponseMapper>> loginResponseMappers =
        new LinkedList<>();
    ProtocolMapperProcessor<SAMLRoleListMapper> roleListMapper = null;

    Set<ProtocolMapperModel> mappings = accessCode.getRequestedProtocolMappers();
    for (ProtocolMapperModel mapping : mappings) {

      ProtocolMapper mapper =
          (ProtocolMapper)
              session
                  .getKeycloakSessionFactory()
                  .getProviderFactory(ProtocolMapper.class, mapping.getProtocolMapper());
      if (mapper == null) continue;
      if (mapper instanceof SAMLAttributeStatementMapper) {
        attributeStatementMappers.add(
            new ProtocolMapperProcessor<SAMLAttributeStatementMapper>(
                (SAMLAttributeStatementMapper) mapper, mapping));
      }
      if (mapper instanceof SAMLLoginResponseMapper) {
        loginResponseMappers.add(
            new ProtocolMapperProcessor<SAMLLoginResponseMapper>(
                (SAMLLoginResponseMapper) mapper, mapping));
      }
      if (mapper instanceof SAMLRoleListMapper) {
        roleListMapper =
            new ProtocolMapperProcessor<SAMLRoleListMapper>((SAMLRoleListMapper) mapper, mapping);
      }
    }

    Document samlDocument = null;
    try {
      ResponseType samlModel = builder.buildModel();
      final AttributeStatementType attributeStatement =
          populateAttributeStatements(
              attributeStatementMappers, session, userSession, clientSession);
      populateRoles(roleListMapper, session, userSession, clientSession, attributeStatement);

      // SAML Spec 2.7.3 AttributeStatement must contain one or more Attribute or EncryptedAttribute
      if (attributeStatement.getAttributes().size() > 0) {
        AssertionType assertion = samlModel.getAssertions().get(0).getAssertion();
        assertion.addStatement(attributeStatement);
      }

      samlModel =
          transformLoginResponse(
              loginResponseMappers, samlModel, session, userSession, clientSession);
      samlDocument = builder.buildDocument(samlModel);
    } catch (Exception e) {
      logger.error("failed", e);
      return ErrorPage.error(session, Messages.FAILED_TO_PROCESS_RESPONSE);
    }

    JaxrsSAML2BindingBuilder bindingBuilder = new JaxrsSAML2BindingBuilder();
    bindingBuilder.relayState(relayState);

    KeyManager keyManager = session.keys();
    KeyManager.ActiveKey keys = keyManager.getActiveKey(realm);

    if (samlClient.requiresRealmSignature()) {
      String canonicalization = samlClient.getCanonicalizationMethod();
      if (canonicalization != null) {
        bindingBuilder.canonicalizationMethod(canonicalization);
      }
      bindingBuilder
          .signatureAlgorithm(samlClient.getSignatureAlgorithm())
          .signWith(keys.getPrivateKey(), keys.getPublicKey(), keys.getCertificate())
          .signDocument();
    }
    if (samlClient.requiresAssertionSignature()) {
      String canonicalization = samlClient.getCanonicalizationMethod();
      if (canonicalization != null) {
        bindingBuilder.canonicalizationMethod(canonicalization);
      }
      bindingBuilder
          .signatureAlgorithm(samlClient.getSignatureAlgorithm())
          .signWith(keys.getPrivateKey(), keys.getPublicKey(), keys.getCertificate())
          .signAssertions();
    }
    if (samlClient.requiresEncryption()) {
      PublicKey publicKey = null;
      try {
        publicKey = SamlProtocolUtils.getEncryptionValidationKey(client);
      } catch (Exception e) {
        logger.error("failed", e);
        return ErrorPage.error(session, Messages.FAILED_TO_PROCESS_RESPONSE);
      }
      bindingBuilder.encrypt(publicKey);
    }
    try {
      return buildAuthenticatedResponse(clientSession, redirectUri, samlDocument, bindingBuilder);
    } catch (Exception e) {
      logger.error("failed", e);
      return ErrorPage.error(session, Messages.FAILED_TO_PROCESS_RESPONSE);
    }
  }

  protected Response buildAuthenticatedResponse(
      ClientSessionModel clientSession,
      String redirectUri,
      Document samlDocument,
      JaxrsSAML2BindingBuilder bindingBuilder)
      throws ConfigurationException, ProcessingException, IOException {
    if (isPostBinding(clientSession)) {
      return bindingBuilder.postBinding(samlDocument).response(redirectUri);
    } else {
      return bindingBuilder.redirectBinding(samlDocument).response(redirectUri);
    }
  }

  public static class ProtocolMapperProcessor<T> {
    public final T mapper;
    public final ProtocolMapperModel model;

    public ProtocolMapperProcessor(T mapper, ProtocolMapperModel model) {
      this.mapper = mapper;
      this.model = model;
    }
  }

  public AttributeStatementType populateAttributeStatements(
      List<ProtocolMapperProcessor<SAMLAttributeStatementMapper>> attributeStatementMappers,
      KeycloakSession session,
      UserSessionModel userSession,
      ClientSessionModel clientSession) {
    AttributeStatementType attributeStatement = new AttributeStatementType();
    for (ProtocolMapperProcessor<SAMLAttributeStatementMapper> processor :
        attributeStatementMappers) {
      processor.mapper.transformAttributeStatement(
          attributeStatement, processor.model, session, userSession, clientSession);
    }

    return attributeStatement;
  }

  public ResponseType transformLoginResponse(
      List<ProtocolMapperProcessor<SAMLLoginResponseMapper>> mappers,
      ResponseType response,
      KeycloakSession session,
      UserSessionModel userSession,
      ClientSessionModel clientSession) {
    for (ProtocolMapperProcessor<SAMLLoginResponseMapper> processor : mappers) {
      response =
          processor.mapper.transformLoginResponse(
              response, processor.model, session, userSession, clientSession);
    }
    return response;
  }

  public void populateRoles(
      ProtocolMapperProcessor<SAMLRoleListMapper> roleListMapper,
      KeycloakSession session,
      UserSessionModel userSession,
      ClientSessionModel clientSession,
      final AttributeStatementType existingAttributeStatement) {
    if (roleListMapper == null) return;
    roleListMapper.mapper.mapRoles(
        existingAttributeStatement, roleListMapper.model, session, userSession, clientSession);
  }

  public static String getLogoutServiceUrl(
      UriInfo uriInfo, ClientModel client, String bindingType) {
    String logoutServiceUrl = null;
    if (SAML_POST_BINDING.equals(bindingType)) {
      logoutServiceUrl = client.getAttribute(SAML_SINGLE_LOGOUT_SERVICE_URL_POST_ATTRIBUTE);
    } else {
      logoutServiceUrl = client.getAttribute(SAML_SINGLE_LOGOUT_SERVICE_URL_REDIRECT_ATTRIBUTE);
    }
    if (logoutServiceUrl == null && client instanceof ClientModel)
      logoutServiceUrl = ((ClientModel) client).getManagementUrl();
    if (logoutServiceUrl == null || logoutServiceUrl.trim().equals("")) return null;
    return ResourceAdminManager.resolveUri(
        uriInfo.getRequestUri(), client.getRootUrl(), logoutServiceUrl);
  }

  @Override
  public Response frontchannelLogout(
      UserSessionModel userSession, ClientSessionModel clientSession) {
    ClientModel client = clientSession.getClient();
    SamlClient samlClient = new SamlClient(client);
    if (!(client instanceof ClientModel)) return null;
    try {
      if (isLogoutPostBindingForClient(clientSession)) {
        String bindingUri = getLogoutServiceUrl(uriInfo, client, SAML_POST_BINDING);
        SAML2LogoutRequestBuilder logoutBuilder =
            createLogoutRequest(bindingUri, clientSession, client);
        JaxrsSAML2BindingBuilder binding = createBindingBuilder(samlClient);
        return binding.postBinding(logoutBuilder.buildDocument()).request(bindingUri);
      } else {
        logger.debug("frontchannel redirect binding");
        String bindingUri = getLogoutServiceUrl(uriInfo, client, SAML_REDIRECT_BINDING);
        SAML2LogoutRequestBuilder logoutBuilder =
            createLogoutRequest(bindingUri, clientSession, client);
        JaxrsSAML2BindingBuilder binding = createBindingBuilder(samlClient);
        return binding.redirectBinding(logoutBuilder.buildDocument()).request(bindingUri);
      }
    } catch (ConfigurationException e) {
      throw new RuntimeException(e);
    } catch (ProcessingException e) {
      throw new RuntimeException(e);
    } catch (IOException e) {
      throw new RuntimeException(e);
    } catch (ParsingException e) {
      throw new RuntimeException(e);
    }
  }

  @Override
  public Response finishLogout(UserSessionModel userSession) {
    logger.debug("finishLogout");
    String logoutBindingUri = userSession.getNote(SAML_LOGOUT_BINDING_URI);
    if (logoutBindingUri == null) {
      logger.error(
          "Can't finish SAML logout as there is no logout binding set.  Please configure the logout service url in the admin console for your client applications.");
      return ErrorPage.error(session, Messages.FAILED_LOGOUT);
    }
    String logoutRelayState = userSession.getNote(SAML_LOGOUT_RELAY_STATE);
    SAML2LogoutResponseBuilder builder = new SAML2LogoutResponseBuilder();
    builder.logoutRequestID(userSession.getNote(SAML_LOGOUT_REQUEST_ID));
    builder.destination(logoutBindingUri);
    builder.issuer(getResponseIssuer(realm));
    JaxrsSAML2BindingBuilder binding = new JaxrsSAML2BindingBuilder();
    binding.relayState(logoutRelayState);
    String signingAlgorithm = userSession.getNote(SAML_LOGOUT_SIGNATURE_ALGORITHM);
    if (signingAlgorithm != null) {
      SignatureAlgorithm algorithm = SignatureAlgorithm.valueOf(signingAlgorithm);
      String canonicalization = userSession.getNote(SAML_LOGOUT_CANONICALIZATION);
      if (canonicalization != null) {
        binding.canonicalizationMethod(canonicalization);
      }
      KeyManager.ActiveKey keys = session.keys().getActiveKey(realm);
      binding
          .signatureAlgorithm(algorithm)
          .signWith(keys.getPrivateKey(), keys.getPublicKey(), keys.getCertificate())
          .signDocument();
    }

    try {
      return buildLogoutResponse(userSession, logoutBindingUri, builder, binding);
    } catch (ConfigurationException e) {
      throw new RuntimeException(e);
    } catch (ProcessingException e) {
      throw new RuntimeException(e);
    } catch (IOException e) {
      throw new RuntimeException(e);
    }
  }

  protected Response buildLogoutResponse(
      UserSessionModel userSession,
      String logoutBindingUri,
      SAML2LogoutResponseBuilder builder,
      JaxrsSAML2BindingBuilder binding)
      throws ConfigurationException, ProcessingException, IOException {
    if (isLogoutPostBindingForInitiator(userSession)) {
      return binding.postBinding(builder.buildDocument()).response(logoutBindingUri);
    } else {
      return binding.redirectBinding(builder.buildDocument()).response(logoutBindingUri);
    }
  }

  @Override
  public void backchannelLogout(UserSessionModel userSession, ClientSessionModel clientSession) {
    ClientModel client = clientSession.getClient();
    SamlClient samlClient = new SamlClient(client);
    String logoutUrl = getLogoutServiceUrl(uriInfo, client, SAML_POST_BINDING);
    if (logoutUrl == null) {
      logger.warnv(
          "Can't do backchannel logout. No SingleLogoutService POST Binding registered for client: {1}",
          client.getClientId());
      return;
    }
    SAML2LogoutRequestBuilder logoutBuilder = createLogoutRequest(logoutUrl, clientSession, client);

    String logoutRequestString = null;
    try {
      JaxrsSAML2BindingBuilder binding = createBindingBuilder(samlClient);
      logoutRequestString = binding.postBinding(logoutBuilder.buildDocument()).encoded();
    } catch (Exception e) {
      logger.warn("failed to send saml logout", e);
      return;
    }

    HttpClient httpClient = session.getProvider(HttpClientProvider.class).getHttpClient();
    for (int i = 0; i < 2; i++) { // follow redirects once
      try {
        List<NameValuePair> formparams = new ArrayList<NameValuePair>();
        formparams.add(
            new BasicNameValuePair(GeneralConstants.SAML_REQUEST_KEY, logoutRequestString));
        formparams.add(
            new BasicNameValuePair("BACK_CHANNEL_LOGOUT", "BACK_CHANNEL_LOGOUT")); // for Picketlink
        // todo remove
        // this
        UrlEncodedFormEntity form = new UrlEncodedFormEntity(formparams, "UTF-8");
        HttpPost post = new HttpPost(logoutUrl);
        post.setEntity(form);
        HttpResponse response = httpClient.execute(post);
        try {
          int status = response.getStatusLine().getStatusCode();
          if (status == 302 && !logoutUrl.endsWith("/")) {
            String redirect = response.getFirstHeader(HttpHeaders.LOCATION).getValue();
            String withSlash = logoutUrl + "/";
            if (withSlash.equals(redirect)) {
              logoutUrl = withSlash;
              continue;
            }
          }
        } finally {
          HttpEntity entity = response.getEntity();
          if (entity != null) {
            InputStream is = entity.getContent();
            if (is != null) is.close();
          }
        }
      } catch (IOException e) {
        logger.warn("failed to send saml logout", e);
      }
      break;
    }
  }

  protected SAML2LogoutRequestBuilder createLogoutRequest(
      String logoutUrl, ClientSessionModel clientSession, ClientModel client) {
    // build userPrincipal with subject used at login
    SAML2LogoutRequestBuilder logoutBuilder =
        new SAML2LogoutRequestBuilder()
            .assertionExpiration(realm.getAccessCodeLifespan())
            .issuer(getResponseIssuer(realm))
            .sessionIndex(clientSession.getId())
            .userPrincipal(
                clientSession.getNote(SAML_NAME_ID), clientSession.getNote(SAML_NAME_ID_FORMAT))
            .destination(logoutUrl);
    return logoutBuilder;
  }

  @Override
  public boolean requireReauthentication(
      UserSessionModel userSession, ClientSessionModel clientSession) {
    // Not yet supported
    return false;
  }

  private JaxrsSAML2BindingBuilder createBindingBuilder(SamlClient samlClient) {
    JaxrsSAML2BindingBuilder binding = new JaxrsSAML2BindingBuilder();
    if (samlClient.requiresRealmSignature()) {
      KeyManager.ActiveKey keys = session.keys().getActiveKey(realm);
      binding
          .signatureAlgorithm(samlClient.getSignatureAlgorithm())
          .signWith(keys.getPrivateKey(), keys.getPublicKey(), keys.getCertificate())
          .signDocument();
    }
    return binding;
  }

  @Override
  public void close() {}
}
  protected AuthOutcome handleLoginResponse(
      ResponseType responseType, OnSessionCreated onCreateSession) {

    AssertionType assertion = null;
    try {
      assertion = AssertionUtil.getAssertion(responseType, deployment.getDecryptionKey());
      if (AssertionUtil.hasExpired(assertion)) {
        return initiateLogin();
      }
    } catch (Exception e) {
      log.error("Error extracting SAML assertion: " + e.getMessage());
      challenge =
          new AuthChallenge() {
            @Override
            public boolean challenge(HttpFacade exchange) {
              SamlAuthenticationError error =
                  new SamlAuthenticationError(SamlAuthenticationError.Reason.EXTRACTION_FAILURE);
              exchange.getRequest().setError(error);
              exchange.getResponse().sendError(403);
              return true;
            }

            @Override
            public int getResponseCode() {
              return 403;
            }
          };
    }

    SubjectType subject = assertion.getSubject();
    SubjectType.STSubType subType = subject.getSubType();
    NameIDType subjectNameID = (NameIDType) subType.getBaseID();
    String principalName = subjectNameID.getValue();

    final Set<String> roles = new HashSet<>();
    MultivaluedHashMap<String, String> attributes = new MultivaluedHashMap<>();
    MultivaluedHashMap<String, String> friendlyAttributes = new MultivaluedHashMap<>();

    Set<StatementAbstractType> statements = assertion.getStatements();
    for (StatementAbstractType statement : statements) {
      if (statement instanceof AttributeStatementType) {
        AttributeStatementType attributeStatement = (AttributeStatementType) statement;
        List<AttributeStatementType.ASTChoiceType> attList = attributeStatement.getAttributes();
        for (AttributeStatementType.ASTChoiceType obj : attList) {
          AttributeType attr = obj.getAttribute();
          if (isRole(attr)) {
            List<Object> attributeValues = attr.getAttributeValue();
            if (attributeValues != null) {
              for (Object attrValue : attributeValues) {
                String role = getAttributeValue(attrValue);
                log.debugv("Add role: {0}", role);
                roles.add(role);
              }
            }
          } else {
            List<Object> attributeValues = attr.getAttributeValue();
            if (attributeValues != null) {
              for (Object attrValue : attributeValues) {
                String value = getAttributeValue(attrValue);
                if (attr.getName() != null) {
                  attributes.add(attr.getName(), value);
                }
                if (attr.getFriendlyName() != null) {
                  friendlyAttributes.add(attr.getFriendlyName(), value);
                }
              }
            }
          }
        }
      }
    }
    if (deployment.getPrincipalNamePolicy() == SamlDeployment.PrincipalNamePolicy.FROM_ATTRIBUTE) {
      if (deployment.getPrincipalAttributeName() != null) {
        String attribute = attributes.getFirst(deployment.getPrincipalAttributeName());
        if (attribute != null) principalName = attribute;
        else {
          attribute = friendlyAttributes.getFirst(deployment.getPrincipalAttributeName());
          if (attribute != null) principalName = attribute;
        }
      }
    }

    AuthnStatementType authn = null;
    for (Object statement : assertion.getStatements()) {
      if (statement instanceof AuthnStatementType) {
        authn = (AuthnStatementType) statement;
        break;
      }
    }

    URI nameFormat = subjectNameID.getFormat();
    String nameFormatString =
        nameFormat == null
            ? JBossSAMLURIConstants.NAMEID_FORMAT_UNSPECIFIED.get()
            : nameFormat.toString();
    final SamlPrincipal principal =
        new SamlPrincipal(
            assertion,
            principalName,
            principalName,
            nameFormatString,
            attributes,
            friendlyAttributes);
    String index = authn == null ? null : authn.getSessionIndex();
    final String sessionIndex = index;
    SamlSession account = new SamlSession(principal, roles, sessionIndex);
    sessionStore.saveAccount(account);
    onCreateSession.onSessionCreated(account);

    // redirect to original request, it will be restored
    String redirectUri = sessionStore.getRedirectUri();
    if (redirectUri != null) {
      facade.getResponse().setHeader("Location", redirectUri);
      facade.getResponse().setStatus(302);
      facade.getResponse().end();
    } else {
      log.debug("IDP initiated invocation");
    }
    log.debug("AUTHENTICATED authn");

    return AuthOutcome.AUTHENTICATED;
  }