Support encrypted PDFs - Resolves #11

Author: ViliusSutkus89

pdf2htmlEX/src/androidTest/assets/encrypted_fontfile3_opentype.pdf

@@ -36,6 +36,7 @@
 import java.io.InputStream;
 import java.io.OutputStream;
 
+import static org.junit.Assert.assertNull;
 import static org.junit.Assert.assertTrue;
 import static org.junit.Assert.fail;
 
@@ -104,4 +105,57 @@ public void conversionTwiceTest() throws IOException {
     conversionTest();
   }
 
+  @Test
+  public synchronized void encryptedPdfTest() throws IOException {
+    pdf2htmlEX converter = new pdf2htmlEX(InstrumentationRegistry.getInstrumentation().getTargetContext());
+
+    // encrypted_fontfile3_opentype.pdf generated using:
+    // qpdf --encrypt sample-user-password sample-owner-password 256 -- fontfile3_opentype.pdf encrypted_fontfile3_opentype.pdf
+    File pdfFile = extractAssetPDF("encrypted_fontfile3_opentype.pdf");
+    File htmlFile;
+    try {
+      htmlFile = converter.setInputPDF(pdfFile)
+              .setOwnerPassword("sample-owner-password")
+              .setUserPassword("sample-user-password")
+              .convert();
+    } catch (IOException | pdf2htmlEX.ConversionFailedException e) {
+      pdfFile.delete();
+      e.printStackTrace();
+      fail("Failed to convert PDF to HTML");
+      return;
+    }
+
+    pdfFile.delete();
+
+    assertTrue("Converted HTML file not found!", htmlFile.exists());
+    assertTrue("Converted HTML file empty!", htmlFile.length() > 0);
+
+    htmlFile.delete();
+  }
+
+  @Test
+  public synchronized void encryptedPdfWrongPasswordTest() throws IOException {
+    pdf2htmlEX converter = new pdf2htmlEX(InstrumentationRegistry.getInstrumentation().getTargetContext());
+
+    // encrypted_fontfile3_opentype.pdf generated using:
+    // qpdf --encrypt sample-user-password sample-owner-password 256 -- fontfile3_opentype.pdf encrypted_fontfile3_opentype.pdf
+    File pdfFile = extractAssetPDF("encrypted_fontfile3_opentype.pdf");
+    File htmlFile = null;
+    try {
+      htmlFile = converter.setInputPDF(pdfFile)
+              .setOwnerPassword("wrong-owner-password")
+              .setUserPassword("wrong-user-password")
+              .convert();
+    } catch (pdf2htmlEX.ConversionFailedException ignored) {
+    } catch (IOException e) {
+      pdfFile.delete();
+      e.printStackTrace();
+      fail("Failed to convert PDF to HTML");
+      return;
+    }
+    pdfFile.delete();
+
+    assertNull("Conversion succeeded when it should have failed because of wrong encryption password!", htmlFile);
+  }
+
 }

pdf2htmlEX/src/androidTest/java/com/viliussutkus89/android/pdf2htmlex/InstrumentedTests.java

@@ -42,6 +42,8 @@ class CCharGC {
       return this->cstr;
     }
 
+    bool isEmpty() const { return this->cstr[0] == '\0'; }
+
     ~CCharGC() {
       env->ReleaseStringUTFChars(this->input, this->cstr);
     }
@@ -88,12 +90,16 @@ Java_com_viliussutkus89_android_pdf2htmlex_pdf2htmlEX_call_1pdf2htmlEX(JNIEnv *e
                                                                jstring dataDir_,
                                                                jstring popplerDir_, jstring tmpDir_,
                                                                jstring inputFile_,
-                                                               jstring outputFile_) {
+                                                               jstring outputFile_,
+                                                               jstring ownerPassword_,
+                                                               jstring userPassword_) {
   CCharGC dataDir(env, dataDir_);
   CCharGC popplerDir(env, popplerDir_);
   CCharGC tmpDir(env, tmpDir_);
   CCharGC inputFile(env, inputFile_);
   CCharGC outputFile(env, outputFile_);
+  CCharGC ownerPassword(env, ownerPassword_);
+  CCharGC userPassword(env, userPassword_);
 
   std::vector<const std::string> args = {
     "libpdf2htmlEX",
@@ -103,6 +109,16 @@ Java_com_viliussutkus89_android_pdf2htmlex_pdf2htmlEX_call_1pdf2htmlEX(JNIEnv *e
     inputFile.c_str(), outputFile.c_str()
   };
 
+  if (!ownerPassword.isEmpty()) {
+    args.push_back("--owner-password");
+    args.push_back(ownerPassword.c_str());
+  }
+
+  if (!userPassword.isEmpty()) {
+    args.push_back("--user-password");
+    args.push_back(userPassword.c_str());
+  }
+
   int argc;
   char **argv;
   int retVal = -1;

pdf2htmlEX/src/main/cpp/pdf2htmlEX.cc

@@ -40,6 +40,8 @@ public final class pdf2htmlEX {
   private File m_outputHtmlsDir;
 
   private File p_inputPDF;
+  private String p_ownerPassword = "";
+  private String p_userPassword = "";
 
   public static class ConversionFailedException extends Exception {
     public ConversionFailedException(String errorMessage) {
@@ -98,6 +100,16 @@ public File convert(@NonNull File inputPDF) throws IOException, ConversionFailed
     return convert();
   }
 
+  public pdf2htmlEX setOwnerPassword(@NonNull String ownerPassword) {
+    this.p_ownerPassword = ownerPassword;
+    return this;
+  }
+
+  public pdf2htmlEX setUserPassword(@NonNull String userPassword) {
+    this.p_userPassword = userPassword;
+    return this;
+  }
+
   public File convert() throws IOException, ConversionFailedException {
     if (null == this.p_inputPDF) {
       throw new ConversionFailedException("No Input PDF given!");
@@ -119,7 +131,9 @@ public File convert() throws IOException, ConversionFailedException {
 
     Integer retVal = call_pdf2htmlEX(m_pdf2htmlEX_dataDir.getAbsolutePath(),
       m_poppler_dataDir.getAbsolutePath(), m_pdf2htmlEX_tmpDir.getAbsolutePath(),
-      this.p_inputPDF.getAbsolutePath(), outputHtml.getAbsolutePath());
+      this.p_inputPDF.getAbsolutePath(), outputHtml.getAbsolutePath(),
+      this.p_ownerPassword, this.p_userPassword
+      );
 
     if (0 != retVal) {
       outputHtml.delete();
@@ -129,7 +143,7 @@ public File convert() throws IOException, ConversionFailedException {
     return outputHtml;
   }
 
-  private native int call_pdf2htmlEX(String dataDir, String popplerDir, String tmpDir, String inputFile, String outputFile);
+  private native int call_pdf2htmlEX(String dataDir, String popplerDir, String tmpDir, String inputFile, String outputFile, String ownerPassword, String userPassword);
 
   // Because Java cannot setenv for the current process
   static native void set_environment_value(String key, String value);