/** * 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.wss4j.dom.message; import java.util.ArrayList; import java.util.Collections; import java.util.List; import javax.security.auth.callback.CallbackHandler; import org.apache.wss4j.common.WSEncryptionPart; import org.apache.wss4j.common.bsp.BSPRule; import org.apache.wss4j.common.crypto.Crypto; import org.apache.wss4j.common.crypto.CryptoFactory; import org.apache.wss4j.common.ext.WSSecurityException; import org.apache.wss4j.common.token.Reference; import org.apache.wss4j.common.token.SecurityTokenReference; import org.apache.wss4j.common.util.XMLUtils; import org.apache.wss4j.dom.WSConstants; import org.apache.wss4j.dom.common.CustomHandler; import org.apache.wss4j.dom.common.KeystoreCallbackHandler; import org.apache.wss4j.dom.common.SOAPUtil; import org.apache.wss4j.dom.common.SecurityTestUtil; import org.apache.wss4j.dom.engine.WSSConfig; import org.apache.wss4j.dom.engine.WSSecurityEngine; import org.apache.wss4j.dom.engine.WSSecurityEngineResult; import org.apache.wss4j.dom.handler.HandlerAction; import org.apache.wss4j.dom.handler.RequestData; import org.apache.wss4j.dom.handler.WSHandlerConstants; import org.apache.wss4j.dom.handler.WSHandlerResult; import org.apache.wss4j.dom.str.STRParser.REFERENCE_TYPE; import org.apache.wss4j.dom.util.WSSecurityUtil; import org.junit.Test; import org.w3c.dom.Document; import org.w3c.dom.Element; import org.w3c.dom.Node; /** * A set of test-cases for signing and verifying SOAP requests. */ public class SignatureTest extends org.junit.Assert { private static final org.slf4j.Logger LOG = org.slf4j.LoggerFactory.getLogger(SignatureTest.class); private WSSecurityEngine secEngine = new WSSecurityEngine(); private CallbackHandler callbackHandler = new KeystoreCallbackHandler(); private Crypto crypto; @org.junit.AfterClass public static void cleanup() throws Exception { SecurityTestUtil.cleanup(); } public SignatureTest() throws Exception { WSSConfig.init(); crypto = CryptoFactory.getInstance(); } /** * The test uses the Issuer Serial key identifier type. * <p/> * * @throws java.lang.Exception Thrown when there is any problem in signing or verification */ @Test public void testX509SignatureIS() throws Exception { Document doc = SOAPUtil.toSOAPPart(SOAPUtil.SAMPLE_SOAP_MSG); WSSecHeader secHeader = new WSSecHeader(doc); secHeader.insertSecurityHeader(); WSSecSignature builder = new WSSecSignature(secHeader); builder.setUserInfo("16c73ab6-b892-458f-abf5-2f875f74882e", "security"); builder.setKeyIdentifierType(WSConstants.ISSUER_SERIAL); LOG.info("Before Signing IS...."); Document signedDoc = builder.build(crypto); if (LOG.isDebugEnabled()) { LOG.debug("Signed message with IssuerSerial key identifier:"); String outputString = XMLUtils.prettyDocumentToString(signedDoc); LOG.debug(outputString); } LOG.info("After Signing IS...."); WSHandlerResult results = verify(signedDoc); WSSecurityEngineResult actionResult = results.getActionResults().get(WSConstants.SIGN).get(0); assertNotNull(actionResult.get(WSSecurityEngineResult.TAG_X509_CERTIFICATE)); assertNotNull(actionResult.get(WSSecurityEngineResult.TAG_X509_REFERENCE_TYPE)); REFERENCE_TYPE referenceType = (REFERENCE_TYPE)actionResult.get(WSSecurityEngineResult.TAG_X509_REFERENCE_TYPE); assertTrue(referenceType == REFERENCE_TYPE.ISSUER_SERIAL); } @Test public void testX509SignatureISAttached() throws Exception { Document doc = SOAPUtil.toSOAPPart(SOAPUtil.SAMPLE_SOAP_MSG); WSSecHeader secHeader = new WSSecHeader(doc); secHeader.insertSecurityHeader(); WSSecSignature builder = new WSSecSignature(secHeader); builder.setUserInfo("16c73ab6-b892-458f-abf5-2f875f74882e", "security"); builder.setKeyIdentifierType(WSConstants.ISSUER_SERIAL); builder.setIncludeSignatureToken(true); LOG.info("Before Signing IS...."); Document signedDoc = builder.build(crypto); if (LOG.isDebugEnabled()) { LOG.debug("Signed message with IssuerSerial key identifier:"); String outputString = XMLUtils.prettyDocumentToString(signedDoc); LOG.debug(outputString); } LOG.info("After Signing IS...."); WSHandlerResult results = verify(signedDoc); WSSecurityEngineResult actionResult = results.getActionResults().get(WSConstants.SIGN).get(0); assertNotNull(actionResult.get(WSSecurityEngineResult.TAG_X509_CERTIFICATE)); assertNotNull(actionResult.get(WSSecurityEngineResult.TAG_X509_REFERENCE_TYPE)); REFERENCE_TYPE referenceType = (REFERENCE_TYPE)actionResult.get(WSSecurityEngineResult.TAG_X509_REFERENCE_TYPE); assertTrue(referenceType == REFERENCE_TYPE.ISSUER_SERIAL); } /** * Test that signs (twice) and verifies a WS-Security envelope. * <p/> * * @throws java.lang.Exception Thrown when there is any problem in signing or verification */ @Test public void testDoubleX509SignatureIS() throws Exception { Document doc = SOAPUtil.toSOAPPart(SOAPUtil.SAMPLE_SOAP_MSG); WSSecHeader secHeader = new WSSecHeader(doc); secHeader.insertSecurityHeader(); WSSecSignature builder = new WSSecSignature(secHeader); builder.setUserInfo("16c73ab6-b892-458f-abf5-2f875f74882e", "security"); builder.build(crypto); Document signedDoc1 = builder.build(crypto); verify(signedDoc1); } /** * Test that signs and verifies a WS-Security envelope * <p/> * * @throws java.lang.Exception Thrown when there is any problem in signing or verification */ @Test public void testIssuerSerialSignature() throws Exception { Document doc = SOAPUtil.toSOAPPart(SOAPUtil.SAMPLE_SOAP_MSG); WSSecHeader secHeader = new WSSecHeader(doc); secHeader.insertSecurityHeader(); WSSecSignature builder = new WSSecSignature(secHeader); builder.setUserInfo("16c73ab6-b892-458f-abf5-2f875f74882e", "security"); LOG.info("Before Signing...."); Document signedDoc = builder.build(crypto); if (LOG.isDebugEnabled()) { LOG.debug("After Signing...."); String outputString = XMLUtils.prettyDocumentToString(signedDoc); LOG.debug(outputString); } verify(signedDoc); } /** * Test that signs and verifies a WS-Security envelope * <p/> * * @throws java.lang.Exception Thrown when there is any problem in signing or verification */ @Test public void testSignatureInclusiveC14N() throws Exception { Document doc = SOAPUtil.toSOAPPart(SOAPUtil.SAMPLE_SOAP_MSG); WSSecHeader secHeader = new WSSecHeader(doc); secHeader.insertSecurityHeader(); WSSecSignature builder = new WSSecSignature(secHeader); builder.setSigCanonicalization(WSConstants.C14N_OMIT_COMMENTS); builder.setUserInfo("16c73ab6-b892-458f-abf5-2f875f74882e", "security"); LOG.info("Before Signing...."); Document signedDoc = builder.build(crypto); if (LOG.isDebugEnabled()) { LOG.debug("After Signing...."); String outputString = XMLUtils.prettyDocumentToString(signedDoc); LOG.debug(outputString); } WSSecurityEngine newEngine = new WSSecurityEngine(); try { newEngine.processSecurityHeader(doc, null, null, crypto); fail("Failure expected on a bad c14n algorithm"); } catch (WSSecurityException ex) { assertTrue(ex.getErrorCode() == WSSecurityException.ErrorCode.INVALID_SECURITY); } RequestData data = new RequestData(); data.setSigVerCrypto(crypto); List<BSPRule> ignoredRules = new ArrayList<>(); ignoredRules.add(BSPRule.R5404); ignoredRules.add(BSPRule.R5406); data.setIgnoredBSPRules(ignoredRules); newEngine.processSecurityHeader(doc, data); } /** * Test that signs and verifies a WS-Security envelope * <p/> * * @throws java.lang.Exception Thrown when there is any problem in signing or verification */ @Test public void testSignatureInclusivePrefixes() throws Exception { Document doc = SOAPUtil.toSOAPPart(SOAPUtil.SAMPLE_SOAP_MSG); WSSecHeader secHeader = new WSSecHeader(doc); secHeader.insertSecurityHeader(); WSSecSignature builder = new WSSecSignature(secHeader); builder.setAddInclusivePrefixes(true); builder.setUserInfo("16c73ab6-b892-458f-abf5-2f875f74882e", "security"); LOG.info("Before Signing...."); Document signedDoc = builder.build(crypto); if (LOG.isDebugEnabled()) { LOG.debug("After Signing...."); String outputString = XMLUtils.prettyDocumentToString(signedDoc); LOG.debug(outputString); } verify(signedDoc); } /** * Test that signs and verifies a WS-Security envelope * <p/> * * @throws java.lang.Exception Thrown when there is any problem in signing or verification */ @Test public void testBSTSignature() throws Exception { Document doc = SOAPUtil.toSOAPPart(SOAPUtil.SAMPLE_SOAP_MSG); WSSecHeader secHeader = new WSSecHeader(doc); secHeader.insertSecurityHeader(); WSSecSignature builder = new WSSecSignature(secHeader); builder.setUserInfo("16c73ab6-b892-458f-abf5-2f875f74882e", "security"); builder.setKeyIdentifierType(WSConstants.BST_DIRECT_REFERENCE); LOG.info("Before Signing...."); Document signedDoc = builder.build(crypto); if (LOG.isDebugEnabled()) { LOG.debug("After Signing...."); String outputString = XMLUtils.prettyDocumentToString(signedDoc); LOG.debug(outputString); } WSHandlerResult results = verify(signedDoc); WSSecurityEngineResult actionResult = results.getActionResults().get(WSConstants.SIGN).get(0); assertNotNull(actionResult.get(WSSecurityEngineResult.TAG_X509_CERTIFICATE)); assertNotNull(actionResult.get(WSSecurityEngineResult.TAG_X509_REFERENCE_TYPE)); REFERENCE_TYPE referenceType = (REFERENCE_TYPE)actionResult.get(WSSecurityEngineResult.TAG_X509_REFERENCE_TYPE); assertTrue(referenceType == REFERENCE_TYPE.DIRECT_REF); } /** * Test that signs and verifies a WS-Security envelope * <p/> * * @throws java.lang.Exception Thrown when there is any problem in signing or verification */ @Test public void testBSTPKIPathSignature() throws Exception { Document doc = SOAPUtil.toSOAPPart(SOAPUtil.SAMPLE_SOAP_MSG); WSSecHeader secHeader = new WSSecHeader(doc); secHeader.insertSecurityHeader(); WSSecSignature builder = new WSSecSignature(secHeader); builder.setUserInfo("wss40", "security"); builder.setKeyIdentifierType(WSConstants.BST_DIRECT_REFERENCE); builder.setUseSingleCertificate(false); LOG.info("Before Signing...."); Crypto pkiCrypto = CryptoFactory.getInstance("wss40.properties"); Document signedDoc = builder.build(pkiCrypto); if (LOG.isDebugEnabled()) { LOG.debug("After PKI Signing...."); String outputString = XMLUtils.prettyDocumentToString(signedDoc); LOG.debug(outputString); } secEngine.processSecurityHeader(doc, null, callbackHandler, pkiCrypto, null); } /** * Test that signs and verifies a WS-Security envelope * <p/> * * @throws java.lang.Exception Thrown when there is any problem in signing or verification */ @Test public void testX509Signature() throws Exception { Document doc = SOAPUtil.toSOAPPart(SOAPUtil.SAMPLE_SOAP_MSG); WSSecHeader secHeader = new WSSecHeader(doc); secHeader.insertSecurityHeader(); WSSecSignature builder = new WSSecSignature(secHeader); builder.setUserInfo("16c73ab6-b892-458f-abf5-2f875f74882e", "security"); builder.setKeyIdentifierType(WSConstants.X509_KEY_IDENTIFIER); LOG.info("Before Signing...."); Document signedDoc = builder.build(crypto); if (LOG.isDebugEnabled()) { LOG.debug("After Signing...."); String outputString = XMLUtils.prettyDocumentToString(signedDoc); LOG.debug(outputString); } WSSecurityEngine newEngine = new WSSecurityEngine(); WSHandlerResult results = newEngine.processSecurityHeader(doc, null, null, crypto); WSSecurityEngineResult actionResult = results.getActionResults().get(WSConstants.SIGN).get(0); assertNotNull(actionResult.get(WSSecurityEngineResult.TAG_X509_CERTIFICATE)); assertNotNull(actionResult.get(WSSecurityEngineResult.TAG_X509_REFERENCE_TYPE)); REFERENCE_TYPE referenceType = (REFERENCE_TYPE)actionResult.get(WSSecurityEngineResult.TAG_X509_REFERENCE_TYPE); assertTrue(referenceType == REFERENCE_TYPE.KEY_IDENTIFIER); } /** * Test that signs and verifies a WS-Security envelope. * The test uses the ThumbprintSHA1 key identifier type. * <p/> * * @throws java.lang.Exception Thrown when there is any problem in signing or verification */ @Test public void testX509SignatureThumb() throws Exception { Document doc = SOAPUtil.toSOAPPart(SOAPUtil.SAMPLE_SOAP_MSG); WSSecHeader secHeader = new WSSecHeader(doc); secHeader.insertSecurityHeader(); WSSecSignature builder = new WSSecSignature(secHeader); builder.setUserInfo("16c73ab6-b892-458f-abf5-2f875f74882e", "security"); builder.setKeyIdentifierType(WSConstants.THUMBPRINT_IDENTIFIER); LOG.info("Before Signing ThumbprintSHA1...."); Document signedDoc = builder.build(crypto); if (LOG.isDebugEnabled()) { LOG.debug("Signed message with ThumbprintSHA1 key identifier:"); String outputString = XMLUtils.prettyDocumentToString(signedDoc); LOG.debug(outputString); } LOG.info("After Signing ThumbprintSHA1...."); WSHandlerResult results = verify(signedDoc); WSSecurityEngineResult actionResult = results.getActionResults().get(WSConstants.SIGN).get(0); assertNotNull(actionResult.get(WSSecurityEngineResult.TAG_X509_CERTIFICATE)); assertNotNull(actionResult.get(WSSecurityEngineResult.TAG_X509_REFERENCE_TYPE)); REFERENCE_TYPE referenceType = (REFERENCE_TYPE)actionResult.get(WSSecurityEngineResult.TAG_X509_REFERENCE_TYPE); assertTrue(referenceType == REFERENCE_TYPE.THUMBPRINT_SHA1); } @Test public void testX509SignatureThumbAttached() throws Exception { Document doc = SOAPUtil.toSOAPPart(SOAPUtil.SAMPLE_SOAP_MSG); WSSecHeader secHeader = new WSSecHeader(doc); secHeader.insertSecurityHeader(); WSSecSignature builder = new WSSecSignature(secHeader); builder.setUserInfo("16c73ab6-b892-458f-abf5-2f875f74882e", "security"); builder.setKeyIdentifierType(WSConstants.THUMBPRINT_IDENTIFIER); builder.setIncludeSignatureToken(true); LOG.info("Before Signing ThumbprintSHA1...."); Document signedDoc = builder.build(crypto); if (LOG.isDebugEnabled()) { LOG.debug("Signed message with ThumbprintSHA1 key identifier:"); String outputString = XMLUtils.prettyDocumentToString(signedDoc); LOG.debug(outputString); } LOG.info("After Signing ThumbprintSHA1...."); WSHandlerResult results = verify(signedDoc); WSSecurityEngineResult actionResult = results.getActionResults().get(WSConstants.SIGN).get(0); assertNotNull(actionResult.get(WSSecurityEngineResult.TAG_X509_CERTIFICATE)); assertNotNull(actionResult.get(WSSecurityEngineResult.TAG_X509_REFERENCE_TYPE)); REFERENCE_TYPE referenceType = (REFERENCE_TYPE)actionResult.get(WSSecurityEngineResult.TAG_X509_REFERENCE_TYPE); assertTrue(referenceType == REFERENCE_TYPE.THUMBPRINT_SHA1); } /** * Test that signs (twice) and verifies a WS-Security envelope. * The test uses the ThumbprintSHA1 key identifier type. * <p/> * * @throws java.lang.Exception Thrown when there is any problem in signing or verification */ @Test public void testDoubleX509SignatureThumb() throws Exception { Document doc = SOAPUtil.toSOAPPart(SOAPUtil.SAMPLE_SOAP_MSG); WSSecHeader secHeader = new WSSecHeader(doc); secHeader.insertSecurityHeader(); WSSecSignature builder = new WSSecSignature(secHeader); builder.setUserInfo("16c73ab6-b892-458f-abf5-2f875f74882e", "security"); builder.setKeyIdentifierType(WSConstants.THUMBPRINT_IDENTIFIER); builder.build(crypto); Document signedDoc1 = builder.build(crypto); verify(signedDoc1); } /** * Test that signs and verifies a Timestamp. The request is then modified so that the * Timestamp has a default (WSU) namespace inserted. The signature validation should still * pass due to c14n (see WSS-181). * * @throws java.lang.Exception Thrown when there is any problem in signing or verification */ @Test public void testValidModifiedSignature() throws Exception { Document doc = SOAPUtil.toSOAPPart(SOAPUtil.SAMPLE_SOAP_MSG); WSSecHeader secHeader = new WSSecHeader(doc); secHeader.insertSecurityHeader(); WSSecSignature builder = new WSSecSignature(secHeader); builder.setUserInfo("16c73ab6-b892-458f-abf5-2f875f74882e", "security"); LOG.info("Before Signing...."); WSSecTimestamp timestamp = new WSSecTimestamp(secHeader); timestamp.setTimeToLive(300); timestamp.build(); WSEncryptionPart encP = new WSEncryptionPart( "Timestamp", WSConstants.WSU_NS, ""); builder.getParts().add(encP); Document signedDoc = builder.build(crypto); Element secHeaderElement = secHeader.getSecurityHeaderElement(); Node timestampNode = secHeaderElement.getElementsByTagNameNS(WSConstants.WSU_NS, "Timestamp").item(0); ((Element)timestampNode).setAttributeNS( WSConstants.XMLNS_NS, "xmlns", WSConstants.WSU_NS ); if (LOG.isDebugEnabled()) { LOG.debug("After Signing...."); String outputString = XMLUtils.prettyDocumentToString(signedDoc); LOG.debug(outputString); } verify(signedDoc); } /** * Sign using a different digest algorithm (SHA-256). * <p/> * * @throws java.lang.Exception Thrown when there is any problem in signing or verification */ @Test public void testX509SignatureSha256() throws Exception { Document doc = SOAPUtil.toSOAPPart(SOAPUtil.SAMPLE_SOAP_MSG); WSSecHeader secHeader = new WSSecHeader(doc); secHeader.insertSecurityHeader(); WSSecSignature builder = new WSSecSignature(secHeader); builder.setUserInfo("16c73ab6-b892-458f-abf5-2f875f74882e", "security"); builder.setKeyIdentifierType(WSConstants.ISSUER_SERIAL); builder.setSignatureAlgorithm("http://www.w3.org/2001/04/xmldsig-more#rsa-sha256"); builder.setDigestAlgo("http://www.w3.org/2001/04/xmlenc#sha256"); LOG.info("Before Signing IS...."); Document signedDoc = builder.build(crypto); if (LOG.isDebugEnabled()) { LOG.debug("Signed message with IssuerSerial key identifier:"); String outputString = XMLUtils.prettyDocumentToString(signedDoc); LOG.debug(outputString); } LOG.info("After Signing IS...."); verify(signedDoc); } /** * A test for "SignatureAction does not set DigestAlgorithm on WSSecSignature instance" */ @Test public void testWSS170() throws Exception { final WSSConfig cfg = WSSConfig.getNewInstance(); final RequestData reqData = new RequestData(); reqData.setWssConfig(cfg); reqData.setUsername("16c73ab6-b892-458f-abf5-2f875f74882e"); java.util.Map<String, Object> config = new java.util.TreeMap<String, Object>(); config.put(WSHandlerConstants.SIG_PROP_FILE, "crypto.properties"); config.put("password", "security"); config.put( WSHandlerConstants.SIG_ALGO, "http://www.w3.org/2001/04/xmldsig-more#rsa-sha256" ); config.put( WSHandlerConstants.SIG_DIGEST_ALGO, "http://www.w3.org/2001/04/xmlenc#sha256" ); reqData.setMsgContext(config); final Document doc = SOAPUtil.toSOAPPart(SOAPUtil.SAMPLE_SOAP_MSG); CustomHandler handler = new CustomHandler(); HandlerAction action = new HandlerAction(WSConstants.SIGN); handler.send( doc, reqData, Collections.singletonList(action), true ); String outputString = XMLUtils.prettyDocumentToString(doc); if (LOG.isDebugEnabled()) { LOG.debug("Signed message:"); LOG.debug(outputString); } assertTrue( outputString.contains("http://www.w3.org/2001/04/xmldsig-more#rsa-sha256") ); assertTrue( outputString.contains("http://www.w3.org/2001/04/xmlenc#sha256") ); verify(doc); } /** * This is a test for WSS-234 - * "When a document contains a comment as its first child element, * wss4j will not find the SOAP body." */ @Test public void testWSS234() throws Exception { Document doc = SOAPUtil.toSOAPPart(SOAPUtil.SAMPLE_SOAP_MSG); WSSecHeader secHeader = new WSSecHeader(doc); secHeader.insertSecurityHeader(); WSSecSignature builder = new WSSecSignature(secHeader); builder.setUserInfo("16c73ab6-b892-458f-abf5-2f875f74882e", "security"); LOG.info("Before Signing...."); Document signedDoc = builder.build(crypto); // Add a comment node as the first node element Node firstChild = signedDoc.getFirstChild(); Node newNode = signedDoc.removeChild(firstChild); Node commentNode = signedDoc.createComment("This is a comment"); signedDoc.appendChild(commentNode); signedDoc.appendChild(newNode); if (LOG.isDebugEnabled()) { LOG.debug("After Signing...."); String outputString = XMLUtils.prettyDocumentToString(signedDoc); LOG.debug(outputString); } verify(signedDoc); } /** * Test that signs and verifies a Timestamp. The Signature element is appended to the security * header, and so appears after the Timestamp element. */ @Test public void testSignedTimestamp() throws Exception { Document doc = SOAPUtil.toSOAPPart(SOAPUtil.SAMPLE_SOAP_MSG); WSSecHeader secHeader = new WSSecHeader(doc); secHeader.insertSecurityHeader(); WSSecTimestamp timestamp = new WSSecTimestamp(secHeader); timestamp.setTimeToLive(300); timestamp.build(); WSSecSignature builder = new WSSecSignature(secHeader); builder.setUserInfo("16c73ab6-b892-458f-abf5-2f875f74882e", "security"); WSEncryptionPart encP = new WSEncryptionPart( "Timestamp", WSConstants.WSU_NS, ""); builder.getParts().add(encP); builder.prepare(crypto); List<javax.xml.crypto.dsig.Reference> referenceList = builder.addReferencesToSign(builder.getParts()); builder.computeSignature(referenceList, false, null); if (LOG.isDebugEnabled()) { LOG.debug("After Signing...."); String outputString = XMLUtils.prettyDocumentToString(doc); LOG.debug(outputString); } verify(doc); } /** * This is a test for WSS-283 - "ClassCastException when signing message with existing * WSSE header containing Text as first child": * * https://issues.apache.org/jira/browse/WSS-283 */ @Test public void testWSS283() throws Exception { Document doc = SOAPUtil.toSOAPPart(SOAPUtil.SAMPLE_SOAP_MSG); WSSecHeader secHeader = new WSSecHeader(doc); Element secHeaderElement = secHeader.insertSecurityHeader(); WSSecSignature builder = new WSSecSignature(secHeader); builder.setUserInfo("16c73ab6-b892-458f-abf5-2f875f74882e", "security"); builder.setKeyIdentifierType(WSConstants.ISSUER_SERIAL); Node textNode = doc.createTextNode("This is a text node"); secHeaderElement.appendChild(textNode); Document signedDoc = builder.build(crypto); if (LOG.isDebugEnabled()) { LOG.debug("Signed message with text node:"); String outputString = XMLUtils.prettyDocumentToString(signedDoc); LOG.debug(outputString); } verify(signedDoc); } /** * Create a signature that uses a custom SecurityTokenReference. */ @Test public void testCustomSTR() throws Exception { Document doc = SOAPUtil.toSOAPPart(SOAPUtil.SAMPLE_SOAP_MSG); WSSecHeader secHeader = new WSSecHeader(doc); secHeader.insertSecurityHeader(); WSSecSignature builder = new WSSecSignature(secHeader); builder.setUserInfo("16c73ab6-b892-458f-abf5-2f875f74882e", "security"); builder.setKeyIdentifierType(WSConstants.ISSUER_SERIAL); LOG.info("Before Signing IS...."); SecurityTokenReference secRef = new SecurityTokenReference(doc); Reference ref = new Reference(doc); ref.setURI("custom-uri"); secRef.setReference(ref); builder.setSecurityTokenReference(secRef); Document signedDoc = builder.build(crypto); if (LOG.isDebugEnabled()) { String outputString = XMLUtils.prettyDocumentToString(signedDoc); LOG.debug(outputString); } } /** * The test uses the Issuer Serial key identifier type. * <p/> * * @throws java.lang.Exception Thrown when there is any problem in signing or verification */ @Test public void testX509SignatureDefaultPassword() throws Exception { Crypto passwordCrypto = CryptoFactory.getInstance("alice.properties"); Document doc = SOAPUtil.toSOAPPart(SOAPUtil.SAMPLE_SOAP_MSG); WSSecHeader secHeader = new WSSecHeader(doc); secHeader.insertSecurityHeader(); WSSecSignature builder = new WSSecSignature(secHeader); builder.setUserInfo(passwordCrypto.getDefaultX509Identifier(), null); builder.setKeyIdentifierType(WSConstants.ISSUER_SERIAL); LOG.info("Before Signing IS...."); Document signedDoc = builder.build(passwordCrypto); if (LOG.isDebugEnabled()) { LOG.debug("Signed message with IssuerSerial key identifier:"); String outputString = XMLUtils.prettyDocumentToString(signedDoc); LOG.debug(outputString); } LOG.info("After Signing IS...."); WSSecurityEngine newEngine = new WSSecurityEngine(); newEngine.processSecurityHeader(doc, null, null, passwordCrypto); } /** * A test for "There is an issue with the position of the <Timestamp> element in the * <Security> header when using WSS4J calling .NET Web Services with WS-Security." */ @Test public void testWSS231() throws Exception { final WSSConfig cfg = WSSConfig.getNewInstance(); final RequestData reqData = new RequestData(); reqData.setWssConfig(cfg); reqData.setUsername("16c73ab6-b892-458f-abf5-2f875f74882e"); java.util.Map<String, Object> config = new java.util.TreeMap<String, Object>(); config.put(WSHandlerConstants.SIG_PROP_FILE, "crypto.properties"); config.put("password", "security"); config.put( WSHandlerConstants.SIGNATURE_PARTS, "{}{" + WSConstants.WSU_NS + "}Timestamp" ); reqData.setMsgContext(config); final Document doc = SOAPUtil.toSOAPPart(SOAPUtil.SAMPLE_SOAP_MSG); CustomHandler handler = new CustomHandler(); List<HandlerAction> actions = new ArrayList<>(); actions.add(new HandlerAction(WSConstants.SIGN)); actions.add(new HandlerAction(WSConstants.TS)); handler.send( doc, reqData, actions, true ); String outputString = XMLUtils.prettyDocumentToString(doc); if (LOG.isDebugEnabled()) { LOG.debug("Signed message:"); LOG.debug(outputString); } WSHandlerResult results = verify(doc); List<Integer> receivedActions = new ArrayList<>(); receivedActions.add(WSConstants.SIGN); receivedActions.add(WSConstants.TS); assertTrue(handler.checkResults(results.getResults(), receivedActions)); } @Test public void testSignatureEncryptTimestampOrder() throws Exception { final WSSConfig cfg = WSSConfig.getNewInstance(); final RequestData reqData = new RequestData(); reqData.setWssConfig(cfg); reqData.setUsername("16c73ab6-b892-458f-abf5-2f875f74882e"); java.util.Map<String, Object> config = new java.util.TreeMap<String, Object>(); config.put(WSHandlerConstants.SIG_PROP_FILE, "crypto.properties"); config.put(WSHandlerConstants.ENC_PROP_FILE, "crypto.properties"); config.put("password", "security"); config.put( WSHandlerConstants.SIGNATURE_PARTS, "{}{" + WSConstants.WSU_NS + "}Timestamp" ); reqData.setMsgContext(config); final Document doc = SOAPUtil.toSOAPPart(SOAPUtil.SAMPLE_SOAP_MSG); CustomHandler handler = new CustomHandler(); List<HandlerAction> actions = new ArrayList<>(); actions.add(new HandlerAction(WSConstants.SIGN)); actions.add(new HandlerAction(WSConstants.ENCR)); actions.add(new HandlerAction(WSConstants.TS)); handler.send( doc, reqData, actions, true ); String outputString = XMLUtils.prettyDocumentToString(doc); if (LOG.isDebugEnabled()) { LOG.debug("Signed message:"); LOG.debug(outputString); } } @Test public void testEncryptSignatureTimestampOrder() throws Exception { final WSSConfig cfg = WSSConfig.getNewInstance(); final RequestData reqData = new RequestData(); reqData.setWssConfig(cfg); reqData.setUsername("16c73ab6-b892-458f-abf5-2f875f74882e"); java.util.Map<String, Object> config = new java.util.TreeMap<String, Object>(); config.put(WSHandlerConstants.SIG_PROP_FILE, "crypto.properties"); config.put(WSHandlerConstants.ENC_PROP_FILE, "crypto.properties"); config.put("password", "security"); config.put( WSHandlerConstants.SIGNATURE_PARTS, "{}{" + WSConstants.WSU_NS + "}Timestamp" ); reqData.setMsgContext(config); final Document doc = SOAPUtil.toSOAPPart(SOAPUtil.SAMPLE_SOAP_MSG); CustomHandler handler = new CustomHandler(); List<HandlerAction> actions = new ArrayList<>(); actions.add(new HandlerAction(WSConstants.ENCR)); actions.add(new HandlerAction(WSConstants.SIGN)); actions.add(new HandlerAction(WSConstants.TS)); handler.send( doc, reqData, actions, true ); String outputString = XMLUtils.prettyDocumentToString(doc); if (LOG.isDebugEnabled()) { LOG.debug("Signed message:"); LOG.debug(outputString); } } @Test public void testWSHandlerSignatureCanonicalization() throws Exception { final WSSConfig cfg = WSSConfig.getNewInstance(); final RequestData reqData = new RequestData(); reqData.setWssConfig(cfg); reqData.setUsername("16c73ab6-b892-458f-abf5-2f875f74882e"); java.util.Map<String, Object> config = new java.util.TreeMap<String, Object>(); config.put(WSHandlerConstants.SIG_PROP_FILE, "crypto.properties"); config.put(WSHandlerConstants.SIG_C14N_ALGO, WSConstants.C14N_WITH_COMMENTS); config.put("password", "security"); reqData.setMsgContext(config); final Document doc = SOAPUtil.toSOAPPart(SOAPUtil.SAMPLE_SOAP_MSG); CustomHandler handler = new CustomHandler(); HandlerAction action = new HandlerAction(WSConstants.SIGN); handler.send( doc, reqData, Collections.singletonList(action), true ); String outputString = XMLUtils.prettyDocumentToString(doc); if (LOG.isDebugEnabled()) { LOG.debug("Signed message:"); LOG.debug(outputString); } RequestData data = new RequestData(); data.setWssConfig(WSSConfig.getNewInstance()); data.setSigVerCrypto(crypto); List<BSPRule> disabledRules = new ArrayList<>(); disabledRules.add(BSPRule.R5404); disabledRules.add(BSPRule.R5406); data.setIgnoredBSPRules(disabledRules); WSSecurityEngine newSecEngine = new WSSecurityEngine(); WSHandlerResult results = newSecEngine.processSecurityHeader(doc, data); assertTrue(handler.checkResults(results.getResults(), Collections.singletonList(WSConstants.SIGN))); } @Test public void testCommentInSOAPBody() throws Exception { Document doc = SOAPUtil.toSOAPPart(SOAPUtil.SAMPLE_SOAP_MSG); WSSecHeader secHeader = new WSSecHeader(doc); secHeader.insertSecurityHeader(); WSSecSignature builder = new WSSecSignature(secHeader); builder.setUserInfo("16c73ab6-b892-458f-abf5-2f875f74882e", "security"); LOG.info("Before Signing...."); Document signedDoc = builder.build(crypto); // Add a comment node Element body = WSSecurityUtil.findBodyElement(signedDoc); Node commentNode = signedDoc.createComment("This is a comment"); body.getFirstChild().appendChild(commentNode); if (LOG.isDebugEnabled()) { String outputString = XMLUtils.prettyDocumentToString(signedDoc); LOG.debug(outputString); } verify(signedDoc); } /** * Verifies the soap envelope. * This method verifies all the signature generated. * * @param env soap envelope * @throws java.lang.Exception Thrown when there is a problem in verification */ private WSHandlerResult verify(Document doc) throws Exception { return secEngine.processSecurityHeader(doc, null, null, crypto); } }