Examples with DocumentSerializer org.apache.xml.security.encryption.DocumentSerializer used on opensource projects

Example 1 with DocumentSerializer

use of org.apache.xml.security.encryption.DocumentSerializer in project santuario-java by apache.

the class SignedEncryptedTest method secureAndVerify.

public void secureAndVerify(TransformerFactory transformerFactory, boolean useDocumentSerializer) throws Exception {
    DocumentBuilder builder = XMLUtils.createDocumentBuilder(false);
    Document document = null;
    try (InputStream is = new ByteArrayInputStream(SAMPLE_MSG.getBytes(StandardCharsets.UTF_8))) {
        document = builder.parse(is);
    }
    // Set up the Key
    KeyPairGenerator rsaKeygen = KeyPairGenerator.getInstance("RSA");
    KeyPair kp = rsaKeygen.generateKeyPair();
    PrivateKey priv = kp.getPrivate();
    PublicKey pub = kp.getPublic();
    XMLSignature sig = new XMLSignature(document, "", XMLSignature.ALGO_ID_SIGNATURE_RSA_SHA1, Transforms.TRANSFORM_C14N_EXCL_OMIT_COMMENTS);
    Element sigElement = sig.getElement();
    document.getDocumentElement().appendChild(sigElement);
    XPathFactory xpf = XPathFactory.newInstance();
    XPath xpath = xpf.newXPath();
    xpath.setNamespaceContext(new DSNamespaceContext());
    Element element = (Element) xpath.evaluate("//*[local-name()='Body']", document, XPathConstants.NODE);
    String id = UUID.randomUUID().toString();
    element.setAttributeNS(null, "Id", id);
    element.setIdAttributeNS(null, "Id", true);
    Transforms transforms = new Transforms(document);
    transforms.addTransform(Transforms.TRANSFORM_C14N_EXCL_OMIT_COMMENTS);
    sig.addDocument("#" + id, transforms, Constants.ALGO_ID_DIGEST_SHA1);
    sig.addKeyInfo(pub);
    sig.sign(priv);
    KeyGenerator keygen = KeyGenerator.getInstance("AES");
    keygen.init(256);
    SecretKey secretKey = keygen.generateKey();
    XMLCipher cipher = XMLCipher.getInstance(XMLCipher.AES_128);
    cipher.init(XMLCipher.ENCRYPT_MODE, secretKey);
    document = cipher.doFinal(document, element, true);
    XMLCipher deCipher = XMLCipher.getInstance(XMLCipher.AES_128);
    if (transformerFactory != null && deCipher.getSerializer() instanceof TransformSerializer) {
        Field f = deCipher.getSerializer().getClass().getDeclaredField("transformerFactory");
        f.setAccessible(true);
        f.set(deCipher.getSerializer(), transformerFactory);
    }
    if (useDocumentSerializer) {
        deCipher.setSerializer(new DocumentSerializer());
    }
    deCipher.init(XMLCipher.DECRYPT_MODE, secretKey);
    deCipher.doFinal(document, element, true);
    XMLSignature xmlSignatureVerifier = new XMLSignature(sigElement, "");
    Assert.assertTrue(xmlSignatureVerifier.checkSignatureValue(pub));
}

Example 2 with DocumentSerializer

use of org.apache.xml.security.encryption.DocumentSerializer in project testcases by coheigea.

the class PerformanceMemoryTest method testRunFirstOutboundEncryptionMemoryPerformance.

