Refactor encryption handling to streamline key computation, centralize version-b…

…ased logic in `setEncryptionMinimumVersion`, and reduce redundancy across `QPDFWriter` methods.

Refactor encryption handling to streamline key computation, centralize version-b…
…ased logic in `setEncryptionMinimumVersion`, and reduce redundancy across `QPDFWriter` methods.
m-holger
1 parent fe122b0b
Showing 4 changed files with 9 additions and 20 deletions
include/qpdf/QPDFWriter.hh
libqpdf/QPDFWriter.cc
libqpdf/QPDF_encryption.cc
qpdf/qpdf.testcov
@@ -523,8 +523,7 @@ class QPDFWriter
     void parseVersion(std::string const& version, int& major, int& minor) const;
     int compareVersions(int major1, int minor1, int major2, int minor2) const;
     void setEncryptionParameters(char const* user_password, char const* owner_password);
-    void setEncryptionParametersInternal(
-        std::string const& user_password, std::string const& encryption_key);
+    void setEncryptionMinimumVersion();
     void setDataKey(int objid);
     int openObject(int objid = 0);
     void closeObject(int objid);
@@ -655,8 +655,8 @@ QPDFWriter::setEncryptionParameters(char const* user_password, char const* owner
 {
     generateID();
     m->encryption->setId1(m->id1);
-    auto encryption_key = m->encryption->compute_parameters(user_password, owner_password);
-    setEncryptionParametersInternal(user_password, encryption_key);
+    m->encryption_key = m->encryption->compute_parameters(user_password, owner_password);
+    setEncryptionMinimumVersion();
 }
 void
@@ -686,11 +686,6 @@ QPDFWriter::copyEncryptionParameters(QPDF&amp; qpdf)
         }
         QTC::TC("qpdf", "QPDFWriter copy encrypt metadata", encrypt_metadata ? 0 : 1);
         QTC::TC("qpdf", "QPDFWriter copy use_aes", m->encrypt_use_aes ? 0 : 1);
-        std::string encryption_key;
-        if (V >= 5) {
-            QTC::TC("qpdf", "QPDFWriter copy V5");
-            encryption_key = qpdf.getEncryptionKey();
-        }
         m->encryption = std::make_unique<QPDF::EncryptionData>(
             V,
@@ -704,7 +699,10 @@ QPDFWriter::copyEncryptionParameters(QPDF&amp; qpdf)
             V < 5 ? "" : encrypt.getKey("/Perms").getStringValue(),
             m->id1, // m->id1 == the other file's id1
             encrypt_metadata);
-        setEncryptionParametersInternal(qpdf.getPaddedUserPassword(), encryption_key);
+        m->encryption_key = V >= 5
+            ? qpdf.getEncryptionKey()
+            : m->encryption->compute_encryption_key(qpdf.getPaddedUserPassword());
+        setEncryptionMinimumVersion();
     }
 }
@@ -779,8 +777,7 @@ QPDFWriter::compareVersions(int major1, int minor1, int major2, int minor2) cons
 }
 void
-QPDFWriter::setEncryptionParametersInternal(
-    std::string const& user_password, std::string const& encryption_key)
+QPDFWriter::setEncryptionMinimumVersion()
 {
     auto const R = m->encryption->getR();
     if (R >= 6) {
@@ -794,12 +791,6 @@ QPDFWriter::setEncryptionParametersInternal(
     } else {
         setMinimumPDFVersion("1.3");
     }
-
-    if (m->encryption->getV() < 5) {
-        m->encryption_key = m->encryption->compute_encryption_key(user_password);
-    } else {
-        m->encryption_key = encryption_key;
-    }
 }
 void
@@ -1201,7 +1201,7 @@ QPDF::EncryptionData::compute_parameters(char const* user_password, char const* 
 {
     if (V < 5) {
         compute_encryption_O_U(user_password, owner_password);
-        return {};
+        return compute_encryption_key(user_password);
     } else {
         return compute_encryption_parameters_V5(user_password, owner_password);
     }
@@ -228,7 +228,6 @@ QPDFWriter extra header text no newline 0
 QPDFWriter extra header text add newline 0
 QPDF bogus 0 offset 0
 QPDF global offset 0
-QPDFWriter copy V5 0
 QPDFWriter increasing extension level 0
 QPDFWriter make Extensions direct 0
 QPDFWriter make ADBE direct 1