From ed4e7cd8a495e092add73ef168ec7ed74ea4bdea Mon Sep 17 00:00:00 2001
From: Jonas Jenwald <jonas.jenwald@gmail.com>
Date: Mon, 14 Sep 2020 13:31:33 +0200
Subject: [PATCH] A couple of small improvements in the "SaveDocument" handler
 in `src/core/worker.js`

 - Check that the "Info"-entry, in the XRef-trailer, is actually a dictionary before accessing it. This is similar to the `PDFDocument.documentInfo` method and follows the general principal of validating data carefully before accessing it, given how often PDF-software may create corrupt PDF files.

 - Slightly simplify the "XFA"-lookup, since there's no point in trying to fetch something from the empty dictionary.
---
 src/core/worker.js | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/src/core/worker.js b/src/core/worker.js
index 4fdd44e923e34..bd6257de07a43 100644
--- a/src/core/worker.js
+++ b/src/core/worker.js
@@ -32,7 +32,7 @@ import {
   VerbosityLevel,
   warn,
 } from "../shared/util.js";
-import { clearPrimitiveCaches, Dict, isDict, Ref } from "./primitives.js";
+import { clearPrimitiveCaches, Dict, Ref } from "./primitives.js";
 import { LocalPdfManager, NetworkPdfManager } from "./pdf_manager.js";
 import { incrementalUpdate } from "./writer.js";
 import { isNodeJS } from "../shared/is_node.js";
@@ -548,8 +548,7 @@ class WorkerMessageHandler {
           return stream.bytes;
         }
 
-        acroForm = isDict(acroForm) ? acroForm : Dict.empty;
-        const xfa = acroForm.get("XFA") || [];
+        const xfa = (acroForm instanceof Dict && acroForm.get("XFA")) || [];
         let xfaDatasets = null;
         if (Array.isArray(xfa)) {
           for (let i = 0, ii = xfa.length; i < ii; i += 2) {
@@ -568,7 +567,7 @@ class WorkerMessageHandler {
           // Get string info from Info in order to compute fileId
           const _info = Object.create(null);
           const xrefInfo = xref.trailer.get("Info") || null;
-          if (xrefInfo) {
+          if (xrefInfo instanceof Dict) {
             xrefInfo.forEach((key, value) => {
               if (isString(key) && isString(value)) {
                 _info[key] = stringToPDFString(value);