Merge pull request #94 from LibrePDF/dss-openpdf

Add includeFileID and overrideFileId properties to PdfStampler.
LibrePDF · Aug 22, 2018 · f42d7e5 · f42d7e5
2 parents 3f6ccac + e629e16
commit f42d7e5
Show file tree

Hide file tree

Showing 2 changed files with 46 additions and 3 deletions.
diff --git a/openpdf/src/main/java/com/lowagie/text/pdf/PdfStamper.java b/openpdf/src/main/java/com/lowagie/text/pdf/PdfStamper.java
@@ -760,4 +760,23 @@ public static PdfStamper createSignature(PdfReader reader, OutputStream os, char
     public Map getPdfLayers() {
     	return stamper.getPdfLayers();
     }
+
+    public void setIncludeFileID(boolean includeFileID) {
+        this.stamper.setIncludeFileID(includeFileID);
+    }
+
+    public boolean isIncludeFileID() {
+        return stamper.isIncludeFileID();
+    }
+
+    public void setOverrideFileId(PdfObject overrideFileId) {
+        this.stamper.setOverrideFileId(overrideFileId);
+    }
+
+    public PdfObject getOverrideFileId() {
+        return stamper.getOverrideFileId();
+    }
+
+
+
 }
diff --git a/openpdf/src/main/java/com/lowagie/text/pdf/PdfStamperImp.java b/openpdf/src/main/java/com/lowagie/text/pdf/PdfStamperImp.java
@@ -94,6 +94,8 @@ class PdfStamperImp extends PdfWriter {
     protected IntHashtable marked;
     protected int initialXrefSize;
     protected PdfAction openAction;
+    private boolean includeFileID = true;
+    private PdfObject overrideFileId = null;
 
     /** Creates new PdfStamperImp.
      * @param reader the read PDF
@@ -319,10 +321,16 @@ else if (producer.indexOf(Document.getProduct()) == -1) {
                 PdfIndirectObject encryptionObject = addToBody(crypto.getEncryptionDictionary(), false);
                 encryption = encryptionObject.getIndirectReference();
             }
-            fileID = crypto.getFileID();
+            if (includeFileID) fileID = crypto.getFileID();
+        }
+        else if (includeFileID) {
+            if (overrideFileId != null) {
+                fileID = overrideFileId;
+            } else {
+                fileID = PdfEncryption.createInfoId(PdfEncryption.createDocumentId());
+            }
+
         }
-        else
-            fileID = PdfEncryption.createInfoId(PdfEncryption.createDocumentId());
         PRIndirectReference iRoot = (PRIndirectReference)reader.trailer.get(PdfName.ROOT);
         PdfIndirectReference root = new PdfIndirectReference(0, getNewObjectNumber(reader, iRoot.getNumber(), 0));
         PdfIndirectReference info = null;
@@ -1690,4 +1698,20 @@ static class PageStamp {
             pageResources.setOriginalResources(resources, stamper.namePtr);
         }
     }
+
+    public boolean isIncludeFileID() {
+        return includeFileID;
+    }
+
+    public void setIncludeFileID(boolean includeFileID) {
+        this.includeFileID = includeFileID;
+    }
+
+    public PdfObject getOverrideFileId() {
+        return overrideFileId;
+    }
+
+    public void setOverrideFileId(PdfObject overrideFileId) {
+        this.overrideFileId = overrideFileId;
+    }
 }