@Test
public void testRunFirstOutboundEncryptionMemoryPerformance() throws Exception {
    System.out.println("Testing Outbound Encryption Memory Performance");
    FileWriter outEncryptionSamplesWriter = new FileWriter("target/encryptionOutMemorySamples.txt", false);
    Serializer documentSerializer = new DocumentSerializer();
    Serializer staxSerializer = new StaxSerializer();
    Serializer transformSerializer = new TransformSerializer();
    for (int i = 1; i <= runs; i++) {
        System.out.println("Run " + i);
        File file = generateLargeXMLFile(i * xmlResizeFactor);
        int startTagCount = countXMLStartTags(file);
        outEncryptionSamplesWriter.write("" + startTagCount);
        long startMem = getUsedMemory();
        MemorySamplerThread mst = new MemorySamplerThread(startMem);
        Thread thread = new Thread(mst);
        thread.setPriority(9);
        thread.start();
        File encryptedFile = doDOMEncryptionOutbound(file, startTagCount, documentSerializer);
        mst.setStop(true);
        thread.join();
        outEncryptionSamplesWriter.write(" " + mst.getMaxUsedMemory());
        startMem = getUsedMemory();
        mst = new MemorySamplerThread(startMem);
        thread = new Thread(mst);
        thread.setPriority(9);
        thread.start();
        doDOMEncryptionOutbound(file, startTagCount, staxSerializer);
        mst.setStop(true);
        thread.join();
        outEncryptionSamplesWriter.write(" " + mst.getMaxUsedMemory());
        startMem = getUsedMemory();
        mst = new MemorySamplerThread(startMem);
        thread = new Thread(mst);
        thread.setPriority(9);
        thread.start();
        doDOMEncryptionOutbound(file, startTagCount, transformSerializer);
        mst.setStop(true);
        thread.join();
        outEncryptionSamplesWriter.write(" " + mst.getMaxUsedMemory());
        encryptedFiles.put(startTagCount, encryptedFile);
        outEncryptionSamplesWriter.write("\n");
    }
    outEncryptionSamplesWriter.close();
}

Example 3 with DocumentSerializer

use of org.apache.xml.security.encryption.DocumentSerializer in project testcases by coheigea.

the class PerformanceMemoryTest method testRunSecondInboundDecryptionMemoryPerformance.

@Test
public void testRunSecondInboundDecryptionMemoryPerformance() throws Exception {
    System.out.println("Testing Inbound Decryption Memory Performance");
    FileWriter inEncryptionSamplesWriter = new FileWriter("target/encryptionInMemorySamples.txt", false);
    Serializer documentSerializer = new DocumentSerializer();
    Serializer staxSerializer = new StaxSerializer();
    Serializer transformSerializer = new TransformSerializer();
    int run = 1;
    Iterator<Map.Entry<Integer, File>> mapIterator = encryptedFiles.entrySet().iterator();
    while (mapIterator.hasNext()) {
        Map.Entry<Integer, File> entry = mapIterator.next();
        System.out.println("Run " + (run++));
        File file = entry.getValue();
        Integer startTagCount = entry.getKey();
        inEncryptionSamplesWriter.write("" + startTagCount);
        long startMem = getUsedMemory();
        MemorySamplerThread mst = new MemorySamplerThread(startMem);
        Thread thread = new Thread(mst);
        thread.setPriority(9);
        thread.start();
        doDOMDecryptionInbound(file, startTagCount, documentSerializer);
        mst.setStop(true);
        thread.join();
        inEncryptionSamplesWriter.write(" " + mst.getMaxUsedMemory());
        startMem = getUsedMemory();
        mst = new MemorySamplerThread(startMem);
        thread = new Thread(mst);
        thread.setPriority(9);
        thread.start();
        doDOMDecryptionInbound(file, startTagCount, staxSerializer);
        mst.setStop(true);
        thread.join();
        inEncryptionSamplesWriter.write(" " + mst.getMaxUsedMemory());
        startMem = getUsedMemory();
        mst = new MemorySamplerThread(startMem);
        thread = new Thread(mst);
        thread.setPriority(9);
        thread.start();
        doDOMDecryptionInbound(file, startTagCount, transformSerializer);
        mst.setStop(true);
        thread.join();
        inEncryptionSamplesWriter.write(" " + mst.getMaxUsedMemory());
        inEncryptionSamplesWriter.write("\n");
    }
    inEncryptionSamplesWriter.close();
}

