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 #ifndef CORE_FPDFAPI_EDIT_CPDF_CREATOR_H_
8 #define CORE_FPDFAPI_EDIT_CPDF_CREATOR_H_
9 
10 #include <map>
11 #include <memory>
12 #include <vector>
13 
14 #include "core/fxcrt/fx_stream.h"
15 #include "core/fxcrt/maybe_owned.h"
16 #include "core/fxcrt/retain_ptr.h"
17 #include "core/fxcrt/unowned_ptr.h"
18 
19 class CPDF_Array;
20 class CPDF_CryptoHandler;
21 class CPDF_SecurityHandler;
22 class CPDF_Dictionary;
23 class CPDF_Document;
24 class CPDF_Object;
25 class CPDF_Parser;
26 
27 #define FPDFCREATE_INCREMENTAL 1
28 #define FPDFCREATE_NO_ORIGINAL 2
29 
30 class CPDF_Creator {
31  public:
32   explicit CPDF_Creator(CPDF_Document* pDoc,
33                         const RetainPtr<IFX_WriteStream>& archive);
34   ~CPDF_Creator();
35 
36   void RemoveSecurity();
37   bool Create(uint32_t flags);
38   int32_t Continue();
39   bool SetFileVersion(int32_t fileVersion);
40 
GetArchive()41   IFX_ArchiveStream* GetArchive() { return m_Archive.get(); }
42 
GetNextObjectNumber()43   uint32_t GetNextObjectNumber() { return ++m_dwLastObjNum; }
GetLastObjectNumber()44   uint32_t GetLastObjectNumber() const { return m_dwLastObjNum; }
45   CPDF_CryptoHandler* GetCryptoHandler();
GetDocument()46   CPDF_Document* GetDocument() const { return m_pDocument.Get(); }
GetIDArray()47   CPDF_Array* GetIDArray() const { return m_pIDArray.get(); }
GetEncryptDict()48   CPDF_Dictionary* GetEncryptDict() const { return m_pEncryptDict.Get(); }
GetEncryptObjectNumber()49   uint32_t GetEncryptObjectNumber() const { return m_dwEncryptObjNum; }
50 
GetObjectOffset(uint32_t objnum)51   uint32_t GetObjectOffset(uint32_t objnum) { return m_ObjectOffsets[objnum]; }
HasObjectNumber(uint32_t objnum)52   bool HasObjectNumber(uint32_t objnum) {
53     return m_ObjectOffsets.find(objnum) != m_ObjectOffsets.end();
54   }
SetObjectOffset(uint32_t objnum,FX_FILESIZE offset)55   void SetObjectOffset(uint32_t objnum, FX_FILESIZE offset) {
56     m_ObjectOffsets[objnum] = offset;
57   }
IsIncremental()58   bool IsIncremental() const { return !!(m_dwFlags & FPDFCREATE_INCREMENTAL); }
IsOriginal()59   bool IsOriginal() const { return !(m_dwFlags & FPDFCREATE_NO_ORIGINAL); }
60 
61  private:
62   void Clear();
63 
64   void InitOldObjNumOffsets();
65   void InitNewObjNumOffsets();
66   void InitID();
67 
68   int32_t WriteDoc_Stage1();
69   int32_t WriteDoc_Stage2();
70   int32_t WriteDoc_Stage3();
71   int32_t WriteDoc_Stage4();
72 
73   bool WriteOldIndirectObject(uint32_t objnum);
74   bool WriteOldObjs();
75   bool WriteNewObjs();
76   bool WriteDirectObj(uint32_t objnum, const CPDF_Object* pObj, bool bEncrypt);
77   bool WriteIndirectObj(uint32_t objnum, const CPDF_Object* pObj);
78 
79   bool WriteStream(const CPDF_Object* pStream,
80                    uint32_t objnum,
81                    CPDF_CryptoHandler* pCrypto);
82 
83   bool IsXRefNeedEnd();
84 
85   UnownedPtr<CPDF_Document> const m_pDocument;
86   UnownedPtr<CPDF_Parser> const m_pParser;
87   bool m_bSecurityChanged;
88   UnownedPtr<CPDF_Dictionary> m_pEncryptDict;
89   uint32_t m_dwEncryptObjNum;
90   fxcrt::MaybeOwned<CPDF_SecurityHandler> m_pSecurityHandler;
91   UnownedPtr<CPDF_Object> m_pMetadata;
92   uint32_t m_dwLastObjNum;
93   std::unique_ptr<IFX_ArchiveStream> m_Archive;
94   FX_FILESIZE m_SavedOffset;
95   int32_t m_iStage;
96   uint32_t m_dwFlags;
97   uint32_t m_CurObjNum;
98   FX_FILESIZE m_XrefStart;
99   std::map<uint32_t, FX_FILESIZE> m_ObjectOffsets;
100   std::vector<uint32_t> m_NewObjNumArray;  // Sorted, ascending.
101   std::unique_ptr<CPDF_Array> m_pIDArray;
102   int32_t m_FileVersion;
103 };
104 
105 #endif  // CORE_FPDFAPI_EDIT_CPDF_CREATOR_H_
106