Revert "Cleanup some numeric code."
[pdfium.git] / core / src / fpdfapi / fpdf_page / fpdf_page_doc.cpp
1 // 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.
4
5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
6
7 #include "../../../include/fpdfapi/fpdf_page.h"
8 #include "../../../include/fpdfapi/fpdf_module.h"
9 #include "../../../include/fdrm/fx_crypt.h"
10 #include "../fpdf_font/font_int.h"
11 #include "pageint.h"
12
13 class CPDF_PageModule : public IPDF_PageModule {
14  public:
15   CPDF_PageModule()
16       : m_StockGrayCS(nullptr, PDFCS_DEVICEGRAY),
17         m_StockRGBCS(nullptr, PDFCS_DEVICERGB),
18         m_StockCMYKCS(nullptr, PDFCS_DEVICECMYK),
19         m_StockPatternCS(nullptr) {}
20
21  private:
22   ~CPDF_PageModule() override {}
23
24   CPDF_DocPageData* CreateDocData(CPDF_Document* pDoc) override {
25     return new CPDF_DocPageData(pDoc);
26   }
27
28   void ReleaseDoc(CPDF_Document* pDoc) override;
29   void ClearDoc(CPDF_Document* pDoc) override;
30
31   CPDF_FontGlobals* GetFontGlobals() override { return &m_FontGlobals; }
32
33   void ClearStockFont(CPDF_Document* pDoc) override {
34     m_FontGlobals.Clear(pDoc);
35   }
36
37   CPDF_ColorSpace* GetStockCS(int family) override;
38   void NotifyCJKAvailable() override;
39
40   CPDF_FontGlobals m_FontGlobals;
41   CPDF_DeviceCS m_StockGrayCS;
42   CPDF_DeviceCS m_StockRGBCS;
43   CPDF_DeviceCS m_StockCMYKCS;
44   CPDF_PatternCS m_StockPatternCS;
45 };
46
47 CPDF_ColorSpace* CPDF_PageModule::GetStockCS(int family) {
48   if (family == PDFCS_DEVICEGRAY) {
49     return &m_StockGrayCS;
50   }
51   if (family == PDFCS_DEVICERGB) {
52     return &m_StockRGBCS;
53   }
54   if (family == PDFCS_DEVICECMYK) {
55     return &m_StockCMYKCS;
56   }
57   if (family == PDFCS_PATTERN) {
58     return &m_StockPatternCS;
59   }
60   return NULL;
61 }
62
63 void CPDF_ModuleMgr::InitPageModule() {
64   m_pPageModule.reset(new CPDF_PageModule);
65 }
66
67 void CPDF_PageModule::ReleaseDoc(CPDF_Document* pDoc) {
68   delete pDoc->GetPageData();
69 }
70 void CPDF_PageModule::ClearDoc(CPDF_Document* pDoc) {
71   pDoc->GetPageData()->Clear(FALSE);
72 }
73 void CPDF_PageModule::NotifyCJKAvailable() {
74   m_FontGlobals.m_CMapManager.ReloadAll();
75 }
76
77 CPDF_Font* CPDF_Document::LoadFont(CPDF_Dictionary* pFontDict) {
78   ASSERT(pFontDict);
79   return GetValidatePageData()->GetFont(pFontDict, FALSE);
80 }
81
82 CPDF_StreamAcc* CPDF_Document::LoadFontFile(CPDF_Stream* pStream) {
83   return GetValidatePageData()->GetFontFileStreamAcc(pStream);
84 }
85
86 CPDF_ColorSpace* _CSFromName(const CFX_ByteString& name);
87 CPDF_ColorSpace* CPDF_Document::LoadColorSpace(CPDF_Object* pCSObj,
88                                                CPDF_Dictionary* pResources) {
89   return GetValidatePageData()->GetColorSpace(pCSObj, pResources);
90 }
91 CPDF_Pattern* CPDF_Document::LoadPattern(CPDF_Object* pPatternObj,
92                                          FX_BOOL bShading,
93                                          const CFX_AffineMatrix* matrix) {
94   return GetValidatePageData()->GetPattern(pPatternObj, bShading, matrix);
95 }
96 CPDF_IccProfile* CPDF_Document::LoadIccProfile(CPDF_Stream* pStream) {
97   return GetValidatePageData()->GetIccProfile(pStream);
98 }
99 CPDF_Image* CPDF_Document::LoadImageF(CPDF_Object* pObj) {
100   if (!pObj) {
101     return NULL;
102   }
103   FXSYS_assert(pObj->GetObjNum());
104   return GetValidatePageData()->GetImage(pObj);
105 }
106 void CPDF_Document::RemoveColorSpaceFromPageData(CPDF_Object* pCSObj) {
107   if (!pCSObj) {
108     return;
109   }
110   GetPageData()->ReleaseColorSpace(pCSObj);
111 }
112 CPDF_DocPageData::CPDF_DocPageData(CPDF_Document* pPDFDoc)
113     : m_pPDFDoc(pPDFDoc), m_bForceClear(FALSE) {}
114
115 CPDF_DocPageData::~CPDF_DocPageData() {
116   Clear(FALSE);
117   Clear(TRUE);
118
119   for (auto& it : m_PatternMap)
120     delete it.second;
121   m_PatternMap.clear();
122
123   for (auto& it : m_FontMap)
124     delete it.second;
125   m_FontMap.clear();
126
127   for (auto& it : m_ColorSpaceMap)
128     delete it.second;
129   m_ColorSpaceMap.clear();
130 }
131
132 void CPDF_DocPageData::Clear(FX_BOOL bForceRelease) {
133   m_bForceClear = bForceRelease;
134
135   for (auto& it : m_PatternMap) {
136     CPDF_CountedPattern* ptData = it.second;
137     if (!ptData->get())
138       continue;
139
140     if (bForceRelease || ptData->use_count() < 2) {
141       ptData->get()->SetForceClear(bForceRelease);
142       ptData->clear();
143     }
144   }
145
146   for (auto& it : m_FontMap) {
147     CPDF_CountedFont* fontData = it.second;
148     if (!fontData->get())
149       continue;
150
151     if (bForceRelease || fontData->use_count() < 2) {
152       fontData->clear();
153     }
154   }
155
156   for (auto& it : m_ColorSpaceMap) {
157     CPDF_CountedColorSpace* csData = it.second;
158     if (!csData->get())
159       continue;
160
161     if (bForceRelease || csData->use_count() < 2) {
162       csData->get()->ReleaseCS();
163       csData->reset(nullptr);
164     }
165   }
166
167   for (auto it = m_IccProfileMap.begin(); it != m_IccProfileMap.end();) {
168     auto curr_it = it++;
169     CPDF_CountedIccProfile* ipData = curr_it->second;
170     if (!ipData->get())
171       continue;
172
173     if (bForceRelease || ipData->use_count() < 2) {
174       for (auto hash_it = m_HashProfileMap.begin();
175            hash_it != m_HashProfileMap.end(); ++hash_it) {
176         if (curr_it->first == hash_it->second) {
177           m_HashProfileMap.erase(hash_it);
178           break;
179         }
180       }
181       delete ipData->get();
182       delete ipData;
183       m_IccProfileMap.erase(curr_it);
184     }
185   }
186
187   for (auto it = m_FontFileMap.begin(); it != m_FontFileMap.end();) {
188     auto curr_it = it++;
189     CPDF_CountedStreamAcc* ftData = curr_it->second;
190     if (!ftData->get())
191       continue;
192
193     if (bForceRelease || ftData->use_count() < 2) {
194       delete ftData->get();
195       delete ftData;
196       m_FontFileMap.erase(curr_it);
197     }
198   }
199
200   for (auto it = m_ImageMap.begin(); it != m_ImageMap.end();) {
201     auto curr_it = it++;
202     CPDF_CountedImage* imageData = curr_it->second;
203     if (!imageData->get())
204       continue;
205
206     if (bForceRelease || imageData->use_count() < 2) {
207       delete imageData->get();
208       delete imageData;
209       m_ImageMap.erase(curr_it);
210     }
211   }
212 }
213
214 CPDF_Font* CPDF_DocPageData::GetFont(CPDF_Dictionary* pFontDict,
215                                      FX_BOOL findOnly) {
216   if (!pFontDict) {
217     return NULL;
218   }
219   if (findOnly) {
220     auto it = m_FontMap.find(pFontDict);
221     if (it != m_FontMap.end() && it->second->get()) {
222       return it->second->AddRef();
223     }
224     return nullptr;
225   }
226
227   CPDF_CountedFont* fontData = nullptr;
228   auto it = m_FontMap.find(pFontDict);
229   if (it != m_FontMap.end()) {
230     fontData = it->second;
231     if (fontData->get()) {
232       return fontData->AddRef();
233     }
234   }
235
236   CPDF_Font* pFont = CPDF_Font::CreateFontF(m_pPDFDoc, pFontDict);
237   if (!pFont) {
238     return nullptr;
239   }
240   if (!fontData) {
241     fontData = new CPDF_CountedFont(pFont);
242     m_FontMap[pFontDict] = fontData;
243   } else {
244     fontData->reset(pFont);
245   }
246   return fontData->AddRef();
247 }
248
249 CPDF_Font* CPDF_DocPageData::GetStandardFont(const CFX_ByteStringC& fontName,
250                                              CPDF_FontEncoding* pEncoding) {
251   if (fontName.IsEmpty())
252     return nullptr;
253
254   for (auto& it : m_FontMap) {
255     CPDF_CountedFont* fontData = it.second;
256     CPDF_Font* pFont = fontData->get();
257     if (!pFont)
258       continue;
259     if (pFont->GetBaseFont() != fontName)
260       continue;
261     if (pFont->IsEmbedded())
262       continue;
263     if (pFont->GetFontType() != PDFFONT_TYPE1)
264       continue;
265     if (pFont->GetFontDict()->KeyExist(FX_BSTRC("Widths")))
266       continue;
267
268     CPDF_Type1Font* pT1Font = pFont->GetType1Font();
269     if (pEncoding && !pT1Font->GetEncoding()->IsIdentical(pEncoding))
270       continue;
271
272     return fontData->AddRef();
273   }
274
275   CPDF_Dictionary* pDict = new CPDF_Dictionary;
276   pDict->SetAtName(FX_BSTRC("Type"), FX_BSTRC("Font"));
277   pDict->SetAtName(FX_BSTRC("Subtype"), FX_BSTRC("Type1"));
278   pDict->SetAtName(FX_BSTRC("BaseFont"), fontName);
279   if (pEncoding) {
280     pDict->SetAt(FX_BSTRC("Encoding"), pEncoding->Realize());
281   }
282   m_pPDFDoc->AddIndirectObject(pDict);
283   CPDF_Font* pFont = CPDF_Font::CreateFontF(m_pPDFDoc, pDict);
284   if (!pFont) {
285     return nullptr;
286   }
287   CPDF_CountedFont* fontData = new CPDF_CountedFont(pFont);
288   m_FontMap[pDict] = fontData;
289   return fontData->AddRef();
290 }
291
292 void CPDF_DocPageData::ReleaseFont(CPDF_Dictionary* pFontDict) {
293   if (!pFontDict)
294     return;
295
296   auto it = m_FontMap.find(pFontDict);
297   if (it == m_FontMap.end())
298     return;
299
300   CPDF_CountedFont* fontData = it->second;
301   if (fontData->get()) {
302     fontData->RemoveRef();
303     if (fontData->use_count() == 0) {
304       fontData->clear();
305     }
306   }
307 }
308
309 CPDF_ColorSpace* CPDF_DocPageData::GetColorSpace(CPDF_Object* pCSObj,
310                                                  CPDF_Dictionary* pResources) {
311   if (!pCSObj)
312     return nullptr;
313
314   if (pCSObj->IsName()) {
315     CFX_ByteString name = pCSObj->GetConstString();
316     CPDF_ColorSpace* pCS = _CSFromName(name);
317     if (!pCS && pResources) {
318       CPDF_Dictionary* pList = pResources->GetDict(FX_BSTRC("ColorSpace"));
319       if (pList) {
320         pCSObj = pList->GetElementValue(name);
321         return GetColorSpace(pCSObj, nullptr);
322       }
323     }
324     if (!pCS || !pResources)
325       return pCS;
326
327     CPDF_Dictionary* pColorSpaces = pResources->GetDict(FX_BSTRC("ColorSpace"));
328     if (!pColorSpaces)
329       return pCS;
330
331     CPDF_Object* pDefaultCS = nullptr;
332     switch (pCS->GetFamily()) {
333       case PDFCS_DEVICERGB:
334         pDefaultCS = pColorSpaces->GetElementValue(FX_BSTRC("DefaultRGB"));
335         break;
336       case PDFCS_DEVICEGRAY:
337         pDefaultCS = pColorSpaces->GetElementValue(FX_BSTRC("DefaultGray"));
338         break;
339       case PDFCS_DEVICECMYK:
340         pDefaultCS = pColorSpaces->GetElementValue(FX_BSTRC("DefaultCMYK"));
341         break;
342     }
343     return pDefaultCS ? GetColorSpace(pDefaultCS, nullptr) : pCS;
344   }
345
346   CPDF_Array* pArray = pCSObj->AsArray();
347   if (!pArray || pArray->GetCount() == 0)
348     return nullptr;
349   if (pArray->GetCount() == 1)
350     return GetColorSpace(pArray->GetElementValue(0), pResources);
351
352   CPDF_CountedColorSpace* csData = nullptr;
353   auto it = m_ColorSpaceMap.find(pCSObj);
354   if (it != m_ColorSpaceMap.end()) {
355     csData = it->second;
356     if (csData->get()) {
357       return csData->AddRef();
358     }
359   }
360
361   CPDF_ColorSpace* pCS = CPDF_ColorSpace::Load(m_pPDFDoc, pArray);
362   if (!pCS)
363     return nullptr;
364
365   if (!csData) {
366     csData = new CPDF_CountedColorSpace(pCS);
367     m_ColorSpaceMap[pCSObj] = csData;
368   } else {
369     csData->reset(pCS);
370   }
371   return csData->AddRef();
372 }
373
374 CPDF_ColorSpace* CPDF_DocPageData::GetCopiedColorSpace(CPDF_Object* pCSObj) {
375   if (!pCSObj)
376     return nullptr;
377
378   auto it = m_ColorSpaceMap.find(pCSObj);
379   if (it != m_ColorSpaceMap.end())
380     return it->second->AddRef();
381
382   return nullptr;
383 }
384
385 void CPDF_DocPageData::ReleaseColorSpace(CPDF_Object* pColorSpace) {
386   if (!pColorSpace)
387     return;
388
389   auto it = m_ColorSpaceMap.find(pColorSpace);
390   if (it == m_ColorSpaceMap.end())
391     return;
392
393   CPDF_CountedColorSpace* csData = it->second;
394   if (csData->get()) {
395     csData->RemoveRef();
396     if (csData->use_count() == 0) {
397       csData->get()->ReleaseCS();
398       csData->reset(nullptr);
399     }
400   }
401 }
402
403 CPDF_Pattern* CPDF_DocPageData::GetPattern(CPDF_Object* pPatternObj,
404                                            FX_BOOL bShading,
405                                            const CFX_AffineMatrix* matrix) {
406   if (!pPatternObj)
407     return nullptr;
408
409   CPDF_CountedPattern* ptData = nullptr;
410   auto it = m_PatternMap.find(pPatternObj);
411   if (it != m_PatternMap.end()) {
412     ptData = it->second;
413     if (ptData->get()) {
414       return ptData->AddRef();
415     }
416   }
417   CPDF_Pattern* pPattern = nullptr;
418   if (bShading) {
419     pPattern =
420         new CPDF_ShadingPattern(m_pPDFDoc, pPatternObj, bShading, matrix);
421   } else {
422     CPDF_Dictionary* pDict = pPatternObj ? pPatternObj->GetDict() : nullptr;
423     if (pDict) {
424       int type = pDict->GetInteger(FX_BSTRC("PatternType"));
425       if (type == 1) {
426         pPattern = new CPDF_TilingPattern(m_pPDFDoc, pPatternObj, matrix);
427       } else if (type == 2) {
428         pPattern =
429             new CPDF_ShadingPattern(m_pPDFDoc, pPatternObj, FALSE, matrix);
430       }
431     }
432   }
433   if (!pPattern)
434     return nullptr;
435
436   if (!ptData) {
437     ptData = new CPDF_CountedPattern(pPattern);
438     m_PatternMap[pPatternObj] = ptData;
439   } else {
440     ptData->reset(pPattern);
441   }
442   return ptData->AddRef();
443 }
444
445 void CPDF_DocPageData::ReleasePattern(CPDF_Object* pPatternObj) {
446   if (!pPatternObj)
447     return;
448
449   auto it = m_PatternMap.find(pPatternObj);
450   if (it == m_PatternMap.end())
451     return;
452
453   CPDF_CountedPattern* ptData = it->second;
454   if (ptData->get()) {
455     ptData->RemoveRef();
456     if (ptData->use_count() == 0) {
457       ptData->clear();
458     }
459   }
460 }
461
462 CPDF_Image* CPDF_DocPageData::GetImage(CPDF_Object* pImageStream) {
463   if (!pImageStream)
464     return nullptr;
465
466   const FX_DWORD dwImageObjNum = pImageStream->GetObjNum();
467   auto it = m_ImageMap.find(dwImageObjNum);
468   if (it != m_ImageMap.end()) {
469     return it->second->AddRef();
470   }
471
472   CPDF_Image* pImage = new CPDF_Image(m_pPDFDoc);
473   pImage->LoadImageF(pImageStream->AsStream(), FALSE);
474
475   CPDF_CountedImage* imageData = new CPDF_CountedImage(pImage);
476   m_ImageMap[dwImageObjNum] = imageData;
477   return imageData->AddRef();
478 }
479
480 void CPDF_DocPageData::ReleaseImage(CPDF_Object* pImageStream) {
481   if (!pImageStream || !pImageStream->GetObjNum())
482     return;
483
484   auto it = m_ImageMap.find(pImageStream->GetObjNum());
485   if (it == m_ImageMap.end())
486     return;
487
488   CPDF_CountedImage* image = it->second;
489   if (!image)
490     return;
491
492   image->RemoveRef();
493   if (image->use_count() == 0) {
494     delete image->get();
495     delete image;
496     m_ImageMap.erase(it);
497   }
498 }
499
500 CPDF_IccProfile* CPDF_DocPageData::GetIccProfile(
501     CPDF_Stream* pIccProfileStream) {
502   if (!pIccProfileStream)
503     return NULL;
504
505   auto it = m_IccProfileMap.find(pIccProfileStream);
506   if (it != m_IccProfileMap.end()) {
507     return it->second->AddRef();
508   }
509
510   CPDF_StreamAcc stream;
511   stream.LoadAllData(pIccProfileStream, FALSE);
512   uint8_t digest[20];
513   CRYPT_SHA1Generate(stream.GetData(), stream.GetSize(), digest);
514   auto hash_it = m_HashProfileMap.find(CFX_ByteStringC(digest, 20));
515   if (hash_it != m_HashProfileMap.end()) {
516     auto it_copied_stream = m_IccProfileMap.find(hash_it->second);
517     return it_copied_stream->second->AddRef();
518   }
519   CPDF_IccProfile* pProfile =
520       new CPDF_IccProfile(stream.GetData(), stream.GetSize());
521   CPDF_CountedIccProfile* ipData = new CPDF_CountedIccProfile(pProfile);
522   m_IccProfileMap[pIccProfileStream] = ipData;
523   m_HashProfileMap[CFX_ByteStringC(digest, 20)] = pIccProfileStream;
524   return ipData->AddRef();
525 }
526
527 void CPDF_DocPageData::ReleaseIccProfile(CPDF_IccProfile* pIccProfile) {
528   ASSERT(pIccProfile);
529
530   for (auto it = m_IccProfileMap.begin(); it != m_IccProfileMap.end(); ++it) {
531     CPDF_CountedIccProfile* profile = it->second;
532     if (profile->get() != pIccProfile)
533       continue;
534
535     profile->RemoveRef();
536     if (profile->use_count() == 0) {
537       delete profile->get();
538       delete profile;
539       m_IccProfileMap.erase(it);
540       return;
541     }
542   }
543 }
544
545 CPDF_StreamAcc* CPDF_DocPageData::GetFontFileStreamAcc(
546     CPDF_Stream* pFontStream) {
547   ASSERT(pFontStream);
548
549   auto it = m_FontFileMap.find(pFontStream);
550   if (it != m_FontFileMap.end())
551     return it->second->AddRef();
552
553   CPDF_Dictionary* pFontDict = pFontStream->GetDict();
554   int32_t org_size = pFontDict->GetInteger(FX_BSTRC("Length1")) +
555                      pFontDict->GetInteger(FX_BSTRC("Length2")) +
556                      pFontDict->GetInteger(FX_BSTRC("Length3"));
557   if (org_size < 0)
558     org_size = 0;
559
560   CPDF_StreamAcc* pFontFile = new CPDF_StreamAcc;
561   pFontFile->LoadAllData(pFontStream, FALSE, org_size);
562
563   CPDF_CountedStreamAcc* ftData = new CPDF_CountedStreamAcc(pFontFile);
564   m_FontFileMap[pFontStream] = ftData;
565   return ftData->AddRef();
566 }
567
568 void CPDF_DocPageData::ReleaseFontFileStreamAcc(CPDF_Stream* pFontStream,
569                                                 FX_BOOL bForce) {
570   if (!pFontStream)
571     return;
572
573   auto it = m_FontFileMap.find(pFontStream);
574   if (it == m_FontFileMap.end())
575     return;
576
577   CPDF_CountedStreamAcc* findData = it->second;
578   if (!findData)
579     return;
580
581   findData->RemoveRef();
582   if (findData->use_count() == 0 || bForce) {
583     delete findData->get();
584     delete findData;
585     m_FontFileMap.erase(it);
586   }
587 }
588
589 CPDF_CountedColorSpace* CPDF_DocPageData::FindColorSpacePtr(
590     CPDF_Object* pCSObj) const {
591   if (!pCSObj)
592     return nullptr;
593
594   auto it = m_ColorSpaceMap.find(pCSObj);
595   return it != m_ColorSpaceMap.end() ? it->second : nullptr;
596 }
597
598 CPDF_CountedPattern* CPDF_DocPageData::FindPatternPtr(
599     CPDF_Object* pPatternObj) const {
600   if (!pPatternObj)
601     return nullptr;
602
603   auto it = m_PatternMap.find(pPatternObj);
604   return it != m_PatternMap.end() ? it->second : nullptr;
605 }