Example 4 with DocumentSerializer

use of org.apache.xml.security.encryption.DocumentSerializer in project testcases by coheigea.

the class PerformanceTimingTest method testRunSecondInboundDecryptionTimePerformance.

@Test
public void testRunSecondInboundDecryptionTimePerformance() throws Exception {
    System.out.println("Testing Inbound Decryption Time Performance");
    FileWriter inEncryptionSamplesWriter = new FileWriter("target/encryptionInTimeSamples.txt", false);
    Serializer documentSerializer = new DocumentSerializer();
    Serializer staxSerializer = new StaxSerializer();
    Serializer transformSerializer = new TransformSerializer();
    int run = 1;
    Iterator<Map.Entry<Integer, File>> mapIterator = encryptedFiles.entrySet().iterator();
    while (mapIterator.hasNext()) {
        Map.Entry<Integer, File> entry = mapIterator.next();
        System.out.println("Run " + (run++));
        File file = entry.getValue();
        Integer startTagCount = entry.getKey();
        inEncryptionSamplesWriter.write("" + startTagCount);
        long start = System.currentTimeMillis();
        doDOMDecryptionInbound(file, startTagCount, documentSerializer);
        inEncryptionSamplesWriter.write(" " + ((System.currentTimeMillis() - start) / 1000.0));
        doGC();
        start = System.currentTimeMillis();
        doDOMDecryptionInbound(file, startTagCount, staxSerializer);
        inEncryptionSamplesWriter.write(" " + ((System.currentTimeMillis() - start) / 1000.0));
        doGC();
        start = System.currentTimeMillis();
        doDOMDecryptionInbound(file, startTagCount, transformSerializer);
        inEncryptionSamplesWriter.write(" " + ((System.currentTimeMillis() - start) / 1000.0));
        doGC();
        inEncryptionSamplesWriter.write("\n");
    }
    inEncryptionSamplesWriter.close();
}

Example 5 with DocumentSerializer

use of org.apache.xml.security.encryption.DocumentSerializer in project testcases by coheigea.

the class PerformanceTimingTest method testRunFirstOutboundEncryptionTimePerformance.

@Test
public void testRunFirstOutboundEncryptionTimePerformance() throws Exception {
    System.out.println("Testing Outbound Encryption Time Performance");
    FileWriter outEncryptionSamplesWriter = new FileWriter("target/encryptionOutTimeSamples.txt", false);
    Serializer documentSerializer = new DocumentSerializer();
    Serializer staxSerializer = new StaxSerializer();
    Serializer transformSerializer = new TransformSerializer();
    for (int i = 1; i <= runs; i++) {
        System.out.println("Run " + i);
        File file = generateLargeXMLFile(i * xmlResizeFactor);
        int startTagCount = countXMLStartTags(file);
        outEncryptionSamplesWriter.write("" + startTagCount);
        long start = System.currentTimeMillis();
        File encryptedFile = doDOMEncryptionOutbound(file, startTagCount, documentSerializer);
        outEncryptionSamplesWriter.write(" " + ((System.currentTimeMillis() - start) / 1000.0));
        doGC();
        start = System.currentTimeMillis();
        doDOMEncryptionOutbound(file, startTagCount, staxSerializer);
        outEncryptionSamplesWriter.write(" " + ((System.currentTimeMillis() - start) / 1000.0));
        doGC();
        start = System.currentTimeMillis();
        doDOMEncryptionOutbound(file, startTagCount, transformSerializer);
        outEncryptionSamplesWriter.write(" " + ((System.currentTimeMillis() - start) / 1000.0));
        doGC();
        encryptedFiles.put(startTagCount, encryptedFile);
        outEncryptionSamplesWriter.write("\n");
    }
    outEncryptionSamplesWriter.close();
}