diff options
Diffstat (limited to 'core/fpdfapi/fpdf_edit/cpdf_creator.h')
-rw-r--r-- | core/fpdfapi/fpdf_edit/cpdf_creator.h | 101 |
1 files changed, 101 insertions, 0 deletions
diff --git a/core/fpdfapi/fpdf_edit/cpdf_creator.h b/core/fpdfapi/fpdf_edit/cpdf_creator.h new file mode 100644 index 0000000000..ec2536f01c --- /dev/null +++ b/core/fpdfapi/fpdf_edit/cpdf_creator.h @@ -0,0 +1,101 @@ +// Copyright 2014 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 + +#ifndef CORE_FPDFAPI_FPDF_EDIT_CPDF_CREATOR_H_ +#define CORE_FPDFAPI_FPDF_EDIT_CPDF_CREATOR_H_ + +#include <memory> + +#include "core/fxcrt/include/fx_basic.h" + +class CPDF_Array; +class CPDF_CryptoHandler; +class CPDF_Dictionary; +class CPDF_Document; +class CPDF_Object; +class CPDF_Parser; +class CPDF_XRefStream; + +#define FPDFCREATE_INCREMENTAL 1 +#define FPDFCREATE_NO_ORIGINAL 2 +#define FPDFCREATE_PROGRESSIVE 4 +#define FPDFCREATE_OBJECTSTREAM 8 + +CFX_ByteTextBuf& operator<<(CFX_ByteTextBuf& buf, const CPDF_Object* pObj); + +class CPDF_Creator { + public: + explicit CPDF_Creator(CPDF_Document* pDoc); + ~CPDF_Creator(); + + void RemoveSecurity(); + bool Create(IFX_StreamWrite* pFile, uint32_t flags = 0); + int32_t Continue(IFX_Pause* pPause = nullptr); + FX_BOOL SetFileVersion(int32_t fileVersion = 17); + + private: + friend class CPDF_ObjectStream; + friend class CPDF_XRefStream; + + bool Create(uint32_t flags); + void ResetStandardSecurity(); + void Clear(); + + void InitOldObjNumOffsets(); + void InitNewObjNumOffsets(); + void InitID(FX_BOOL bDefault = TRUE); + + void AppendNewObjNum(uint32_t objbum); + int32_t AppendObjectNumberToXRef(uint32_t objnum); + + int32_t WriteDoc_Stage1(IFX_Pause* pPause); + int32_t WriteDoc_Stage2(IFX_Pause* pPause); + int32_t WriteDoc_Stage3(IFX_Pause* pPause); + int32_t WriteDoc_Stage4(IFX_Pause* pPause); + + int32_t WriteOldIndirectObject(uint32_t objnum); + int32_t WriteOldObjs(IFX_Pause* pPause); + int32_t WriteNewObjs(FX_BOOL bIncremental, IFX_Pause* pPause); + int32_t WriteIndirectObj(const CPDF_Object* pObj); + int32_t WriteDirectObj(uint32_t objnum, + const CPDF_Object* pObj, + FX_BOOL bEncrypt = TRUE); + int32_t WriteIndirectObjectToStream(const CPDF_Object* pObj); + int32_t WriteIndirectObj(uint32_t objnum, const CPDF_Object* pObj); + int32_t WriteIndirectObjectToStream(uint32_t objnum, + const uint8_t* pBuffer, + uint32_t dwSize); + + int32_t WriteStream(const CPDF_Object* pStream, + uint32_t objnum, + CPDF_CryptoHandler* pCrypto); + + CPDF_Document* const m_pDocument; + CPDF_Parser* const m_pParser; + FX_BOOL m_bSecurityChanged; + CPDF_Dictionary* m_pEncryptDict; + uint32_t m_dwEncryptObjNum; + FX_BOOL m_bEncryptCloned; + CPDF_CryptoHandler* m_pCryptoHandler; + // Whether this owns the crypto handler |m_pCryptoHandler|. + FX_BOOL m_bLocalCryptoHandler; + CPDF_Object* m_pMetadata; + std::unique_ptr<CPDF_XRefStream> m_pXRefStream; + int32_t m_ObjectStreamSize; + uint32_t m_dwLastObjNum; + CFX_FileBufferArchive m_File; + FX_FILESIZE m_Offset; + int32_t m_iStage; + uint32_t m_dwFlags; + FX_POSITION m_Pos; + FX_FILESIZE m_XrefStart; + CFX_FileSizeListArray m_ObjectOffset; + CFX_ArrayTemplate<uint32_t> m_NewObjNumArray; + std::unique_ptr<CPDF_Array, ReleaseDeleter<CPDF_Array>> m_pIDArray; + int32_t m_FileVersion; +}; + +#endif // CORE_FPDFAPI_FPDF_EDIT_CPDF_CREATOR_H_ |