blob: 777310bb2b59778424eb6286cfbc6e9878b27f9a [file] [log] [blame]
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -07001// Copyright 2014 PDFium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
Lei Zhanga6d9f0e2015-06-13 00:48:38 -07004
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -07005// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
6
Lei Zhangb4e7f302015-11-06 15:52:32 -08007#include "public/fpdf_ppo.h"
8
Dan Sinclair3ebd1212016-03-09 09:59:23 -05009#include <map>
Lei Zhangaa8bf7e2015-12-24 19:13:32 -080010#include <memory>
Dan Sinclair3ebd1212016-03-09 09:59:23 -050011#include <vector>
Lei Zhangaa8bf7e2015-12-24 19:13:32 -080012
Dan Sinclairaa403d32016-03-15 14:57:22 -040013#include "core/fpdfapi/fpdf_parser/include/cpdf_array.h"
14#include "core/fpdfapi/fpdf_parser/include/cpdf_document.h"
15#include "core/fpdfapi/fpdf_parser/include/cpdf_name.h"
16#include "core/fpdfapi/fpdf_parser/include/cpdf_number.h"
17#include "core/fpdfapi/fpdf_parser/include/cpdf_reference.h"
Dan Sinclair584b1e62016-03-21 09:15:45 -040018#include "core/fpdfapi/fpdf_parser/include/cpdf_stream.h"
Dan Sinclairaa403d32016-03-15 14:57:22 -040019#include "core/fpdfapi/fpdf_parser/include/cpdf_string.h"
Lei Zhangbde53d22015-11-12 22:21:30 -080020#include "fpdfsdk/include/fsdk_define.h"
Tom Sepez11d93552016-02-09 09:55:54 -080021#include "third_party/base/stl_util.h"
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -070022
Nico Weber9d8ec5a2015-08-04 13:00:21 -070023class CPDF_PageOrganizer {
24 public:
tsepezc3255f52016-03-25 14:52:27 -070025 using ObjectNumberMap = std::map<uint32_t, uint32_t>;
Nico Weber9d8ec5a2015-08-04 13:00:21 -070026 CPDF_PageOrganizer();
27 ~CPDF_PageOrganizer();
Lei Zhanga6d9f0e2015-06-13 00:48:38 -070028
Nico Weber9d8ec5a2015-08-04 13:00:21 -070029 FX_BOOL PDFDocInit(CPDF_Document* pDestPDFDoc, CPDF_Document* pSrcPDFDoc);
30 FX_BOOL ExportPage(CPDF_Document* pSrcPDFDoc,
Tom Sepez62a70f92016-03-21 15:00:20 -070031 std::vector<uint16_t>* pPageNums,
Nico Weber9d8ec5a2015-08-04 13:00:21 -070032 CPDF_Document* pDestPDFDoc,
33 int nIndex);
34 CPDF_Object* PageDictGetInheritableTag(CPDF_Dictionary* pDict,
35 CFX_ByteString nSrctag);
36 FX_BOOL UpdateReference(CPDF_Object* pObj,
37 CPDF_Document* pDoc,
Tom Sepezc7e4c4f2015-11-20 09:45:24 -080038 ObjectNumberMap* pObjNumberMap);
tsepezc3255f52016-03-25 14:52:27 -070039 uint32_t GetNewObjId(CPDF_Document* pDoc,
Tom Sepezc7e4c4f2015-11-20 09:45:24 -080040 ObjectNumberMap* pObjNumberMap,
41 CPDF_Reference* pRef);
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -070042};
43
Nico Weber9d8ec5a2015-08-04 13:00:21 -070044CPDF_PageOrganizer::CPDF_PageOrganizer() {}
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -070045
Nico Weber9d8ec5a2015-08-04 13:00:21 -070046CPDF_PageOrganizer::~CPDF_PageOrganizer() {}
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -070047
Nico Weber9d8ec5a2015-08-04 13:00:21 -070048FX_BOOL CPDF_PageOrganizer::PDFDocInit(CPDF_Document* pDestPDFDoc,
49 CPDF_Document* pSrcPDFDoc) {
50 if (!pDestPDFDoc || !pSrcPDFDoc)
Tom Sepezc7e4c4f2015-11-20 09:45:24 -080051 return FALSE;
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -070052
Nico Weber9d8ec5a2015-08-04 13:00:21 -070053 CPDF_Dictionary* pNewRoot = pDestPDFDoc->GetRoot();
54 if (!pNewRoot)
Tom Sepez2f2ffec2015-07-23 14:42:09 -070055 return FALSE;
Nico Weber9d8ec5a2015-08-04 13:00:21 -070056
Nico Weber9d8ec5a2015-08-04 13:00:21 -070057 CPDF_Dictionary* DInfoDict = pDestPDFDoc->GetInfo();
Nico Weber9d8ec5a2015-08-04 13:00:21 -070058 if (!DInfoDict)
59 return FALSE;
60
61 CFX_ByteString producerstr;
62 producerstr.Format("PDFium");
Lei Zhang4880d1a2015-12-18 17:05:11 -080063 DInfoDict->SetAt("Producer", new CPDF_String(producerstr, FALSE));
Nico Weber9d8ec5a2015-08-04 13:00:21 -070064
Wei Li9b761132016-01-29 15:44:20 -080065 CFX_ByteString cbRootType = pNewRoot->GetStringBy("Type", "");
tsepez9f2970c2016-04-01 10:23:04 -070066 if (cbRootType.IsEmpty())
Nico Weber9d8ec5a2015-08-04 13:00:21 -070067 pNewRoot->SetAt("Type", new CPDF_Name("Catalog"));
Nico Weber9d8ec5a2015-08-04 13:00:21 -070068
tsepezbd567552016-03-29 14:51:50 -070069 CPDF_Object* pElement = pNewRoot->GetObjectBy("Pages");
Tom Sepezc7e4c4f2015-11-20 09:45:24 -080070 CPDF_Dictionary* pNewPages =
71 pElement ? ToDictionary(pElement->GetDirect()) : nullptr;
Nico Weber9d8ec5a2015-08-04 13:00:21 -070072 if (!pNewPages) {
73 pNewPages = new CPDF_Dictionary;
tsepezc3255f52016-03-25 14:52:27 -070074 uint32_t NewPagesON = pDestPDFDoc->AddIndirectObject(pNewPages);
Nico Weber9d8ec5a2015-08-04 13:00:21 -070075 pNewRoot->SetAt("Pages", new CPDF_Reference(pDestPDFDoc, NewPagesON));
76 }
77
Wei Li9b761132016-01-29 15:44:20 -080078 CFX_ByteString cbPageType = pNewPages->GetStringBy("Type", "");
tsepez9f2970c2016-04-01 10:23:04 -070079 if (cbPageType == "") {
Nico Weber9d8ec5a2015-08-04 13:00:21 -070080 pNewPages->SetAt("Type", new CPDF_Name("Pages"));
81 }
82
Wei Li9b761132016-01-29 15:44:20 -080083 CPDF_Array* pKeysArray = pNewPages->GetArrayBy("Kids");
Tom Sepezc7e4c4f2015-11-20 09:45:24 -080084 if (!pKeysArray) {
Nico Weber9d8ec5a2015-08-04 13:00:21 -070085 CPDF_Array* pNewKids = new CPDF_Array;
tsepezc3255f52016-03-25 14:52:27 -070086 uint32_t Kidsobjnum = pDestPDFDoc->AddIndirectObject(pNewKids);
Nico Weber9d8ec5a2015-08-04 13:00:21 -070087
Tom Sepezc7e4c4f2015-11-20 09:45:24 -080088 pNewPages->SetAt("Kids", new CPDF_Reference(pDestPDFDoc, Kidsobjnum));
Nico Weber9d8ec5a2015-08-04 13:00:21 -070089 pNewPages->SetAt("Count", new CPDF_Number(0));
90 }
91
Tom Sepezc7e4c4f2015-11-20 09:45:24 -080092 return TRUE;
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -070093}
94
Nico Weber9d8ec5a2015-08-04 13:00:21 -070095FX_BOOL CPDF_PageOrganizer::ExportPage(CPDF_Document* pSrcPDFDoc,
Tom Sepez62a70f92016-03-21 15:00:20 -070096 std::vector<uint16_t>* pPageNums,
Nico Weber9d8ec5a2015-08-04 13:00:21 -070097 CPDF_Document* pDestPDFDoc,
98 int nIndex) {
99 int curpage = nIndex;
Lei Zhangaa8bf7e2015-12-24 19:13:32 -0800100 std::unique_ptr<ObjectNumberMap> pObjNumberMap(new ObjectNumberMap);
Tom Sepez11d93552016-02-09 09:55:54 -0800101 int nSize = pdfium::CollectionSize<int>(*pPageNums);
102 for (int i = 0; i < nSize; ++i) {
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700103 CPDF_Dictionary* pCurPageDict = pDestPDFDoc->CreateNewPage(curpage);
Tom Sepez11d93552016-02-09 09:55:54 -0800104 CPDF_Dictionary* pSrcPageDict = pSrcPDFDoc->GetPage(pPageNums->at(i) - 1);
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800105 if (!pSrcPageDict || !pCurPageDict)
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700106 return FALSE;
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700107
Lei Zhang4880d1a2015-12-18 17:05:11 -0800108 // Clone the page dictionary
Oliver Chang3f1c71f2016-01-11 08:45:31 -0800109 for (const auto& it : *pSrcPageDict) {
110 const CFX_ByteString& cbSrcKeyStr = it.first;
111 CPDF_Object* pObj = it.second;
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700112 if (cbSrcKeyStr.Compare(("Type")) && cbSrcKeyStr.Compare(("Parent"))) {
tsepez4c3debb2016-04-08 12:20:38 -0700113 if (pCurPageDict->KeyExist(cbSrcKeyStr.AsStringC()))
114 pCurPageDict->RemoveAt(cbSrcKeyStr.AsStringC());
115 pCurPageDict->SetAt(cbSrcKeyStr.AsStringC(), pObj->Clone());
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700116 }
117 }
118
Lei Zhang4880d1a2015-12-18 17:05:11 -0800119 // inheritable item
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800120 CPDF_Object* pInheritable = nullptr;
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700121 // 1 MediaBox //required
122 if (!pCurPageDict->KeyExist("MediaBox")) {
123 pInheritable = PageDictGetInheritableTag(pSrcPageDict, "MediaBox");
124 if (!pInheritable) {
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800125 // Search the "CropBox" from source page dictionary,
126 // if not exists,we take the letter size.
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700127 pInheritable = PageDictGetInheritableTag(pSrcPageDict, "CropBox");
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800128 if (pInheritable) {
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700129 pCurPageDict->SetAt("MediaBox", pInheritable->Clone());
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800130 } else {
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700131 // Make the default size to be letter size (8.5'x11')
132 CPDF_Array* pArray = new CPDF_Array;
133 pArray->AddNumber(0);
134 pArray->AddNumber(0);
135 pArray->AddNumber(612);
136 pArray->AddNumber(792);
137 pCurPageDict->SetAt("MediaBox", pArray);
138 }
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800139 } else {
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700140 pCurPageDict->SetAt("MediaBox", pInheritable->Clone());
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800141 }
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700142 }
143 // 2 Resources //required
144 if (!pCurPageDict->KeyExist("Resources")) {
145 pInheritable = PageDictGetInheritableTag(pSrcPageDict, "Resources");
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800146 if (!pInheritable)
Tom Sepez2f2ffec2015-07-23 14:42:09 -0700147 return FALSE;
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700148 pCurPageDict->SetAt("Resources", pInheritable->Clone());
149 }
150 // 3 CropBox //Optional
151 if (!pCurPageDict->KeyExist("CropBox")) {
152 pInheritable = PageDictGetInheritableTag(pSrcPageDict, "CropBox");
153 if (pInheritable)
154 pCurPageDict->SetAt("CropBox", pInheritable->Clone());
155 }
156 // 4 Rotate //Optional
157 if (!pCurPageDict->KeyExist("Rotate")) {
158 pInheritable = PageDictGetInheritableTag(pSrcPageDict, "Rotate");
159 if (pInheritable)
160 pCurPageDict->SetAt("Rotate", pInheritable->Clone());
161 }
162
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700163 // Update the reference
tsepezc3255f52016-03-25 14:52:27 -0700164 uint32_t dwOldPageObj = pSrcPageDict->GetObjNum();
165 uint32_t dwNewPageObj = pCurPageDict->GetObjNum();
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700166
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800167 (*pObjNumberMap)[dwOldPageObj] = dwNewPageObj;
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700168
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800169 UpdateReference(pCurPageDict, pDestPDFDoc, pObjNumberMap.get());
170 ++curpage;
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700171 }
172
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700173 return TRUE;
174}
175
176CPDF_Object* CPDF_PageOrganizer::PageDictGetInheritableTag(
177 CPDF_Dictionary* pDict,
178 CFX_ByteString nSrctag) {
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800179 if (!pDict || nSrctag.IsEmpty())
180 return nullptr;
181 if (!pDict->KeyExist("Parent") || !pDict->KeyExist("Type"))
182 return nullptr;
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700183
tsepezbd567552016-03-29 14:51:50 -0700184 CPDF_Object* pType = pDict->GetObjectBy("Type")->GetDirect();
Dan Sinclair710c9092015-10-21 15:46:10 -0400185 if (!ToName(pType))
186 return nullptr;
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700187 if (pType->GetString().Compare("Page"))
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800188 return nullptr;
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700189
tsepezbd567552016-03-29 14:51:50 -0700190 CPDF_Dictionary* pp = ToDictionary(pDict->GetObjectBy("Parent")->GetDirect());
Dan Sinclairf1251c12015-10-20 16:24:45 -0400191 if (!pp)
192 return nullptr;
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700193
194 if (pDict->KeyExist((const char*)nSrctag))
tsepezbd567552016-03-29 14:51:50 -0700195 return pDict->GetObjectBy((const char*)nSrctag);
Dan Sinclairf1251c12015-10-20 16:24:45 -0400196
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700197 while (pp) {
198 if (pp->KeyExist((const char*)nSrctag))
tsepezbd567552016-03-29 14:51:50 -0700199 return pp->GetObjectBy((const char*)nSrctag);
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800200 if (!pp->KeyExist("Parent"))
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700201 break;
tsepezbd567552016-03-29 14:51:50 -0700202 pp = ToDictionary(pp->GetObjectBy("Parent")->GetDirect());
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700203 }
Dan Sinclairf1251c12015-10-20 16:24:45 -0400204 return nullptr;
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700205}
206
207FX_BOOL CPDF_PageOrganizer::UpdateReference(CPDF_Object* pObj,
208 CPDF_Document* pDoc,
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800209 ObjectNumberMap* pObjNumberMap) {
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700210 switch (pObj->GetType()) {
Tom Sepez8e5cd192016-01-26 13:20:26 -0800211 case CPDF_Object::REFERENCE: {
Dan Sinclairbf81c142015-10-26 16:54:39 -0400212 CPDF_Reference* pReference = pObj->AsReference();
tsepezc3255f52016-03-25 14:52:27 -0700213 uint32_t newobjnum = GetNewObjId(pDoc, pObjNumberMap, pReference);
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700214 if (newobjnum == 0)
Tom Sepez2f2ffec2015-07-23 14:42:09 -0700215 return FALSE;
Dan Sinclairbf81c142015-10-26 16:54:39 -0400216 pReference->SetRef(pDoc, newobjnum);
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700217 break;
218 }
Tom Sepez8e5cd192016-01-26 13:20:26 -0800219 case CPDF_Object::DICTIONARY: {
Dan Sinclairf1251c12015-10-20 16:24:45 -0400220 CPDF_Dictionary* pDict = pObj->AsDictionary();
Oliver Changbd292ae2016-01-13 18:46:09 -0800221 auto it = pDict->begin();
222 while (it != pDict->end()) {
223 const CFX_ByteString& key = it->first;
224 CPDF_Object* pNextObj = it->second;
225 ++it;
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700226 if (!FXSYS_strcmp(key, "Parent") || !FXSYS_strcmp(key, "Prev") ||
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800227 !FXSYS_strcmp(key, "First")) {
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700228 continue;
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800229 }
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700230 if (pNextObj) {
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800231 if (!UpdateReference(pNextObj, pDoc, pObjNumberMap))
tsepez4c3debb2016-04-08 12:20:38 -0700232 pDict->RemoveAt(key.AsStringC());
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800233 } else {
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700234 return FALSE;
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800235 }
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700236 }
237 break;
238 }
Tom Sepez8e5cd192016-01-26 13:20:26 -0800239 case CPDF_Object::ARRAY: {
Dan Sinclair2b11dc12015-10-22 15:02:06 -0400240 CPDF_Array* pArray = pObj->AsArray();
Wei Lie1aebd42016-04-11 10:02:09 -0700241 for (size_t i = 0; i < pArray->GetCount(); ++i) {
tsepezbd567552016-03-29 14:51:50 -0700242 CPDF_Object* pNextObj = pArray->GetObjectAt(i);
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800243 if (!pNextObj)
244 return FALSE;
245 if (!UpdateReference(pNextObj, pDoc, pObjNumberMap))
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700246 return FALSE;
247 }
248 break;
249 }
Tom Sepez8e5cd192016-01-26 13:20:26 -0800250 case CPDF_Object::STREAM: {
Dan Sinclairaa435ba2015-10-22 16:45:48 -0400251 CPDF_Stream* pStream = pObj->AsStream();
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700252 CPDF_Dictionary* pDict = pStream->GetDict();
253 if (pDict) {
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800254 if (!UpdateReference(pDict, pDoc, pObjNumberMap))
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700255 return FALSE;
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800256 } else {
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700257 return FALSE;
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800258 }
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700259 break;
260 }
261 default:
262 break;
263 }
264
265 return TRUE;
266}
267
tsepezc3255f52016-03-25 14:52:27 -0700268uint32_t CPDF_PageOrganizer::GetNewObjId(CPDF_Document* pDoc,
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800269 ObjectNumberMap* pObjNumberMap,
270 CPDF_Reference* pRef) {
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700271 if (!pRef)
272 return 0;
273
tsepezc3255f52016-03-25 14:52:27 -0700274 uint32_t dwObjnum = pRef->GetRefObjNum();
275 uint32_t dwNewObjNum = 0;
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800276 const auto it = pObjNumberMap->find(dwObjnum);
277 if (it != pObjNumberMap->end())
278 dwNewObjNum = it->second;
279 if (dwNewObjNum)
280 return dwNewObjNum;
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700281
282 CPDF_Object* pDirect = pRef->GetDirect();
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800283 if (!pDirect)
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700284 return 0;
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700285
286 CPDF_Object* pClone = pDirect->Clone();
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800287 if (!pClone)
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700288 return 0;
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700289
Dan Sinclairf1251c12015-10-20 16:24:45 -0400290 if (CPDF_Dictionary* pDictClone = pClone->AsDictionary()) {
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700291 if (pDictClone->KeyExist("Type")) {
Wei Li9b761132016-01-29 15:44:20 -0800292 CFX_ByteString strType = pDictClone->GetStringBy("Type");
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700293 if (!FXSYS_stricmp(strType, "Pages")) {
294 pDictClone->Release();
295 return 4;
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800296 }
297 if (!FXSYS_stricmp(strType, "Page")) {
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700298 pDictClone->Release();
299 return 0;
300 }
301 }
302 }
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700303 dwNewObjNum = pDoc->AddIndirectObject(pClone);
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800304 (*pObjNumberMap)[dwObjnum] = dwNewObjNum;
305 if (!UpdateReference(pClone, pDoc, pObjNumberMap)) {
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700306 pClone->Release();
307 return 0;
308 }
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800309 return dwNewObjNum;
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700310}
311
312FPDF_BOOL ParserPageRangeString(CFX_ByteString rangstring,
Tom Sepez62a70f92016-03-21 15:00:20 -0700313 std::vector<uint16_t>* pageArray,
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700314 int nCount) {
315 if (rangstring.GetLength() != 0) {
316 rangstring.Remove(' ');
317 int nLength = rangstring.GetLength();
318 CFX_ByteString cbCompareString("0123456789-,");
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800319 for (int i = 0; i < nLength; ++i) {
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700320 if (cbCompareString.Find(rangstring[i]) == -1)
321 return FALSE;
322 }
323 CFX_ByteString cbMidRange;
324 int nStringFrom = 0;
325 int nStringTo = 0;
326 while (nStringTo < nLength) {
327 nStringTo = rangstring.Find(',', nStringFrom);
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800328 if (nStringTo == -1)
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700329 nStringTo = nLength;
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700330 cbMidRange = rangstring.Mid(nStringFrom, nStringTo - nStringFrom);
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700331 int nMid = cbMidRange.Find('-');
332 if (nMid == -1) {
333 long lPageNum = atol(cbMidRange);
334 if (lPageNum <= 0 || lPageNum > nCount)
335 return FALSE;
Tom Sepez62a70f92016-03-21 15:00:20 -0700336 pageArray->push_back((uint16_t)lPageNum);
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700337 } else {
338 int nStartPageNum = atol(cbMidRange.Mid(0, nMid));
339 if (nStartPageNum == 0)
340 return FALSE;
341
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800342 ++nMid;
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700343 int nEnd = cbMidRange.GetLength() - nMid;
344 if (nEnd == 0)
345 return FALSE;
346
347 int nEndPageNum = atol(cbMidRange.Mid(nMid, nEnd));
348 if (nStartPageNum < 0 || nStartPageNum > nEndPageNum ||
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800349 nEndPageNum > nCount) {
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700350 return FALSE;
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800351 }
352 for (int i = nStartPageNum; i <= nEndPageNum; ++i) {
Tom Sepez11d93552016-02-09 09:55:54 -0800353 pageArray->push_back(i);
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800354 }
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700355 }
356 nStringFrom = nStringTo + 1;
357 }
358 }
359 return TRUE;
360}
361
362DLLEXPORT FPDF_BOOL STDCALL FPDF_ImportPages(FPDF_DOCUMENT dest_doc,
363 FPDF_DOCUMENT src_doc,
364 FPDF_BYTESTRING pagerange,
365 int index) {
Tom Sepez471a1032015-10-15 16:17:18 -0700366 CPDF_Document* pDestDoc = CPDFDocumentFromFPDFDocument(dest_doc);
367 if (!dest_doc)
368 return FALSE;
369
370 CPDF_Document* pSrcDoc = CPDFDocumentFromFPDFDocument(src_doc);
371 if (!pSrcDoc)
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700372 return FALSE;
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800373
Tom Sepez62a70f92016-03-21 15:00:20 -0700374 std::vector<uint16_t> pageArray;
Tom Sepez471a1032015-10-15 16:17:18 -0700375 int nCount = pSrcDoc->GetPageCount();
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700376 if (pagerange) {
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800377 if (!ParserPageRangeString(pagerange, &pageArray, nCount))
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700378 return FALSE;
379 } else {
Tom Sepezc7e4c4f2015-11-20 09:45:24 -0800380 for (int i = 1; i <= nCount; ++i) {
Tom Sepez11d93552016-02-09 09:55:54 -0800381 pageArray.push_back(i);
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700382 }
383 }
384
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700385 CPDF_PageOrganizer pageOrg;
Tom Sepez471a1032015-10-15 16:17:18 -0700386 pageOrg.PDFDocInit(pDestDoc, pSrcDoc);
387 return pageOrg.ExportPage(pSrcDoc, &pageArray, pDestDoc, index);
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700388}
389
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700390DLLEXPORT FPDF_BOOL STDCALL FPDF_CopyViewerPreferences(FPDF_DOCUMENT dest_doc,
391 FPDF_DOCUMENT src_doc) {
Tom Sepez471a1032015-10-15 16:17:18 -0700392 CPDF_Document* pDstDoc = CPDFDocumentFromFPDFDocument(dest_doc);
393 if (!pDstDoc)
394 return FALSE;
395
396 CPDF_Document* pSrcDoc = CPDFDocumentFromFPDFDocument(src_doc);
397 if (!pSrcDoc)
398 return FALSE;
399
400 CPDF_Dictionary* pSrcDict = pSrcDoc->GetRoot();
Wei Li9b761132016-01-29 15:44:20 -0800401 pSrcDict = pSrcDict->GetDictBy("ViewerPreferences");
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700402 if (!pSrcDict)
403 return FALSE;
Tom Sepez471a1032015-10-15 16:17:18 -0700404
405 CPDF_Dictionary* pDstDict = pDstDoc->GetRoot();
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700406 if (!pDstDict)
407 return FALSE;
Tom Sepez471a1032015-10-15 16:17:18 -0700408
Lei Zhangd983b092015-12-14 16:58:33 -0800409 pDstDict->SetAt("ViewerPreferences", pSrcDict->Clone(TRUE));
Nico Weber9d8ec5a2015-08-04 13:00:21 -0700410 return TRUE;
411}