From 9064542b5f23bde3e7d6826b5b5e8b0a6f356434 Mon Sep 17 00:00:00 2001
From: Jay Berkenbilt <ejb@ql.org>
Date: Sun, 15 May 2022 13:10:10 -0400
Subject: Add private methods for reserving specific objects

---
 libqpdf/QPDF.cc             | 11 +++++++++++
 libqpdf/QPDFObjectHandle.cc |  9 +++++++--
 2 files changed, 18 insertions(+), 2 deletions(-)

(limited to 'libqpdf')

diff --git a/libqpdf/QPDF.cc b/libqpdf/QPDF.cc
index 2658dd26..1807a838 100644
--- a/libqpdf/QPDF.cc
+++ b/libqpdf/QPDF.cc
@@ -2155,6 +2155,17 @@ QPDF::makeIndirectObject(QPDFObjectHandle oh)
         this, next.getObj(), next.getGen());
 }
 
+QPDFObjectHandle
+QPDF::reserveObjectIfNotExists(int objid, int gen)
+{
+    QPDFObjGen og(objid, gen);
+    if ((!this->m->obj_cache.count(og)) && (!this->m->xref_table.count(og))) {
+        resolve(objid, gen);
+        replaceObject(objid, gen, QPDFObjectHandle::Factory::makeReserved());
+    }
+    return getObjectByID(objid, gen);
+}
+
 QPDFObjectHandle
 QPDF::getObjectByObjGen(QPDFObjGen const& og)
 {
diff --git a/libqpdf/QPDFObjectHandle.cc b/libqpdf/QPDFObjectHandle.cc
index 4a61f595..4a0a69ae 100644
--- a/libqpdf/QPDFObjectHandle.cc
+++ b/libqpdf/QPDFObjectHandle.cc
@@ -2775,14 +2775,19 @@ QPDFObjectHandle::newReserved(QPDF* qpdf)
 {
     // Reserve a spot for this object by assigning it an object
     // number, but then return an unresolved handle to the object.
-    QPDFObjectHandle reserved =
-        qpdf->makeIndirectObject(QPDFObjectHandle(new QPDF_Reserved()));
+    QPDFObjectHandle reserved = qpdf->makeIndirectObject(makeReserved());
     QPDFObjectHandle result =
         newIndirect(qpdf, reserved.objid, reserved.generation);
     result.reserved = true;
     return result;
 }
 
+QPDFObjectHandle
+QPDFObjectHandle::makeReserved()
+{
+    return QPDFObjectHandle(new QPDF_Reserved());
+}
+
 void
 QPDFObjectHandle::setObjectDescription(
     QPDF* owning_qpdf, std::string const& object_description)
-- 
cgit v1.2.3-54-g00ecf