From 764ec513eecbebd12781bcc96ce81ed5e736ee92 Mon Sep 17 00:00:00 2001 From: Dan Sinclair Date: Mon, 14 Mar 2016 13:35:12 -0400 Subject: Move core/src/ up to core/. This CL moves the core/src/ files up to core/ and fixes up the include guards, includes and build files. R=tsepez@chromium.org Review URL: https://codereview.chromium.org/1800523005 . --- core/fpdfapi/fpdf_edit/fpdf_edit_create.cpp | 2076 +++++++++++++++++++++++++++ 1 file changed, 2076 insertions(+) create mode 100644 core/fpdfapi/fpdf_edit/fpdf_edit_create.cpp (limited to 'core/fpdfapi/fpdf_edit/fpdf_edit_create.cpp') diff --git a/core/fpdfapi/fpdf_edit/fpdf_edit_create.cpp b/core/fpdfapi/fpdf_edit/fpdf_edit_create.cpp new file mode 100644 index 0000000000..2dc2e3eb8e --- /dev/null +++ b/core/fpdfapi/fpdf_edit/fpdf_edit_create.cpp @@ -0,0 +1,2076 @@ +// 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 + +#include "core/fpdfapi/fpdf_edit/editint.h" + +#include + +#include "core/fpdfapi/fpdf_parser/cpdf_standard_crypto_handler.h" +#include "core/fpdfapi/fpdf_parser/cpdf_standard_security_handler.h" +#include "core/include/fpdfapi/cpdf_array.h" +#include "core/include/fpdfapi/cpdf_dictionary.h" +#include "core/include/fpdfapi/cpdf_document.h" +#include "core/include/fpdfapi/cpdf_parser.h" +#include "core/include/fpdfapi/cpdf_reference.h" +#include "core/include/fpdfapi/cpdf_string.h" +#include "core/include/fpdfapi/fpdf_parser_decode.h" +#include "core/include/fpdfapi/fpdf_serial.h" +#include "core/include/fpdfapi/ipdf_crypto_handler.h" +#include "core/include/fxcrt/fx_ext.h" +#include "third_party/base/stl_util.h" + +#define PDF_OBJECTSTREAM_MAXLENGTH (256 * 1024) +#define PDF_XREFSTREAM_MAXSIZE 10000 + +// TODO(ochang): Make helper for appending "objnum 0 R ". + +namespace { + +int32_t PDF_CreatorAppendObject(const CPDF_Object* pObj, + CFX_FileBufferArchive* pFile, + FX_FILESIZE& offset) { + int32_t len = 0; + if (!pObj) { + if (pFile->AppendString(" null") < 0) { + return -1; + } + offset += 5; + return 1; + } + switch (pObj->GetType()) { + case CPDF_Object::NULLOBJ: + if (pFile->AppendString(" null") < 0) { + return -1; + } + offset += 5; + break; + case CPDF_Object::BOOLEAN: + case CPDF_Object::NUMBER: + if (pFile->AppendString(" ") < 0) { + return -1; + } + if ((len = pFile->AppendString(pObj->GetString())) < 0) { + return -1; + } + offset += len + 1; + break; + case CPDF_Object::STRING: { + CFX_ByteString str = pObj->GetString(); + FX_BOOL bHex = pObj->AsString()->IsHex(); + if ((len = pFile->AppendString(PDF_EncodeString(str, bHex))) < 0) { + return -1; + } + offset += len; + break; + } + case CPDF_Object::NAME: { + if (pFile->AppendString("/") < 0) { + return -1; + } + CFX_ByteString str = pObj->GetString(); + if ((len = pFile->AppendString(PDF_NameEncode(str))) < 0) { + return -1; + } + offset += len + 1; + break; + } + case CPDF_Object::REFERENCE: { + if (pFile->AppendString(" ") < 0) + return -1; + if ((len = pFile->AppendDWord(pObj->AsReference()->GetRefObjNum())) < 0) + return -1; + if (pFile->AppendString(" 0 R ") < 0) + return -1; + offset += len + 6; + break; + } + case CPDF_Object::ARRAY: { + if (pFile->AppendString("[") < 0) { + return -1; + } + offset += 1; + const CPDF_Array* p = pObj->AsArray(); + for (FX_DWORD i = 0; i < p->GetCount(); i++) { + CPDF_Object* pElement = p->GetElement(i); + if (pElement->GetObjNum()) { + if (pFile->AppendString(" ") < 0) { + return -1; + } + if ((len = pFile->AppendDWord(pElement->GetObjNum())) < 0) { + return -1; + } + if (pFile->AppendString(" 0 R") < 0) { + return -1; + } + offset += len + 5; + } else { + if (PDF_CreatorAppendObject(pElement, pFile, offset) < 0) { + return -1; + } + } + } + if (pFile->AppendString("]") < 0) { + return -1; + } + offset += 1; + break; + } + case CPDF_Object::DICTIONARY: { + if (pFile->AppendString("<<") < 0) { + return -1; + } + offset += 2; + const CPDF_Dictionary* p = pObj->AsDictionary(); + for (const auto& it : *p) { + const CFX_ByteString& key = it.first; + CPDF_Object* pValue = it.second; + if (pFile->AppendString("/") < 0) { + return -1; + } + if ((len = pFile->AppendString(PDF_NameEncode(key))) < 0) { + return -1; + } + offset += len + 1; + if (pValue->GetObjNum()) { + if (pFile->AppendString(" ") < 0) { + return -1; + } + if ((len = pFile->AppendDWord(pValue->GetObjNum())) < 0) { + return -1; + } + if (pFile->AppendString(" 0 R") < 0) { + return -1; + } + offset += len + 5; + } else { + if (PDF_CreatorAppendObject(pValue, pFile, offset) < 0) { + return -1; + } + } + } + if (pFile->AppendString(">>") < 0) { + return -1; + } + offset += 2; + break; + } + case CPDF_Object::STREAM: { + const CPDF_Stream* p = pObj->AsStream(); + if (PDF_CreatorAppendObject(p->GetDict(), pFile, offset) < 0) { + return -1; + } + if (pFile->AppendString("stream\r\n") < 0) { + return -1; + } + offset += 8; + CPDF_StreamAcc acc; + acc.LoadAllData(p, TRUE); + if (pFile->AppendBlock(acc.GetData(), acc.GetSize()) < 0) { + return -1; + } + offset += acc.GetSize(); + if ((len = pFile->AppendString("\r\nendstream")) < 0) { + return -1; + } + offset += len; + break; + } + default: + ASSERT(FALSE); + break; + } + return 1; +} + +int32_t PDF_CreatorWriteTrailer(CPDF_Document* pDocument, + CFX_FileBufferArchive* pFile, + CPDF_Array* pIDArray, + FX_BOOL bCompress) { + FX_FILESIZE offset = 0; + int32_t len = 0; + FXSYS_assert(pDocument && pFile); + CPDF_Parser* pParser = (CPDF_Parser*)pDocument->GetParser(); + if (pParser) { + CPDF_Dictionary* p = pParser->GetTrailer(); + for (const auto& it : *p) { + const CFX_ByteString& key = it.first; + CPDF_Object* pValue = it.second; + if (key == "Encrypt" || key == "Size" || key == "Filter" || + key == "Index" || key == "Length" || key == "Prev" || key == "W" || + key == "XRefStm" || key == "Type" || key == "ID") { + continue; + } + if (bCompress && key == "DecodeParms") { + continue; + } + if (pFile->AppendString(("/")) < 0) { + return -1; + } + if ((len = pFile->AppendString(PDF_NameEncode(key))) < 0) { + return -1; + } + offset += len + 1; + if (pValue->GetObjNum()) { + if (pFile->AppendString(" ") < 0) { + return -1; + } + if ((len = pFile->AppendDWord(pValue->GetObjNum())) < 0) { + return -1; + } + if (pFile->AppendString(" 0 R ") < 0) { + return -1; + } + offset += len + 6; + } else { + if (PDF_CreatorAppendObject(pValue, pFile, offset) < 0) { + return -1; + } + } + } + if (pIDArray) { + if (pFile->AppendString(("/ID")) < 0) { + return -1; + } + offset += 3; + if (PDF_CreatorAppendObject(pIDArray, pFile, offset) < 0) { + return -1; + } + } + return offset; + } + if (pFile->AppendString("\r\n/Root ") < 0) { + return -1; + } + if ((len = pFile->AppendDWord(pDocument->GetRoot()->GetObjNum())) < 0) { + return -1; + } + if (pFile->AppendString(" 0 R\r\n") < 0) { + return -1; + } + offset += len + 14; + if (pDocument->GetInfo()) { + if (pFile->AppendString("/Info ") < 0) { + return -1; + } + if ((len = pFile->AppendDWord(pDocument->GetInfo()->GetObjNum())) < 0) { + return -1; + } + if (pFile->AppendString(" 0 R\r\n") < 0) { + return -1; + } + offset += len + 12; + } + if (pIDArray) { + if (pFile->AppendString(("/ID")) < 0) { + return -1; + } + offset += 3; + if (PDF_CreatorAppendObject(pIDArray, pFile, offset) < 0) { + return -1; + } + } + return offset; +} + +int32_t PDF_CreatorWriteEncrypt(const CPDF_Dictionary* pEncryptDict, + FX_DWORD dwObjNum, + CFX_FileBufferArchive* pFile) { + if (!pEncryptDict) { + return 0; + } + FXSYS_assert(pFile); + FX_FILESIZE offset = 0; + int32_t len = 0; + if (pFile->AppendString("/Encrypt") < 0) { + return -1; + } + offset += 8; + if (pFile->AppendString(" ") < 0) { + return -1; + } + if ((len = pFile->AppendDWord(dwObjNum)) < 0) { + return -1; + } + if (pFile->AppendString(" 0 R ") < 0) { + return -1; + } + offset += len + 6; + return offset; +} + +std::vector PDF_GenerateFileID(FX_DWORD dwSeed1, FX_DWORD dwSeed2) { + std::vector buffer(sizeof(FX_DWORD) * 4); + FX_DWORD* pBuffer = reinterpret_cast(buffer.data()); + void* pContext = FX_Random_MT_Start(dwSeed1); + for (int i = 0; i < 2; ++i) + *pBuffer++ = FX_Random_MT_Generate(pContext); + FX_Random_MT_Close(pContext); + pContext = FX_Random_MT_Start(dwSeed2); + for (int i = 0; i < 2; ++i) + *pBuffer++ = FX_Random_MT_Generate(pContext); + FX_Random_MT_Close(pContext); + return buffer; +} + +void AppendIndex0(CFX_ByteTextBuf& buffer, bool bFirstObject) { + buffer.AppendByte(0); + buffer.AppendByte(0); + buffer.AppendByte(0); + buffer.AppendByte(0); + buffer.AppendByte(0); + const uint8_t byte = bFirstObject ? 0xFF : 0; + buffer.AppendByte(byte); + buffer.AppendByte(byte); +} + +void AppendIndex1(CFX_ByteTextBuf& buffer, FX_FILESIZE offset) { + buffer.AppendByte(1); + buffer.AppendByte(FX_GETBYTEOFFSET24(offset)); + buffer.AppendByte(FX_GETBYTEOFFSET16(offset)); + buffer.AppendByte(FX_GETBYTEOFFSET8(offset)); + buffer.AppendByte(FX_GETBYTEOFFSET0(offset)); + buffer.AppendByte(0); + buffer.AppendByte(0); +} + +void AppendIndex2(CFX_ByteTextBuf& buffer, FX_DWORD objnum, int32_t index) { + buffer.AppendByte(2); + buffer.AppendByte(FX_GETBYTEOFFSET24(objnum)); + buffer.AppendByte(FX_GETBYTEOFFSET16(objnum)); + buffer.AppendByte(FX_GETBYTEOFFSET8(objnum)); + buffer.AppendByte(FX_GETBYTEOFFSET0(objnum)); + buffer.AppendByte(FX_GETBYTEOFFSET8(index)); + buffer.AppendByte(FX_GETBYTEOFFSET0(index)); +} + +bool IsXRefNeedEnd(CPDF_XRefStream* pXRef, FX_DWORD flag) { + if (!(flag & FPDFCREATE_INCREMENTAL)) + return false; + + FX_DWORD iCount = 0; + for (const auto& pair : pXRef->m_IndexArray) + iCount += pair.count; + + return iCount >= PDF_XREFSTREAM_MAXSIZE; +} + +int32_t OutputIndex(CFX_FileBufferArchive* pFile, FX_FILESIZE offset) { + if (sizeof(offset) > 4) { + if (FX_GETBYTEOFFSET32(offset)) { + if (pFile->AppendByte(FX_GETBYTEOFFSET56(offset)) < 0) + return -1; + if (pFile->AppendByte(FX_GETBYTEOFFSET48(offset)) < 0) + return -1; + if (pFile->AppendByte(FX_GETBYTEOFFSET40(offset)) < 0) + return -1; + if (pFile->AppendByte(FX_GETBYTEOFFSET32(offset)) < 0) + return -1; + } + } + if (pFile->AppendByte(FX_GETBYTEOFFSET24(offset)) < 0) + return -1; + if (pFile->AppendByte(FX_GETBYTEOFFSET16(offset)) < 0) + return -1; + if (pFile->AppendByte(FX_GETBYTEOFFSET8(offset)) < 0) + return -1; + if (pFile->AppendByte(FX_GETBYTEOFFSET0(offset)) < 0) + return -1; + if (pFile->AppendByte(0) < 0) + return -1; + return 0; +} + +class CPDF_FlateEncoder { + public: + CPDF_FlateEncoder(); + ~CPDF_FlateEncoder(); + FX_BOOL Initialize(CPDF_Stream* pStream, FX_BOOL bFlateEncode); + FX_BOOL Initialize(const uint8_t* pBuffer, + FX_DWORD size, + FX_BOOL bFlateEncode, + FX_BOOL bXRefStream = FALSE); + void CloneDict(); + uint8_t* m_pData; + FX_DWORD m_dwSize; + CPDF_Dictionary* m_pDict; + FX_BOOL m_bCloned; + FX_BOOL m_bNewData; + CPDF_StreamAcc m_Acc; +}; +CPDF_FlateEncoder::CPDF_FlateEncoder() { + m_pData = NULL; + m_dwSize = 0; + m_pDict = NULL; + m_bCloned = FALSE; + m_bNewData = FALSE; +} +void CPDF_FlateEncoder::CloneDict() { + if (!m_bCloned) { + m_pDict = ToDictionary(m_pDict->Clone()); + ASSERT(m_pDict); + m_bCloned = TRUE; + } +} +FX_BOOL CPDF_FlateEncoder::Initialize(CPDF_Stream* pStream, + FX_BOOL bFlateEncode) { + m_Acc.LoadAllData(pStream, TRUE); + if ((pStream && pStream->GetDict() && + pStream->GetDict()->KeyExist("Filter")) || + !bFlateEncode) { + if (pStream->GetDict()->KeyExist("Filter") && !bFlateEncode) { + CPDF_StreamAcc destAcc; + destAcc.LoadAllData(pStream); + m_dwSize = destAcc.GetSize(); + m_pData = (uint8_t*)destAcc.DetachData(); + m_pDict = ToDictionary(pStream->GetDict()->Clone()); + m_pDict->RemoveAt("Filter"); + m_bNewData = TRUE; + m_bCloned = TRUE; + } else { + m_pData = (uint8_t*)m_Acc.GetData(); + m_dwSize = m_Acc.GetSize(); + m_pDict = pStream->GetDict(); + } + return TRUE; + } + m_pData = NULL; + m_dwSize = 0; + m_bNewData = TRUE; + m_bCloned = TRUE; + ::FlateEncode(m_Acc.GetData(), m_Acc.GetSize(), m_pData, m_dwSize); + m_pDict = ToDictionary(pStream->GetDict()->Clone()); + m_pDict->SetAtInteger("Length", m_dwSize); + m_pDict->SetAtName("Filter", "FlateDecode"); + m_pDict->RemoveAt("DecodeParms"); + return TRUE; +} +FX_BOOL CPDF_FlateEncoder::Initialize(const uint8_t* pBuffer, + FX_DWORD size, + FX_BOOL bFlateEncode, + FX_BOOL bXRefStream) { + if (!bFlateEncode) { + m_pData = (uint8_t*)pBuffer; + m_dwSize = size; + return TRUE; + } + m_bNewData = TRUE; + if (bXRefStream) { + ::FlateEncode(pBuffer, size, 12, 1, 8, 7, m_pData, m_dwSize); + } else { + ::FlateEncode(pBuffer, size, m_pData, m_dwSize); + } + return TRUE; +} +CPDF_FlateEncoder::~CPDF_FlateEncoder() { + if (m_bCloned && m_pDict) { + m_pDict->Release(); + } + if (m_bNewData) { + FX_Free(m_pData); + } +} +class CPDF_Encryptor { + public: + CPDF_Encryptor(); + ~CPDF_Encryptor(); + FX_BOOL Initialize(IPDF_CryptoHandler* pHandler, + int objnum, + uint8_t* src_data, + FX_DWORD src_size); + uint8_t* m_pData; + FX_DWORD m_dwSize; + FX_BOOL m_bNewBuf; +}; +CPDF_Encryptor::CPDF_Encryptor() { + m_pData = NULL; + m_dwSize = 0; + m_bNewBuf = FALSE; +} +FX_BOOL CPDF_Encryptor::Initialize(IPDF_CryptoHandler* pHandler, + int objnum, + uint8_t* src_data, + FX_DWORD src_size) { + if (src_size == 0) { + return TRUE; + } + if (!pHandler) { + m_pData = (uint8_t*)src_data; + m_dwSize = src_size; + m_bNewBuf = FALSE; + return TRUE; + } + m_dwSize = pHandler->EncryptGetSize(objnum, 0, src_data, src_size); + m_pData = FX_Alloc(uint8_t, m_dwSize); + pHandler->EncryptContent(objnum, 0, src_data, src_size, m_pData, m_dwSize); + m_bNewBuf = TRUE; + return TRUE; +} +CPDF_Encryptor::~CPDF_Encryptor() { + if (m_bNewBuf) { + FX_Free(m_pData); + } +} + +} // namespace + +CPDF_ObjectStream::CPDF_ObjectStream() : m_dwObjNum(0), m_index(0) {} +FX_BOOL CPDF_ObjectStream::Start() { + m_ObjNumArray.RemoveAll(); + m_OffsetArray.RemoveAll(); + m_Buffer.Clear(); + m_dwObjNum = 0; + m_index = 0; + return TRUE; +} +int32_t CPDF_ObjectStream::CompressIndirectObject(FX_DWORD dwObjNum, + const CPDF_Object* pObj) { + m_ObjNumArray.Add(dwObjNum); + m_OffsetArray.Add(m_Buffer.GetLength()); + m_Buffer << pObj; + return 1; +} +int32_t CPDF_ObjectStream::CompressIndirectObject(FX_DWORD dwObjNum, + const uint8_t* pBuffer, + FX_DWORD dwSize) { + m_ObjNumArray.Add(dwObjNum); + m_OffsetArray.Add(m_Buffer.GetLength()); + m_Buffer.AppendBlock(pBuffer, dwSize); + return 1; +} +FX_FILESIZE CPDF_ObjectStream::End(CPDF_Creator* pCreator) { + FXSYS_assert(pCreator); + if (m_ObjNumArray.GetSize() == 0) { + return 0; + } + CFX_FileBufferArchive* pFile = &pCreator->m_File; + IPDF_CryptoHandler* pHandler = pCreator->m_pCryptoHandler; + FX_FILESIZE ObjOffset = pCreator->m_Offset; + if (!m_dwObjNum) { + m_dwObjNum = ++pCreator->m_dwLastObjNum; + } + CFX_ByteTextBuf tempBuffer; + int32_t iCount = m_ObjNumArray.GetSize(); + for (int32_t i = 0; i < iCount; i++) { + tempBuffer << m_ObjNumArray.ElementAt(i) << " " + << m_OffsetArray.ElementAt(i) << " "; + } + FX_FILESIZE& offset = pCreator->m_Offset; + int32_t len = pFile->AppendDWord(m_dwObjNum); + if (len < 0) { + return -1; + } + offset += len; + if ((len = pFile->AppendString(" 0 obj\r\n<AppendDWord((FX_DWORD)iCount)) < 0) { + return -1; + } + offset += len; + if (pFile->AppendString("/First ") < 0) { + return -1; + } + if ((len = pFile->AppendDWord((FX_DWORD)tempBuffer.GetLength())) < 0) { + return -1; + } + if (pFile->AppendString("/Length ") < 0) { + return -1; + } + offset += len + 15; + if (!pCreator->m_bCompress && !pHandler) { + if ((len = pFile->AppendDWord( + (FX_DWORD)(tempBuffer.GetLength() + m_Buffer.GetLength()))) < 0) { + return -1; + } + offset += len; + if ((len = pFile->AppendString(">>stream\r\n")) < 0) { + return -1; + } + if (pFile->AppendBlock(tempBuffer.GetBuffer(), tempBuffer.GetLength()) < + 0) { + return -1; + } + if (pFile->AppendBlock(m_Buffer.GetBuffer(), m_Buffer.GetLength()) < 0) { + return -1; + } + offset += len + tempBuffer.GetLength() + m_Buffer.GetLength(); + } else { + tempBuffer << m_Buffer; + CPDF_FlateEncoder encoder; + encoder.Initialize(tempBuffer.GetBuffer(), tempBuffer.GetLength(), + pCreator->m_bCompress); + CPDF_Encryptor encryptor; + encryptor.Initialize(pHandler, m_dwObjNum, encoder.m_pData, + encoder.m_dwSize); + if ((len = pFile->AppendDWord(encryptor.m_dwSize)) < 0) { + return -1; + } + offset += len; + if (pCreator->m_bCompress) { + if (pFile->AppendString("/Filter /FlateDecode") < 0) { + return -1; + } + offset += 20; + } + if ((len = pFile->AppendString(">>stream\r\n")) < 0) { + return -1; + } + if (pFile->AppendBlock(encryptor.m_pData, encryptor.m_dwSize) < 0) { + return -1; + } + offset += len + encryptor.m_dwSize; + } + if ((len = pFile->AppendString("\r\nendstream\r\nendobj\r\n")) < 0) { + return -1; + } + offset += len; + return ObjOffset; +} +CPDF_XRefStream::CPDF_XRefStream() + : m_PrevOffset(0), m_dwTempObjNum(0), m_iSeg(0) {} +FX_BOOL CPDF_XRefStream::Start() { + m_IndexArray.clear(); + m_Buffer.Clear(); + m_iSeg = 0; + return TRUE; +} +int32_t CPDF_XRefStream::CompressIndirectObject(FX_DWORD dwObjNum, + const CPDF_Object* pObj, + CPDF_Creator* pCreator) { + if (!pCreator) { + return 0; + } + m_ObjStream.CompressIndirectObject(dwObjNum, pObj); + if (m_ObjStream.m_ObjNumArray.GetSize() < pCreator->m_ObjectStreamSize && + m_ObjStream.m_Buffer.GetLength() < PDF_OBJECTSTREAM_MAXLENGTH) { + return 1; + } + return EndObjectStream(pCreator); +} +int32_t CPDF_XRefStream::CompressIndirectObject(FX_DWORD dwObjNum, + const uint8_t* pBuffer, + FX_DWORD dwSize, + CPDF_Creator* pCreator) { + if (!pCreator) { + return 0; + } + m_ObjStream.CompressIndirectObject(dwObjNum, pBuffer, dwSize); + if (m_ObjStream.m_ObjNumArray.GetSize() < pCreator->m_ObjectStreamSize && + m_ObjStream.m_Buffer.GetLength() < PDF_OBJECTSTREAM_MAXLENGTH) { + return 1; + } + return EndObjectStream(pCreator); +} + +int32_t CPDF_XRefStream::EndObjectStream(CPDF_Creator* pCreator, FX_BOOL bEOF) { + FX_FILESIZE objOffset = 0; + if (bEOF) { + objOffset = m_ObjStream.End(pCreator); + if (objOffset < 0) { + return -1; + } + } + FX_DWORD& dwObjStmNum = m_ObjStream.m_dwObjNum; + if (!dwObjStmNum) { + dwObjStmNum = ++pCreator->m_dwLastObjNum; + } + int32_t iSize = m_ObjStream.m_ObjNumArray.GetSize(); + size_t iSeg = m_IndexArray.size(); + if (!(pCreator->m_dwFlags & FPDFCREATE_INCREMENTAL)) { + if (m_dwTempObjNum == 0) { + AppendIndex0(m_Buffer, true); + m_dwTempObjNum++; + } + FX_DWORD end_num = m_IndexArray.back().objnum + m_IndexArray.back().count; + int index = 0; + for (; m_dwTempObjNum < end_num; m_dwTempObjNum++) { + FX_FILESIZE* offset = pCreator->m_ObjectOffset.GetPtrAt(m_dwTempObjNum); + if (offset) { + if (index >= iSize || + m_dwTempObjNum != m_ObjStream.m_ObjNumArray[index]) { + AppendIndex1(m_Buffer, *offset); + } else { + AppendIndex2(m_Buffer, dwObjStmNum, index++); + } + } else { + AppendIndex0(m_Buffer, false); + } + } + if (iSize > 0 && bEOF) { + pCreator->m_ObjectOffset.Add(dwObjStmNum, 1); + pCreator->m_ObjectOffset[dwObjStmNum] = objOffset; + } + m_iSeg = iSeg; + if (bEOF) { + m_ObjStream.Start(); + } + return 1; + } + for (auto it = m_IndexArray.begin() + m_iSeg; it != m_IndexArray.end(); + ++it) { + for (FX_DWORD m = it->objnum; m < it->objnum + it->count; ++m) { + if (m_ObjStream.m_index >= iSize || + m != m_ObjStream.m_ObjNumArray.ElementAt(it - m_IndexArray.begin())) { + AppendIndex1(m_Buffer, pCreator->m_ObjectOffset[m]); + } else { + AppendIndex2(m_Buffer, dwObjStmNum, m_ObjStream.m_index++); + } + } + } + if (iSize > 0 && bEOF) { + AppendIndex1(m_Buffer, objOffset); + m_IndexArray.push_back({dwObjStmNum, 1}); + iSeg += 1; + } + m_iSeg = iSeg; + if (bEOF) { + m_ObjStream.Start(); + } + return 1; +} +FX_BOOL CPDF_XRefStream::GenerateXRefStream(CPDF_Creator* pCreator, + FX_BOOL bEOF) { + FX_FILESIZE offset_tmp = pCreator->m_Offset; + FX_DWORD objnum = ++pCreator->m_dwLastObjNum; + CFX_FileBufferArchive* pFile = &pCreator->m_File; + FX_BOOL bIncremental = (pCreator->m_dwFlags & FPDFCREATE_INCREMENTAL) != 0; + if (bIncremental) { + AddObjectNumberToIndexArray(objnum); + } else { + for (; m_dwTempObjNum < pCreator->m_dwLastObjNum; m_dwTempObjNum++) { + FX_FILESIZE* offset = pCreator->m_ObjectOffset.GetPtrAt(m_dwTempObjNum); + if (offset) { + AppendIndex1(m_Buffer, *offset); + } else { + AppendIndex0(m_Buffer, false); + } + } + } + AppendIndex1(m_Buffer, offset_tmp); + FX_FILESIZE& offset = pCreator->m_Offset; + int32_t len = pFile->AppendDWord(objnum); + if (len < 0) { + return FALSE; + } + offset += len; + if ((len = pFile->AppendString(" 0 obj\r\n<AppendDWord(0)) < 0) { + return FALSE; + } + if ((len = pFile->AppendString(" ")) < 0) { + return FALSE; + } + offset += len + 1; + if ((len = pFile->AppendDWord(objnum + 1)) < 0) { + return FALSE; + } + offset += len; + } else { + for (const auto& pair : m_IndexArray) { + if ((len = pFile->AppendDWord(pair.objnum)) < 0) { + return FALSE; + } + if (pFile->AppendString(" ") < 0) { + return FALSE; + } + offset += len + 1; + if ((len = pFile->AppendDWord(pair.count)) < 0) { + return FALSE; + } + if (pFile->AppendString(" ") < 0) { + return FALSE; + } + offset += len + 1; + } + } + if (pFile->AppendString("]/Size ") < 0) { + return FALSE; + } + if ((len = pFile->AppendDWord(objnum + 1)) < 0) { + return FALSE; + } + offset += len + 7; + if (m_PrevOffset > 0) { + if (pFile->AppendString("/Prev ") < 0) { + return FALSE; + } + FX_CHAR offset_buf[20]; + FXSYS_memset(offset_buf, 0, sizeof(offset_buf)); + FXSYS_i64toa(m_PrevOffset, offset_buf, 10); + int32_t len = (int32_t)FXSYS_strlen(offset_buf); + if (pFile->AppendBlock(offset_buf, len) < 0) { + return FALSE; + } + offset += len + 6; + } + FX_BOOL bPredictor = TRUE; + CPDF_FlateEncoder encoder; + encoder.Initialize(m_Buffer.GetBuffer(), m_Buffer.GetLength(), + pCreator->m_bCompress, bPredictor); + if (pCreator->m_bCompress) { + if (pFile->AppendString("/Filter /FlateDecode") < 0) { + return FALSE; + } + offset += 20; + if (bPredictor) { + if ((len = pFile->AppendString( + "/DecodeParms<>")) < 0) { + return FALSE; + } + offset += len; + } + } + if (pFile->AppendString("/Length ") < 0) { + return FALSE; + } + if ((len = pFile->AppendDWord(encoder.m_dwSize)) < 0) { + return FALSE; + } + offset += len + 8; + if (bEOF) { + if ((len = PDF_CreatorWriteTrailer(pCreator->m_pDocument, pFile, + pCreator->m_pIDArray, + pCreator->m_bCompress)) < 0) { + return FALSE; + } + offset += len; + if (pCreator->m_pEncryptDict) { + FX_DWORD dwEncryptObjNum = pCreator->m_pEncryptDict->GetObjNum(); + if (dwEncryptObjNum == 0) { + dwEncryptObjNum = pCreator->m_dwEnryptObjNum; + } + if ((len = PDF_CreatorWriteEncrypt(pCreator->m_pEncryptDict, + dwEncryptObjNum, pFile)) < 0) { + return FALSE; + } + offset += len; + } + } + if ((len = pFile->AppendString(">>stream\r\n")) < 0) { + return FALSE; + } + offset += len; + if (pFile->AppendBlock(encoder.m_pData, encoder.m_dwSize) < 0) { + return FALSE; + } + if ((len = pFile->AppendString("\r\nendstream\r\nendobj\r\n")) < 0) { + return FALSE; + } + offset += encoder.m_dwSize + len; + m_PrevOffset = offset_tmp; + return TRUE; +} +FX_BOOL CPDF_XRefStream::End(CPDF_Creator* pCreator, FX_BOOL bEOF) { + if (EndObjectStream(pCreator, bEOF) < 0) { + return FALSE; + } + return GenerateXRefStream(pCreator, bEOF); +} +FX_BOOL CPDF_XRefStream::EndXRefStream(CPDF_Creator* pCreator) { + if (!(pCreator->m_dwFlags & FPDFCREATE_INCREMENTAL)) { + AppendIndex0(m_Buffer, true); + for (FX_DWORD i = 1; i < pCreator->m_dwLastObjNum + 1; i++) { + FX_FILESIZE* offset = pCreator->m_ObjectOffset.GetPtrAt(i); + if (offset) { + AppendIndex1(m_Buffer, *offset); + } else { + AppendIndex0(m_Buffer, false); + } + } + } else { + for (const auto& pair : m_IndexArray) { + for (FX_DWORD j = pair.objnum; j < pair.objnum + pair.count; ++j) + AppendIndex1(m_Buffer, pCreator->m_ObjectOffset[j]); + } + } + return GenerateXRefStream(pCreator, FALSE); +} +void CPDF_XRefStream::AddObjectNumberToIndexArray(FX_DWORD objnum) { + if (m_IndexArray.empty()) { + m_IndexArray.push_back({objnum, 1}); + return; + } + FX_DWORD next_objnum = m_IndexArray.back().objnum + m_IndexArray.back().count; + if (objnum == next_objnum) + m_IndexArray.back().count += 1; + else + m_IndexArray.push_back({objnum, 1}); +} +CPDF_Creator::CPDF_Creator(CPDF_Document* pDoc) { + m_pDocument = pDoc; + m_pParser = (CPDF_Parser*)pDoc->m_pParser; + m_bCompress = TRUE; + if (m_pParser) { + m_pEncryptDict = m_pParser->GetEncryptDict(); + m_pCryptoHandler = m_pParser->GetCryptoHandler(); + } else { + m_pEncryptDict = NULL; + m_pCryptoHandler = NULL; + } + m_bSecurityChanged = FALSE; + m_bStandardSecurity = FALSE; + m_pMetadata = NULL; + m_bEncryptCloned = FALSE; + m_bEncryptMetadata = FALSE; + m_Offset = 0; + m_iStage = -1; + m_dwFlags = 0; + m_Pos = NULL; + m_XrefStart = 0; + m_pXRefStream = NULL; + m_ObjectStreamSize = 200; + m_dwLastObjNum = m_pDocument->GetLastObjNum(); + m_pIDArray = NULL; + m_FileVersion = 0; + m_dwEnryptObjNum = 0; + m_bNewCrypto = FALSE; +} +CPDF_Creator::~CPDF_Creator() { + ResetStandardSecurity(); + if (m_bEncryptCloned && m_pEncryptDict) { + m_pEncryptDict->Release(); + m_pEncryptDict = NULL; + } + Clear(); +} + +int32_t CPDF_Creator::WriteIndirectObjectToStream(const CPDF_Object* pObj) { + if (!m_pXRefStream) + return 1; + + FX_DWORD objnum = pObj->GetObjNum(); + if (m_pParser && m_pParser->GetObjectGenNum(objnum) > 0) + return 1; + + if (pObj->IsNumber()) + return 1; + + CPDF_Dictionary* pDict = pObj->GetDict(); + if (pObj->IsStream()) { + if (pDict && pDict->GetStringBy("Type") == "XRef") + return 0; + return 1; + } + + if (pDict) { + if (pDict == m_pDocument->m_pRootDict || pDict == m_pEncryptDict) + return 1; + if (pDict->IsSignatureDict()) + return 1; + if (pDict->GetStringBy("Type") == "Page") + return 1; + } + + m_pXRefStream->AddObjectNumberToIndexArray(objnum); + if (m_pXRefStream->CompressIndirectObject(objnum, pObj, this) < 0) + return -1; + if (!IsXRefNeedEnd(m_pXRefStream, m_dwFlags)) + return 0; + if (!m_pXRefStream->End(this)) + return -1; + if (!m_pXRefStream->Start()) + return -1; + return 0; +} +int32_t CPDF_Creator::WriteIndirectObjectToStream(FX_DWORD objnum, + const uint8_t* pBuffer, + FX_DWORD dwSize) { + if (!m_pXRefStream) { + return 1; + } + m_pXRefStream->AddObjectNumberToIndexArray(objnum); + int32_t iRet = + m_pXRefStream->CompressIndirectObject(objnum, pBuffer, dwSize, this); + if (iRet < 1) { + return iRet; + } + if (!IsXRefNeedEnd(m_pXRefStream, m_dwFlags)) { + return 0; + } + if (!m_pXRefStream->End(this)) { + return -1; + } + if (!m_pXRefStream->Start()) { + return -1; + } + return 0; +} +int32_t CPDF_Creator::AppendObjectNumberToXRef(FX_DWORD objnum) { + if (!m_pXRefStream) { + return 1; + } + m_pXRefStream->AddObjectNumberToIndexArray(objnum); + if (!IsXRefNeedEnd(m_pXRefStream, m_dwFlags)) { + return 0; + } + if (!m_pXRefStream->End(this)) { + return -1; + } + if (!m_pXRefStream->Start()) { + return -1; + } + return 0; +} +int32_t CPDF_Creator::WriteStream(const CPDF_Object* pStream, + FX_DWORD objnum, + IPDF_CryptoHandler* pCrypto) { + CPDF_FlateEncoder encoder; + encoder.Initialize(const_cast(pStream->AsStream()), + pStream == m_pMetadata ? FALSE : m_bCompress); + CPDF_Encryptor encryptor; + if (!encryptor.Initialize(pCrypto, objnum, encoder.m_pData, + encoder.m_dwSize)) { + return -1; + } + if ((FX_DWORD)encoder.m_pDict->GetIntegerBy("Length") != encryptor.m_dwSize) { + encoder.CloneDict(); + encoder.m_pDict->SetAtInteger("Length", encryptor.m_dwSize); + } + if (WriteDirectObj(objnum, encoder.m_pDict) < 0) { + return -1; + } + int len = m_File.AppendString("stream\r\n"); + if (len < 0) { + return -1; + } + m_Offset += len; + if (m_File.AppendBlock(encryptor.m_pData, encryptor.m_dwSize) < 0) { + return -1; + } + m_Offset += encryptor.m_dwSize; + if ((len = m_File.AppendString("\r\nendstream")) < 0) { + return -1; + } + m_Offset += len; + return 1; +} +int32_t CPDF_Creator::WriteIndirectObj(FX_DWORD objnum, + const CPDF_Object* pObj) { + int32_t len = m_File.AppendDWord(objnum); + if (len < 0) + return -1; + + m_Offset += len; + if ((len = m_File.AppendString(" 0 obj\r\n")) < 0) + return -1; + + m_Offset += len; + if (pObj->IsStream()) { + IPDF_CryptoHandler* pHandler = nullptr; + pHandler = + (pObj == m_pMetadata && !m_bEncryptMetadata) ? NULL : m_pCryptoHandler; + if (WriteStream(pObj, objnum, pHandler) < 0) + return -1; + } else { + if (WriteDirectObj(objnum, pObj) < 0) + return -1; + } + if ((len = m_File.AppendString("\r\nendobj\r\n")) < 0) + return -1; + + m_Offset += len; + if (AppendObjectNumberToXRef(objnum) < 0) + return -1; + return 0; +} +int32_t CPDF_Creator::WriteIndirectObj(const CPDF_Object* pObj) { + int32_t iRet = WriteIndirectObjectToStream(pObj); + if (iRet < 1) { + return iRet; + } + return WriteIndirectObj(pObj->GetObjNum(), pObj); +} +int32_t CPDF_Creator::WriteDirectObj(FX_DWORD objnum, + const CPDF_Object* pObj, + FX_BOOL bEncrypt) { + int32_t len = 0; + if (!pObj) { + if (m_File.AppendString(" null") < 0) { + return -1; + } + m_Offset += 5; + return 1; + } + switch (pObj->GetType()) { + case CPDF_Object::NULLOBJ: + if (m_File.AppendString(" null") < 0) { + return -1; + } + m_Offset += 5; + break; + case CPDF_Object::BOOLEAN: + case CPDF_Object::NUMBER: + if (m_File.AppendString(" ") < 0) { + return -1; + } + if ((len = m_File.AppendString(pObj->GetString())) < 0) { + return -1; + } + m_Offset += len + 1; + break; + case CPDF_Object::STRING: { + CFX_ByteString str = pObj->GetString(); + FX_BOOL bHex = pObj->AsString()->IsHex(); + if (!m_pCryptoHandler || !bEncrypt) { + CFX_ByteString content = PDF_EncodeString(str, bHex); + if ((len = m_File.AppendString(content)) < 0) { + return -1; + } + m_Offset += len; + break; + } + CPDF_Encryptor encryptor; + encryptor.Initialize(m_pCryptoHandler, objnum, (uint8_t*)str.c_str(), + str.GetLength()); + CFX_ByteString content = PDF_EncodeString( + CFX_ByteString((const FX_CHAR*)encryptor.m_pData, encryptor.m_dwSize), + bHex); + if ((len = m_File.AppendString(content)) < 0) { + return -1; + } + m_Offset += len; + break; + } + case CPDF_Object::STREAM: { + CPDF_FlateEncoder encoder; + encoder.Initialize(const_cast(pObj->AsStream()), + m_bCompress); + CPDF_Encryptor encryptor; + IPDF_CryptoHandler* pHandler = m_pCryptoHandler; + encryptor.Initialize(pHandler, objnum, encoder.m_pData, encoder.m_dwSize); + if ((FX_DWORD)encoder.m_pDict->GetIntegerBy("Length") != + encryptor.m_dwSize) { + encoder.CloneDict(); + encoder.m_pDict->SetAtInteger("Length", encryptor.m_dwSize); + } + if (WriteDirectObj(objnum, encoder.m_pDict) < 0) { + return -1; + } + if ((len = m_File.AppendString("stream\r\n")) < 0) { + return -1; + } + m_Offset += len; + if (m_File.AppendBlock(encryptor.m_pData, encryptor.m_dwSize) < 0) { + return -1; + } + m_Offset += encryptor.m_dwSize; + if ((len = m_File.AppendString("\r\nendstream")) < 0) { + return -1; + } + m_Offset += len; + break; + } + case CPDF_Object::NAME: { + if (m_File.AppendString("/") < 0) { + return -1; + } + CFX_ByteString str = pObj->GetString(); + if ((len = m_File.AppendString(PDF_NameEncode(str))) < 0) { + return -1; + } + m_Offset += len + 1; + break; + } + case CPDF_Object::REFERENCE: { + if (m_File.AppendString(" ") < 0) + return -1; + if ((len = m_File.AppendDWord(pObj->AsReference()->GetRefObjNum())) < 0) + return -1; + if (m_File.AppendString(" 0 R") < 0) + return -1; + m_Offset += len + 5; + break; + } + case CPDF_Object::ARRAY: { + if (m_File.AppendString("[") < 0) { + return -1; + } + m_Offset += 1; + const CPDF_Array* p = pObj->AsArray(); + for (FX_DWORD i = 0; i < p->GetCount(); i++) { + CPDF_Object* pElement = p->GetElement(i); + if (pElement->GetObjNum()) { + if (m_File.AppendString(" ") < 0) { + return -1; + } + if ((len = m_File.AppendDWord(pElement->GetObjNum())) < 0) { + return -1; + } + if (m_File.AppendString(" 0 R") < 0) { + return -1; + } + m_Offset += len + 5; + } else { + if (WriteDirectObj(objnum, pElement) < 0) { + return -1; + } + } + } + if (m_File.AppendString("]") < 0) { + return -1; + } + m_Offset += 1; + break; + } + case CPDF_Object::DICTIONARY: { + if (!m_pCryptoHandler || pObj == m_pEncryptDict) { + return PDF_CreatorAppendObject(pObj, &m_File, m_Offset); + } + if (m_File.AppendString("<<") < 0) { + return -1; + } + m_Offset += 2; + const CPDF_Dictionary* p = pObj->AsDictionary(); + bool bSignDict = p->IsSignatureDict(); + for (const auto& it : *p) { + FX_BOOL bSignValue = FALSE; + const CFX_ByteString& key = it.first; + CPDF_Object* pValue = it.second; + if (m_File.AppendString("/") < 0) { + return -1; + } + if ((len = m_File.AppendString(PDF_NameEncode(key))) < 0) { + return -1; + } + m_Offset += len + 1; + if (bSignDict && key == "Contents") { + bSignValue = TRUE; + } + if (pValue->GetObjNum()) { + if (m_File.AppendString(" ") < 0) { + return -1; + } + if ((len = m_File.AppendDWord(pValue->GetObjNum())) < 0) { + return -1; + } + if (m_File.AppendString(" 0 R ") < 0) { + return -1; + } + m_Offset += len + 6; + } else { + if (WriteDirectObj(objnum, pValue, !bSignValue) < 0) { + return -1; + } + } + } + if (m_File.AppendString(">>") < 0) { + return -1; + } + m_Offset += 2; + break; + } + } + return 1; +} +int32_t CPDF_Creator::WriteOldIndirectObject(FX_DWORD objnum) { + if (m_pParser->IsObjectFreeOrNull(objnum)) + return 0; + + m_ObjectOffset[objnum] = m_Offset; + FX_BOOL bExistInMap = + pdfium::ContainsKey(m_pDocument->m_IndirectObjs, objnum); + const uint8_t object_type = m_pParser->GetObjectType(objnum); + bool bObjStm = (object_type == 2) && m_pEncryptDict && !m_pXRefStream; + if (m_pParser->IsVersionUpdated() || m_bSecurityChanged || bExistInMap || + bObjStm) { + CPDF_Object* pObj = m_pDocument->GetIndirectObject(objnum); + if (!pObj) { + m_ObjectOffset[objnum] = 0; + return 0; + } + if (WriteIndirectObj(pObj)) { + return -1; + } + if (!bExistInMap) { + m_pDocument->ReleaseIndirectObject(objnum); + } + } else { + uint8_t* pBuffer; + FX_DWORD size; + m_pParser->GetIndirectBinary(objnum, pBuffer, size); + if (!pBuffer) { + return 0; + } + if (object_type == 2) { + if (m_pXRefStream) { + if (WriteIndirectObjectToStream(objnum, pBuffer, size) < 0) { + FX_Free(pBuffer); + return -1; + } + } else { + int32_t len = m_File.AppendDWord(objnum); + if (len < 0) { + return -1; + } + if (m_File.AppendString(" 0 obj ") < 0) { + return -1; + } + m_Offset += len + 7; + if (m_File.AppendBlock(pBuffer, size) < 0) { + return -1; + } + m_Offset += size; + if (m_File.AppendString("\r\nendobj\r\n") < 0) { + return -1; + } + m_Offset += 10; + } + } else { + if (m_File.AppendBlock(pBuffer, size) < 0) { + return -1; + } + m_Offset += size; + if (AppendObjectNumberToXRef(objnum) < 0) { + return -1; + } + } + FX_Free(pBuffer); + } + return 1; +} +int32_t CPDF_Creator::WriteOldObjs(IFX_Pause* pPause) { + FX_DWORD nLastObjNum = m_pParser->GetLastObjNum(); + if (!m_pParser->IsValidObjectNumber(nLastObjNum)) + return 0; + + FX_DWORD objnum = (FX_DWORD)(uintptr_t)m_Pos; + for (; objnum <= nLastObjNum; ++objnum) { + int32_t iRet = WriteOldIndirectObject(objnum); + if (iRet < 0) + return iRet; + + if (!iRet) + continue; + + if (pPause && pPause->NeedToPauseNow()) { + m_Pos = (void*)(uintptr_t)(objnum + 1); + return 1; + } + } + return 0; +} +int32_t CPDF_Creator::WriteNewObjs(FX_BOOL bIncremental, IFX_Pause* pPause) { + int32_t iCount = m_NewObjNumArray.GetSize(); + int32_t index = (int32_t)(uintptr_t)m_Pos; + while (index < iCount) { + FX_DWORD objnum = m_NewObjNumArray.ElementAt(index); + auto it = m_pDocument->m_IndirectObjs.find(objnum); + if (it == m_pDocument->m_IndirectObjs.end()) { + ++index; + continue; + } + m_ObjectOffset[objnum] = m_Offset; + if (WriteIndirectObj(it->second)) { + return -1; + } + index++; + if (pPause && pPause->NeedToPauseNow()) { + m_Pos = (FX_POSITION)(uintptr_t)index; + return 1; + } + } + return 0; +} +void CPDF_Creator::InitOldObjNumOffsets() { + if (!m_pParser) { + return; + } + FX_DWORD j = 0; + FX_DWORD dwStart = 0; + FX_DWORD dwEnd = m_pParser->GetLastObjNum(); + while (dwStart <= dwEnd) { + while (dwStart <= dwEnd && m_pParser->IsObjectFreeOrNull(dwStart)) + dwStart++; + + if (dwStart > dwEnd) + break; + + j = dwStart; + while (j <= dwEnd && !m_pParser->IsObjectFreeOrNull(j)) + j++; + + m_ObjectOffset.Add(dwStart, j - dwStart); + dwStart = j; + } +} +void CPDF_Creator::InitNewObjNumOffsets() { + FX_BOOL bIncremental = (m_dwFlags & FPDFCREATE_INCREMENTAL) != 0; + FX_BOOL bNoOriginal = (m_dwFlags & FPDFCREATE_NO_ORIGINAL) != 0; + for (const auto& pair : m_pDocument->m_IndirectObjs) { + const FX_DWORD objnum = pair.first; + const CPDF_Object* pObj = pair.second; + if (pObj->GetObjNum() == -1) + continue; + if (bIncremental) { + if (!pObj->IsModified()) + continue; + } else if (m_pParser && m_pParser->IsValidObjectNumber(objnum) && + m_pParser->GetObjectType(objnum)) { + continue; + } + AppendNewObjNum(objnum); + } + + int32_t iCount = m_NewObjNumArray.GetSize(); + if (iCount == 0) { + return; + } + int32_t i = 0; + FX_DWORD dwStartObjNum = 0; + FX_BOOL bCrossRefValid = m_pParser && m_pParser->GetLastXRefOffset() > 0; + while (i < iCount) { + dwStartObjNum = m_NewObjNumArray.ElementAt(i); + if ((bIncremental && (bNoOriginal || bCrossRefValid)) || + !m_ObjectOffset.GetPtrAt(dwStartObjNum)) { + break; + } + i++; + } + if (i >= iCount) { + return; + } + FX_DWORD dwLastObjNum = dwStartObjNum; + i++; + FX_BOOL bNewStart = FALSE; + for (; i < iCount; i++) { + FX_DWORD dwCurObjNum = m_NewObjNumArray.ElementAt(i); + bool bExist = m_pParser && m_pParser->IsValidObjectNumber(dwCurObjNum) && + m_ObjectOffset.GetPtrAt(dwCurObjNum); + if (bExist || dwCurObjNum - dwLastObjNum > 1) { + if (!bNewStart) + m_ObjectOffset.Add(dwStartObjNum, dwLastObjNum - dwStartObjNum + 1); + dwStartObjNum = dwCurObjNum; + } + if (bNewStart) { + dwStartObjNum = dwCurObjNum; + } + bNewStart = bExist; + dwLastObjNum = dwCurObjNum; + } + m_ObjectOffset.Add(dwStartObjNum, dwLastObjNum - dwStartObjNum + 1); +} +void CPDF_Creator::AppendNewObjNum(FX_DWORD objbum) { + int32_t iStart = 0, iFind = 0; + int32_t iEnd = m_NewObjNumArray.GetUpperBound(); + while (iStart <= iEnd) { + int32_t iMid = (iStart + iEnd) / 2; + FX_DWORD dwMid = m_NewObjNumArray.ElementAt(iMid); + if (objbum < dwMid) { + iEnd = iMid - 1; + } else { + if (iMid == iEnd) { + iFind = iMid + 1; + break; + } + FX_DWORD dwNext = m_NewObjNumArray.ElementAt(iMid + 1); + if (objbum < dwNext) { + iFind = iMid + 1; + break; + } + iStart = iMid + 1; + } + } + m_NewObjNumArray.InsertAt(iFind, objbum); +} +int32_t CPDF_Creator::WriteDoc_Stage1(IFX_Pause* pPause) { + FXSYS_assert(m_iStage > -1 || m_iStage < 20); + if (m_iStage == 0) { + if (!m_pParser) { + m_dwFlags &= ~FPDFCREATE_INCREMENTAL; + } + if (m_bSecurityChanged && (m_dwFlags & FPDFCREATE_NO_ORIGINAL) == 0) { + m_dwFlags &= ~FPDFCREATE_INCREMENTAL; + } + CPDF_Dictionary* pDict = m_pDocument->GetRoot(); + m_pMetadata = pDict ? pDict->GetElementValue("Metadata") : NULL; + if (m_dwFlags & FPDFCREATE_OBJECTSTREAM) { + m_pXRefStream = new CPDF_XRefStream; + m_pXRefStream->Start(); + if ((m_dwFlags & FPDFCREATE_INCREMENTAL) != 0 && m_pParser) { + FX_FILESIZE prev = m_pParser->GetLastXRefOffset(); + m_pXRefStream->m_PrevOffset = prev; + } + } + m_iStage = 10; + } + if (m_iStage == 10) { + if ((m_dwFlags & FPDFCREATE_INCREMENTAL) == 0) { + if (m_File.AppendString("%PDF-1.") < 0) { + return -1; + } + m_Offset += 7; + int32_t version = 7; + if (m_FileVersion) { + version = m_FileVersion; + } else if (m_pParser) { + version = m_pParser->GetFileVersion(); + } + int32_t len = m_File.AppendDWord(version % 10); + if (len < 0) { + return -1; + } + m_Offset += len; + if ((len = m_File.AppendString("\r\n%\xA1\xB3\xC5\xD7\r\n")) < 0) { + return -1; + } + m_Offset += len; + InitOldObjNumOffsets(); + m_iStage = 20; + } else { + IFX_FileRead* pSrcFile = m_pParser->GetFileAccess(); + m_Offset = pSrcFile->GetSize(); + m_Pos = (void*)(uintptr_t)m_Offset; + m_iStage = 15; + } + } + if (m_iStage == 15) { + if ((m_dwFlags & FPDFCREATE_NO_ORIGINAL) == 0 && m_Pos) { + IFX_FileRead* pSrcFile = m_pParser->GetFileAccess(); + uint8_t buffer[4096]; + FX_DWORD src_size = (FX_DWORD)(uintptr_t)m_Pos; + while (src_size) { + FX_DWORD block_size = src_size > 4096 ? 4096 : src_size; + if (!pSrcFile->ReadBlock(buffer, m_Offset - src_size, block_size)) { + return -1; + } + if (m_File.AppendBlock(buffer, block_size) < 0) { + return -1; + } + src_size -= block_size; + if (pPause && pPause->NeedToPauseNow()) { + m_Pos = (void*)(uintptr_t)src_size; + return 1; + } + } + } + if ((m_dwFlags & FPDFCREATE_NO_ORIGINAL) == 0 && + m_pParser->GetLastXRefOffset() == 0) { + InitOldObjNumOffsets(); + FX_DWORD dwEnd = m_pParser->GetLastObjNum(); + bool bObjStm = (m_dwFlags & FPDFCREATE_OBJECTSTREAM) != 0; + for (FX_DWORD objnum = 0; objnum <= dwEnd; objnum++) { + if (m_pParser->IsObjectFreeOrNull(objnum)) + continue; + + m_ObjectOffset[objnum] = m_pParser->GetObjectPositionOrZero(objnum); + if (bObjStm) { + m_pXRefStream->AddObjectNumberToIndexArray(objnum); + } + } + if (bObjStm) { + m_pXRefStream->EndXRefStream(this); + m_pXRefStream->Start(); + } + } + m_iStage = 20; + } + InitNewObjNumOffsets(); + return m_iStage; +} +int32_t CPDF_Creator::WriteDoc_Stage2(IFX_Pause* pPause) { + FXSYS_assert(m_iStage >= 20 || m_iStage < 30); + if (m_iStage == 20) { + if ((m_dwFlags & FPDFCREATE_INCREMENTAL) == 0 && m_pParser) { + m_Pos = (void*)(uintptr_t)0; + m_iStage = 21; + } else { + m_iStage = 25; + } + } + if (m_iStage == 21) { + int32_t iRet = WriteOldObjs(pPause); + if (iRet) { + return iRet; + } + m_iStage = 25; + } + if (m_iStage == 25) { + m_Pos = (void*)(uintptr_t)0; + m_iStage = 26; + } + if (m_iStage == 26) { + int32_t iRet = + WriteNewObjs((m_dwFlags & FPDFCREATE_INCREMENTAL) != 0, pPause); + if (iRet) { + return iRet; + } + m_iStage = 27; + } + if (m_iStage == 27) { + if (NULL != m_pEncryptDict && 0 == m_pEncryptDict->GetObjNum()) { + m_dwLastObjNum += 1; + FX_FILESIZE saveOffset = m_Offset; + if (WriteIndirectObj(m_dwLastObjNum, m_pEncryptDict) < 0) { + return -1; + } + m_ObjectOffset.Add(m_dwLastObjNum, 1); + m_ObjectOffset[m_dwLastObjNum] = saveOffset; + m_dwEnryptObjNum = m_dwLastObjNum; + if (m_dwFlags & FPDFCREATE_INCREMENTAL) { + m_NewObjNumArray.Add(m_dwLastObjNum); + } + } + m_iStage = 80; + } + return m_iStage; +} +int32_t CPDF_Creator::WriteDoc_Stage3(IFX_Pause* pPause) { + FXSYS_assert(m_iStage >= 80 || m_iStage < 90); + FX_DWORD dwLastObjNum = m_dwLastObjNum; + if (m_iStage == 80) { + m_XrefStart = m_Offset; + if (m_dwFlags & FPDFCREATE_OBJECTSTREAM) { + m_pXRefStream->End(this, TRUE); + m_XrefStart = m_pXRefStream->m_PrevOffset; + m_iStage = 90; + } else if ((m_dwFlags & FPDFCREATE_INCREMENTAL) == 0 || + !m_pParser->IsXRefStream()) { + if ((m_dwFlags & FPDFCREATE_INCREMENTAL) == 0 || + m_pParser->GetLastXRefOffset() == 0) { + CFX_ByteString str; + str = m_ObjectOffset.GetPtrAt(1) + ? "xref\r\n" + : "xref\r\n0 1\r\n0000000000 65535 f\r\n"; + if (m_File.AppendString(str) < 0) { + return -1; + } + m_Pos = (void*)(uintptr_t)1; + m_iStage = 81; + } else { + if (m_File.AppendString("xref\r\n") < 0) { + return -1; + } + m_Pos = (void*)(uintptr_t)0; + m_iStage = 82; + } + } else { + m_iStage = 90; + } + } + if (m_iStage == 81) { + CFX_ByteString str; + FX_DWORD i = (FX_DWORD)(uintptr_t)m_Pos, j; + while (i <= dwLastObjNum) { + while (i <= dwLastObjNum && !m_ObjectOffset.GetPtrAt(i)) { + i++; + } + if (i > dwLastObjNum) { + break; + } + j = i; + while (j <= dwLastObjNum && m_ObjectOffset.GetPtrAt(j)) { + j++; + } + if (i == 1) { + str.Format("0 %d\r\n0000000000 65535 f\r\n", j); + } else { + str.Format("%d %d\r\n", i, j - i); + } + if (m_File.AppendBlock(str.c_str(), str.GetLength()) < 0) { + return -1; + } + while (i < j) { + str.Format("%010d 00000 n\r\n", m_ObjectOffset[i++]); + if (m_File.AppendBlock(str.c_str(), str.GetLength()) < 0) { + return -1; + } + } + if (i > dwLastObjNum) { + break; + } + if (pPause && pPause->NeedToPauseNow()) { + m_Pos = (void*)(uintptr_t)i; + return 1; + } + } + m_iStage = 90; + } + if (m_iStage == 82) { + CFX_ByteString str; + int32_t iCount = m_NewObjNumArray.GetSize(); + int32_t i = (int32_t)(uintptr_t)m_Pos; + while (i < iCount) { + int32_t j = i; + FX_DWORD objnum = m_NewObjNumArray.ElementAt(i); + while (j < iCount) { + if (++j == iCount) { + break; + } + FX_DWORD dwCurrent = m_NewObjNumArray.ElementAt(j); + if (dwCurrent - objnum > 1) { + break; + } + objnum = dwCurrent; + } + objnum = m_NewObjNumArray.ElementAt(i); + if (objnum == 1) { + str.Format("0 %d\r\n0000000000 65535 f\r\n", j - i + 1); + } else { + str.Format("%d %d\r\n", objnum, j - i); + } + if (m_File.AppendBlock(str.c_str(), str.GetLength()) < 0) { + return -1; + } + while (i < j) { + objnum = m_NewObjNumArray.ElementAt(i++); + str.Format("%010d 00000 n\r\n", m_ObjectOffset[objnum]); + if (m_File.AppendBlock(str.c_str(), str.GetLength()) < 0) { + return -1; + } + } + if (pPause && (i % 100) == 0 && pPause->NeedToPauseNow()) { + m_Pos = (void*)(uintptr_t)i; + return 1; + } + } + m_iStage = 90; + } + return m_iStage; +} + +int32_t CPDF_Creator::WriteDoc_Stage4(IFX_Pause* pPause) { + FXSYS_assert(m_iStage >= 90); + if ((m_dwFlags & FPDFCREATE_OBJECTSTREAM) == 0) { + FX_BOOL bXRefStream = + (m_dwFlags & FPDFCREATE_INCREMENTAL) != 0 && m_pParser->IsXRefStream(); + if (!bXRefStream) { + if (m_File.AppendString("trailer\r\n<<") < 0) { + return -1; + } + } else { + if (m_File.AppendDWord(m_pDocument->m_LastObjNum + 1) < 0) { + return -1; + } + if (m_File.AppendString(" 0 obj <<") < 0) { + return -1; + } + } + if (m_pParser) { + CPDF_Dictionary* p = m_pParser->GetTrailer(); + for (const auto& it : *p) { + const CFX_ByteString& key = it.first; + CPDF_Object* pValue = it.second; + // TODO(ochang): Consolidate with similar check in + // PDF_CreatorWriteTrailer. + if (key == "Encrypt" || key == "Size" || key == "Filter" || + key == "Index" || key == "Length" || key == "Prev" || key == "W" || + key == "XRefStm" || key == "ID") { + continue; + } + if (m_File.AppendString(("/")) < 0) { + return -1; + } + if (m_File.AppendString(PDF_NameEncode(key)) < 0) { + return -1; + } + if (pValue->GetObjNum()) { + if (m_File.AppendString(" ") < 0) { + return -1; + } + if (m_File.AppendDWord(pValue->GetObjNum()) < 0) { + return -1; + } + if (m_File.AppendString(" 0 R ") < 0) { + return -1; + } + } else { + FX_FILESIZE offset = 0; + if (PDF_CreatorAppendObject(pValue, &m_File, offset) < 0) { + return -1; + } + } + } + } else { + if (m_File.AppendString("\r\n/Root ") < 0) { + return -1; + } + if (m_File.AppendDWord(m_pDocument->m_pRootDict->GetObjNum()) < 0) { + return -1; + } + if (m_File.AppendString(" 0 R\r\n") < 0) { + return -1; + } + if (m_pDocument->m_pInfoDict) { + if (m_File.AppendString("/Info ") < 0) { + return -1; + } + if (m_File.AppendDWord(m_pDocument->m_pInfoDict->GetObjNum()) < 0) { + return -1; + } + if (m_File.AppendString(" 0 R\r\n") < 0) { + return -1; + } + } + } + if (m_pEncryptDict) { + if (m_File.AppendString("/Encrypt") < 0) { + return -1; + } + FX_DWORD dwObjNum = m_pEncryptDict->GetObjNum(); + if (dwObjNum == 0) { + dwObjNum = m_pDocument->GetLastObjNum() + 1; + } + if (m_File.AppendString(" ") < 0) { + return -1; + } + if (m_File.AppendDWord(dwObjNum) < 0) { + return -1; + } + if (m_File.AppendString(" 0 R ") < 0) { + return -1; + } + } + if (m_File.AppendString("/Size ") < 0) { + return -1; + } + if (m_File.AppendDWord(m_dwLastObjNum + (bXRefStream ? 2 : 1)) < 0) { + return -1; + } + if ((m_dwFlags & FPDFCREATE_INCREMENTAL) != 0) { + FX_FILESIZE prev = m_pParser->GetLastXRefOffset(); + if (prev) { + if (m_File.AppendString("/Prev ") < 0) { + return -1; + } + FX_CHAR offset_buf[20]; + FXSYS_memset(offset_buf, 0, sizeof(offset_buf)); + FXSYS_i64toa(prev, offset_buf, 10); + if (m_File.AppendBlock(offset_buf, FXSYS_strlen(offset_buf)) < 0) { + return -1; + } + } + } + if (m_pIDArray) { + if (m_File.AppendString(("/ID")) < 0) { + return -1; + } + FX_FILESIZE offset = 0; + if (PDF_CreatorAppendObject(m_pIDArray, &m_File, offset) < 0) { + return -1; + } + } + if (!bXRefStream) { + if (m_File.AppendString(">>") < 0) { + return -1; + } + } else { + if (m_File.AppendString("/W[0 4 1]/Index[") < 0) { + return -1; + } + if ((m_dwFlags & FPDFCREATE_INCREMENTAL) != 0 && m_pParser && + m_pParser->GetLastXRefOffset() == 0) { + FX_DWORD i = 0; + for (i = 0; i < m_dwLastObjNum; i++) { + if (!m_ObjectOffset.GetPtrAt(i)) { + continue; + } + if (m_File.AppendDWord(i) < 0) { + return -1; + } + if (m_File.AppendString(" 1 ") < 0) { + return -1; + } + } + if (m_File.AppendString("]/Length ") < 0) { + return -1; + } + if (m_File.AppendDWord(m_dwLastObjNum * 5) < 0) { + return -1; + } + if (m_File.AppendString(">>stream\r\n") < 0) { + return -1; + } + for (i = 0; i < m_dwLastObjNum; i++) { + FX_FILESIZE* offset = m_ObjectOffset.GetPtrAt(i); + if (!offset) { + continue; + } + OutputIndex(&m_File, *offset); + } + } else { + int count = m_NewObjNumArray.GetSize(); + int32_t i = 0; + for (i = 0; i < count; i++) { + FX_DWORD objnum = m_NewObjNumArray.ElementAt(i); + if (m_File.AppendDWord(objnum) < 0) { + return -1; + } + if (m_File.AppendString(" 1 ") < 0) { + return -1; + } + } + if (m_File.AppendString("]/Length ") < 0) { + return -1; + } + if (m_File.AppendDWord(count * 5) < 0) { + return -1; + } + if (m_File.AppendString(">>stream\r\n") < 0) { + return -1; + } + for (i = 0; i < count; i++) { + FX_DWORD objnum = m_NewObjNumArray.ElementAt(i); + FX_FILESIZE offset = m_ObjectOffset[objnum]; + OutputIndex(&m_File, offset); + } + } + if (m_File.AppendString("\r\nendstream") < 0) { + return -1; + } + } + } + if (m_File.AppendString("\r\nstartxref\r\n") < 0) { + return -1; + } + FX_CHAR offset_buf[20]; + FXSYS_memset(offset_buf, 0, sizeof(offset_buf)); + FXSYS_i64toa(m_XrefStart, offset_buf, 10); + if (m_File.AppendBlock(offset_buf, FXSYS_strlen(offset_buf)) < 0) { + return -1; + } + if (m_File.AppendString("\r\n%%EOF\r\n") < 0) { + return -1; + } + m_File.Flush(); + return m_iStage = 100; +} +void CPDF_Creator::Clear() { + delete m_pXRefStream; + m_pXRefStream = NULL; + m_File.Clear(); + m_NewObjNumArray.RemoveAll(); + if (m_pIDArray) { + m_pIDArray->Release(); + m_pIDArray = NULL; + } +} + +bool CPDF_Creator::Create(IFX_StreamWrite* pFile, FX_DWORD flags) { + m_File.AttachFile(pFile); + return Create(flags); +} + +bool CPDF_Creator::Create(FX_DWORD flags) { + m_dwFlags = flags; + m_iStage = 0; + m_Offset = 0; + m_dwLastObjNum = m_pDocument->GetLastObjNum(); + m_ObjectOffset.Clear(); + m_NewObjNumArray.RemoveAll(); + InitID(); + if (flags & FPDFCREATE_PROGRESSIVE) { + return true; + } + return Continue(NULL) > -1; +} +void CPDF_Creator::InitID(FX_BOOL bDefault) { + CPDF_Array* pOldIDArray = m_pParser ? m_pParser->GetIDArray() : NULL; + FX_BOOL bNewId = !m_pIDArray; + if (!m_pIDArray) { + m_pIDArray = new CPDF_Array; + CPDF_Object* pID1 = pOldIDArray ? pOldIDArray->GetElement(0) : NULL; + if (pID1) { + m_pIDArray->Add(pID1->Clone()); + } else { + std::vector buffer = + PDF_GenerateFileID((FX_DWORD)(uintptr_t) this, m_dwLastObjNum); + CFX_ByteStringC bsBuffer(buffer.data(), buffer.size()); + m_pIDArray->Add(new CPDF_String(bsBuffer, TRUE), m_pDocument); + } + } + if (!bDefault) { + return; + } + if (pOldIDArray) { + CPDF_Object* pID2 = pOldIDArray->GetElement(1); + if ((m_dwFlags & FPDFCREATE_INCREMENTAL) && m_pEncryptDict && pID2) { + m_pIDArray->Add(pID2->Clone()); + return; + } + std::vector buffer = + PDF_GenerateFileID((FX_DWORD)(uintptr_t) this, m_dwLastObjNum); + CFX_ByteStringC bsBuffer(buffer.data(), buffer.size()); + m_pIDArray->Add(new CPDF_String(bsBuffer, TRUE), m_pDocument); + return; + } + m_pIDArray->Add(m_pIDArray->GetElement(0)->Clone()); + if (m_pEncryptDict && !pOldIDArray && m_pParser && bNewId) { + if (m_pEncryptDict->GetStringBy("Filter") == "Standard") { + CPDF_StandardSecurityHandler handler; + CFX_ByteString user_pass = m_pParser->GetPassword(); + FX_DWORD flag = PDF_ENCRYPT_CONTENT; + handler.OnCreate(m_pEncryptDict, m_pIDArray, (const uint8_t*)user_pass, + user_pass.GetLength(), flag); + if (m_bNewCrypto) { + delete m_pCryptoHandler; + } + m_pCryptoHandler = new CPDF_StandardCryptoHandler; + m_pCryptoHandler->Init(m_pEncryptDict, &handler); + m_bNewCrypto = TRUE; + m_bSecurityChanged = TRUE; + } + } +} +int32_t CPDF_Creator::Continue(IFX_Pause* pPause) { + if (m_iStage < 0) { + return m_iStage; + } + int32_t iRet = 0; + while (m_iStage < 100) { + if (m_iStage < 20) { + iRet = WriteDoc_Stage1(pPause); + } else if (m_iStage < 30) { + iRet = WriteDoc_Stage2(pPause); + } else if (m_iStage < 90) { + iRet = WriteDoc_Stage3(pPause); + } else { + iRet = WriteDoc_Stage4(pPause); + } + if (iRet < m_iStage) { + break; + } + } + if (iRet < 1 || m_iStage == 100) { + m_iStage = -1; + Clear(); + return iRet > 99 ? 0 : (iRet < 1 ? -1 : iRet); + } + return m_iStage; +} +FX_BOOL CPDF_Creator::SetFileVersion(int32_t fileVersion) { + if (fileVersion < 10 || fileVersion > 17) { + return FALSE; + } + m_FileVersion = fileVersion; + return TRUE; +} +void CPDF_Creator::RemoveSecurity() { + ResetStandardSecurity(); + m_bSecurityChanged = TRUE; + m_pEncryptDict = NULL; + m_pCryptoHandler = NULL; +} +void CPDF_Creator::ResetStandardSecurity() { + if (m_bStandardSecurity || m_bNewCrypto) { + delete m_pCryptoHandler; + m_pCryptoHandler = NULL; + } + m_bNewCrypto = FALSE; + if (!m_bStandardSecurity) { + return; + } + if (m_pEncryptDict) { + m_pEncryptDict->Release(); + m_pEncryptDict = NULL; + } + m_bStandardSecurity = FALSE; +} -- cgit v1.2.3