| /** |
| * 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.xml.security.test.stax.performance; |
| |
| import org.apache.xml.security.encryption.XMLCipher; |
| import org.apache.xml.security.exceptions.XMLSecurityException; |
| import org.apache.xml.security.signature.XMLSignature; |
| import org.apache.xml.security.stax.ext.*; |
| import org.apache.xml.security.stax.securityToken.SecurityTokenConstants; |
| import org.apache.xml.security.test.stax.utils.XmlReaderToWriter; |
| import org.apache.xml.security.transforms.Transforms; |
| import org.apache.xml.security.utils.XMLUtils; |
| import org.junit.Before; |
| import org.junit.BeforeClass; |
| import org.w3c.dom.Document; |
| import org.w3c.dom.Element; |
| |
| import javax.crypto.KeyGenerator; |
| import javax.xml.namespace.QName; |
| import javax.xml.parsers.DocumentBuilder; |
| import javax.xml.stream.XMLInputFactory; |
| import javax.xml.stream.XMLStreamConstants; |
| import javax.xml.stream.XMLStreamReader; |
| import javax.xml.stream.XMLStreamWriter; |
| |
| import java.io.*; |
| import java.nio.charset.StandardCharsets; |
| import java.security.Key; |
| import java.security.KeyStore; |
| import java.security.cert.X509Certificate; |
| import java.util.ArrayList; |
| import java.util.List; |
| |
| /** |
| */ |
| public abstract class AbstractPerformanceTest { |
| |
| private static Key encryptionSymKey; |
| protected XMLInputFactory xmlInputFactory; |
| protected Key key; |
| protected X509Certificate cert; |
| private OutboundXMLSec outboundSignatureXMLSec; |
| private InboundXMLSec inboundSignatureXMLSec; |
| private OutboundXMLSec outboundEncryptionXMLSec; |
| private InboundXMLSec inboundDecryptionXMLSec; |
| |
| @BeforeClass |
| public static void genKey() throws Exception { |
| KeyGenerator keygen = KeyGenerator.getInstance("AES"); |
| keygen.init(256); |
| encryptionSymKey = keygen.generateKey(); |
| } |
| |
| @Before |
| public void setUp() throws Exception { |
| xmlInputFactory = XMLInputFactory.newInstance(); |
| xmlInputFactory.setProperty(XMLInputFactory.IS_COALESCING, false); |
| xmlInputFactory.setProperty(XMLInputFactory.SUPPORT_DTD, false); |
| |
| org.apache.xml.security.Init.init(); |
| |
| KeyStore keyStore = KeyStore.getInstance("jks"); |
| keyStore.load( |
| this.getClass().getClassLoader().getResource("transmitter.jks").openStream(), |
| "default".toCharArray() |
| ); |
| key = keyStore.getKey("transmitter", "default".toCharArray()); |
| cert = (X509Certificate) keyStore.getCertificate("transmitter"); |
| |
| setUpOutboundSignatureXMLSec(); |
| setUpInboundSignatureXMLSec(); |
| |
| setUpOutboundEncryptionXMLSec(); |
| setUpInboundEncryptionXMLSec(); |
| } |
| |
| protected File generateLargeXMLFile(int factor) throws Exception { |
| File path = getTmpFilePath(); |
| path.mkdirs(); |
| File target = new File(path, "tmp.xml"); |
| FileWriter fileWriter = new FileWriter(target, false); |
| fileWriter.write("<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n" + |
| "<test xmlns=\"http://www.example.com\">"); |
| fileWriter.close(); |
| FileOutputStream fileOutputStream = new FileOutputStream(target, true); |
| for (int i = 0; i < factor; i++) { |
| int read = 0; |
| byte[] buffer = new byte[4096]; |
| InputStream inputStream = this.getClass().getClassLoader().getResourceAsStream( |
| "org/w3c/www/interop/xmlenc-core-11/plaintext.xml"); |
| while ((read = inputStream.read(buffer)) != -1) { |
| fileOutputStream.write(buffer, 0, read); |
| } |
| inputStream.close(); |
| } |
| fileWriter = new FileWriter(target, true); |
| fileWriter.write("</test>"); |
| fileWriter.close(); |
| |
| fileOutputStream.close(); |
| |
| return target; |
| } |
| |
| protected int countXMLStartTags(File file) throws Exception { |
| int i = 0; |
| FileInputStream fileInputStream = new FileInputStream(file); |
| XMLStreamReader xmlStreamReader = xmlInputFactory.createXMLStreamReader(fileInputStream); |
| while (xmlStreamReader.hasNext()) { |
| xmlStreamReader.next(); |
| switch (xmlStreamReader.getEventType()) { |
| case XMLStreamConstants.START_ELEMENT: |
| i++; |
| break; |
| } |
| } |
| xmlStreamReader.close(); |
| fileInputStream.close(); |
| return i; |
| } |
| |
| protected abstract File getTmpFilePath(); |
| |
| protected void setUpOutboundSignatureXMLSec() throws XMLSecurityException { |
| XMLSecurityProperties xmlSecurityProperties = new XMLSecurityProperties(); |
| List<XMLSecurityConstants.Action> actions = new ArrayList<XMLSecurityConstants.Action>(); |
| actions.add(XMLSecurityConstants.SIGNATURE); |
| xmlSecurityProperties.setActions(actions); |
| xmlSecurityProperties.setSignatureKeyIdentifier(SecurityTokenConstants.KeyIdentifier_X509KeyIdentifier); |
| |
| xmlSecurityProperties.setSignatureKey(key); |
| xmlSecurityProperties.setSignatureCerts(new X509Certificate[]{cert}); |
| xmlSecurityProperties.setSignatureCanonicalizationAlgorithm("http://www.w3.org/TR/2001/REC-xml-c14n-20010315"); |
| |
| SecurePart securePart = new SecurePart( |
| new QName("http://www.example.com", "test"), |
| SecurePart.Modifier.Element, |
| new String[]{ |
| "http://www.w3.org/2000/09/xmldsig#enveloped-signature", |
| "http://www.w3.org/TR/2001/REC-xml-c14n-20010315" |
| }, |
| "http://www.w3.org/2000/09/xmldsig#sha1" |
| ); |
| xmlSecurityProperties.addSignaturePart(securePart); |
| |
| outboundSignatureXMLSec = XMLSec.getOutboundXMLSec(xmlSecurityProperties); |
| } |
| |
| protected void setUpInboundSignatureXMLSec() throws XMLSecurityException { |
| XMLSecurityProperties inboundProperties = new XMLSecurityProperties(); |
| inboundProperties.setSignatureVerificationKey(cert.getPublicKey()); |
| inboundSignatureXMLSec = XMLSec.getInboundWSSec(inboundProperties); |
| } |
| |
| protected void setUpOutboundEncryptionXMLSec() throws XMLSecurityException { |
| XMLSecurityProperties xmlSecurityProperties = new XMLSecurityProperties(); |
| List<XMLSecurityConstants.Action> actions = new ArrayList<XMLSecurityConstants.Action>(); |
| actions.add(XMLSecurityConstants.ENCRYPT); |
| xmlSecurityProperties.setActions(actions); |
| xmlSecurityProperties.setEncryptionKey(encryptionSymKey); |
| xmlSecurityProperties.setEncryptionSymAlgorithm("http://www.w3.org/2001/04/xmlenc#aes256-cbc"); |
| |
| SecurePart securePart = new SecurePart( |
| new QName("http://www.example.com", "test"), |
| SecurePart.Modifier.Element |
| ); |
| xmlSecurityProperties.addEncryptionPart(securePart); |
| |
| outboundEncryptionXMLSec = XMLSec.getOutboundXMLSec(xmlSecurityProperties); |
| } |
| |
| protected void setUpInboundEncryptionXMLSec() throws XMLSecurityException { |
| XMLSecurityProperties inboundProperties = new XMLSecurityProperties(); |
| inboundProperties.setDecryptionKey(encryptionSymKey); |
| inboundDecryptionXMLSec = XMLSec.getInboundWSSec(inboundProperties); |
| } |
| |
| protected File doStreamingSignatureOutbound(File file, int tagCount) throws Exception { |
| |
| final File signedFile = new File(getTmpFilePath(), "signature-stax-" + tagCount + ".xml"); |
| OutputStream outputStream = new FileOutputStream(signedFile); |
| XMLStreamWriter xmlStreamWriter = outboundSignatureXMLSec.processOutMessage(outputStream, StandardCharsets.UTF_8.name()); |
| |
| InputStream inputStream = new FileInputStream(file); |
| XMLStreamReader xmlStreamReader = xmlInputFactory.createXMLStreamReader(inputStream); |
| |
| XmlReaderToWriter.writeAll(xmlStreamReader, xmlStreamWriter); |
| xmlStreamWriter.close(); |
| outputStream.close(); |
| xmlStreamReader.close(); |
| inputStream.close(); |
| return signedFile; |
| } |
| |
| protected void doStreamingSignatureInbound(File file, int tagCount) throws Exception { |
| |
| InputStream inputStream = new FileInputStream(file); |
| XMLStreamReader xmlStreamReader = xmlInputFactory.createXMLStreamReader(inputStream); |
| XMLStreamReader securityStreamReader = inboundSignatureXMLSec.processInMessage(xmlStreamReader); |
| |
| while (securityStreamReader.hasNext()) { |
| securityStreamReader.next(); |
| } |
| xmlStreamReader.close(); |
| inputStream.close(); |
| securityStreamReader.close(); |
| } |
| |
| protected void doDOMSignatureOutbound(File file, int tagCount) throws Exception { |
| |
| DocumentBuilder builder = XMLUtils.createDocumentBuilder(false); |
| Document document = builder.parse(file); |
| |
| XMLSignature sig = new XMLSignature(document, "", "http://www.w3.org/2000/09/xmldsig#rsa-sha1"); |
| Element root = document.getDocumentElement(); |
| root.insertBefore(sig.getElement(), root.getFirstChild()); |
| |
| Transforms transforms = new Transforms(document); |
| transforms.addTransform(Transforms.TRANSFORM_ENVELOPED_SIGNATURE); |
| transforms.addTransform(Transforms.TRANSFORM_C14N_OMIT_COMMENTS); |
| sig.addDocument("", transforms, "http://www.w3.org/2000/09/xmldsig#sha1"); |
| |
| sig.sign(key); |
| sig.addKeyInfo(cert); |
| |
| XMLUtils.outputDOM(document, new BufferedOutputStream(new FileOutputStream(new File(getTmpFilePath(), "signature-dom-" + tagCount + ".xml")))); |
| } |
| |
| protected void doDOMSignatureInbound(File file, int tagCount) throws Exception { |
| |
| DocumentBuilder builder = XMLUtils.createDocumentBuilder(false); |
| Document document = builder.parse(file); |
| |
| Element signatureElement = (Element) document.getElementsByTagNameNS("http://www.w3.org/2000/09/xmldsig#", "Signature").item(0); |
| ((Element) signatureElement.getParentNode()).setIdAttributeNS(null, "Id", true); |
| XMLSignature xmlSignature = new XMLSignature(signatureElement, "", true); |
| xmlSignature.checkSignatureValue(cert); |
| } |
| |
| protected File doStreamingEncryptionOutbound(File file, int tagCount) throws Exception { |
| |
| final File signedFile = new File(getTmpFilePath(), "encryption-stax-" + tagCount + ".xml"); |
| OutputStream outputStream = new FileOutputStream(signedFile); |
| XMLStreamWriter xmlStreamWriter = outboundEncryptionXMLSec.processOutMessage(outputStream, StandardCharsets.UTF_8.name()); |
| |
| InputStream inputStream = new FileInputStream(file); |
| XMLStreamReader xmlStreamReader = xmlInputFactory.createXMLStreamReader(inputStream); |
| |
| XmlReaderToWriter.writeAll(xmlStreamReader, xmlStreamWriter); |
| xmlStreamWriter.close(); |
| outputStream.close(); |
| xmlStreamReader.close(); |
| inputStream.close(); |
| return signedFile; |
| } |
| |
| protected void doStreamingDecryptionInbound(File file, int tagCount) throws Exception { |
| |
| InputStream inputStream = new FileInputStream(file); |
| XMLStreamReader xmlStreamReader = xmlInputFactory.createXMLStreamReader(inputStream); |
| XMLStreamReader securityStreamReader = inboundDecryptionXMLSec.processInMessage(xmlStreamReader); |
| |
| while (securityStreamReader.hasNext()) { |
| securityStreamReader.next(); |
| } |
| xmlStreamReader.close(); |
| inputStream.close(); |
| securityStreamReader.close(); |
| } |
| |
| protected void doDOMEncryptionOutbound(File file, int tagCount) throws Exception { |
| |
| DocumentBuilder builder = XMLUtils.createDocumentBuilder(false); |
| Document document = builder.parse(file); |
| |
| XMLCipher cipher = XMLCipher.getInstance("http://www.w3.org/2001/04/xmlenc#aes256-cbc"); |
| cipher.init(XMLCipher.ENCRYPT_MODE, encryptionSymKey); |
| document = cipher.doFinal(document, document.getDocumentElement()); |
| |
| XMLUtils.outputDOM(document, new BufferedOutputStream(new FileOutputStream(new File(getTmpFilePath(), "encryption-dom-" + tagCount + ".xml")))); |
| } |
| |
| protected void doDOMDecryptionInbound(File file, int tagCount) throws Exception { |
| |
| DocumentBuilder builder = XMLUtils.createDocumentBuilder(false); |
| Document document = builder.parse(file); |
| |
| XMLCipher cipher = XMLCipher.getInstance("http://www.w3.org/2001/04/xmlenc#aes256-cbc"); |
| cipher.init(XMLCipher.DECRYPT_MODE, encryptionSymKey); |
| cipher.doFinal(document, document.getDocumentElement()); |
| } |
| } |