You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@cxf.apache.org by co...@apache.org on 2014/07/25 11:23:53 UTC
[1/2] Adding some WS-Federation request creation unit tests
Repository: cxf-fediz
Updated Branches:
refs/heads/master 87e87a056 -> 4a0b86496
http://git-wip-us.apache.org/repos/asf/cxf-fediz/blob/4a0b8649/plugins/core/src/test/java/org/apache/cxf/fediz/core/federation/FederationResponseTest.java
----------------------------------------------------------------------
diff --git a/plugins/core/src/test/java/org/apache/cxf/fediz/core/federation/FederationResponseTest.java b/plugins/core/src/test/java/org/apache/cxf/fediz/core/federation/FederationResponseTest.java
new file mode 100644
index 0000000..f194e11
--- /dev/null
+++ b/plugins/core/src/test/java/org/apache/cxf/fediz/core/federation/FederationResponseTest.java
@@ -0,0 +1,1351 @@
+/**
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements. See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership. The ASF licenses this file
+ * to you 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 org.apache.cxf.fediz.core.federation;
+
+import java.io.File;
+import java.io.IOException;
+import java.math.BigInteger;
+import java.net.URL;
+import java.security.cert.X509Certificate;
+import java.util.Collections;
+import java.util.List;
+
+import javax.security.auth.callback.CallbackHandler;
+import javax.security.auth.callback.UnsupportedCallbackException;
+
+import org.w3c.dom.Document;
+import org.w3c.dom.Element;
+import org.w3c.dom.Node;
+import org.apache.cxf.fediz.common.STSUtil;
+import org.apache.cxf.fediz.common.SecurityTestUtil;
+import org.apache.cxf.fediz.core.AbstractSAMLCallbackHandler;
+import org.apache.cxf.fediz.core.AbstractSAMLCallbackHandler.MultiValue;
+import org.apache.cxf.fediz.core.Claim;
+import org.apache.cxf.fediz.core.ClaimTypes;
+import org.apache.cxf.fediz.core.FederationConstants;
+import org.apache.cxf.fediz.core.KeystoreCallbackHandler;
+import org.apache.cxf.fediz.core.SAML1CallbackHandler;
+import org.apache.cxf.fediz.core.SAML2CallbackHandler;
+import org.apache.cxf.fediz.core.TokenValidator;
+import org.apache.cxf.fediz.core.config.FederationProtocol;
+import org.apache.cxf.fediz.core.config.FedizConfigurator;
+import org.apache.cxf.fediz.core.config.FedizContext;
+import org.apache.cxf.fediz.core.config.Protocol;
+import org.apache.cxf.fediz.core.config.jaxb.AudienceUris;
+import org.apache.cxf.fediz.core.config.jaxb.CertificateStores;
+import org.apache.cxf.fediz.core.config.jaxb.ContextConfig;
+import org.apache.cxf.fediz.core.config.jaxb.FederationProtocolType;
+import org.apache.cxf.fediz.core.config.jaxb.KeyStoreType;
+import org.apache.cxf.fediz.core.config.jaxb.TrustManagersType;
+import org.apache.cxf.fediz.core.config.jaxb.TrustedIssuerType;
+import org.apache.cxf.fediz.core.config.jaxb.TrustedIssuers;
+import org.apache.cxf.fediz.core.config.jaxb.ValidationType;
+import org.apache.cxf.fediz.core.exception.ProcessingException;
+import org.apache.cxf.fediz.core.exception.ProcessingException.TYPE;
+import org.apache.cxf.fediz.core.processor.FederationProcessorImpl;
+import org.apache.cxf.fediz.core.processor.FedizProcessor;
+import org.apache.cxf.fediz.core.processor.FedizRequest;
+import org.apache.cxf.fediz.core.processor.FedizResponse;
+import org.apache.wss4j.common.WSEncryptionPart;
+import org.apache.wss4j.common.crypto.Crypto;
+import org.apache.wss4j.common.crypto.CryptoFactory;
+import org.apache.wss4j.common.crypto.CryptoType;
+import org.apache.wss4j.common.ext.WSPasswordCallback;
+import org.apache.wss4j.common.ext.WSSecurityException;
+import org.apache.wss4j.common.saml.SAMLCallback;
+import org.apache.wss4j.common.saml.SAMLUtil;
+import org.apache.wss4j.common.saml.SamlAssertionWrapper;
+import org.apache.wss4j.common.saml.bean.AudienceRestrictionBean;
+import org.apache.wss4j.common.saml.bean.ConditionsBean;
+import org.apache.wss4j.common.saml.builder.SAML1Constants;
+import org.apache.wss4j.common.saml.builder.SAML2Constants;
+import org.apache.wss4j.common.util.DOM2Writer;
+import org.apache.wss4j.dom.WSConstants;
+import org.apache.wss4j.dom.message.WSSecEncrypt;
+import org.joda.time.DateTime;
+import org.junit.AfterClass;
+import org.junit.Assert;
+import org.junit.BeforeClass;
+
+import static org.junit.Assert.fail;
+
+/**
+ * Some tests for the WS-Federation "FederationProcessor".
+ */
+public class FederationResponseTest {
+ static final String TEST_USER = "alice";
+ static final String TEST_RSTR_ISSUER = "FedizSTSIssuer";
+ static final String TEST_AUDIENCE = "https://localhost/fedizhelloworld";
+
+ private static final String CONFIG_FILE = "fediz_test_config.xml";
+
+ private static Crypto crypto;
+ private static CallbackHandler cbPasswordHandler;
+ private static FedizConfigurator configurator;
+
+
+ @BeforeClass
+ public static void init() {
+ try {
+ crypto = CryptoFactory.getInstance("signature.properties");
+ cbPasswordHandler = new KeystoreCallbackHandler();
+ getFederationConfigurator();
+ } catch (Exception e) {
+ e.printStackTrace();
+ }
+ Assert.assertNotNull(configurator);
+
+ }
+
+ @AfterClass
+ public static void cleanup() {
+ SecurityTestUtil.cleanup();
+ }
+
+
+ private static FedizConfigurator getFederationConfigurator() {
+ if (configurator != null) {
+ return configurator;
+ }
+ try {
+ configurator = new FedizConfigurator();
+ final URL resource = Thread.currentThread().getContextClassLoader()
+ .getResource(CONFIG_FILE);
+ File f = new File(resource.toURI());
+ configurator.loadConfig(f);
+ return configurator;
+ } catch (Exception e) {
+ e.printStackTrace();
+ return null;
+ }
+ }
+
+
+ /**
+ * Validate RSTR without RequestedSecurityToken element
+ */
+ @org.junit.Test
+ public void validateRSTRWithoutToken() throws Exception {
+ Document doc = STSUtil.toSOAPPart(STSUtil.SAMPLE_RSTR_COLL_MSG);
+
+ FedizRequest wfReq = new FedizRequest();
+ wfReq.setAction(FederationConstants.ACTION_SIGNIN);
+ wfReq.setResponseToken(DOM2Writer.nodeToString(doc));
+
+ configurator = null;
+ FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
+
+ FedizProcessor wfProc = new FederationProcessorImpl();
+ try {
+ wfProc.processRequest(wfReq, config);
+ fail("Failure expected on missing security token in RSTR");
+ } catch (ProcessingException ex) {
+ if (!TYPE.BAD_REQUEST.equals(ex.getType())) {
+ fail("Expected ProcessingException with BAD_REQUEST type");
+ }
+ }
+ }
+
+ /**
+ * Validate FederationRequest with unknown action
+ */
+ @org.junit.Test
+ public void validateRequestUnknownAction() throws Exception {
+ Document doc = STSUtil.toSOAPPart(STSUtil.SAMPLE_RSTR_COLL_MSG);
+
+ FedizRequest wfReq = new FedizRequest();
+ wfReq.setAction("gugus");
+ wfReq.setResponseToken(DOM2Writer.nodeToString(doc));
+
+ configurator = null;
+ FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
+
+ FedizProcessor wfProc = new FederationProcessorImpl();
+ try {
+ wfProc.processRequest(wfReq, config);
+ fail("Failure expected due to invalid action");
+ } catch (ProcessingException ex) {
+ if (!TYPE.INVALID_REQUEST.equals(ex.getType())) {
+ fail("Expected ProcessingException with INVALID_REQUEST type");
+ }
+ }
+ }
+
+ /**
+ *Validate FederationRequest with invalid RSTR/wresult
+ */
+ @org.junit.Test
+ public void validateSignInInvalidWResult() throws Exception {
+ FedizRequest wfReq = new FedizRequest();
+ wfReq.setAction(FederationConstants.ACTION_SIGNIN);
+ wfReq.setResponseToken("gugus");
+
+ configurator = null;
+ FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
+
+ FedizProcessor wfProc = new FederationProcessorImpl();
+ try {
+ wfProc.processRequest(wfReq, config);
+ fail("Failure expected due to invalid wresult");
+ } catch (ProcessingException ex) {
+ if (!TYPE.INVALID_REQUEST.equals(ex.getType())) {
+ fail("Expected ProcessingException with INVALID_REQUEST type");
+ }
+ }
+ }
+
+ @org.junit.Test
+ public void validateTokenAndCreateMetadata() throws Exception {
+ validateSAML2Token();
+ FederationMetaDataTest other = new FederationMetaDataTest();
+ other.validateMetaDataWithAlias();
+ }
+
+ /**
+ * Validate SAML 2 token which includes the role attribute with 2 values
+ * Roles are encoded as a multi-value saml attribute
+ */
+ @org.junit.Test
+ public void validateSAML2Token() throws Exception {
+ SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
+ callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
+ callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
+ callbackHandler.setIssuer(TEST_RSTR_ISSUER);
+ callbackHandler.setSubjectName(TEST_USER);
+ ConditionsBean cp = new ConditionsBean();
+ AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
+ audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
+ cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
+ callbackHandler.setConditions(cp);
+
+ SAMLCallback samlCallback = new SAMLCallback();
+ SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
+ SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
+ String rstr = createSamlToken(assertion, "mystskey", true);
+
+ FedizRequest wfReq = new FedizRequest();
+ wfReq.setAction(FederationConstants.ACTION_SIGNIN);
+ wfReq.setResponseToken(rstr);
+
+ configurator = null;
+ FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
+
+ FedizProcessor wfProc = new FederationProcessorImpl();
+ FedizResponse wfRes = wfProc.processRequest(wfReq, config);
+
+ Assert.assertEquals("Principal name wrong", TEST_USER,
+ wfRes.getUsername());
+ Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
+ Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles()
+ .size());
+ Assert.assertEquals("Audience wrong", TEST_AUDIENCE, wfRes.getAudience());
+ assertClaims(wfRes.getClaims(), callbackHandler.getRoleAttributeName());
+
+ }
+
+ /**
+ * Validate SAML 2 token which includes the role attribute with 2 values
+ * Roles are encoded as a multi-value saml attribute
+ * Not RequestedSecurityTokenCollection in this test, default in all others
+ */
+ @org.junit.Test
+ public void validateSAML2TokenRSTR() throws Exception {
+ SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
+ callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
+ callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
+ callbackHandler.setIssuer(TEST_RSTR_ISSUER);
+ callbackHandler.setSubjectName(TEST_USER);
+ ConditionsBean cp = new ConditionsBean();
+ AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
+ audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
+ cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
+ callbackHandler.setConditions(cp);
+
+ SAMLCallback samlCallback = new SAMLCallback();
+ SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
+ SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
+ String rstr = createSamlToken(assertion, "mystskey", true, STSUtil.SAMPLE_RSTR_MSG);
+
+ FedizRequest wfReq = new FedizRequest();
+ wfReq.setAction(FederationConstants.ACTION_SIGNIN);
+ wfReq.setResponseToken(rstr);
+
+ configurator = null;
+ FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
+
+ FedizProcessor wfProc = new FederationProcessorImpl();
+ FedizResponse wfRes = wfProc.processRequest(wfReq, config);
+
+ Assert.assertEquals("Principal name wrong", TEST_USER,
+ wfRes.getUsername());
+ Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
+ Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles()
+ .size());
+ Assert.assertEquals("Audience wrong", TEST_AUDIENCE, wfRes.getAudience());
+ }
+
+ /**
+ * Validate SAML 2 token which doesn't include the role SAML attribute
+ */
+ @org.junit.Test
+ public void validateSAML2TokenWithoutRoles() throws Exception {
+ SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
+ callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
+ callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
+ callbackHandler.setIssuer(TEST_RSTR_ISSUER);
+ callbackHandler.setSubjectName(TEST_USER);
+ callbackHandler.setRoles(null);
+ ConditionsBean cp = new ConditionsBean();
+ AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
+ audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
+ cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
+ callbackHandler.setConditions(cp);
+
+ SAMLCallback samlCallback = new SAMLCallback();
+ SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
+ SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
+ String rstr = createSamlToken(assertion, "mystskey", true);
+
+ FedizRequest wfReq = new FedizRequest();
+ wfReq.setAction(FederationConstants.ACTION_SIGNIN);
+ wfReq.setResponseToken(rstr);
+
+ configurator = null;
+ FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
+
+ FedizProcessor wfProc = new FederationProcessorImpl();
+ FedizResponse wfRes = wfProc.processRequest(wfReq, config);
+
+ Assert.assertEquals("Principal name wrong", TEST_USER,
+ wfRes.getUsername());
+ Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
+ Assert.assertEquals("No roles must be found", null, wfRes.getRoles());
+ Assert.assertEquals("Audience wrong", TEST_AUDIENCE, wfRes.getAudience());
+ }
+
+ /**
+ * Validate SAML 2 token where role information is provided
+ * within another SAML attribute
+ */
+ @org.junit.Test
+ public void validateSAML2TokenDifferentRoleURI() throws Exception {
+ SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
+ callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
+ callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
+ callbackHandler.setIssuer(TEST_RSTR_ISSUER);
+ callbackHandler.setSubjectName(TEST_USER);
+ callbackHandler.setRoleAttributeName("http://schemas.mycompany.com/claims/role");
+ ConditionsBean cp = new ConditionsBean();
+ AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
+ audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
+ cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
+ callbackHandler.setConditions(cp);
+
+ SAMLCallback samlCallback = new SAMLCallback();
+ SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
+ SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
+ String rstr = createSamlToken(assertion, "mystskey", true);
+
+ FedizRequest wfReq = new FedizRequest();
+ wfReq.setAction(FederationConstants.ACTION_SIGNIN);
+ wfReq.setResponseToken(rstr);
+
+ configurator = null;
+ FedizContext config = getFederationConfigurator().getFedizContext("CUSTOMROLEURI");
+
+ FedizProcessor wfProc = new FederationProcessorImpl();
+ FedizResponse wfRes = wfProc.processRequest(wfReq, config);
+
+ Assert.assertEquals("Principal name wrong", TEST_USER, wfRes.getUsername());
+ Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
+ Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles().size());
+ Assert.assertEquals("Audience wrong", TEST_AUDIENCE, wfRes.getAudience());
+ assertClaims(wfRes.getClaims(), callbackHandler.getRoleAttributeName());
+ }
+
+ /**
+ * Validate SAML 1 token where role information is provided
+ * within another SAML attribute
+ */
+ @org.junit.Test
+ public void validateSAML1TokenDifferentRoleURI() throws Exception {
+ SAML1CallbackHandler callbackHandler = new SAML1CallbackHandler();
+ callbackHandler.setStatement(SAML1CallbackHandler.Statement.ATTR);
+ callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
+ callbackHandler.setIssuer(TEST_RSTR_ISSUER);
+ callbackHandler.setSubjectName(TEST_USER);
+ callbackHandler.setRoleAttributeName("http://schemas.mycompany.com/claims/role");
+ ConditionsBean cp = new ConditionsBean();
+ AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
+ audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
+ cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
+ callbackHandler.setConditions(cp);
+
+ SAMLCallback samlCallback = new SAMLCallback();
+ SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
+ SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
+ String rstr = createSamlToken(assertion, "mystskey", true);
+
+ FedizRequest wfReq = new FedizRequest();
+ wfReq.setAction(FederationConstants.ACTION_SIGNIN);
+ wfReq.setResponseToken(rstr);
+
+ configurator = null;
+ FedizContext config = getFederationConfigurator().getFedizContext("CUSTOMROLEURI");
+
+ FedizProcessor wfProc = new FederationProcessorImpl();
+ FedizResponse wfRes = wfProc.processRequest(wfReq, config);
+
+ Assert.assertEquals("Principal name wrong", TEST_USER, wfRes.getUsername());
+ Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
+ Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles().size());
+ Assert.assertEquals("Audience wrong", TEST_AUDIENCE, wfRes.getAudience());
+ assertClaims(wfRes.getClaims(), callbackHandler.getRoleAttributeName());
+ }
+
+ /**
+ * Validate SAML 2 token which includes role attribute
+ * but RoleURI is not configured
+ */
+ @org.junit.Test
+ public void validateSAML2TokenRoleURINotConfigured() throws Exception {
+ SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
+ callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
+ callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
+ callbackHandler.setIssuer(TEST_RSTR_ISSUER);
+ callbackHandler.setSubjectName(TEST_USER);
+ ConditionsBean cp = new ConditionsBean();
+ AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
+ audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
+ cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
+ callbackHandler.setConditions(cp);
+
+ SAMLCallback samlCallback = new SAMLCallback();
+ SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
+ SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
+ String rstr = createSamlToken(assertion, "mystskey", true);
+
+ FedizRequest wfReq = new FedizRequest();
+ wfReq.setAction(FederationConstants.ACTION_SIGNIN);
+ wfReq.setResponseToken(rstr);
+
+ configurator = null;
+ FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
+ config.getProtocol().setRoleURI(null);
+
+ FedizProcessor wfProc = new FederationProcessorImpl();
+ FedizResponse wfRes = wfProc.processRequest(wfReq, config);
+
+ Assert.assertEquals("Principal name wrong", TEST_USER,
+ wfRes.getUsername());
+ Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
+ Assert.assertEquals("Two roles must be found", null, wfRes.getRoles());
+ Assert.assertEquals("Audience wrong", TEST_AUDIENCE, wfRes.getAudience());
+ }
+
+ /**
+ * Validate SAML 1.1 token which includes the role attribute with 2 values
+ * Roles are encoded as a multi-value saml attribute
+ */
+ @org.junit.Test
+ public void validateSAML1Token() throws Exception {
+ SAML1CallbackHandler callbackHandler = new SAML1CallbackHandler();
+ callbackHandler.setStatement(SAML1CallbackHandler.Statement.ATTR);
+ callbackHandler.setConfirmationMethod(SAML1Constants.CONF_BEARER);
+ callbackHandler.setIssuer(TEST_RSTR_ISSUER);
+ callbackHandler.setSubjectName(TEST_USER);
+ ConditionsBean cp = new ConditionsBean();
+ AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
+ audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
+ cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
+ callbackHandler.setConditions(cp);
+
+ SAMLCallback samlCallback = new SAMLCallback();
+ SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
+ SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
+ String rstr = createSamlToken(assertion, "mystskey", true);
+
+ FedizRequest wfReq = new FedizRequest();
+ wfReq.setAction(FederationConstants.ACTION_SIGNIN);
+ wfReq.setResponseToken(rstr);
+
+ configurator = null;
+ FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
+
+ FedizProcessor wfProc = new FederationProcessorImpl();
+ FedizResponse wfRes = wfProc.processRequest(wfReq, config);
+
+ Assert.assertEquals("Principal name wrong", TEST_USER,
+ wfRes.getUsername());
+ Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
+ Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles()
+ .size());
+ Assert.assertEquals("Audience wrong", TEST_AUDIENCE, wfRes.getAudience());
+ assertClaims(wfRes.getClaims(), callbackHandler.getRoleAttributeName());
+ }
+
+ /**
+ * Validate SAML 1.1 token which includes the role attribute with 2 values
+ * Roles are encoded as a multi-value saml attribute
+ * Token embedded in RSTR 2005/02 - WS Federation 1.0
+ */
+ @org.junit.Test
+ public void validateSAML1TokenWSFed10() throws Exception {
+ SAML1CallbackHandler callbackHandler = new SAML1CallbackHandler();
+ callbackHandler.setStatement(SAML1CallbackHandler.Statement.ATTR);
+ callbackHandler.setConfirmationMethod(SAML1Constants.CONF_BEARER);
+ callbackHandler.setIssuer(TEST_RSTR_ISSUER);
+ callbackHandler.setSubjectName(TEST_USER);
+ ConditionsBean cp = new ConditionsBean();
+ AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
+ audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
+ cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
+ callbackHandler.setConditions(cp);
+
+ SAMLCallback samlCallback = new SAMLCallback();
+ SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
+ SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
+ String rstr = createSamlToken(assertion, "mystskey", true, STSUtil.SAMPLE_RSTR_2005_02_MSG);
+ FedizRequest wfReq = new FedizRequest();
+ wfReq.setAction(FederationConstants.ACTION_SIGNIN);
+ wfReq.setResponseToken(rstr);
+
+ configurator = null;
+ FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
+
+ FedizProcessor wfProc = new FederationProcessorImpl();
+ FedizResponse wfRes = wfProc.processRequest(wfReq, config);
+
+ Assert.assertEquals("Principal name wrong", TEST_USER,
+ wfRes.getUsername());
+ Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
+ Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles()
+ .size());
+ Assert.assertEquals("Audience wrong", TEST_AUDIENCE, wfRes.getAudience());
+ }
+
+ /**
+ * Validate SAML 2 token which includes the role attribute with 2 values
+ * Roles are encoded as a multiple saml attributes with the same name
+ */
+ @org.junit.Test
+ public void validateSAML2TokenRoleMultiAttributes() throws Exception {
+ SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
+ callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
+ callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
+ callbackHandler.setIssuer(TEST_RSTR_ISSUER);
+ callbackHandler.setSubjectName(TEST_USER);
+ callbackHandler.setMultiValueType(MultiValue.MULTI_ATTR);
+ ConditionsBean cp = new ConditionsBean();
+ AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
+ audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
+ cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
+ callbackHandler.setConditions(cp);
+
+ SAMLCallback samlCallback = new SAMLCallback();
+ SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
+ SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
+ String rstr = createSamlToken(assertion, "mystskey", true);
+
+ FedizRequest wfReq = new FedizRequest();
+ wfReq.setAction(FederationConstants.ACTION_SIGNIN);
+ wfReq.setResponseToken(rstr);
+
+ configurator = null;
+ FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
+
+ FedizProcessor wfProc = new FederationProcessorImpl();
+ FedizResponse wfRes = wfProc.processRequest(wfReq, config);
+
+ Assert.assertEquals("Principal name wrong", TEST_USER,
+ wfRes.getUsername());
+ Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
+ Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles()
+ .size());
+ assertClaims(wfRes.getClaims(), callbackHandler.getRoleAttributeName());
+ }
+
+ /**
+ * Validate SAML 2 token which includes the role attribute with 2 values
+ * Roles are encoded as a single saml attribute with encoded value
+ */
+ @org.junit.Test
+ public void validateSAML2TokenRoleEncodedValue() throws Exception {
+ SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
+ callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
+ callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
+ callbackHandler.setIssuer(TEST_RSTR_ISSUER);
+ callbackHandler.setSubjectName(TEST_USER);
+ callbackHandler.setMultiValueType(MultiValue.ENC_VALUE);
+ ConditionsBean cp = new ConditionsBean();
+ AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
+ audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
+ cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
+ callbackHandler.setConditions(cp);
+
+ SAMLCallback samlCallback = new SAMLCallback();
+ SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
+ SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
+ String rstr = createSamlToken(assertion, "mystskey", true);
+
+ FedizRequest wfReq = new FedizRequest();
+ wfReq.setAction(FederationConstants.ACTION_SIGNIN);
+ wfReq.setResponseToken(rstr);
+
+ configurator = null;
+ FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
+ FederationProtocol fp = (FederationProtocol)config.getProtocol();
+ fp.setRoleDelimiter(",");
+
+ FedizProcessor wfProc = new FederationProcessorImpl();
+ FedizResponse wfRes = wfProc.processRequest(wfReq, config);
+
+ Assert.assertEquals("Principal name wrong", TEST_USER,
+ wfRes.getUsername());
+ Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
+ Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles()
+ .size());
+ assertClaims(wfRes.getClaims(), callbackHandler.getRoleAttributeName());
+ }
+
+ /**
+ * Validate SAML 2 token which includes the role attribute with 2 values
+ * The configured subject of the trusted issuer doesn't match with
+ * the issuer of the SAML token
+ *
+ * Ignored because PeerTrust ignores subject attribute
+ */
+ @org.junit.Test
+ @org.junit.Ignore
+ public void validateSAML2TokenUntrustedIssuer() throws Exception {
+ SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
+ callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
+ callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
+ callbackHandler.setIssuer(TEST_RSTR_ISSUER);
+ callbackHandler.setSubjectName(TEST_USER);
+ ConditionsBean cp = new ConditionsBean();
+ AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
+ audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
+ cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
+ callbackHandler.setConditions(cp);
+
+ SAMLCallback samlCallback = new SAMLCallback();
+ SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
+ SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
+
+ String rstr = createSamlToken(assertion, "mystskey", true);
+ FedizRequest wfReq = new FedizRequest();
+ wfReq.setAction(FederationConstants.ACTION_SIGNIN);
+ wfReq.setResponseToken(rstr);
+
+ // Load and update the config to enforce an error
+ configurator = null;
+ FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
+ config.getTrustedIssuers().get(0).setSubject("wrong-issuer-name");
+
+ FedizProcessor wfProc = new FederationProcessorImpl();
+ try {
+ wfProc.processRequest(wfReq, config);
+ Assert.fail("Processing must fail because of untrusted issuer configured");
+ } catch (ProcessingException ex) {
+ if (!TYPE.ISSUER_NOT_TRUSTED.equals(ex.getType())) {
+ fail("Expected ProcessingException with ISSUER_NOT_TRUSTED type");
+ }
+ }
+ }
+
+ /**
+ * Validate SAML 2 token which includes the role attribute with 2 values
+ * The configured subject of the trusted issuer doesn't match with
+ * the issuer of the SAML token
+ */
+ @org.junit.Test
+ public void validateUnsignedSAML2Token() throws Exception {
+ SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
+ callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
+ callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
+ callbackHandler.setIssuer(TEST_RSTR_ISSUER);
+ callbackHandler.setSubjectName(TEST_USER);
+ ConditionsBean cp = new ConditionsBean();
+ AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
+ audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
+ cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
+ callbackHandler.setConditions(cp);
+
+ SAMLCallback samlCallback = new SAMLCallback();
+ SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
+ SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
+
+ String rstr = createSamlToken(assertion, "mystskey", false);
+ FedizRequest wfReq = new FedizRequest();
+ wfReq.setAction(FederationConstants.ACTION_SIGNIN);
+ wfReq.setResponseToken(rstr);
+
+ // Load and update the config to enforce an error
+ configurator = null;
+ FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
+
+ FedizProcessor wfProc = new FederationProcessorImpl();
+ try {
+ wfProc.processRequest(wfReq, config);
+ Assert.fail("Processing must fail because of missing signature");
+ } catch (ProcessingException ex) {
+ if (!TYPE.TOKEN_NO_SIGNATURE.equals(ex.getType())) {
+ fail("Expected ProcessingException with TOKEN_NO_SIGNATURE type");
+ }
+ }
+ }
+
+ /**
+ * Validate SAML 2 token twice which causes an exception
+ * due to replay attack
+ */
+ @org.junit.Test
+ public void testReplayAttack() throws Exception {
+ SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
+ callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
+ callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
+ callbackHandler.setIssuer(TEST_RSTR_ISSUER);
+ callbackHandler.setSubjectName(TEST_USER);
+ ConditionsBean cp = new ConditionsBean();
+ AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
+ audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
+ cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
+ callbackHandler.setConditions(cp);
+
+ SAMLCallback samlCallback = new SAMLCallback();
+ SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
+ SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
+
+ String rstr = createSamlToken(assertion, "mystskey", true);
+
+ FedizRequest wfReq = new FedizRequest();
+ wfReq.setAction(FederationConstants.ACTION_SIGNIN);
+ wfReq.setResponseToken(rstr);
+
+ configurator = null;
+ FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
+
+ FedizProcessor wfProc = new FederationProcessorImpl();
+ FedizResponse wfRes = wfProc.processRequest(wfReq, config);
+ Assert.assertEquals("Principal name wrong", TEST_USER,
+ wfRes.getUsername());
+ Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
+
+ wfProc = new FederationProcessorImpl();
+ try {
+ wfProc.processRequest(wfReq, config);
+ fail("Failure expected on a replay attack");
+ } catch (ProcessingException ex) {
+ if (!TYPE.TOKEN_REPLAY.equals(ex.getType())) {
+ fail("Expected ProcessingException with TOKEN_REPLAY type");
+ }
+ }
+ }
+
+
+ /**
+ * Validate SAML 2 token which includes the role attribute with 2 values
+ * The configured subject of the trusted issuer doesn't match with
+ * the issuer of the SAML token
+ */
+ @org.junit.Test
+ public void validateSAML2TokenSeveralCertStore() throws Exception {
+ SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
+ callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
+ callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
+ callbackHandler.setIssuer(TEST_RSTR_ISSUER);
+ callbackHandler.setSubjectName(TEST_USER);
+ ConditionsBean cp = new ConditionsBean();
+ AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
+ audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
+ cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
+ callbackHandler.setConditions(cp);
+
+ SAMLCallback samlCallback = new SAMLCallback();
+ SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
+ SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
+
+ String rstr = createSamlToken(assertion, "mystskey", true);
+ FedizRequest wfReq = new FedizRequest();
+ wfReq.setAction(FederationConstants.ACTION_SIGNIN);
+ wfReq.setResponseToken(rstr);
+
+ // Load and update the config to enforce an error
+ configurator = null;
+ FedizContext config = getFederationConfigurator().getFedizContext("ROOT2");
+
+ FedizProcessor wfProc = new FederationProcessorImpl();
+ FedizResponse wfRes = wfProc.processRequest(wfReq, config);
+
+ Assert.assertEquals("Principal name wrong", TEST_USER,
+ wfRes.getUsername());
+ Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
+ Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles()
+ .size());
+ }
+
+ /**
+ * Validate SAML 2 token which includes the role attribute with 2 values
+ * The configured subject of the trusted issuer doesn't match with
+ * the issuer of the SAML token
+ */
+ @org.junit.Test
+ public void validateSAML2TokenSeveralCertStoreTrustedIssuer() throws Exception {
+ SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
+ callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
+ callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
+ callbackHandler.setIssuer(TEST_RSTR_ISSUER);
+ callbackHandler.setSubjectName(TEST_USER);
+ ConditionsBean cp = new ConditionsBean();
+ AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
+ audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
+ cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
+ callbackHandler.setConditions(cp);
+
+ SAMLCallback samlCallback = new SAMLCallback();
+ SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
+ SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
+
+ String rstr = createSamlToken(assertion, "mystskey", true);
+ FedizRequest wfReq = new FedizRequest();
+ wfReq.setAction(FederationConstants.ACTION_SIGNIN);
+ wfReq.setResponseToken(rstr);
+
+ // Load and update the config to enforce an error
+ configurator = null;
+ FedizContext config = getFederationConfigurator().getFedizContext("ROOT3");
+
+ FedizProcessor wfProc = new FederationProcessorImpl();
+ FedizResponse wfRes = wfProc.processRequest(wfReq, config);
+
+ Assert.assertEquals("Principal name wrong", TEST_USER,
+ wfRes.getUsername());
+ Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
+ Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles()
+ .size());
+ }
+
+ /**
+ * Validate SAML 2 token which is expired
+ */
+ @org.junit.Test
+ public void validateSAML2TokenExpired() throws Exception {
+ SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
+ callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
+ callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
+ callbackHandler.setIssuer(TEST_RSTR_ISSUER);
+ callbackHandler.setSubjectName(TEST_USER);
+ ConditionsBean cp = new ConditionsBean();
+ DateTime currentTime = new DateTime();
+ currentTime = currentTime.minusSeconds(60);
+ cp.setNotAfter(currentTime);
+ currentTime = new DateTime();
+ currentTime = currentTime.minusSeconds(300);
+ cp.setNotBefore(currentTime);
+ AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
+ audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
+ cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
+ callbackHandler.setConditions(cp);
+
+ SAMLCallback samlCallback = new SAMLCallback();
+ SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
+ SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
+
+ String rstr = createSamlToken(assertion, "mystskey", true);
+
+ FedizRequest wfReq = new FedizRequest();
+ wfReq.setAction(FederationConstants.ACTION_SIGNIN);
+ wfReq.setResponseToken(rstr);
+
+ configurator = null;
+ FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
+
+ FedizProcessor wfProc = new FederationProcessorImpl();
+ try {
+ wfProc.processRequest(wfReq, config);
+ fail("Failure expected on expired SAML token");
+ } catch (ProcessingException ex) {
+ if (!TYPE.TOKEN_EXPIRED.equals(ex.getType())) {
+ fail("Expected ProcessingException with TOKEN_EXPIRED type");
+ }
+ }
+ }
+
+ /**
+ * Validate SAML 2 token which is not yet valid (in 30 seconds)
+ * but within the maximum clock skew range (60 seconds)
+ */
+ @org.junit.Test
+ public void validateSAML2TokenClockSkewRange() throws Exception {
+ SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
+ callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
+ callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
+ callbackHandler.setIssuer(TEST_RSTR_ISSUER);
+ callbackHandler.setSubjectName(TEST_USER);
+ ConditionsBean cp = new ConditionsBean();
+ DateTime currentTime = new DateTime();
+ currentTime = currentTime.plusSeconds(300);
+ cp.setNotAfter(currentTime);
+ currentTime = new DateTime();
+ currentTime = currentTime.plusSeconds(30);
+ cp.setNotBefore(currentTime);
+ AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
+ audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
+ cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
+ callbackHandler.setConditions(cp);
+
+ SAMLCallback samlCallback = new SAMLCallback();
+ SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
+ SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
+
+ String rstr = createSamlToken(assertion, "mystskey", true);
+
+ FedizRequest wfReq = new FedizRequest();
+ wfReq.setAction(FederationConstants.ACTION_SIGNIN);
+ wfReq.setResponseToken(rstr);
+
+ configurator = null;
+ FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
+ config.setMaximumClockSkew(BigInteger.valueOf(60));
+
+ FedizProcessor wfProc = new FederationProcessorImpl();
+ FedizResponse wfRes = wfProc.processRequest(wfReq, config);
+
+ Assert.assertEquals("Principal name wrong", TEST_USER,
+ wfRes.getUsername());
+ Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
+ Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles()
+ .size());
+ }
+
+ /**
+ * "Validate" SAML 2 token with a custom token validator
+ * If a validator is configured it precedes the SAMLTokenValidator as part of Fediz
+ */
+ @org.junit.Test
+ public void validateSAML2TokenCustomValidator() throws Exception {
+ SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
+ callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
+ callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
+ callbackHandler.setIssuer(TEST_RSTR_ISSUER);
+ callbackHandler.setSubjectName(TEST_USER);
+ ConditionsBean cp = new ConditionsBean();
+ AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
+ audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
+ cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
+ callbackHandler.setConditions(cp);
+
+ SAMLCallback samlCallback = new SAMLCallback();
+ SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
+ SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
+
+ String rstr = createSamlToken(assertion, "mystskey", true);
+
+ FedizRequest wfReq = new FedizRequest();
+ wfReq.setAction(FederationConstants.ACTION_SIGNIN);
+ wfReq.setResponseToken(rstr);
+
+ configurator = null;
+ FedizContext config = getFederationConfigurator().getFedizContext("CUSTTOK");
+ Protocol protocol = config.getProtocol();
+ List<TokenValidator> validators = protocol.getTokenValidators();
+ Assert.assertEquals("Two validators must be found", 2, validators.size());
+ Assert.assertEquals("First validator must be custom validator",
+ CustomValidator.class.getName(), validators.get(0).getClass().getName());
+
+ FedizProcessor wfProc = new FederationProcessorImpl();
+ FedizResponse wfRes = wfProc.processRequest(wfReq, config);
+
+ Assert.assertEquals("Principal name wrong", TEST_USER,
+ wfRes.getUsername());
+ Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
+ }
+
+ /**
+ * "Validate" SAML 2 token with a custom token validator
+ * If a validator is configured it precedes the SAMLTokenValidator as part of Fediz
+ */
+ @org.junit.Test
+ public void validateSAML2TokenMaxClockSkewNotDefined() throws Exception {
+ SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
+ callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
+ callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
+ callbackHandler.setIssuer(TEST_RSTR_ISSUER);
+ callbackHandler.setSubjectName(TEST_USER);
+ ConditionsBean cp = new ConditionsBean();
+ AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
+ audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
+ cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
+ callbackHandler.setConditions(cp);
+
+ SAMLCallback samlCallback = new SAMLCallback();
+ SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
+ SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
+
+ String rstr = createSamlToken(assertion, "mystskey", true);
+
+ FedizRequest wfReq = new FedizRequest();
+ wfReq.setAction(FederationConstants.ACTION_SIGNIN);
+ wfReq.setResponseToken(rstr);
+
+ configurator = null;
+ FedizContext config = getFederationConfigurator().getFedizContext("NOCLOCKSKEW");
+
+ FedizProcessor wfProc = new FederationProcessorImpl();
+ FedizResponse wfRes = wfProc.processRequest(wfReq, config);
+
+ Assert.assertEquals("Principal name wrong", TEST_USER,
+ wfRes.getUsername());
+ Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
+ Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles()
+ .size());
+ Assert.assertEquals("Audience wrong", TEST_AUDIENCE, wfRes.getAudience());
+ }
+
+ /**
+ * Validate an encrypted SAML 2 token which includes the role attribute with 2 values
+ * Roles are encoded as a multi-value saml attribute
+ */
+ @org.junit.Test
+ public void validateEncryptedSAML2Token() throws Exception {
+ SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
+ callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
+ callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
+ callbackHandler.setIssuer(TEST_RSTR_ISSUER);
+ callbackHandler.setSubjectName(TEST_USER);
+ ConditionsBean cp = new ConditionsBean();
+ AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
+ audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
+ cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
+ callbackHandler.setConditions(cp);
+
+ SAMLCallback samlCallback = new SAMLCallback();
+ SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
+ SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
+
+ String rstr = encryptAndSignToken(assertion);
+
+ FedizRequest wfReq = new FedizRequest();
+ wfReq.setAction(FederationConstants.ACTION_SIGNIN);
+ wfReq.setResponseToken(rstr);
+
+ configurator = null;
+ FedizContext config =
+ getFederationConfigurator().getFedizContext("ROOT_DECRYPTION");
+
+ FedizProcessor wfProc = new FederationProcessorImpl();
+ FedizResponse wfRes = wfProc.processRequest(wfReq, config);
+
+ Assert.assertEquals("Principal name wrong", TEST_USER,
+ wfRes.getUsername());
+ Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
+ Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles()
+ .size());
+ Assert.assertEquals("Audience wrong", TEST_AUDIENCE, wfRes.getAudience());
+ assertClaims(wfRes.getClaims(), callbackHandler.getRoleAttributeName());
+ }
+
+ /**
+ * Validate a HolderOfKey SAML 2 token
+ */
+ @org.junit.Test
+ public void validateHOKSAML2Token() throws Exception {
+ SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
+ callbackHandler.setStatement(SAML2CallbackHandler.Statement.AUTHN);
+ callbackHandler.setConfirmationMethod(SAML2Constants.CONF_HOLDER_KEY);
+ callbackHandler.setIssuer(TEST_RSTR_ISSUER);
+ callbackHandler.setSubjectName(TEST_USER);
+ ConditionsBean cp = new ConditionsBean();
+ AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
+ audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
+ cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
+ callbackHandler.setConditions(cp);
+
+ Crypto clientCrypto = CryptoFactory.getInstance("client-crypto.properties");
+ CryptoType cryptoType = new CryptoType(CryptoType.TYPE.ALIAS);
+ cryptoType.setAlias("myclientkey");
+ X509Certificate[] certs = clientCrypto.getX509Certificates(cryptoType);
+ callbackHandler.setCerts(certs);
+
+ SAMLCallback samlCallback = new SAMLCallback();
+ SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
+ SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
+
+ WSPasswordCallback[] cb = {
+ new WSPasswordCallback("mystskey", WSPasswordCallback.SIGNATURE)
+ };
+ cbPasswordHandler.handle(cb);
+ String password = cb[0].getPassword();
+
+ assertion.signAssertion("mystskey", password, crypto, false);
+
+ Document doc = STSUtil.toSOAPPart(STSUtil.SAMPLE_RSTR_COLL_MSG);
+ Element token = assertion.toDOM(doc);
+
+ Element e = FederationResponseTest.findElement(doc, "RequestedSecurityToken",
+ FederationConstants.WS_TRUST_13_NS);
+ if (e == null) {
+ e = FederationResponseTest.findElement(doc, "RequestedSecurityToken",
+ FederationConstants.WS_TRUST_2005_02_NS);
+ }
+ e.appendChild(token);
+
+ String rstr = DOM2Writer.nodeToString(doc);
+
+ FedizRequest wfReq = new FedizRequest();
+ wfReq.setAction(FederationConstants.ACTION_SIGNIN);
+ wfReq.setResponseToken(rstr);
+
+ configurator = null;
+ FedizContext config =
+ getFederationConfigurator().getFedizContext("ROOT_DECRYPTION");
+
+ FedizProcessor wfProc = new FederationProcessorImpl();
+ try {
+ wfProc.processRequest(wfReq, config);
+ fail("Failure expected on missing client certs");
+ } catch (ProcessingException ex) {
+ // expected
+ }
+
+ // Now set client certs
+ wfReq.setCerts(certs);
+ wfProc.processRequest(wfReq, config);
+ }
+
+ @org.junit.Test
+ public void validateSAML2TokenWithConfigCreatedWithAPI() throws Exception {
+
+ ContextConfig config = new ContextConfig();
+
+ config.setName("whatever");
+
+ // Configure certificate store
+ CertificateStores certStores = new CertificateStores();
+ TrustManagersType tm0 = new TrustManagersType();
+ KeyStoreType ks0 = new KeyStoreType();
+ ks0.setType("JKS");
+ ks0.setPassword("storepass");
+ ks0.setFile("ststrust.jks");
+ tm0.setKeyStore(ks0);
+ certStores.getTrustManager().add(tm0);
+ config.setCertificateStores(certStores);
+
+ // Configure trusted IDP
+ TrustedIssuers trustedIssuers = new TrustedIssuers();
+ TrustedIssuerType ti0 = new TrustedIssuerType();
+ ti0.setCertificateValidation(ValidationType.CHAIN_TRUST);
+ ti0.setName("FedizSTSIssuer");
+ ti0.setSubject(".*CN=www.sts.com.*");
+ trustedIssuers.getIssuer().add(ti0);
+ config.setTrustedIssuers(trustedIssuers);
+
+ FederationProtocolType protocol = new FederationProtocolType();
+ config.setProtocol(protocol);
+
+ AudienceUris audienceUris = new AudienceUris();
+ audienceUris.getAudienceItem().add("https://localhost/fedizhelloworld");
+ config.setAudienceUris(audienceUris);
+
+ protocol.setRoleURI("http://schemas.xmlsoap.org/ws/2005/05/identity/claims/role");
+
+ FedizContext fedContext = new FedizContext(config);
+
+ SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
+ callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
+ callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
+ callbackHandler.setIssuer(TEST_RSTR_ISSUER);
+ callbackHandler.setSubjectName(TEST_USER);
+ ConditionsBean cp = new ConditionsBean();
+ AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
+ audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
+ cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
+ callbackHandler.setConditions(cp);
+
+ SAMLCallback samlCallback = new SAMLCallback();
+ SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
+ SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
+
+ String rstr = createSamlToken(assertion, "mystskey", true, STSUtil.SAMPLE_RSTR_MSG);
+
+ FedizRequest wfReq = new FedizRequest();
+ wfReq.setAction(FederationConstants.ACTION_SIGNIN);
+ wfReq.setResponseToken(rstr);
+
+ FedizProcessor wfProc = new FederationProcessorImpl();
+ FedizResponse wfRes = wfProc.processRequest(wfReq, fedContext);
+
+ Assert.assertEquals("Principal name wrong", TEST_USER,
+ wfRes.getUsername());
+ Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
+ Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles()
+ .size());
+ Assert.assertEquals("Audience wrong", TEST_AUDIENCE, wfRes.getAudience());
+
+ fedContext.close();
+
+ }
+
+
+ private String encryptAndSignToken(
+ SamlAssertionWrapper assertion
+ ) throws Exception {
+
+ WSPasswordCallback[] cb = {
+ new WSPasswordCallback("mystskey", WSPasswordCallback.SIGNATURE)
+ };
+ cbPasswordHandler.handle(cb);
+ String password = cb[0].getPassword();
+
+ assertion.signAssertion("mystskey", password, crypto, false);
+
+ Document doc = STSUtil.toSOAPPart(STSUtil.SAMPLE_RSTR_COLL_MSG);
+ Element token = assertion.toDOM(doc);
+
+ Element e = FederationResponseTest.findElement(doc, "RequestedSecurityToken",
+ FederationConstants.WS_TRUST_13_NS);
+ if (e == null) {
+ e = FederationResponseTest.findElement(doc, "RequestedSecurityToken",
+ FederationConstants.WS_TRUST_2005_02_NS);
+ }
+ e.appendChild(token);
+
+ WSSecEncrypt builder = new WSSecEncrypt();
+ builder.setUserInfo("mystskey");
+
+ builder.setKeyIdentifierType(WSConstants.ISSUER_SERIAL);
+ builder.setSymmetricEncAlgorithm(WSConstants.AES_128);
+ builder.setKeyEncAlgo(WSConstants.KEYTRANSPORT_RSA15);
+ builder.setEmbedEncryptedKey(true);
+
+ WSEncryptionPart encryptionPart = new WSEncryptionPart(assertion.getId(), "Element");
+ encryptionPart.setElement(token);
+
+ Crypto encrCrypto = CryptoFactory.getInstance("signature.properties");
+ builder.prepare(token.getOwnerDocument(), encrCrypto);
+ builder.encryptForRef(null, Collections.singletonList(encryptionPart));
+
+ // return doc.getDocumentElement();
+ return DOM2Writer.nodeToString(doc);
+ }
+
+ private String createSamlToken(SamlAssertionWrapper assertion, String alias, boolean sign)
+ throws IOException, UnsupportedCallbackException, WSSecurityException, Exception {
+ return createSamlToken(assertion, alias, sign, STSUtil.SAMPLE_RSTR_COLL_MSG);
+ }
+
+ private String createSamlToken(SamlAssertionWrapper assertion, String alias, boolean sign, String rstr)
+ throws IOException, UnsupportedCallbackException, WSSecurityException, Exception {
+ WSPasswordCallback[] cb = {
+ new WSPasswordCallback(alias, WSPasswordCallback.SIGNATURE)
+ };
+ cbPasswordHandler.handle(cb);
+ String password = cb[0].getPassword();
+
+ if (sign) {
+ assertion.signAssertion(alias, password, crypto, false);
+ }
+ Document doc = STSUtil.toSOAPPart(rstr);
+ Element token = assertion.toDOM(doc);
+
+ Element e = FederationResponseTest.findElement(doc, "RequestedSecurityToken",
+ FederationConstants.WS_TRUST_13_NS);
+ if (e == null) {
+ e = FederationResponseTest.findElement(doc, "RequestedSecurityToken",
+ FederationConstants.WS_TRUST_2005_02_NS);
+ }
+ e.appendChild(token);
+ return DOM2Writer.nodeToString(doc);
+ }
+
+
+
+
+ /**
+ * Returns the first element that matches <code>name</code> and
+ * <code>namespace</code>. <p/> This is a replacement for a XPath lookup
+ * <code>//name</code> with the given namespace. It's somewhat faster than
+ * XPath, and we do not deal with prefixes, just with the real namespace URI
+ *
+ * @param startNode Where to start the search
+ * @param name Local name of the element
+ * @param namespace Namespace URI of the element
+ * @return The found element or <code>null</code>
+ */
+ public static Element findElement(Node startNode, String name, String namespace) {
+ //
+ // Replace the formerly recursive implementation with a depth-first-loop
+ // lookup
+ //
+ if (startNode == null) {
+ return null;
+ }
+ Node startParent = startNode.getParentNode();
+ Node processedNode = null;
+
+ while (startNode != null) {
+ // start node processing at this point
+ if (startNode.getNodeType() == Node.ELEMENT_NODE
+ && startNode.getLocalName().equals(name)) {
+ String ns = startNode.getNamespaceURI();
+ if (ns != null && ns.equals(namespace)) {
+ return (Element)startNode;
+ }
+
+ if ((namespace == null || namespace.length() == 0)
+ && (ns == null || ns.length() == 0)) {
+ return (Element)startNode;
+ }
+ }
+ processedNode = startNode;
+ startNode = startNode.getFirstChild();
+
+ // no child, this node is done.
+ if (startNode == null) {
+ // close node processing, get sibling
+ startNode = processedNode.getNextSibling();
+ }
+ // no more siblings, get parent, all children
+ // of parent are processed.
+ while (startNode == null) {
+ processedNode = processedNode.getParentNode();
+ if (processedNode == startParent) {
+ return null;
+ }
+ // close parent node processing (processed node now)
+ startNode = processedNode.getNextSibling();
+ }
+ }
+ return null;
+ }
+
+ private void assertClaims(List<Claim> claims, String roleClaimType) {
+ for (Claim c : claims) {
+ Assert.assertTrue("Invalid ClaimType URI: " + c.getClaimType(),
+ c.getClaimType().equals(roleClaimType)
+ || c.getClaimType().equals(ClaimTypes.COUNTRY)
+ || c.getClaimType().equals(AbstractSAMLCallbackHandler.CLAIM_TYPE_LANGUAGE)
+ );
+ }
+ }
+
+
+}
http://git-wip-us.apache.org/repos/asf/cxf-fediz/blob/4a0b8649/plugins/core/src/test/resources/fediz_test_config.xml
----------------------------------------------------------------------
diff --git a/plugins/core/src/test/resources/fediz_test_config.xml b/plugins/core/src/test/resources/fediz_test_config.xml
index 48622a7..ab53717 100644
--- a/plugins/core/src/test/resources/fediz_test_config.xml
+++ b/plugins/core/src/test/resources/fediz_test_config.xml
@@ -22,7 +22,6 @@
<roleDelimiter>;</roleDelimiter>
<roleURI>http://schemas.xmlsoap.org/ws/2005/05/identity/claims/role</roleURI>
<authenticationType value="some auth type" type="String" />
- <homeRealm type="Class">org.apache.fediz.realm.MyHomeRealm.class</homeRealm>
<freshness>10000</freshness>
<reply>reply value</reply>
<request>REQUEST</request>
[2/2] git commit: Adding some WS-Federation request creation unit
tests
Posted by co...@apache.org.
Adding some WS-Federation request creation unit tests
Project: http://git-wip-us.apache.org/repos/asf/cxf-fediz/repo
Commit: http://git-wip-us.apache.org/repos/asf/cxf-fediz/commit/4a0b8649
Tree: http://git-wip-us.apache.org/repos/asf/cxf-fediz/tree/4a0b8649
Diff: http://git-wip-us.apache.org/repos/asf/cxf-fediz/diff/4a0b8649
Branch: refs/heads/master
Commit: 4a0b8649605320008a74278bb518a1f0bf37c55c
Parents: 87e87a0
Author: Colm O hEigeartaigh <co...@apache.org>
Authored: Fri Jul 25 10:23:26 2014 +0100
Committer: Colm O hEigeartaigh <co...@apache.org>
Committed: Fri Jul 25 10:23:26 2014 +0100
----------------------------------------------------------------------
.../fediz/core/federation/CustomValidator.java | 6 +-
.../federation/FederationProcessorTest.java | 1351 ------------------
.../core/federation/FederationRequestTest.java | 130 ++
.../core/federation/FederationResponseTest.java | 1351 ++++++++++++++++++
.../src/test/resources/fediz_test_config.xml | 1 -
5 files changed, 1484 insertions(+), 1355 deletions(-)
----------------------------------------------------------------------
http://git-wip-us.apache.org/repos/asf/cxf-fediz/blob/4a0b8649/plugins/core/src/test/java/org/apache/cxf/fediz/core/federation/CustomValidator.java
----------------------------------------------------------------------
diff --git a/plugins/core/src/test/java/org/apache/cxf/fediz/core/federation/CustomValidator.java b/plugins/core/src/test/java/org/apache/cxf/fediz/core/federation/CustomValidator.java
index 9f988bb..30faea8 100644
--- a/plugins/core/src/test/java/org/apache/cxf/fediz/core/federation/CustomValidator.java
+++ b/plugins/core/src/test/java/org/apache/cxf/fediz/core/federation/CustomValidator.java
@@ -44,11 +44,11 @@ public class CustomValidator implements TokenValidator {
FedizContext config
) {
return new TokenValidatorResponse(null,
- FederationProcessorTest.TEST_USER,
- FederationProcessorTest.TEST_RSTR_ISSUER,
+ FederationResponseTest.TEST_USER,
+ FederationResponseTest.TEST_RSTR_ISSUER,
null,
null,
- FederationProcessorTest.TEST_AUDIENCE);
+ FederationResponseTest.TEST_AUDIENCE);
}
}
http://git-wip-us.apache.org/repos/asf/cxf-fediz/blob/4a0b8649/plugins/core/src/test/java/org/apache/cxf/fediz/core/federation/FederationProcessorTest.java
----------------------------------------------------------------------
diff --git a/plugins/core/src/test/java/org/apache/cxf/fediz/core/federation/FederationProcessorTest.java b/plugins/core/src/test/java/org/apache/cxf/fediz/core/federation/FederationProcessorTest.java
deleted file mode 100644
index 09a8921..0000000
--- a/plugins/core/src/test/java/org/apache/cxf/fediz/core/federation/FederationProcessorTest.java
+++ /dev/null
@@ -1,1351 +0,0 @@
-/**
- * Licensed to the Apache Software Foundation (ASF) under one
- * or more contributor license agreements. See the NOTICE file
- * distributed with this work for additional information
- * regarding copyright ownership. The ASF licenses this file
- * to you 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 org.apache.cxf.fediz.core.federation;
-
-import java.io.File;
-import java.io.IOException;
-import java.math.BigInteger;
-import java.net.URL;
-import java.security.cert.X509Certificate;
-import java.util.Collections;
-import java.util.List;
-
-import javax.security.auth.callback.CallbackHandler;
-import javax.security.auth.callback.UnsupportedCallbackException;
-
-import org.w3c.dom.Document;
-import org.w3c.dom.Element;
-import org.w3c.dom.Node;
-import org.apache.cxf.fediz.common.STSUtil;
-import org.apache.cxf.fediz.common.SecurityTestUtil;
-import org.apache.cxf.fediz.core.AbstractSAMLCallbackHandler;
-import org.apache.cxf.fediz.core.AbstractSAMLCallbackHandler.MultiValue;
-import org.apache.cxf.fediz.core.Claim;
-import org.apache.cxf.fediz.core.ClaimTypes;
-import org.apache.cxf.fediz.core.FederationConstants;
-import org.apache.cxf.fediz.core.KeystoreCallbackHandler;
-import org.apache.cxf.fediz.core.SAML1CallbackHandler;
-import org.apache.cxf.fediz.core.SAML2CallbackHandler;
-import org.apache.cxf.fediz.core.TokenValidator;
-import org.apache.cxf.fediz.core.config.FederationProtocol;
-import org.apache.cxf.fediz.core.config.FedizConfigurator;
-import org.apache.cxf.fediz.core.config.FedizContext;
-import org.apache.cxf.fediz.core.config.Protocol;
-import org.apache.cxf.fediz.core.config.jaxb.AudienceUris;
-import org.apache.cxf.fediz.core.config.jaxb.CertificateStores;
-import org.apache.cxf.fediz.core.config.jaxb.ContextConfig;
-import org.apache.cxf.fediz.core.config.jaxb.FederationProtocolType;
-import org.apache.cxf.fediz.core.config.jaxb.KeyStoreType;
-import org.apache.cxf.fediz.core.config.jaxb.TrustManagersType;
-import org.apache.cxf.fediz.core.config.jaxb.TrustedIssuerType;
-import org.apache.cxf.fediz.core.config.jaxb.TrustedIssuers;
-import org.apache.cxf.fediz.core.config.jaxb.ValidationType;
-import org.apache.cxf.fediz.core.exception.ProcessingException;
-import org.apache.cxf.fediz.core.exception.ProcessingException.TYPE;
-import org.apache.cxf.fediz.core.processor.FederationProcessorImpl;
-import org.apache.cxf.fediz.core.processor.FedizProcessor;
-import org.apache.cxf.fediz.core.processor.FedizRequest;
-import org.apache.cxf.fediz.core.processor.FedizResponse;
-import org.apache.wss4j.common.WSEncryptionPart;
-import org.apache.wss4j.common.crypto.Crypto;
-import org.apache.wss4j.common.crypto.CryptoFactory;
-import org.apache.wss4j.common.crypto.CryptoType;
-import org.apache.wss4j.common.ext.WSPasswordCallback;
-import org.apache.wss4j.common.ext.WSSecurityException;
-import org.apache.wss4j.common.saml.SAMLCallback;
-import org.apache.wss4j.common.saml.SAMLUtil;
-import org.apache.wss4j.common.saml.SamlAssertionWrapper;
-import org.apache.wss4j.common.saml.bean.AudienceRestrictionBean;
-import org.apache.wss4j.common.saml.bean.ConditionsBean;
-import org.apache.wss4j.common.saml.builder.SAML1Constants;
-import org.apache.wss4j.common.saml.builder.SAML2Constants;
-import org.apache.wss4j.common.util.DOM2Writer;
-import org.apache.wss4j.dom.WSConstants;
-import org.apache.wss4j.dom.message.WSSecEncrypt;
-import org.joda.time.DateTime;
-import org.junit.AfterClass;
-import org.junit.Assert;
-import org.junit.BeforeClass;
-
-import static org.junit.Assert.fail;
-
-/**
- * Some tests for the WS-Federation "FederationProcessor".
- */
-public class FederationProcessorTest {
- static final String TEST_USER = "alice";
- static final String TEST_RSTR_ISSUER = "FedizSTSIssuer";
- static final String TEST_AUDIENCE = "https://localhost/fedizhelloworld";
-
- private static final String CONFIG_FILE = "fediz_test_config.xml";
-
- private static Crypto crypto;
- private static CallbackHandler cbPasswordHandler;
- private static FedizConfigurator configurator;
-
-
- @BeforeClass
- public static void init() {
- try {
- crypto = CryptoFactory.getInstance("signature.properties");
- cbPasswordHandler = new KeystoreCallbackHandler();
- getFederationConfigurator();
- } catch (Exception e) {
- e.printStackTrace();
- }
- Assert.assertNotNull(configurator);
-
- }
-
- @AfterClass
- public static void cleanup() {
- SecurityTestUtil.cleanup();
- }
-
-
- private static FedizConfigurator getFederationConfigurator() {
- if (configurator != null) {
- return configurator;
- }
- try {
- configurator = new FedizConfigurator();
- final URL resource = Thread.currentThread().getContextClassLoader()
- .getResource(CONFIG_FILE);
- File f = new File(resource.toURI());
- configurator.loadConfig(f);
- return configurator;
- } catch (Exception e) {
- e.printStackTrace();
- return null;
- }
- }
-
-
- /**
- * Validate RSTR without RequestedSecurityToken element
- */
- @org.junit.Test
- public void validateRSTRWithoutToken() throws Exception {
- Document doc = STSUtil.toSOAPPart(STSUtil.SAMPLE_RSTR_COLL_MSG);
-
- FedizRequest wfReq = new FedizRequest();
- wfReq.setAction(FederationConstants.ACTION_SIGNIN);
- wfReq.setResponseToken(DOM2Writer.nodeToString(doc));
-
- configurator = null;
- FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
-
- FedizProcessor wfProc = new FederationProcessorImpl();
- try {
- wfProc.processRequest(wfReq, config);
- fail("Failure expected on missing security token in RSTR");
- } catch (ProcessingException ex) {
- if (!TYPE.BAD_REQUEST.equals(ex.getType())) {
- fail("Expected ProcessingException with BAD_REQUEST type");
- }
- }
- }
-
- /**
- * Validate FederationRequest with unknown action
- */
- @org.junit.Test
- public void validateRequestUnknownAction() throws Exception {
- Document doc = STSUtil.toSOAPPart(STSUtil.SAMPLE_RSTR_COLL_MSG);
-
- FedizRequest wfReq = new FedizRequest();
- wfReq.setAction("gugus");
- wfReq.setResponseToken(DOM2Writer.nodeToString(doc));
-
- configurator = null;
- FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
-
- FedizProcessor wfProc = new FederationProcessorImpl();
- try {
- wfProc.processRequest(wfReq, config);
- fail("Failure expected due to invalid action");
- } catch (ProcessingException ex) {
- if (!TYPE.INVALID_REQUEST.equals(ex.getType())) {
- fail("Expected ProcessingException with INVALID_REQUEST type");
- }
- }
- }
-
- /**
- *Validate FederationRequest with invalid RSTR/wresult
- */
- @org.junit.Test
- public void validateSignInInvalidWResult() throws Exception {
- FedizRequest wfReq = new FedizRequest();
- wfReq.setAction(FederationConstants.ACTION_SIGNIN);
- wfReq.setResponseToken("gugus");
-
- configurator = null;
- FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
-
- FedizProcessor wfProc = new FederationProcessorImpl();
- try {
- wfProc.processRequest(wfReq, config);
- fail("Failure expected due to invalid wresult");
- } catch (ProcessingException ex) {
- if (!TYPE.INVALID_REQUEST.equals(ex.getType())) {
- fail("Expected ProcessingException with INVALID_REQUEST type");
- }
- }
- }
-
- @org.junit.Test
- public void validateTokenAndCreateMetadata() throws Exception {
- validateSAML2Token();
- FederationMetaDataTest other = new FederationMetaDataTest();
- other.validateMetaDataWithAlias();
- }
-
- /**
- * Validate SAML 2 token which includes the role attribute with 2 values
- * Roles are encoded as a multi-value saml attribute
- */
- @org.junit.Test
- public void validateSAML2Token() throws Exception {
- SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
- callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
- callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
- callbackHandler.setIssuer(TEST_RSTR_ISSUER);
- callbackHandler.setSubjectName(TEST_USER);
- ConditionsBean cp = new ConditionsBean();
- AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
- audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
- cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
- callbackHandler.setConditions(cp);
-
- SAMLCallback samlCallback = new SAMLCallback();
- SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
- SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
- String rstr = createSamlToken(assertion, "mystskey", true);
-
- FedizRequest wfReq = new FedizRequest();
- wfReq.setAction(FederationConstants.ACTION_SIGNIN);
- wfReq.setResponseToken(rstr);
-
- configurator = null;
- FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
-
- FedizProcessor wfProc = new FederationProcessorImpl();
- FedizResponse wfRes = wfProc.processRequest(wfReq, config);
-
- Assert.assertEquals("Principal name wrong", TEST_USER,
- wfRes.getUsername());
- Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
- Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles()
- .size());
- Assert.assertEquals("Audience wrong", TEST_AUDIENCE, wfRes.getAudience());
- assertClaims(wfRes.getClaims(), callbackHandler.getRoleAttributeName());
-
- }
-
- /**
- * Validate SAML 2 token which includes the role attribute with 2 values
- * Roles are encoded as a multi-value saml attribute
- * Not RequestedSecurityTokenCollection in this test, default in all others
- */
- @org.junit.Test
- public void validateSAML2TokenRSTR() throws Exception {
- SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
- callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
- callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
- callbackHandler.setIssuer(TEST_RSTR_ISSUER);
- callbackHandler.setSubjectName(TEST_USER);
- ConditionsBean cp = new ConditionsBean();
- AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
- audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
- cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
- callbackHandler.setConditions(cp);
-
- SAMLCallback samlCallback = new SAMLCallback();
- SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
- SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
- String rstr = createSamlToken(assertion, "mystskey", true, STSUtil.SAMPLE_RSTR_MSG);
-
- FedizRequest wfReq = new FedizRequest();
- wfReq.setAction(FederationConstants.ACTION_SIGNIN);
- wfReq.setResponseToken(rstr);
-
- configurator = null;
- FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
-
- FedizProcessor wfProc = new FederationProcessorImpl();
- FedizResponse wfRes = wfProc.processRequest(wfReq, config);
-
- Assert.assertEquals("Principal name wrong", TEST_USER,
- wfRes.getUsername());
- Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
- Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles()
- .size());
- Assert.assertEquals("Audience wrong", TEST_AUDIENCE, wfRes.getAudience());
- }
-
- /**
- * Validate SAML 2 token which doesn't include the role SAML attribute
- */
- @org.junit.Test
- public void validateSAML2TokenWithoutRoles() throws Exception {
- SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
- callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
- callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
- callbackHandler.setIssuer(TEST_RSTR_ISSUER);
- callbackHandler.setSubjectName(TEST_USER);
- callbackHandler.setRoles(null);
- ConditionsBean cp = new ConditionsBean();
- AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
- audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
- cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
- callbackHandler.setConditions(cp);
-
- SAMLCallback samlCallback = new SAMLCallback();
- SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
- SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
- String rstr = createSamlToken(assertion, "mystskey", true);
-
- FedizRequest wfReq = new FedizRequest();
- wfReq.setAction(FederationConstants.ACTION_SIGNIN);
- wfReq.setResponseToken(rstr);
-
- configurator = null;
- FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
-
- FedizProcessor wfProc = new FederationProcessorImpl();
- FedizResponse wfRes = wfProc.processRequest(wfReq, config);
-
- Assert.assertEquals("Principal name wrong", TEST_USER,
- wfRes.getUsername());
- Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
- Assert.assertEquals("No roles must be found", null, wfRes.getRoles());
- Assert.assertEquals("Audience wrong", TEST_AUDIENCE, wfRes.getAudience());
- }
-
- /**
- * Validate SAML 2 token where role information is provided
- * within another SAML attribute
- */
- @org.junit.Test
- public void validateSAML2TokenDifferentRoleURI() throws Exception {
- SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
- callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
- callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
- callbackHandler.setIssuer(TEST_RSTR_ISSUER);
- callbackHandler.setSubjectName(TEST_USER);
- callbackHandler.setRoleAttributeName("http://schemas.mycompany.com/claims/role");
- ConditionsBean cp = new ConditionsBean();
- AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
- audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
- cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
- callbackHandler.setConditions(cp);
-
- SAMLCallback samlCallback = new SAMLCallback();
- SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
- SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
- String rstr = createSamlToken(assertion, "mystskey", true);
-
- FedizRequest wfReq = new FedizRequest();
- wfReq.setAction(FederationConstants.ACTION_SIGNIN);
- wfReq.setResponseToken(rstr);
-
- configurator = null;
- FedizContext config = getFederationConfigurator().getFedizContext("CUSTOMROLEURI");
-
- FedizProcessor wfProc = new FederationProcessorImpl();
- FedizResponse wfRes = wfProc.processRequest(wfReq, config);
-
- Assert.assertEquals("Principal name wrong", TEST_USER, wfRes.getUsername());
- Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
- Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles().size());
- Assert.assertEquals("Audience wrong", TEST_AUDIENCE, wfRes.getAudience());
- assertClaims(wfRes.getClaims(), callbackHandler.getRoleAttributeName());
- }
-
- /**
- * Validate SAML 1 token where role information is provided
- * within another SAML attribute
- */
- @org.junit.Test
- public void validateSAML1TokenDifferentRoleURI() throws Exception {
- SAML1CallbackHandler callbackHandler = new SAML1CallbackHandler();
- callbackHandler.setStatement(SAML1CallbackHandler.Statement.ATTR);
- callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
- callbackHandler.setIssuer(TEST_RSTR_ISSUER);
- callbackHandler.setSubjectName(TEST_USER);
- callbackHandler.setRoleAttributeName("http://schemas.mycompany.com/claims/role");
- ConditionsBean cp = new ConditionsBean();
- AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
- audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
- cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
- callbackHandler.setConditions(cp);
-
- SAMLCallback samlCallback = new SAMLCallback();
- SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
- SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
- String rstr = createSamlToken(assertion, "mystskey", true);
-
- FedizRequest wfReq = new FedizRequest();
- wfReq.setAction(FederationConstants.ACTION_SIGNIN);
- wfReq.setResponseToken(rstr);
-
- configurator = null;
- FedizContext config = getFederationConfigurator().getFedizContext("CUSTOMROLEURI");
-
- FedizProcessor wfProc = new FederationProcessorImpl();
- FedizResponse wfRes = wfProc.processRequest(wfReq, config);
-
- Assert.assertEquals("Principal name wrong", TEST_USER, wfRes.getUsername());
- Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
- Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles().size());
- Assert.assertEquals("Audience wrong", TEST_AUDIENCE, wfRes.getAudience());
- assertClaims(wfRes.getClaims(), callbackHandler.getRoleAttributeName());
- }
-
- /**
- * Validate SAML 2 token which includes role attribute
- * but RoleURI is not configured
- */
- @org.junit.Test
- public void validateSAML2TokenRoleURINotConfigured() throws Exception {
- SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
- callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
- callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
- callbackHandler.setIssuer(TEST_RSTR_ISSUER);
- callbackHandler.setSubjectName(TEST_USER);
- ConditionsBean cp = new ConditionsBean();
- AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
- audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
- cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
- callbackHandler.setConditions(cp);
-
- SAMLCallback samlCallback = new SAMLCallback();
- SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
- SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
- String rstr = createSamlToken(assertion, "mystskey", true);
-
- FedizRequest wfReq = new FedizRequest();
- wfReq.setAction(FederationConstants.ACTION_SIGNIN);
- wfReq.setResponseToken(rstr);
-
- configurator = null;
- FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
- config.getProtocol().setRoleURI(null);
-
- FedizProcessor wfProc = new FederationProcessorImpl();
- FedizResponse wfRes = wfProc.processRequest(wfReq, config);
-
- Assert.assertEquals("Principal name wrong", TEST_USER,
- wfRes.getUsername());
- Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
- Assert.assertEquals("Two roles must be found", null, wfRes.getRoles());
- Assert.assertEquals("Audience wrong", TEST_AUDIENCE, wfRes.getAudience());
- }
-
- /**
- * Validate SAML 1.1 token which includes the role attribute with 2 values
- * Roles are encoded as a multi-value saml attribute
- */
- @org.junit.Test
- public void validateSAML1Token() throws Exception {
- SAML1CallbackHandler callbackHandler = new SAML1CallbackHandler();
- callbackHandler.setStatement(SAML1CallbackHandler.Statement.ATTR);
- callbackHandler.setConfirmationMethod(SAML1Constants.CONF_BEARER);
- callbackHandler.setIssuer(TEST_RSTR_ISSUER);
- callbackHandler.setSubjectName(TEST_USER);
- ConditionsBean cp = new ConditionsBean();
- AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
- audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
- cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
- callbackHandler.setConditions(cp);
-
- SAMLCallback samlCallback = new SAMLCallback();
- SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
- SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
- String rstr = createSamlToken(assertion, "mystskey", true);
-
- FedizRequest wfReq = new FedizRequest();
- wfReq.setAction(FederationConstants.ACTION_SIGNIN);
- wfReq.setResponseToken(rstr);
-
- configurator = null;
- FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
-
- FedizProcessor wfProc = new FederationProcessorImpl();
- FedizResponse wfRes = wfProc.processRequest(wfReq, config);
-
- Assert.assertEquals("Principal name wrong", TEST_USER,
- wfRes.getUsername());
- Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
- Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles()
- .size());
- Assert.assertEquals("Audience wrong", TEST_AUDIENCE, wfRes.getAudience());
- assertClaims(wfRes.getClaims(), callbackHandler.getRoleAttributeName());
- }
-
- /**
- * Validate SAML 1.1 token which includes the role attribute with 2 values
- * Roles are encoded as a multi-value saml attribute
- * Token embedded in RSTR 2005/02 - WS Federation 1.0
- */
- @org.junit.Test
- public void validateSAML1TokenWSFed10() throws Exception {
- SAML1CallbackHandler callbackHandler = new SAML1CallbackHandler();
- callbackHandler.setStatement(SAML1CallbackHandler.Statement.ATTR);
- callbackHandler.setConfirmationMethod(SAML1Constants.CONF_BEARER);
- callbackHandler.setIssuer(TEST_RSTR_ISSUER);
- callbackHandler.setSubjectName(TEST_USER);
- ConditionsBean cp = new ConditionsBean();
- AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
- audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
- cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
- callbackHandler.setConditions(cp);
-
- SAMLCallback samlCallback = new SAMLCallback();
- SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
- SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
- String rstr = createSamlToken(assertion, "mystskey", true, STSUtil.SAMPLE_RSTR_2005_02_MSG);
- FedizRequest wfReq = new FedizRequest();
- wfReq.setAction(FederationConstants.ACTION_SIGNIN);
- wfReq.setResponseToken(rstr);
-
- configurator = null;
- FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
-
- FedizProcessor wfProc = new FederationProcessorImpl();
- FedizResponse wfRes = wfProc.processRequest(wfReq, config);
-
- Assert.assertEquals("Principal name wrong", TEST_USER,
- wfRes.getUsername());
- Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
- Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles()
- .size());
- Assert.assertEquals("Audience wrong", TEST_AUDIENCE, wfRes.getAudience());
- }
-
- /**
- * Validate SAML 2 token which includes the role attribute with 2 values
- * Roles are encoded as a multiple saml attributes with the same name
- */
- @org.junit.Test
- public void validateSAML2TokenRoleMultiAttributes() throws Exception {
- SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
- callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
- callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
- callbackHandler.setIssuer(TEST_RSTR_ISSUER);
- callbackHandler.setSubjectName(TEST_USER);
- callbackHandler.setMultiValueType(MultiValue.MULTI_ATTR);
- ConditionsBean cp = new ConditionsBean();
- AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
- audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
- cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
- callbackHandler.setConditions(cp);
-
- SAMLCallback samlCallback = new SAMLCallback();
- SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
- SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
- String rstr = createSamlToken(assertion, "mystskey", true);
-
- FedizRequest wfReq = new FedizRequest();
- wfReq.setAction(FederationConstants.ACTION_SIGNIN);
- wfReq.setResponseToken(rstr);
-
- configurator = null;
- FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
-
- FedizProcessor wfProc = new FederationProcessorImpl();
- FedizResponse wfRes = wfProc.processRequest(wfReq, config);
-
- Assert.assertEquals("Principal name wrong", TEST_USER,
- wfRes.getUsername());
- Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
- Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles()
- .size());
- assertClaims(wfRes.getClaims(), callbackHandler.getRoleAttributeName());
- }
-
- /**
- * Validate SAML 2 token which includes the role attribute with 2 values
- * Roles are encoded as a single saml attribute with encoded value
- */
- @org.junit.Test
- public void validateSAML2TokenRoleEncodedValue() throws Exception {
- SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
- callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
- callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
- callbackHandler.setIssuer(TEST_RSTR_ISSUER);
- callbackHandler.setSubjectName(TEST_USER);
- callbackHandler.setMultiValueType(MultiValue.ENC_VALUE);
- ConditionsBean cp = new ConditionsBean();
- AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
- audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
- cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
- callbackHandler.setConditions(cp);
-
- SAMLCallback samlCallback = new SAMLCallback();
- SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
- SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
- String rstr = createSamlToken(assertion, "mystskey", true);
-
- FedizRequest wfReq = new FedizRequest();
- wfReq.setAction(FederationConstants.ACTION_SIGNIN);
- wfReq.setResponseToken(rstr);
-
- configurator = null;
- FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
- FederationProtocol fp = (FederationProtocol)config.getProtocol();
- fp.setRoleDelimiter(",");
-
- FedizProcessor wfProc = new FederationProcessorImpl();
- FedizResponse wfRes = wfProc.processRequest(wfReq, config);
-
- Assert.assertEquals("Principal name wrong", TEST_USER,
- wfRes.getUsername());
- Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
- Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles()
- .size());
- assertClaims(wfRes.getClaims(), callbackHandler.getRoleAttributeName());
- }
-
- /**
- * Validate SAML 2 token which includes the role attribute with 2 values
- * The configured subject of the trusted issuer doesn't match with
- * the issuer of the SAML token
- *
- * Ignored because PeerTrust ignores subject attribute
- */
- @org.junit.Test
- @org.junit.Ignore
- public void validateSAML2TokenUntrustedIssuer() throws Exception {
- SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
- callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
- callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
- callbackHandler.setIssuer(TEST_RSTR_ISSUER);
- callbackHandler.setSubjectName(TEST_USER);
- ConditionsBean cp = new ConditionsBean();
- AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
- audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
- cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
- callbackHandler.setConditions(cp);
-
- SAMLCallback samlCallback = new SAMLCallback();
- SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
- SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
-
- String rstr = createSamlToken(assertion, "mystskey", true);
- FedizRequest wfReq = new FedizRequest();
- wfReq.setAction(FederationConstants.ACTION_SIGNIN);
- wfReq.setResponseToken(rstr);
-
- // Load and update the config to enforce an error
- configurator = null;
- FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
- config.getTrustedIssuers().get(0).setSubject("wrong-issuer-name");
-
- FedizProcessor wfProc = new FederationProcessorImpl();
- try {
- wfProc.processRequest(wfReq, config);
- Assert.fail("Processing must fail because of untrusted issuer configured");
- } catch (ProcessingException ex) {
- if (!TYPE.ISSUER_NOT_TRUSTED.equals(ex.getType())) {
- fail("Expected ProcessingException with ISSUER_NOT_TRUSTED type");
- }
- }
- }
-
- /**
- * Validate SAML 2 token which includes the role attribute with 2 values
- * The configured subject of the trusted issuer doesn't match with
- * the issuer of the SAML token
- */
- @org.junit.Test
- public void validateUnsignedSAML2Token() throws Exception {
- SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
- callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
- callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
- callbackHandler.setIssuer(TEST_RSTR_ISSUER);
- callbackHandler.setSubjectName(TEST_USER);
- ConditionsBean cp = new ConditionsBean();
- AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
- audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
- cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
- callbackHandler.setConditions(cp);
-
- SAMLCallback samlCallback = new SAMLCallback();
- SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
- SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
-
- String rstr = createSamlToken(assertion, "mystskey", false);
- FedizRequest wfReq = new FedizRequest();
- wfReq.setAction(FederationConstants.ACTION_SIGNIN);
- wfReq.setResponseToken(rstr);
-
- // Load and update the config to enforce an error
- configurator = null;
- FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
-
- FedizProcessor wfProc = new FederationProcessorImpl();
- try {
- wfProc.processRequest(wfReq, config);
- Assert.fail("Processing must fail because of missing signature");
- } catch (ProcessingException ex) {
- if (!TYPE.TOKEN_NO_SIGNATURE.equals(ex.getType())) {
- fail("Expected ProcessingException with TOKEN_NO_SIGNATURE type");
- }
- }
- }
-
- /**
- * Validate SAML 2 token twice which causes an exception
- * due to replay attack
- */
- @org.junit.Test
- public void testReplayAttack() throws Exception {
- SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
- callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
- callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
- callbackHandler.setIssuer(TEST_RSTR_ISSUER);
- callbackHandler.setSubjectName(TEST_USER);
- ConditionsBean cp = new ConditionsBean();
- AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
- audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
- cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
- callbackHandler.setConditions(cp);
-
- SAMLCallback samlCallback = new SAMLCallback();
- SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
- SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
-
- String rstr = createSamlToken(assertion, "mystskey", true);
-
- FedizRequest wfReq = new FedizRequest();
- wfReq.setAction(FederationConstants.ACTION_SIGNIN);
- wfReq.setResponseToken(rstr);
-
- configurator = null;
- FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
-
- FedizProcessor wfProc = new FederationProcessorImpl();
- FedizResponse wfRes = wfProc.processRequest(wfReq, config);
- Assert.assertEquals("Principal name wrong", TEST_USER,
- wfRes.getUsername());
- Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
-
- wfProc = new FederationProcessorImpl();
- try {
- wfProc.processRequest(wfReq, config);
- fail("Failure expected on a replay attack");
- } catch (ProcessingException ex) {
- if (!TYPE.TOKEN_REPLAY.equals(ex.getType())) {
- fail("Expected ProcessingException with TOKEN_REPLAY type");
- }
- }
- }
-
-
- /**
- * Validate SAML 2 token which includes the role attribute with 2 values
- * The configured subject of the trusted issuer doesn't match with
- * the issuer of the SAML token
- */
- @org.junit.Test
- public void validateSAML2TokenSeveralCertStore() throws Exception {
- SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
- callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
- callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
- callbackHandler.setIssuer(TEST_RSTR_ISSUER);
- callbackHandler.setSubjectName(TEST_USER);
- ConditionsBean cp = new ConditionsBean();
- AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
- audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
- cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
- callbackHandler.setConditions(cp);
-
- SAMLCallback samlCallback = new SAMLCallback();
- SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
- SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
-
- String rstr = createSamlToken(assertion, "mystskey", true);
- FedizRequest wfReq = new FedizRequest();
- wfReq.setAction(FederationConstants.ACTION_SIGNIN);
- wfReq.setResponseToken(rstr);
-
- // Load and update the config to enforce an error
- configurator = null;
- FedizContext config = getFederationConfigurator().getFedizContext("ROOT2");
-
- FedizProcessor wfProc = new FederationProcessorImpl();
- FedizResponse wfRes = wfProc.processRequest(wfReq, config);
-
- Assert.assertEquals("Principal name wrong", TEST_USER,
- wfRes.getUsername());
- Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
- Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles()
- .size());
- }
-
- /**
- * Validate SAML 2 token which includes the role attribute with 2 values
- * The configured subject of the trusted issuer doesn't match with
- * the issuer of the SAML token
- */
- @org.junit.Test
- public void validateSAML2TokenSeveralCertStoreTrustedIssuer() throws Exception {
- SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
- callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
- callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
- callbackHandler.setIssuer(TEST_RSTR_ISSUER);
- callbackHandler.setSubjectName(TEST_USER);
- ConditionsBean cp = new ConditionsBean();
- AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
- audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
- cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
- callbackHandler.setConditions(cp);
-
- SAMLCallback samlCallback = new SAMLCallback();
- SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
- SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
-
- String rstr = createSamlToken(assertion, "mystskey", true);
- FedizRequest wfReq = new FedizRequest();
- wfReq.setAction(FederationConstants.ACTION_SIGNIN);
- wfReq.setResponseToken(rstr);
-
- // Load and update the config to enforce an error
- configurator = null;
- FedizContext config = getFederationConfigurator().getFedizContext("ROOT3");
-
- FedizProcessor wfProc = new FederationProcessorImpl();
- FedizResponse wfRes = wfProc.processRequest(wfReq, config);
-
- Assert.assertEquals("Principal name wrong", TEST_USER,
- wfRes.getUsername());
- Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
- Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles()
- .size());
- }
-
- /**
- * Validate SAML 2 token which is expired
- */
- @org.junit.Test
- public void validateSAML2TokenExpired() throws Exception {
- SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
- callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
- callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
- callbackHandler.setIssuer(TEST_RSTR_ISSUER);
- callbackHandler.setSubjectName(TEST_USER);
- ConditionsBean cp = new ConditionsBean();
- DateTime currentTime = new DateTime();
- currentTime = currentTime.minusSeconds(60);
- cp.setNotAfter(currentTime);
- currentTime = new DateTime();
- currentTime = currentTime.minusSeconds(300);
- cp.setNotBefore(currentTime);
- AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
- audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
- cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
- callbackHandler.setConditions(cp);
-
- SAMLCallback samlCallback = new SAMLCallback();
- SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
- SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
-
- String rstr = createSamlToken(assertion, "mystskey", true);
-
- FedizRequest wfReq = new FedizRequest();
- wfReq.setAction(FederationConstants.ACTION_SIGNIN);
- wfReq.setResponseToken(rstr);
-
- configurator = null;
- FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
-
- FedizProcessor wfProc = new FederationProcessorImpl();
- try {
- wfProc.processRequest(wfReq, config);
- fail("Failure expected on expired SAML token");
- } catch (ProcessingException ex) {
- if (!TYPE.TOKEN_EXPIRED.equals(ex.getType())) {
- fail("Expected ProcessingException with TOKEN_EXPIRED type");
- }
- }
- }
-
- /**
- * Validate SAML 2 token which is not yet valid (in 30 seconds)
- * but within the maximum clock skew range (60 seconds)
- */
- @org.junit.Test
- public void validateSAML2TokenClockSkewRange() throws Exception {
- SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
- callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
- callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
- callbackHandler.setIssuer(TEST_RSTR_ISSUER);
- callbackHandler.setSubjectName(TEST_USER);
- ConditionsBean cp = new ConditionsBean();
- DateTime currentTime = new DateTime();
- currentTime = currentTime.plusSeconds(300);
- cp.setNotAfter(currentTime);
- currentTime = new DateTime();
- currentTime = currentTime.plusSeconds(30);
- cp.setNotBefore(currentTime);
- AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
- audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
- cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
- callbackHandler.setConditions(cp);
-
- SAMLCallback samlCallback = new SAMLCallback();
- SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
- SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
-
- String rstr = createSamlToken(assertion, "mystskey", true);
-
- FedizRequest wfReq = new FedizRequest();
- wfReq.setAction(FederationConstants.ACTION_SIGNIN);
- wfReq.setResponseToken(rstr);
-
- configurator = null;
- FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
- config.setMaximumClockSkew(BigInteger.valueOf(60));
-
- FedizProcessor wfProc = new FederationProcessorImpl();
- FedizResponse wfRes = wfProc.processRequest(wfReq, config);
-
- Assert.assertEquals("Principal name wrong", TEST_USER,
- wfRes.getUsername());
- Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
- Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles()
- .size());
- }
-
- /**
- * "Validate" SAML 2 token with a custom token validator
- * If a validator is configured it precedes the SAMLTokenValidator as part of Fediz
- */
- @org.junit.Test
- public void validateSAML2TokenCustomValidator() throws Exception {
- SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
- callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
- callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
- callbackHandler.setIssuer(TEST_RSTR_ISSUER);
- callbackHandler.setSubjectName(TEST_USER);
- ConditionsBean cp = new ConditionsBean();
- AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
- audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
- cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
- callbackHandler.setConditions(cp);
-
- SAMLCallback samlCallback = new SAMLCallback();
- SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
- SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
-
- String rstr = createSamlToken(assertion, "mystskey", true);
-
- FedizRequest wfReq = new FedizRequest();
- wfReq.setAction(FederationConstants.ACTION_SIGNIN);
- wfReq.setResponseToken(rstr);
-
- configurator = null;
- FedizContext config = getFederationConfigurator().getFedizContext("CUSTTOK");
- Protocol protocol = config.getProtocol();
- List<TokenValidator> validators = protocol.getTokenValidators();
- Assert.assertEquals("Two validators must be found", 2, validators.size());
- Assert.assertEquals("First validator must be custom validator",
- CustomValidator.class.getName(), validators.get(0).getClass().getName());
-
- FedizProcessor wfProc = new FederationProcessorImpl();
- FedizResponse wfRes = wfProc.processRequest(wfReq, config);
-
- Assert.assertEquals("Principal name wrong", TEST_USER,
- wfRes.getUsername());
- Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
- }
-
- /**
- * "Validate" SAML 2 token with a custom token validator
- * If a validator is configured it precedes the SAMLTokenValidator as part of Fediz
- */
- @org.junit.Test
- public void validateSAML2TokenMaxClockSkewNotDefined() throws Exception {
- SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
- callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
- callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
- callbackHandler.setIssuer(TEST_RSTR_ISSUER);
- callbackHandler.setSubjectName(TEST_USER);
- ConditionsBean cp = new ConditionsBean();
- AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
- audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
- cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
- callbackHandler.setConditions(cp);
-
- SAMLCallback samlCallback = new SAMLCallback();
- SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
- SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
-
- String rstr = createSamlToken(assertion, "mystskey", true);
-
- FedizRequest wfReq = new FedizRequest();
- wfReq.setAction(FederationConstants.ACTION_SIGNIN);
- wfReq.setResponseToken(rstr);
-
- configurator = null;
- FedizContext config = getFederationConfigurator().getFedizContext("NOCLOCKSKEW");
-
- FedizProcessor wfProc = new FederationProcessorImpl();
- FedizResponse wfRes = wfProc.processRequest(wfReq, config);
-
- Assert.assertEquals("Principal name wrong", TEST_USER,
- wfRes.getUsername());
- Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
- Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles()
- .size());
- Assert.assertEquals("Audience wrong", TEST_AUDIENCE, wfRes.getAudience());
- }
-
- /**
- * Validate an encrypted SAML 2 token which includes the role attribute with 2 values
- * Roles are encoded as a multi-value saml attribute
- */
- @org.junit.Test
- public void validateEncryptedSAML2Token() throws Exception {
- SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
- callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
- callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
- callbackHandler.setIssuer(TEST_RSTR_ISSUER);
- callbackHandler.setSubjectName(TEST_USER);
- ConditionsBean cp = new ConditionsBean();
- AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
- audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
- cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
- callbackHandler.setConditions(cp);
-
- SAMLCallback samlCallback = new SAMLCallback();
- SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
- SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
-
- String rstr = encryptAndSignToken(assertion);
-
- FedizRequest wfReq = new FedizRequest();
- wfReq.setAction(FederationConstants.ACTION_SIGNIN);
- wfReq.setResponseToken(rstr);
-
- configurator = null;
- FedizContext config =
- getFederationConfigurator().getFedizContext("ROOT_DECRYPTION");
-
- FedizProcessor wfProc = new FederationProcessorImpl();
- FedizResponse wfRes = wfProc.processRequest(wfReq, config);
-
- Assert.assertEquals("Principal name wrong", TEST_USER,
- wfRes.getUsername());
- Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
- Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles()
- .size());
- Assert.assertEquals("Audience wrong", TEST_AUDIENCE, wfRes.getAudience());
- assertClaims(wfRes.getClaims(), callbackHandler.getRoleAttributeName());
- }
-
- /**
- * Validate a HolderOfKey SAML 2 token
- */
- @org.junit.Test
- public void validateHOKSAML2Token() throws Exception {
- SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
- callbackHandler.setStatement(SAML2CallbackHandler.Statement.AUTHN);
- callbackHandler.setConfirmationMethod(SAML2Constants.CONF_HOLDER_KEY);
- callbackHandler.setIssuer(TEST_RSTR_ISSUER);
- callbackHandler.setSubjectName(TEST_USER);
- ConditionsBean cp = new ConditionsBean();
- AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
- audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
- cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
- callbackHandler.setConditions(cp);
-
- Crypto clientCrypto = CryptoFactory.getInstance("client-crypto.properties");
- CryptoType cryptoType = new CryptoType(CryptoType.TYPE.ALIAS);
- cryptoType.setAlias("myclientkey");
- X509Certificate[] certs = clientCrypto.getX509Certificates(cryptoType);
- callbackHandler.setCerts(certs);
-
- SAMLCallback samlCallback = new SAMLCallback();
- SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
- SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
-
- WSPasswordCallback[] cb = {
- new WSPasswordCallback("mystskey", WSPasswordCallback.SIGNATURE)
- };
- cbPasswordHandler.handle(cb);
- String password = cb[0].getPassword();
-
- assertion.signAssertion("mystskey", password, crypto, false);
-
- Document doc = STSUtil.toSOAPPart(STSUtil.SAMPLE_RSTR_COLL_MSG);
- Element token = assertion.toDOM(doc);
-
- Element e = FederationProcessorTest.findElement(doc, "RequestedSecurityToken",
- FederationConstants.WS_TRUST_13_NS);
- if (e == null) {
- e = FederationProcessorTest.findElement(doc, "RequestedSecurityToken",
- FederationConstants.WS_TRUST_2005_02_NS);
- }
- e.appendChild(token);
-
- String rstr = DOM2Writer.nodeToString(doc);
-
- FedizRequest wfReq = new FedizRequest();
- wfReq.setAction(FederationConstants.ACTION_SIGNIN);
- wfReq.setResponseToken(rstr);
-
- configurator = null;
- FedizContext config =
- getFederationConfigurator().getFedizContext("ROOT_DECRYPTION");
-
- FedizProcessor wfProc = new FederationProcessorImpl();
- try {
- wfProc.processRequest(wfReq, config);
- fail("Failure expected on missing client certs");
- } catch (ProcessingException ex) {
- // expected
- }
-
- // Now set client certs
- wfReq.setCerts(certs);
- wfProc.processRequest(wfReq, config);
- }
-
- @org.junit.Test
- public void validateSAML2TokenWithConfigCreatedWithAPI() throws Exception {
-
- ContextConfig config = new ContextConfig();
-
- config.setName("whatever");
-
- // Configure certificate store
- CertificateStores certStores = new CertificateStores();
- TrustManagersType tm0 = new TrustManagersType();
- KeyStoreType ks0 = new KeyStoreType();
- ks0.setType("JKS");
- ks0.setPassword("storepass");
- ks0.setFile("ststrust.jks");
- tm0.setKeyStore(ks0);
- certStores.getTrustManager().add(tm0);
- config.setCertificateStores(certStores);
-
- // Configure trusted IDP
- TrustedIssuers trustedIssuers = new TrustedIssuers();
- TrustedIssuerType ti0 = new TrustedIssuerType();
- ti0.setCertificateValidation(ValidationType.CHAIN_TRUST);
- ti0.setName("FedizSTSIssuer");
- ti0.setSubject(".*CN=www.sts.com.*");
- trustedIssuers.getIssuer().add(ti0);
- config.setTrustedIssuers(trustedIssuers);
-
- FederationProtocolType protocol = new FederationProtocolType();
- config.setProtocol(protocol);
-
- AudienceUris audienceUris = new AudienceUris();
- audienceUris.getAudienceItem().add("https://localhost/fedizhelloworld");
- config.setAudienceUris(audienceUris);
-
- protocol.setRoleURI("http://schemas.xmlsoap.org/ws/2005/05/identity/claims/role");
-
- FedizContext fedContext = new FedizContext(config);
-
- SAML2CallbackHandler callbackHandler = new SAML2CallbackHandler();
- callbackHandler.setStatement(SAML2CallbackHandler.Statement.ATTR);
- callbackHandler.setConfirmationMethod(SAML2Constants.CONF_BEARER);
- callbackHandler.setIssuer(TEST_RSTR_ISSUER);
- callbackHandler.setSubjectName(TEST_USER);
- ConditionsBean cp = new ConditionsBean();
- AudienceRestrictionBean audienceRestriction = new AudienceRestrictionBean();
- audienceRestriction.getAudienceURIs().add(TEST_AUDIENCE);
- cp.setAudienceRestrictions(Collections.singletonList(audienceRestriction));
- callbackHandler.setConditions(cp);
-
- SAMLCallback samlCallback = new SAMLCallback();
- SAMLUtil.doSAMLCallback(callbackHandler, samlCallback);
- SamlAssertionWrapper assertion = new SamlAssertionWrapper(samlCallback);
-
- String rstr = createSamlToken(assertion, "mystskey", true, STSUtil.SAMPLE_RSTR_MSG);
-
- FedizRequest wfReq = new FedizRequest();
- wfReq.setAction(FederationConstants.ACTION_SIGNIN);
- wfReq.setResponseToken(rstr);
-
- FedizProcessor wfProc = new FederationProcessorImpl();
- FedizResponse wfRes = wfProc.processRequest(wfReq, fedContext);
-
- Assert.assertEquals("Principal name wrong", TEST_USER,
- wfRes.getUsername());
- Assert.assertEquals("Issuer wrong", TEST_RSTR_ISSUER, wfRes.getIssuer());
- Assert.assertEquals("Two roles must be found", 2, wfRes.getRoles()
- .size());
- Assert.assertEquals("Audience wrong", TEST_AUDIENCE, wfRes.getAudience());
-
- fedContext.close();
-
- }
-
-
- private String encryptAndSignToken(
- SamlAssertionWrapper assertion
- ) throws Exception {
-
- WSPasswordCallback[] cb = {
- new WSPasswordCallback("mystskey", WSPasswordCallback.SIGNATURE)
- };
- cbPasswordHandler.handle(cb);
- String password = cb[0].getPassword();
-
- assertion.signAssertion("mystskey", password, crypto, false);
-
- Document doc = STSUtil.toSOAPPart(STSUtil.SAMPLE_RSTR_COLL_MSG);
- Element token = assertion.toDOM(doc);
-
- Element e = FederationProcessorTest.findElement(doc, "RequestedSecurityToken",
- FederationConstants.WS_TRUST_13_NS);
- if (e == null) {
- e = FederationProcessorTest.findElement(doc, "RequestedSecurityToken",
- FederationConstants.WS_TRUST_2005_02_NS);
- }
- e.appendChild(token);
-
- WSSecEncrypt builder = new WSSecEncrypt();
- builder.setUserInfo("mystskey");
-
- builder.setKeyIdentifierType(WSConstants.ISSUER_SERIAL);
- builder.setSymmetricEncAlgorithm(WSConstants.AES_128);
- builder.setKeyEncAlgo(WSConstants.KEYTRANSPORT_RSA15);
- builder.setEmbedEncryptedKey(true);
-
- WSEncryptionPart encryptionPart = new WSEncryptionPart(assertion.getId(), "Element");
- encryptionPart.setElement(token);
-
- Crypto encrCrypto = CryptoFactory.getInstance("signature.properties");
- builder.prepare(token.getOwnerDocument(), encrCrypto);
- builder.encryptForRef(null, Collections.singletonList(encryptionPart));
-
- // return doc.getDocumentElement();
- return DOM2Writer.nodeToString(doc);
- }
-
- private String createSamlToken(SamlAssertionWrapper assertion, String alias, boolean sign)
- throws IOException, UnsupportedCallbackException, WSSecurityException, Exception {
- return createSamlToken(assertion, alias, sign, STSUtil.SAMPLE_RSTR_COLL_MSG);
- }
-
- private String createSamlToken(SamlAssertionWrapper assertion, String alias, boolean sign, String rstr)
- throws IOException, UnsupportedCallbackException, WSSecurityException, Exception {
- WSPasswordCallback[] cb = {
- new WSPasswordCallback(alias, WSPasswordCallback.SIGNATURE)
- };
- cbPasswordHandler.handle(cb);
- String password = cb[0].getPassword();
-
- if (sign) {
- assertion.signAssertion(alias, password, crypto, false);
- }
- Document doc = STSUtil.toSOAPPart(rstr);
- Element token = assertion.toDOM(doc);
-
- Element e = FederationProcessorTest.findElement(doc, "RequestedSecurityToken",
- FederationConstants.WS_TRUST_13_NS);
- if (e == null) {
- e = FederationProcessorTest.findElement(doc, "RequestedSecurityToken",
- FederationConstants.WS_TRUST_2005_02_NS);
- }
- e.appendChild(token);
- return DOM2Writer.nodeToString(doc);
- }
-
-
-
-
- /**
- * Returns the first element that matches <code>name</code> and
- * <code>namespace</code>. <p/> This is a replacement for a XPath lookup
- * <code>//name</code> with the given namespace. It's somewhat faster than
- * XPath, and we do not deal with prefixes, just with the real namespace URI
- *
- * @param startNode Where to start the search
- * @param name Local name of the element
- * @param namespace Namespace URI of the element
- * @return The found element or <code>null</code>
- */
- public static Element findElement(Node startNode, String name, String namespace) {
- //
- // Replace the formerly recursive implementation with a depth-first-loop
- // lookup
- //
- if (startNode == null) {
- return null;
- }
- Node startParent = startNode.getParentNode();
- Node processedNode = null;
-
- while (startNode != null) {
- // start node processing at this point
- if (startNode.getNodeType() == Node.ELEMENT_NODE
- && startNode.getLocalName().equals(name)) {
- String ns = startNode.getNamespaceURI();
- if (ns != null && ns.equals(namespace)) {
- return (Element)startNode;
- }
-
- if ((namespace == null || namespace.length() == 0)
- && (ns == null || ns.length() == 0)) {
- return (Element)startNode;
- }
- }
- processedNode = startNode;
- startNode = startNode.getFirstChild();
-
- // no child, this node is done.
- if (startNode == null) {
- // close node processing, get sibling
- startNode = processedNode.getNextSibling();
- }
- // no more siblings, get parent, all children
- // of parent are processed.
- while (startNode == null) {
- processedNode = processedNode.getParentNode();
- if (processedNode == startParent) {
- return null;
- }
- // close parent node processing (processed node now)
- startNode = processedNode.getNextSibling();
- }
- }
- return null;
- }
-
- private void assertClaims(List<Claim> claims, String roleClaimType) {
- for (Claim c : claims) {
- Assert.assertTrue("Invalid ClaimType URI: " + c.getClaimType(),
- c.getClaimType().equals(roleClaimType)
- || c.getClaimType().equals(ClaimTypes.COUNTRY)
- || c.getClaimType().equals(AbstractSAMLCallbackHandler.CLAIM_TYPE_LANGUAGE)
- );
- }
- }
-
-
-}
http://git-wip-us.apache.org/repos/asf/cxf-fediz/blob/4a0b8649/plugins/core/src/test/java/org/apache/cxf/fediz/core/federation/FederationRequestTest.java
----------------------------------------------------------------------
diff --git a/plugins/core/src/test/java/org/apache/cxf/fediz/core/federation/FederationRequestTest.java b/plugins/core/src/test/java/org/apache/cxf/fediz/core/federation/FederationRequestTest.java
new file mode 100644
index 0000000..b48521b
--- /dev/null
+++ b/plugins/core/src/test/java/org/apache/cxf/fediz/core/federation/FederationRequestTest.java
@@ -0,0 +1,130 @@
+/**
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements. See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership. The ASF licenses this file
+ * to you 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 org.apache.cxf.fediz.core.federation;
+
+import java.io.File;
+import java.net.URL;
+
+import javax.servlet.http.HttpServletRequest;
+import javax.xml.parsers.DocumentBuilderFactory;
+
+import org.apache.cxf.fediz.common.SecurityTestUtil;
+import org.apache.cxf.fediz.core.config.FedizConfigurator;
+import org.apache.cxf.fediz.core.config.FedizContext;
+import org.apache.cxf.fediz.core.processor.FederationProcessorImpl;
+import org.apache.cxf.fediz.core.processor.FedizProcessor;
+import org.apache.cxf.fediz.core.processor.RedirectionResponse;
+import org.easymock.EasyMock;
+import org.junit.AfterClass;
+import org.junit.Assert;
+import org.junit.BeforeClass;
+
+/**
+ * Some tests for creating WS-Federation requests using the FederationProcessorImpl
+ */
+public class FederationRequestTest {
+ static final String TEST_USER = "alice";
+ static final String TEST_RSTR_ISSUER = "FedizSTSIssuer";
+ static final String TEST_REQUEST_URL = "https://localhost/fedizhelloworld/";
+ static final String TEST_REQUEST_URI = "/fedizhelloworld";
+ static final String TEST_IDP_ISSUER = "http://url_to_the_issuer";
+
+ private static final String CONFIG_FILE = "fediz_test_config.xml";
+
+ private static FedizConfigurator configurator;
+ private static DocumentBuilderFactory docBuilderFactory;
+
+ static {
+ docBuilderFactory = DocumentBuilderFactory.newInstance();
+ docBuilderFactory.setNamespaceAware(true);
+ }
+
+
+ @BeforeClass
+ public static void init() {
+ getFederationConfigurator();
+ Assert.assertNotNull(configurator);
+ }
+
+ @AfterClass
+ public static void cleanup() {
+ SecurityTestUtil.cleanup();
+ }
+
+
+ private static FedizConfigurator getFederationConfigurator() {
+ if (configurator != null) {
+ return configurator;
+ }
+ try {
+ configurator = new FedizConfigurator();
+ final URL resource = Thread.currentThread().getContextClassLoader()
+ .getResource(CONFIG_FILE);
+ File f = new File(resource.toURI());
+ configurator.loadConfig(f);
+ return configurator;
+ } catch (Exception e) {
+ e.printStackTrace();
+ return null;
+ }
+ }
+
+ @org.junit.Test
+ public void createFederationSignInRequest() throws Exception {
+ // Mock up a Request
+ FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
+
+ HttpServletRequest req = EasyMock.createMock(HttpServletRequest.class);
+ EasyMock.expect(req.getRequestURL()).andReturn(new StringBuffer(TEST_REQUEST_URL)).times(1, 2);
+ EasyMock.expect(req.getContextPath()).andReturn(TEST_REQUEST_URI);
+ EasyMock.replay(req);
+
+ FedizProcessor wfProc = new FederationProcessorImpl();
+ RedirectionResponse response = wfProc.createSignInRequest(req, config);
+
+ String redirectionURL = response.getRedirectionURL();
+ Assert.assertTrue(redirectionURL.startsWith(TEST_IDP_ISSUER));
+ Assert.assertTrue(redirectionURL.contains("wa=wsignin1.0"));
+ Assert.assertTrue(redirectionURL.contains("wreq=REQUEST"));
+ Assert.assertTrue(redirectionURL.contains("wfresh=10000"));
+ Assert.assertTrue(redirectionURL.contains("wct="));
+ Assert.assertTrue(redirectionURL.contains("wtrealm=target+realm"));
+ Assert.assertTrue(redirectionURL.contains("wreply="));
+ }
+
+ @org.junit.Test
+ public void createFederationSignOutRequest() throws Exception {
+ // Mock up a Request
+ FedizContext config = getFederationConfigurator().getFedizContext("ROOT");
+
+ HttpServletRequest req = EasyMock.createMock(HttpServletRequest.class);
+ EasyMock.expect(req.getRequestURL()).andReturn(new StringBuffer(TEST_REQUEST_URL)).times(1, 2);
+ EasyMock.expect(req.getContextPath()).andReturn(TEST_REQUEST_URI);
+ EasyMock.replay(req);
+
+ FedizProcessor wfProc = new FederationProcessorImpl();
+ RedirectionResponse response = wfProc.createSignOutRequest(req, config);
+
+ String redirectionURL = response.getRedirectionURL();
+ Assert.assertTrue(redirectionURL.startsWith(TEST_IDP_ISSUER));
+ Assert.assertTrue(redirectionURL.contains("wa=wsignout1.0"));
+ }
+
+}
\ No newline at end of file