You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@pdfbox.apache.org by ti...@apache.org on 2014/10/29 20:03:28 UTC
svn commit: r1635256 -
/pdfbox/trunk/pdfbox/src/test/java/org/apache/pdfbox/encryption/TestSymmetricKeyEncryption.java
Author: tilman
Date: Wed Oct 29 19:03:28 2014
New Revision: 1635256
URL: http://svn.apache.org/r1635256
Log:
PDFBOX-2456: symetric key test for 40 and 128 bit length, as suggested by Ralf Hauser
Added:
pdfbox/trunk/pdfbox/src/test/java/org/apache/pdfbox/encryption/TestSymmetricKeyEncryption.java (with props)
Added: pdfbox/trunk/pdfbox/src/test/java/org/apache/pdfbox/encryption/TestSymmetricKeyEncryption.java
URL: http://svn.apache.org/viewvc/pdfbox/trunk/pdfbox/src/test/java/org/apache/pdfbox/encryption/TestSymmetricKeyEncryption.java?rev=1635256&view=auto
==============================================================================
--- pdfbox/trunk/pdfbox/src/test/java/org/apache/pdfbox/encryption/TestSymmetricKeyEncryption.java (added)
+++ pdfbox/trunk/pdfbox/src/test/java/org/apache/pdfbox/encryption/TestSymmetricKeyEncryption.java Wed Oct 29 19:03:28 2014
@@ -0,0 +1,297 @@
+/*
+ * 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.pdfbox.encryption;
+
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+import java.io.File;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStreamWriter;
+import java.io.Writer;
+import javax.crypto.Cipher;
+
+import junit.framework.TestCase;
+import static junit.framework.TestCase.fail;
+
+import org.apache.pdfbox.cos.COSStream;
+import org.apache.pdfbox.pdmodel.PDDocument;
+import org.apache.pdfbox.pdmodel.encryption.AccessPermission;
+import org.apache.pdfbox.pdmodel.encryption.DecryptionMaterial;
+import org.apache.pdfbox.pdmodel.encryption.StandardDecryptionMaterial;
+import org.apache.pdfbox.pdmodel.encryption.StandardProtectionPolicy;
+import org.apache.pdfbox.util.PDFTextStripper;
+import org.junit.Assert;
+
+/**
+ * Tests for public key encryption.
+ *
+ * @author Ralf Hauser
+ */
+public class TestSymmetricKeyEncryption extends TestCase
+{
+ private final File testResultsDir = new File("target/test-output/crypto");
+
+ private AccessPermission permission1;
+ private AccessPermission permission2;
+
+ private int sizePriorToEncryption = -1;
+ private int sizePriorToEncryptionInnerSubfile = -1;
+
+ static final String PASSWORD = "1234567890abcdefghijk1234567890abcdefghijk";
+
+ static byte[] inputFileAsByteArray = null;
+ static byte[] inputInner = null;
+
+ static String textContent = null;
+
+ static int page0size = -1;
+
+ /**
+ * Simple test document that gets encrypted by the test cases.
+ */
+ private PDDocument document;
+
+ /**
+ * {@inheritDoc}
+ */
+ @Override
+ protected void setUp() throws Exception
+ {
+ testResultsDir.mkdirs();
+
+ if (Cipher.getMaxAllowedKeyLength("AES") != Integer.MAX_VALUE)
+ {
+ // we need strong encryption for these tests
+ fail("JCE unlimited strength jurisdiction policy files are not installed");
+ }
+
+ permission1 = new AccessPermission();
+ permission1.setCanAssembleDocument(false);
+ permission1.setCanExtractContent(false);
+ permission1.setCanExtractForAccessibility(true);
+ permission1.setCanFillInForm(false);
+ permission1.setCanModify(false);
+ permission1.setCanModifyAnnotations(false);
+ permission1.setCanPrint(false);
+ permission1.setCanPrintDegraded(false);
+ permission1.setReadOnly();
+
+ permission2 = new AccessPermission();
+ permission2.setCanAssembleDocument(false);
+ permission2.setCanExtractContent(false);
+ permission2.setCanExtractForAccessibility(true);
+ permission2.setCanFillInForm(false);
+ permission2.setCanModify(false);
+ permission2.setCanModifyAnnotations(false);
+ permission2.setCanPrint(true); // it is true now !
+ permission2.setCanPrintDegraded(false);
+
+ String testFileName = "test.pdf";
+ inputFileAsByteArray = getFileAsByteArray(testFileName);
+ try
+ {
+ sizePriorToEncryption = inputFileAsByteArray.length;
+ document = PDDocument.load(new ByteArrayInputStream(inputFileAsByteArray));
+ boolean extractText = false;
+ if (extractText)
+ {
+ PDFTextStripper stripper = new PDFTextStripper();
+ stripper.setForceParsing(true);
+ // stripper.setSortByPosition( sort );
+ // stripper.setShouldSeparateByBeads( separateBeads );
+ stripper.setStartPage(0);
+ stripper.setEndPage(10);
+ ByteArrayOutputStream baos = new ByteArrayOutputStream();
+ Writer output = new OutputStreamWriter(baos);
+ stripper.writeText(document, output);
+ textContent = new String(baos.toByteArray());
+ // content "" ;(
+ }
+ COSStream contentStream = document.getPage(0).getContentStream();
+ page0size = (int) contentStream.getFilteredLength();// was 2
+ // contentStream.size();
+ }
+ catch (Exception t)
+ {
+ System.err.println(testFileName + " " + t.getMessage());
+ t.printStackTrace();
+ throw t;
+ }
+ }
+
+ public byte[] getFileAsByteArray(String testFileName) throws Exception,
+ IOException
+ {
+ InputStream is = TestSymmetricKeyEncryption.class
+ .getResourceAsStream(testFileName);
+ try
+ {
+ try
+ {
+ ByteArrayOutputStream baos = new ByteArrayOutputStream();
+ while (true)
+ {
+ int c = is.read();
+ if (c == -1)
+ {
+ break;
+ }
+ baos.write(c);
+ }
+ baos.close();
+ return baos.toByteArray();
+ }
+ finally
+ {
+ is.close();
+ }
+ }
+ catch (Exception t)
+ {
+ System.err.println(testFileName + " " + t.getMessage());
+ t.printStackTrace();
+ throw t;
+ }
+ finally
+ {
+ is.close();
+ }
+ }
+
+ /**
+ * {@inheritDoc}
+ */
+ @Override
+ protected void tearDown() throws Exception
+ {
+ document.close();
+ }
+
+ /**
+ * Protect a document with a public certificate and try to open it with the
+ * corresponding private certificate.
+ *
+ * @throws Exception If there is an unexpected error during the test.
+ */
+ public void testProtection() throws Exception
+ {
+ testSymmEncrForKeySize(40, sizePriorToEncryption, document, PASSWORD, permission1);
+ document = PDDocument.load(new ByteArrayInputStream(inputFileAsByteArray));
+ testSymmEncrForKeySize(128, sizePriorToEncryption, document, PASSWORD, permission1);
+
+ //TODO
+ // 1) check permissions
+ // 2) 256 key length
+ //document = PDDocument.load(new ByteArrayInputStream(inputFileAsByteArray));
+ //testSymmEncrForKeySize(256, sizePriorToEncryption, document, PASSWORD);
+ }
+
+ public void testSymmEncrForKeySize(int keyLength,
+ int sizePriorToEncr, PDDocument doc, String password, AccessPermission permission) throws IOException
+ {
+ PDDocument encryptedDoc = encrypt(keyLength, sizePriorToEncr, doc, "", permission);
+
+ try
+ {
+ Assert.assertTrue(encryptedDoc.isEncrypted());
+ DecryptionMaterial decryptionMaterial = new StandardDecryptionMaterial(PASSWORD);
+ encryptedDoc.openProtection(decryptionMaterial);
+
+ AccessPermission newPermission = encryptedDoc.getCurrentAccessPermission();
+
+ File pdfFile = new File(testResultsDir, keyLength + "-bit-decrypted.pdf");
+ encryptedDoc.save(pdfFile);
+ long sizeAfterDecr = pdfFile.length();
+ {
+ // for some reason, they are not identical :( 12263 vs 12418
+ // Assert.assertTrue(
+ // keyLength
+ // + "bit decrypted pdf should have same size as plain one",
+ // sizeAfterDecr == sizePriorToEncr);
+ }
+ // difference already at position 8
+ // for (int i = 0; i < 500 // byteArray.length
+ // ; i++) {
+ // byte b = byteArrayDecr[i];
+ // byte c = input[i];
+ // Assert.assertTrue(keyLength
+ // + "bit decrypted: character different in pos " +
+ // i+" of "+sizeAfterDecr,
+ // b == c);
+ // }
+ COSStream contentStreamDecr = encryptedDoc.getPage(0).getContentStream();
+ int decrSizePage0 = (int) contentStreamDecr.getFilteredLength();// was
+ // 2
+ // contentStream.size();
+ Assert.assertTrue(
+ keyLength
+ + "bit decrypted pdf page 0 should have same size as plain one",
+ page0size == decrSizePage0);
+
+ boolean canAssembleDocument = newPermission.canAssembleDocument();
+ boolean canExtractContent = newPermission.canExtractContent();
+ boolean canExtractForAccessibility = newPermission
+ .canExtractForAccessibility();
+ boolean canFillInForm = newPermission.canFillInForm();
+ boolean canModify = newPermission.canModify();
+ boolean canModifyAnnotations = newPermission.canModifyAnnotations();
+ boolean canPrint = newPermission.canPrint();
+ boolean canPrintDegraded = newPermission.canPrintDegraded();
+ encryptedDoc.close();
+// Assert.assertFalse(canAssembleDocument);
+// Assert.assertFalse(canExtractContent);
+// Assert.assertTrue(canExtractForAccessibility);
+// Assert.assertFalse(canFillInForm);
+// Assert.assertFalse(canModify);
+// Assert.assertFalse(canModifyAnnotations);
+// Assert.assertFalse(canPrint);
+// Assert.assertFalse(canPrintDegraded);
+ }
+ finally
+ {
+ encryptedDoc.close();
+ }
+ }
+
+ public PDDocument encrypt(int keyLength, int sizePriorToEncr,
+ PDDocument doc, String spec, AccessPermission permission) throws IOException
+ {
+ AccessPermission ap = new AccessPermission();
+ StandardProtectionPolicy spp = new StandardProtectionPolicy(PASSWORD, PASSWORD, ap);
+ spp.setEncryptionKeyLength(keyLength);
+ spp.setPermissions(permission);
+ doc.protect(spp);
+
+ File pdfFile = new File(testResultsDir, keyLength + "-bit-encrypted.pdf");
+
+ doc.save(pdfFile);
+ doc.close();
+ long sizeEncrypted = pdfFile.length();
+ PDDocument encryptedDoc = PDDocument.load(pdfFile);
+ Assert.assertTrue(keyLength
+ + "-bit encrypted pdf should not have same size as plain one",
+ sizeEncrypted != sizePriorToEncr);
+// COSStream contentStream = encrypted.getPage(0).getContentStream();
+// int encrPage0size = (int) contentStream.getFilteredLength();// was 2
+// // contentStream.size();
+
+ return encryptedDoc;
+ }
+
+
+}
Propchange: pdfbox/trunk/pdfbox/src/test/java/org/apache/pdfbox/encryption/TestSymmetricKeyEncryption.java
------------------------------------------------------------------------------
svn:eol-style = native