1 files changed, 79 insertions, 0 deletions
diff --git a/core/fpdfapi/parser/cpdf_indirect_object_holder.cpp b/core/fpdfapi/parser/cpdf_indirect_object_holder.cpp
new file mode 100644
index 0000000000..6e549de5a7
--- /dev/null
+++ b/core/fpdfapi/parser/cpdf_indirect_object_holder.cpp
@@ -0,0 +1,79 @@
+// Copyright 2016 PDFium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
+
+#include "core/fpdfapi/parser/cpdf_indirect_object_holder.h"
+
+#include "core/fpdfapi/parser/cpdf_object.h"
+#include "core/fpdfapi/parser/cpdf_parser.h"
+
+CPDF_IndirectObjectHolder::CPDF_IndirectObjectHolder() : m_LastObjNum(0) {}
+
+CPDF_IndirectObjectHolder::~CPDF_IndirectObjectHolder() {}
+
+CPDF_Object* CPDF_IndirectObjectHolder::GetIndirectObject(
+    uint32_t objnum) const {
+  auto it = m_IndirectObjs.find(objnum);
+  return it != m_IndirectObjs.end() ? it->second.get() : nullptr;
+}
+
+CPDF_Object* CPDF_IndirectObjectHolder::GetOrParseIndirectObject(
+    uint32_t objnum) {
+  if (objnum == 0)
+    return nullptr;
+
+  CPDF_Object* pObj = GetIndirectObject(objnum);
+  if (pObj)
+    return pObj->GetObjNum() != CPDF_Object::kInvalidObjNum ? pObj : nullptr;
+
+  pObj = ParseIndirectObject(objnum);
+  if (!pObj)
+    return nullptr;
+
+  pObj->m_ObjNum = objnum;
+  m_LastObjNum = std::max(m_LastObjNum, objnum);
+  m_IndirectObjs[objnum].reset(pObj);
+  return pObj;
+}
+
+CPDF_Object* CPDF_IndirectObjectHolder::ParseIndirectObject(uint32_t objnum) {
+  return nullptr;
+}
+
+uint32_t CPDF_IndirectObjectHolder::AddIndirectObject(CPDF_Object* pObj) {
+  if (pObj->m_ObjNum)
+    return pObj->m_ObjNum;
+
+  m_LastObjNum++;
+  m_IndirectObjs[m_LastObjNum].release();  // TODO(tsepez): stop this leak.
+  m_IndirectObjs[m_LastObjNum].reset(pObj);
+  pObj->m_ObjNum = m_LastObjNum;
+  return m_LastObjNum;
+}
+
+bool CPDF_IndirectObjectHolder::ReplaceIndirectObjectIfHigherGeneration(
+    uint32_t objnum,
+    CPDF_Object* pObj) {
+  if (!objnum || !pObj)
+    return false;
+
+  CPDF_Object* pOldObj = GetIndirectObject(objnum);
+  if (pOldObj && pObj->GetGenNum() <= pOldObj->GetGenNum()) {
+    delete pObj;
+    return false;
+  }
+  pObj->m_ObjNum = objnum;
+  m_IndirectObjs[objnum].reset(pObj);
+  m_LastObjNum = std::max(m_LastObjNum, objnum);
+  return true;
+}
+
+void CPDF_IndirectObjectHolder::ReleaseIndirectObject(uint32_t objnum) {
+  CPDF_Object* pObj = GetIndirectObject(objnum);
+  if (!pObj || pObj->GetObjNum() == CPDF_Object::kInvalidObjNum)
+    return;
+
+  m_IndirectObjs.erase(objnum);
+}