summaryrefslogtreecommitdiff
path: root/core/fpdfapi/edit/fpdf_edit_create.cpp
diff options
context:
space:
mode:
authorDan Sinclair <dsinclair@chromium.org>2017-05-09 12:36:08 -0400
committerChromium commit bot <commit-bot@chromium.org>2017-05-09 17:48:44 +0000
commita0b1954794b7a8a279ba00787628cc7f8d1419d9 (patch)
tree4556b78b9550dbb80baa52e0eee016395f223c4e /core/fpdfapi/edit/fpdf_edit_create.cpp
parente02b2bc896b0872ce1a9d0997946b07dc76b9f7b (diff)
downloadpdfium-a0b1954794b7a8a279ba00787628cc7f8d1419d9.tar.xz
Rename fpdf_edit_create.cpp to cpdf_creator.cpp
This Cl renames the fpdf_edit_create file to better match the cpdf_creator content. The CPDF_ObjectStream and CPDF_XRefStream code is moved out to their own .cpp files. Needed anonymous classes are split out as well. Change-Id: Ic83fb319ce28c816db82a4286e22032dc68811bc Reviewed-on: https://pdfium-review.googlesource.com/5171 Commit-Queue: dsinclair <dsinclair@chromium.org> Reviewed-by: Tom Sepez <tsepez@chromium.org> Reviewed-by: Nicolás Peña <npm@chromium.org>
Diffstat (limited to 'core/fpdfapi/edit/fpdf_edit_create.cpp')
-rw-r--r--core/fpdfapi/edit/fpdf_edit_create.cpp1638
1 files changed, 0 insertions, 1638 deletions
diff --git a/core/fpdfapi/edit/fpdf_edit_create.cpp b/core/fpdfapi/edit/fpdf_edit_create.cpp
deleted file mode 100644
index 42fe13d4e9..0000000000
--- a/core/fpdfapi/edit/fpdf_edit_create.cpp
+++ /dev/null
@@ -1,1638 +0,0 @@
-// 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 <memory>
-#include <vector>
-
-#include "core/fpdfapi/edit/cpdf_creator.h"
-#include "core/fpdfapi/edit/cpdf_objectstream.h"
-#include "core/fpdfapi/edit/cpdf_xrefstream.h"
-#include "core/fpdfapi/parser/cpdf_array.h"
-#include "core/fpdfapi/parser/cpdf_crypto_handler.h"
-#include "core/fpdfapi/parser/cpdf_dictionary.h"
-#include "core/fpdfapi/parser/cpdf_document.h"
-#include "core/fpdfapi/parser/cpdf_name.h"
-#include "core/fpdfapi/parser/cpdf_number.h"
-#include "core/fpdfapi/parser/cpdf_parser.h"
-#include "core/fpdfapi/parser/cpdf_reference.h"
-#include "core/fpdfapi/parser/cpdf_security_handler.h"
-#include "core/fpdfapi/parser/cpdf_stream.h"
-#include "core/fpdfapi/parser/cpdf_stream_acc.h"
-#include "core/fpdfapi/parser/cpdf_string.h"
-#include "core/fpdfapi/parser/fpdf_parser_decode.h"
-#include "core/fpdfapi/parser/fpdf_parser_utility.h"
-#include "core/fxcrt/cfx_maybe_owned.h"
-#include "core/fxcrt/fx_extension.h"
-#include "third_party/base/numerics/safe_math.h"
-#include "third_party/base/ptr_util.h"
-#include "third_party/base/stl_util.h"
-
-namespace {
-
-const uint32_t kXRefStreamMaxSize = 10000;
-const int kObjectStreamMaxLength = 256 * 1024;
-
-int32_t WriteTrailer(CPDF_Document* pDocument,
- CFX_FileBufferArchive* pFile,
- CPDF_Array* pIDArray) {
- FX_FILESIZE offset = 0;
- int32_t len = 0;
- CPDF_Parser* pParser = 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.get();
- if (key == "Encrypt" || key == "Size" || key == "Filter" ||
- key == "Index" || key == "Length" || key == "Prev" || key == "W" ||
- key == "XRefStm" || key == "Type" || key == "ID") {
- continue;
- }
- if (key == "DecodeParms")
- continue;
- if (pFile->AppendString(("/")) < 0)
- return -1;
- if ((len = pFile->AppendString(PDF_NameEncode(key).AsStringC())) < 0)
- return -1;
-
- offset += len + 1;
- if (!pValue->IsInline()) {
- 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 (!pValue->WriteTo(pFile, &offset))
- return -1;
- }
- }
- if (pIDArray) {
- if (pFile->AppendString(("/ID")) < 0)
- return -1;
-
- offset += 3;
- if (!pIDArray->WriteTo(pFile, &offset))
- 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 (!pIDArray->WriteTo(pFile, &offset))
- return -1;
- }
- return offset;
-}
-
-int32_t WriteEncryptDictObjectReference(uint32_t dwObjNum,
- CFX_FileBufferArchive* pFile) {
- 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<uint8_t> GenerateFileID(uint32_t dwSeed1, uint32_t dwSeed2) {
- std::vector<uint8_t> buffer(sizeof(uint32_t) * 4);
- uint32_t* pBuffer = reinterpret_cast<uint32_t*>(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(static_cast<uint8_t>(offset >> 24));
- buffer.AppendByte(static_cast<uint8_t>(offset >> 16));
- buffer.AppendByte(static_cast<uint8_t>(offset >> 8));
- buffer.AppendByte(static_cast<uint8_t>(offset));
- buffer.AppendByte(0);
- buffer.AppendByte(0);
-}
-
-void AppendIndex2(CFX_ByteTextBuf& buffer, uint32_t objnum, int32_t index) {
- buffer.AppendByte(2);
- buffer.AppendByte(static_cast<uint8_t>(objnum >> 24));
- buffer.AppendByte(static_cast<uint8_t>(objnum >> 16));
- buffer.AppendByte(static_cast<uint8_t>(objnum >> 8));
- buffer.AppendByte(static_cast<uint8_t>(objnum));
- buffer.AppendByte(static_cast<uint8_t>(index >> 8));
- buffer.AppendByte(static_cast<uint8_t>(index));
-}
-
-int32_t OutputIndex(CFX_FileBufferArchive* pFile, FX_FILESIZE offset) {
- if (pFile->AppendByte(static_cast<uint8_t>(offset >> 24)) < 0)
- return -1;
- if (pFile->AppendByte(static_cast<uint8_t>(offset >> 16)) < 0)
- return -1;
- if (pFile->AppendByte(static_cast<uint8_t>(offset >> 8)) < 0)
- return -1;
- if (pFile->AppendByte(static_cast<uint8_t>(offset)) < 0)
- return -1;
- if (pFile->AppendByte(0) < 0)
- return -1;
- return 0;
-}
-
-class CPDF_FlateEncoder {
- public:
- CPDF_FlateEncoder(CPDF_Stream* pStream, bool bFlateEncode);
- CPDF_FlateEncoder(const uint8_t* pBuffer,
- uint32_t size,
- bool bFlateEncode,
- bool bXRefStream);
- ~CPDF_FlateEncoder();
-
- void CloneDict();
-
- uint32_t m_dwSize;
- CFX_MaybeOwned<uint8_t, FxFreeDeleter> m_pData;
- CFX_MaybeOwned<CPDF_Dictionary> m_pDict;
- CFX_RetainPtr<CPDF_StreamAcc> m_pAcc;
-};
-
-CPDF_FlateEncoder::CPDF_FlateEncoder(CPDF_Stream* pStream, bool bFlateEncode)
- : m_dwSize(0), m_pAcc(pdfium::MakeRetain<CPDF_StreamAcc>(pStream)) {
- m_pAcc->LoadAllData(true);
- bool bHasFilter = pStream && pStream->HasFilter();
- if (bHasFilter && !bFlateEncode) {
- auto pDestAcc = pdfium::MakeRetain<CPDF_StreamAcc>(pStream);
- pDestAcc->LoadAllData();
- m_dwSize = pDestAcc->GetSize();
- m_pData = pDestAcc->DetachData();
- m_pDict = ToDictionary(pStream->GetDict()->Clone());
- m_pDict->RemoveFor("Filter");
- return;
- }
- if (bHasFilter || !bFlateEncode) {
- m_pData = const_cast<uint8_t*>(m_pAcc->GetData());
- m_dwSize = m_pAcc->GetSize();
- m_pDict = pStream->GetDict();
- return;
- }
- // TODO(thestig): Move to Init() and check return value.
- uint8_t* buffer = nullptr;
- ::FlateEncode(m_pAcc->GetData(), m_pAcc->GetSize(), &buffer, &m_dwSize);
- m_pData = std::unique_ptr<uint8_t, FxFreeDeleter>(buffer);
- m_pDict = ToDictionary(pStream->GetDict()->Clone());
- m_pDict->SetNewFor<CPDF_Number>("Length", static_cast<int>(m_dwSize));
- m_pDict->SetNewFor<CPDF_Name>("Filter", "FlateDecode");
- m_pDict->RemoveFor("DecodeParms");
-}
-
-CPDF_FlateEncoder::CPDF_FlateEncoder(const uint8_t* pBuffer,
- uint32_t size,
- bool bFlateEncode,
- bool bXRefStream)
- : m_dwSize(0) {
- if (!bFlateEncode) {
- m_pData = const_cast<uint8_t*>(pBuffer);
- m_dwSize = size;
- return;
- }
- uint8_t* buffer = nullptr;
- // TODO(thestig): Move to Init() and check return value.
- if (bXRefStream)
- ::PngEncode(pBuffer, size, &buffer, &m_dwSize);
- else
- ::FlateEncode(pBuffer, size, &buffer, &m_dwSize);
- m_pData = std::unique_ptr<uint8_t, FxFreeDeleter>(buffer);
-}
-
-CPDF_FlateEncoder::~CPDF_FlateEncoder() {}
-
-void CPDF_FlateEncoder::CloneDict() {
- if (m_pDict.IsOwned())
- return;
-
- m_pDict = ToDictionary(m_pDict->Clone());
- ASSERT(m_pDict.IsOwned());
-}
-
-class CPDF_Encryptor {
- public:
- CPDF_Encryptor(CPDF_CryptoHandler* pHandler,
- int objnum,
- uint8_t* src_data,
- uint32_t src_size);
- ~CPDF_Encryptor();
-
- uint8_t* m_pData;
- uint32_t m_dwSize;
- bool m_bNewBuf;
-};
-
-CPDF_Encryptor::CPDF_Encryptor(CPDF_CryptoHandler* pHandler,
- int objnum,
- uint8_t* src_data,
- uint32_t src_size)
- : m_pData(nullptr), m_dwSize(0), m_bNewBuf(false) {
- if (src_size == 0)
- return;
-
- if (!pHandler) {
- m_pData = (uint8_t*)src_data;
- m_dwSize = src_size;
- return;
- }
- 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;
-}
-
-CPDF_Encryptor::~CPDF_Encryptor() {
- if (m_bNewBuf)
- FX_Free(m_pData);
-}
-
-} // namespace
-
-CPDF_ObjectStream::CPDF_ObjectStream() : m_dwObjNum(0), m_index(0) {}
-
-CPDF_ObjectStream::~CPDF_ObjectStream() {}
-
-void CPDF_ObjectStream::Start() {
- m_Items.clear();
- m_Buffer.Clear();
- m_dwObjNum = 0;
- m_index = 0;
-}
-
-void CPDF_ObjectStream::CompressIndirectObject(uint32_t dwObjNum,
- const CPDF_Object* pObj) {
- m_Items.push_back({dwObjNum, m_Buffer.GetLength()});
- m_Buffer << pObj;
-}
-
-void CPDF_ObjectStream::CompressIndirectObject(uint32_t dwObjNum,
- const uint8_t* pBuffer,
- uint32_t dwSize) {
- m_Items.push_back({dwObjNum, m_Buffer.GetLength()});
- m_Buffer.AppendBlock(pBuffer, dwSize);
-}
-
-FX_FILESIZE CPDF_ObjectStream::End(CPDF_Creator* pCreator) {
- ASSERT(pCreator);
- if (m_Items.empty())
- return 0;
-
- CFX_FileBufferArchive* pFile = pCreator->GetFile();
- FX_FILESIZE ObjOffset = pCreator->GetOffset();
- if (!m_dwObjNum)
- m_dwObjNum = pCreator->GetNextObjectNumber();
-
- CFX_ByteTextBuf tempBuffer;
- for (const auto& pair : m_Items)
- tempBuffer << pair.objnum << " " << pair.offset << " ";
-
- int32_t len = pFile->AppendDWord(m_dwObjNum);
- if (len < 0)
- return -1;
-
- pCreator->IncrementOffset(len);
- if ((len = pFile->AppendString(" 0 obj\r\n<</Type /ObjStm /N ")) < 0)
- return -1;
-
- pCreator->IncrementOffset(len);
- uint32_t iCount = pdfium::CollectionSize<uint32_t>(m_Items);
- if ((len = pFile->AppendDWord(iCount)) < 0)
- return -1;
-
- pCreator->IncrementOffset(len);
- if (pFile->AppendString("/First ") < 0)
- return -1;
- if ((len = pFile->AppendDWord((uint32_t)tempBuffer.GetLength())) < 0)
- return -1;
- if (pFile->AppendString("/Length ") < 0)
- return -1;
-
- pCreator->IncrementOffset(len + 15);
-
- tempBuffer << m_Buffer;
- CPDF_FlateEncoder encoder(tempBuffer.GetBuffer(), tempBuffer.GetLength(),
- true, false);
- CPDF_Encryptor encryptor(pCreator->GetCryptoHandler(), m_dwObjNum,
- encoder.m_pData.Get(), encoder.m_dwSize);
- if ((len = pFile->AppendDWord(encryptor.m_dwSize)) < 0)
- return -1;
-
- pCreator->IncrementOffset(len);
- if (pFile->AppendString("/Filter /FlateDecode") < 0)
- return -1;
-
- pCreator->IncrementOffset(20);
- if ((len = pFile->AppendString(">>stream\r\n")) < 0)
- return -1;
- if (pFile->AppendBlock(encryptor.m_pData, encryptor.m_dwSize) < 0)
- return -1;
-
- pCreator->IncrementOffset(len + encryptor.m_dwSize);
- if ((len = pFile->AppendString("\r\nendstream\r\nendobj\r\n")) < 0)
- return -1;
-
- pCreator->IncrementOffset(len);
- return ObjOffset;
-}
-
-CPDF_XRefStream::CPDF_XRefStream()
- : m_PrevOffset(0), m_dwTempObjNum(0), m_iSeg(0) {}
-
-CPDF_XRefStream::~CPDF_XRefStream() {}
-
-bool CPDF_XRefStream::Start() {
- m_IndexArray.clear();
- m_Buffer.Clear();
- m_iSeg = 0;
- return true;
-}
-
-int32_t CPDF_XRefStream::CompressIndirectObject(uint32_t dwObjNum,
- const CPDF_Object* pObj,
- CPDF_Creator* pCreator) {
- if (!pCreator)
- return 0;
-
- m_ObjStream.CompressIndirectObject(dwObjNum, pObj);
- if (pdfium::CollectionSize<int32_t>(m_ObjStream.m_Items) <
- pCreator->GetObjectStreamSize() &&
- m_ObjStream.m_Buffer.GetLength() < kObjectStreamMaxLength) {
- return 1;
- }
- return EndObjectStream(pCreator, true);
-}
-
-int32_t CPDF_XRefStream::CompressIndirectObject(uint32_t dwObjNum,
- const uint8_t* pBuffer,
- uint32_t dwSize,
- CPDF_Creator* pCreator) {
- if (!pCreator)
- return 0;
-
- m_ObjStream.CompressIndirectObject(dwObjNum, pBuffer, dwSize);
- if (pdfium::CollectionSize<int32_t>(m_ObjStream.m_Items) <
- pCreator->GetObjectStreamSize() &&
- m_ObjStream.m_Buffer.GetLength() < kObjectStreamMaxLength) {
- return 1;
- }
- return EndObjectStream(pCreator, true);
-}
-
-int32_t CPDF_XRefStream::EndObjectStream(CPDF_Creator* pCreator, bool bEOF) {
- FX_FILESIZE objOffset = 0;
- if (bEOF) {
- objOffset = m_ObjStream.End(pCreator);
- if (objOffset < 0)
- return -1;
- }
-
- uint32_t& dwObjStmNum = m_ObjStream.m_dwObjNum;
- if (!dwObjStmNum)
- dwObjStmNum = pCreator->GetNextObjectNumber();
-
- int32_t iSize = pdfium::CollectionSize<int32_t>(m_ObjStream.m_Items);
- size_t iSeg = m_IndexArray.size();
- if (!pCreator->IsIncremental()) {
- if (m_dwTempObjNum == 0) {
- AppendIndex0(m_Buffer, true);
- m_dwTempObjNum++;
- }
- uint32_t end_num = m_IndexArray.back().objnum + m_IndexArray.back().count;
- int index = 0;
- for (; m_dwTempObjNum < end_num; m_dwTempObjNum++) {
- if (pCreator->HasObjectNumber(m_dwTempObjNum)) {
- if (index >= iSize ||
- m_dwTempObjNum != m_ObjStream.m_Items[index].objnum) {
- AppendIndex1(m_Buffer, pCreator->GetObjectOffset(m_dwTempObjNum));
- } else {
- AppendIndex2(m_Buffer, dwObjStmNum, index++);
- }
- } else {
- AppendIndex0(m_Buffer, false);
- }
- }
- if (iSize > 0 && bEOF)
- pCreator->SetObjectOffset(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 (uint32_t m = it->objnum; m < it->objnum + it->count; ++m) {
- if (m_ObjStream.m_index >= iSize ||
- m != m_ObjStream.m_Items[it - m_IndexArray.begin()].objnum) {
- AppendIndex1(m_Buffer, pCreator->GetObjectOffset(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;
-}
-
-bool CPDF_XRefStream::GenerateXRefStream(CPDF_Creator* pCreator, bool bEOF) {
- FX_FILESIZE offset_tmp = pCreator->GetOffset();
- uint32_t objnum = pCreator->GetNextObjectNumber();
- CFX_FileBufferArchive* pFile = pCreator->GetFile();
- if (pCreator->IsIncremental()) {
- AddObjectNumberToIndexArray(objnum);
- } else {
- for (; m_dwTempObjNum < pCreator->GetLastObjectNumber(); m_dwTempObjNum++) {
- if (pCreator->HasObjectNumber(m_dwTempObjNum))
- AppendIndex1(m_Buffer, pCreator->GetObjectOffset(m_dwTempObjNum));
- else
- AppendIndex0(m_Buffer, false);
- }
- }
-
- AppendIndex1(m_Buffer, offset_tmp);
-
- int32_t len = pFile->AppendDWord(objnum);
- if (len < 0)
- return false;
-
- pCreator->IncrementOffset(len);
- if ((len = pFile->AppendString(" 0 obj\r\n<</Type /XRef/W[1 4 2]/Index[")) <
- 0) {
- return false;
- }
- pCreator->IncrementOffset(len);
- if (!pCreator->IsIncremental()) {
- if ((len = pFile->AppendDWord(0)) < 0)
- return false;
- if ((len = pFile->AppendString(" ")) < 0)
- return false;
-
- pCreator->IncrementOffset(len + 1);
- if ((len = pFile->AppendDWord(objnum + 1)) < 0)
- return false;
-
- pCreator->IncrementOffset(len);
- } else {
- for (const auto& pair : m_IndexArray) {
- if ((len = pFile->AppendDWord(pair.objnum)) < 0)
- return false;
- if (pFile->AppendString(" ") < 0)
- return false;
-
- pCreator->IncrementOffset(len + 1);
- if ((len = pFile->AppendDWord(pair.count)) < 0)
- return false;
- if (pFile->AppendString(" ") < 0)
- return false;
-
- pCreator->IncrementOffset(len + 1);
- }
- }
- if (pFile->AppendString("]/Size ") < 0)
- return false;
- if ((len = pFile->AppendDWord(objnum + 1)) < 0)
- return false;
-
- pCreator->IncrementOffset(len + 7);
- if (m_PrevOffset > 0) {
- if (pFile->AppendString("/Prev ") < 0)
- return false;
-
- char offset_buf[20];
- memset(offset_buf, 0, sizeof(offset_buf));
- FXSYS_i64toa(m_PrevOffset, offset_buf, 10);
- int32_t offset_len = (int32_t)FXSYS_strlen(offset_buf);
- if (pFile->AppendBlock(offset_buf, offset_len) < 0)
- return false;
-
- pCreator->IncrementOffset(offset_len + 6);
- }
-
- CPDF_FlateEncoder encoder(m_Buffer.GetBuffer(), m_Buffer.GetLength(), true,
- true);
- if (pFile->AppendString("/Filter /FlateDecode") < 0)
- return false;
-
- pCreator->IncrementOffset(20);
- if ((len = pFile->AppendString("/DecodeParms<</Columns 7/Predictor 12>>")) <
- 0) {
- return false;
- }
-
- pCreator->IncrementOffset(len);
- if (pFile->AppendString("/Length ") < 0)
- return false;
- if ((len = pFile->AppendDWord(encoder.m_dwSize)) < 0)
- return false;
-
- pCreator->IncrementOffset(len + 8);
- if (bEOF) {
- if ((len = WriteTrailer(pCreator->GetDocument(), pFile,
- pCreator->GetIDArray())) < 0) {
- return false;
- }
- pCreator->IncrementOffset(len);
-
- if (CPDF_Dictionary* encryptDict = pCreator->GetEncryptDict()) {
- uint32_t dwEncryptObjNum = encryptDict->GetObjNum();
- if (dwEncryptObjNum == 0)
- dwEncryptObjNum = pCreator->GetEncryptObjectNumber();
- if ((len = WriteEncryptDictObjectReference(dwEncryptObjNum, pFile)) < 0)
- return false;
- pCreator->IncrementOffset(len);
- }
- }
- if ((len = pFile->AppendString(">>stream\r\n")) < 0)
- return false;
-
- pCreator->IncrementOffset(len);
- if (pFile->AppendBlock(encoder.m_pData.Get(), encoder.m_dwSize) < 0)
- return false;
- if ((len = pFile->AppendString("\r\nendstream\r\nendobj\r\n")) < 0)
- return false;
-
- pCreator->IncrementOffset(encoder.m_dwSize + len);
- m_PrevOffset = offset_tmp;
- return true;
-}
-
-bool CPDF_XRefStream::End(CPDF_Creator* pCreator, bool bEOF) {
- if (EndObjectStream(pCreator, bEOF) < 0)
- return false;
- return GenerateXRefStream(pCreator, bEOF);
-}
-
-bool CPDF_XRefStream::EndXRefStream(CPDF_Creator* pCreator) {
- if (!pCreator->IsIncremental()) {
- AppendIndex0(m_Buffer, true);
- for (uint32_t i = 1; i < pCreator->GetLastObjectNumber() + 1; i++) {
- if (pCreator->HasObjectNumber(i))
- AppendIndex1(m_Buffer, pCreator->GetObjectOffset(i));
- else
- AppendIndex0(m_Buffer, false);
- }
- } else {
- for (const auto& pair : m_IndexArray) {
- for (uint32_t j = pair.objnum; j < pair.objnum + pair.count; ++j)
- AppendIndex1(m_Buffer, pCreator->GetObjectOffset(j));
- }
- }
- return GenerateXRefStream(pCreator, false);
-}
-
-void CPDF_XRefStream::AddObjectNumberToIndexArray(uint32_t objnum) {
- if (m_IndexArray.empty()) {
- m_IndexArray.push_back({objnum, 1});
- return;
- }
-
- uint32_t 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(pDoc->GetParser()),
- m_bSecurityChanged(false),
- m_pEncryptDict(m_pParser ? m_pParser->GetEncryptDict() : nullptr),
- m_dwEncryptObjNum(0),
- m_pCryptoHandler(m_pParser ? m_pParser->GetCryptoHandler() : nullptr),
- m_pMetadata(nullptr),
- m_ObjectStreamSize(200),
- m_dwLastObjNum(m_pDocument->GetLastObjNum()),
- m_Offset(0),
- m_SavedOffset(0),
- m_iStage(-1),
- m_dwFlags(0),
- m_CurObjNum(0),
- m_XrefStart(0),
- m_pIDArray(nullptr),
- m_FileVersion(0) {}
-
-CPDF_Creator::~CPDF_Creator() {
- Clear();
-}
-
-bool CPDF_Creator::IsXRefNeedEnd() {
- if (!IsIncremental())
- return false;
-
- uint32_t iCount = 0;
- for (const auto& pair : m_pXRefStream->m_IndexArray)
- iCount += pair.count;
-
- return iCount >= kXRefStreamMaxSize;
-}
-
-int32_t CPDF_Creator::WriteIndirectObjectToStream(const CPDF_Object* pObj) {
- if (!m_pXRefStream)
- return 1;
-
- uint32_t 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->GetStringFor("Type") == "XRef")
- return 0;
- return 1;
- }
-
- if (pDict) {
- if (pDict == m_pDocument->GetRoot() || pDict == m_pEncryptDict)
- return 1;
- if (pDict->IsSignatureDict())
- return 1;
- if (pDict->GetStringFor("Type") == "Page")
- return 1;
- }
-
- m_pXRefStream->AddObjectNumberToIndexArray(objnum);
- if (m_pXRefStream->CompressIndirectObject(objnum, pObj, this) < 0)
- return -1;
- if (!IsXRefNeedEnd())
- return 0;
- if (!m_pXRefStream->End(this, false))
- return -1;
- if (!m_pXRefStream->Start())
- return -1;
- return 0;
-}
-int32_t CPDF_Creator::WriteIndirectObjectToStream(uint32_t objnum,
- const uint8_t* pBuffer,
- uint32_t 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())
- return 0;
- if (!m_pXRefStream->End(this, false))
- return -1;
- if (!m_pXRefStream->Start())
- return -1;
- return 0;
-}
-
-int32_t CPDF_Creator::AppendObjectNumberToXRef(uint32_t objnum) {
- if (!m_pXRefStream)
- return 1;
-
- m_pXRefStream->AddObjectNumberToIndexArray(objnum);
- if (!IsXRefNeedEnd())
- return 0;
- if (!m_pXRefStream->End(this, false))
- return -1;
- if (!m_pXRefStream->Start())
- return -1;
- return 0;
-}
-
-int32_t CPDF_Creator::WriteStream(const CPDF_Object* pStream,
- uint32_t objnum,
- CPDF_CryptoHandler* pCrypto) {
- CPDF_FlateEncoder encoder(const_cast<CPDF_Stream*>(pStream->AsStream()),
- pStream != m_pMetadata);
- CPDF_Encryptor encryptor(pCrypto, objnum, encoder.m_pData.Get(),
- encoder.m_dwSize);
- if (static_cast<uint32_t>(encoder.m_pDict->GetIntegerFor("Length")) !=
- encryptor.m_dwSize) {
- encoder.CloneDict();
- encoder.m_pDict->SetNewFor<CPDF_Number>(
- "Length", static_cast<int>(encryptor.m_dwSize));
- }
- if (WriteDirectObj(objnum, encoder.m_pDict.Get()) < 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(uint32_t 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()) {
- CPDF_CryptoHandler* pHandler =
- pObj != m_pMetadata ? m_pCryptoHandler.Get() : nullptr;
- 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(uint32_t objnum,
- const CPDF_Object* pObj,
- bool bEncrypt) {
- switch (pObj->GetType()) {
- case CPDF_Object::BOOLEAN:
- case CPDF_Object::NAME:
- case CPDF_Object::NULLOBJ:
- case CPDF_Object::NUMBER:
- case CPDF_Object::REFERENCE:
- if (!pObj->WriteTo(&m_File, &m_Offset))
- return -1;
- break;
-
- case CPDF_Object::STRING: {
- CFX_ByteString str = pObj->GetString();
- bool bHex = pObj->AsString()->IsHex();
- if (!m_pCryptoHandler || !bEncrypt) {
- if (!pObj->WriteTo(&m_File, &m_Offset))
- return -1;
- break;
- }
- CPDF_Encryptor encryptor(m_pCryptoHandler.Get(), objnum,
- (uint8_t*)str.c_str(), str.GetLength());
- CFX_ByteString content = PDF_EncodeString(
- CFX_ByteString((const char*)encryptor.m_pData, encryptor.m_dwSize),
- bHex);
- int32_t len = m_File.AppendString(content.AsStringC());
- if (len < 0)
- return -1;
-
- m_Offset += len;
- break;
- }
- case CPDF_Object::STREAM: {
- CPDF_FlateEncoder encoder(const_cast<CPDF_Stream*>(pObj->AsStream()),
- true);
- CPDF_Encryptor encryptor(m_pCryptoHandler.Get(), objnum,
- encoder.m_pData.Get(), encoder.m_dwSize);
- if (static_cast<uint32_t>(encoder.m_pDict->GetIntegerFor("Length")) !=
- encryptor.m_dwSize) {
- encoder.CloneDict();
- encoder.m_pDict->SetNewFor<CPDF_Number>(
- "Length", static_cast<int>(encryptor.m_dwSize));
- }
- if (WriteDirectObj(objnum, encoder.m_pDict.Get()) < 0)
- return -1;
-
- int32_t 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;
- len = m_File.AppendString("\r\nendstream");
- if (len < 0)
- return -1;
-
- m_Offset += len;
- break;
- }
- case CPDF_Object::ARRAY: {
- if (m_File.AppendString("[") < 0)
- return -1;
-
- m_Offset += 1;
- const CPDF_Array* p = pObj->AsArray();
- for (size_t i = 0; i < p->GetCount(); i++) {
- CPDF_Object* pElement = p->GetObjectAt(i);
- if (!pElement->IsInline()) {
- if (m_File.AppendString(" ") < 0)
- return -1;
-
- int32_t len = m_File.AppendDWord(pElement->GetObjNum());
- if (len < 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) {
- if (!pObj->WriteTo(&m_File, &m_Offset))
- return -1;
- break;
- }
-
- 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) {
- bool bSignValue = false;
- const CFX_ByteString& key = it.first;
- CPDF_Object* pValue = it.second.get();
- if (m_File.AppendString("/") < 0)
- return -1;
-
- int32_t len = m_File.AppendString(PDF_NameEncode(key).AsStringC());
- if (len < 0)
- return -1;
-
- m_Offset += len + 1;
- if (bSignDict && key == "Contents")
- bSignValue = true;
-
- if (!pValue->IsInline()) {
- if (m_File.AppendString(" ") < 0)
- return -1;
-
- len = m_File.AppendDWord(pValue->GetObjNum());
- if (len < 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(uint32_t objnum) {
- if (m_pParser->IsObjectFreeOrNull(objnum))
- return 0;
-
- m_ObjectOffsets[objnum] = m_Offset;
- bool bExistInMap = !!m_pDocument->GetIndirectObject(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->GetOrParseIndirectObject(objnum);
- if (!pObj) {
- m_ObjectOffsets.erase(objnum);
- return 0;
- }
- if (WriteIndirectObj(pObj))
- return -1;
- if (!bExistInMap)
- m_pDocument->DeleteIndirectObject(objnum);
- } else {
- uint8_t* pBuffer;
- uint32_t 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() {
- uint32_t nLastObjNum = m_pParser->GetLastObjNum();
- if (!m_pParser->IsValidObjectNumber(nLastObjNum))
- return 0;
-
- for (uint32_t objnum = m_CurObjNum; objnum <= nLastObjNum; ++objnum) {
- int32_t iRet = WriteOldIndirectObject(objnum);
- if (iRet < 0)
- return iRet;
- if (!iRet)
- continue;
- }
- return 0;
-}
-
-int32_t CPDF_Creator::WriteNewObjs(bool bIncremental) {
- uint32_t iCount = pdfium::CollectionSize<uint32_t>(m_NewObjNumArray);
- uint32_t index = m_CurObjNum;
- while (index < iCount) {
- uint32_t objnum = m_NewObjNumArray[index];
- CPDF_Object* pObj = m_pDocument->GetIndirectObject(objnum);
- if (!pObj) {
- ++index;
- continue;
- }
- m_ObjectOffsets[objnum] = m_Offset;
- if (WriteIndirectObj(pObj))
- return -1;
-
- index++;
- }
- return 0;
-}
-
-void CPDF_Creator::InitOldObjNumOffsets() {
- if (!m_pParser)
- return;
-
- uint32_t j = 0;
- uint32_t dwStart = 0;
- uint32_t 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++;
-
- dwStart = j;
- }
-}
-
-void CPDF_Creator::InitNewObjNumOffsets() {
- for (const auto& pair : *m_pDocument) {
- const uint32_t objnum = pair.first;
- const CPDF_Object* pObj = pair.second.get();
- if (IsIncremental() || pObj->GetObjNum() == CPDF_Object::kInvalidObjNum)
- continue;
- if (m_pParser && m_pParser->IsValidObjectNumber(objnum) &&
- m_pParser->GetObjectType(objnum)) {
- continue;
- }
- m_NewObjNumArray.insert(std::lower_bound(m_NewObjNumArray.begin(),
- m_NewObjNumArray.end(), objnum),
- objnum);
- }
-}
-
-int32_t CPDF_Creator::WriteDoc_Stage1() {
- 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->GetDirectObjectFor("Metadata") : nullptr;
- if (m_dwFlags & FPDFCREATE_OBJECTSTREAM) {
- m_pXRefStream = pdfium::MakeUnique<CPDF_XRefStream>();
- m_pXRefStream->Start();
- if (IsIncremental() && m_pParser) {
- FX_FILESIZE prev = m_pParser->GetLastXRefOffset();
- m_pXRefStream->m_PrevOffset = prev;
- }
- }
- m_iStage = 10;
- }
- if (m_iStage == 10) {
- if (!IsIncremental()) {
- 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 {
- CFX_RetainPtr<IFX_SeekableReadStream> pSrcFile =
- m_pParser->GetFileAccess();
- m_Offset = pSrcFile->GetSize();
- m_SavedOffset = m_Offset;
- m_iStage = 15;
- }
- }
- if (m_iStage == 15) {
- if ((m_dwFlags & FPDFCREATE_NO_ORIGINAL) == 0 && m_SavedOffset > 0) {
- CFX_RetainPtr<IFX_SeekableReadStream> pSrcFile =
- m_pParser->GetFileAccess();
- std::vector<uint8_t> buffer(4096);
- FX_FILESIZE src_size = m_SavedOffset;
- while (src_size) {
- uint32_t block_size = src_size > 4096 ? 4096 : src_size;
- if (!pSrcFile->ReadBlock(buffer.data(), m_Offset - src_size,
- block_size)) {
- return -1;
- }
- if (m_File.AppendBlock(buffer.data(), block_size) < 0)
- return -1;
-
- src_size -= block_size;
- }
- }
- if ((m_dwFlags & FPDFCREATE_NO_ORIGINAL) == 0 &&
- m_pParser->GetLastXRefOffset() == 0) {
- InitOldObjNumOffsets();
- uint32_t dwEnd = m_pParser->GetLastObjNum();
- bool bObjStm = (m_dwFlags & FPDFCREATE_OBJECTSTREAM) != 0;
- for (uint32_t objnum = 0; objnum <= dwEnd; objnum++) {
- if (m_pParser->IsObjectFreeOrNull(objnum))
- continue;
-
- m_ObjectOffsets[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() {
- ASSERT(m_iStage >= 20 || m_iStage < 30);
- if (m_iStage == 20) {
- if (!IsIncremental() && m_pParser) {
- m_CurObjNum = 0;
- m_iStage = 21;
- } else {
- m_iStage = 25;
- }
- }
- if (m_iStage == 21) {
- int32_t iRet = WriteOldObjs();
- if (iRet)
- return iRet;
-
- m_iStage = 25;
- }
- if (m_iStage == 25) {
- m_CurObjNum = 0;
- m_iStage = 26;
- }
- if (m_iStage == 26) {
- int32_t iRet = WriteNewObjs(IsIncremental());
- if (iRet)
- return iRet;
-
- m_iStage = 27;
- }
- if (m_iStage == 27) {
- if (m_pEncryptDict && m_pEncryptDict->IsInline()) {
- m_dwLastObjNum += 1;
- FX_FILESIZE saveOffset = m_Offset;
- if (WriteIndirectObj(m_dwLastObjNum, m_pEncryptDict) < 0)
- return -1;
-
- m_ObjectOffsets[m_dwLastObjNum] = saveOffset;
- m_dwEncryptObjNum = m_dwLastObjNum;
- if (IsIncremental())
- m_NewObjNumArray.push_back(m_dwLastObjNum);
- }
- m_iStage = 80;
- }
- return m_iStage;
-}
-
-int32_t CPDF_Creator::WriteDoc_Stage3() {
- ASSERT(m_iStage >= 80 || m_iStage < 90);
- uint32_t 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 (!IsIncremental() || !m_pParser->IsXRefStream()) {
- if (!IsIncremental() || m_pParser->GetLastXRefOffset() == 0) {
- CFX_ByteString str;
- str = pdfium::ContainsKey(m_ObjectOffsets, 1)
- ? "xref\r\n"
- : "xref\r\n0 1\r\n0000000000 65535 f\r\n";
- if (m_File.AppendString(str.AsStringC()) < 0)
- return -1;
-
- m_CurObjNum = 1;
- m_iStage = 81;
- } else {
- if (m_File.AppendString("xref\r\n") < 0)
- return -1;
-
- m_CurObjNum = 0;
- m_iStage = 82;
- }
- } else {
- m_iStage = 90;
- }
- }
- if (m_iStage == 81) {
- CFX_ByteString str;
- uint32_t i = m_CurObjNum;
- uint32_t j;
- while (i <= dwLastObjNum) {
- while (i <= dwLastObjNum && !pdfium::ContainsKey(m_ObjectOffsets, i))
- i++;
-
- if (i > dwLastObjNum)
- break;
-
- j = i;
- while (j <= dwLastObjNum && pdfium::ContainsKey(m_ObjectOffsets, 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_ObjectOffsets[i++]);
- if (m_File.AppendBlock(str.c_str(), str.GetLength()) < 0)
- return -1;
- }
- if (i > dwLastObjNum)
- break;
- }
- m_iStage = 90;
- }
- if (m_iStage == 82) {
- CFX_ByteString str;
- uint32_t iCount = pdfium::CollectionSize<uint32_t>(m_NewObjNumArray);
- uint32_t i = m_CurObjNum;
- while (i < iCount) {
- size_t j = i;
- uint32_t objnum = m_NewObjNumArray[i];
- while (j < iCount) {
- if (++j == iCount)
- break;
- uint32_t dwCurrent = m_NewObjNumArray[j];
- if (dwCurrent - objnum > 1)
- break;
- objnum = dwCurrent;
- }
- objnum = m_NewObjNumArray[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[i++];
- str.Format("%010d 00000 n\r\n", m_ObjectOffsets[objnum]);
- if (m_File.AppendBlock(str.c_str(), str.GetLength()) < 0)
- return -1;
- }
- }
- m_iStage = 90;
- }
- return m_iStage;
-}
-
-int32_t CPDF_Creator::WriteDoc_Stage4() {
- ASSERT(m_iStage >= 90);
- if ((m_dwFlags & FPDFCREATE_OBJECTSTREAM) == 0) {
- bool bXRefStream = IsIncremental() && m_pParser->IsXRefStream();
- if (!bXRefStream) {
- if (m_File.AppendString("trailer\r\n<<") < 0)
- return -1;
- } else {
- if (m_File.AppendDWord(m_pDocument->GetLastObjNum() + 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.get();
- // TODO(ochang): Consolidate with similar check in WriteTrailer.
- 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).AsStringC()) < 0)
- return -1;
- if (!pValue->IsInline()) {
- 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 (!pValue->WriteTo(&m_File, &offset))
- return -1;
- }
- }
- } else {
- if (m_File.AppendString("\r\n/Root ") < 0)
- return -1;
- if (m_File.AppendDWord(m_pDocument->GetRoot()->GetObjNum()) < 0)
- return -1;
- if (m_File.AppendString(" 0 R\r\n") < 0)
- return -1;
- if (m_pDocument->GetInfo()) {
- if (m_File.AppendString("/Info ") < 0)
- return -1;
- if (m_File.AppendDWord(m_pDocument->GetInfo()->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;
-
- uint32_t 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 (IsIncremental()) {
- FX_FILESIZE prev = m_pParser->GetLastXRefOffset();
- if (prev) {
- if (m_File.AppendString("/Prev ") < 0)
- return -1;
-
- char offset_buf[20];
- 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 (!m_pIDArray->WriteTo(&m_File, &offset))
- 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 (IsIncremental() && m_pParser && m_pParser->GetLastXRefOffset() == 0) {
- uint32_t i = 0;
- for (i = 0; i < m_dwLastObjNum; i++) {
- if (!pdfium::ContainsKey(m_ObjectOffsets, 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++) {
- auto it = m_ObjectOffsets.find(i);
- if (it == m_ObjectOffsets.end())
- continue;
- OutputIndex(&m_File, it->second);
- }
- } else {
- size_t count = m_NewObjNumArray.size();
- size_t i = 0;
- for (i = 0; i < count; i++) {
- if (m_File.AppendDWord(m_NewObjNumArray[i]) < 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++)
- OutputIndex(&m_File, m_ObjectOffsets[m_NewObjNumArray[i]]);
- }
- if (m_File.AppendString("\r\nendstream") < 0)
- return -1;
- }
- }
-
- if (m_File.AppendString("\r\nstartxref\r\n") < 0)
- return -1;
-
- char offset_buf[20];
- 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() {
- m_pXRefStream.reset();
- m_File.Clear();
- m_NewObjNumArray.clear();
- m_pIDArray.reset();
-}
-
-bool CPDF_Creator::Create(const CFX_RetainPtr<IFX_WriteStream>& pFile,
- uint32_t flags) {
- m_File.AttachFile(pFile);
- return Create(flags);
-}
-
-bool CPDF_Creator::Create(uint32_t flags) {
- m_dwFlags = flags;
- m_iStage = 0;
- m_Offset = 0;
- m_dwLastObjNum = m_pDocument->GetLastObjNum();
- m_ObjectOffsets.clear();
- m_NewObjNumArray.clear();
- InitID();
- if (flags & FPDFCREATE_PROGRESSIVE)
- return true;
- return Continue() > -1;
-}
-
-void CPDF_Creator::InitID(bool bDefault) {
- CPDF_Array* pOldIDArray = m_pParser ? m_pParser->GetIDArray() : nullptr;
- bool bNewId = !m_pIDArray;
- if (bNewId) {
- m_pIDArray = pdfium::MakeUnique<CPDF_Array>();
- CPDF_Object* pID1 = pOldIDArray ? pOldIDArray->GetObjectAt(0) : nullptr;
- if (pID1) {
- m_pIDArray->Add(pID1->Clone());
- } else {
- std::vector<uint8_t> buffer =
- GenerateFileID((uint32_t)(uintptr_t)this, m_dwLastObjNum);
- CFX_ByteString bsBuffer(buffer.data(), buffer.size());
- m_pIDArray->AddNew<CPDF_String>(bsBuffer, true);
- }
- }
- if (!bDefault)
- return;
-
- if (pOldIDArray) {
- CPDF_Object* pID2 = pOldIDArray->GetObjectAt(1);
- if (IsIncremental() && m_pEncryptDict && pID2) {
- m_pIDArray->Add(pID2->Clone());
- return;
- }
- std::vector<uint8_t> buffer =
- GenerateFileID((uint32_t)(uintptr_t)this, m_dwLastObjNum);
- CFX_ByteString bsBuffer(buffer.data(), buffer.size());
- m_pIDArray->AddNew<CPDF_String>(bsBuffer, true);
- return;
- }
-
- m_pIDArray->Add(m_pIDArray->GetObjectAt(0)->Clone());
- if (m_pEncryptDict && !pOldIDArray && m_pParser && bNewId) {
- if (m_pEncryptDict->GetStringFor("Filter") == "Standard") {
- CFX_ByteString user_pass = m_pParser->GetPassword();
- uint32_t flag = PDF_ENCRYPT_CONTENT;
- CPDF_SecurityHandler handler;
- handler.OnCreate(m_pEncryptDict, m_pIDArray.get(), user_pass.raw_str(),
- user_pass.GetLength(), flag);
- m_pCryptoHandler = pdfium::MakeRetain<CPDF_CryptoHandler>();
- m_pCryptoHandler->Init(m_pEncryptDict, &handler);
- m_bSecurityChanged = true;
- }
- }
-}
-
-int32_t CPDF_Creator::Continue() {
- if (m_iStage < 0)
- return m_iStage;
-
- int32_t iRet = 0;
- while (m_iStage < 100) {
- if (m_iStage < 20)
- iRet = WriteDoc_Stage1();
- else if (m_iStage < 30)
- iRet = WriteDoc_Stage2();
- else if (m_iStage < 90)
- iRet = WriteDoc_Stage3();
- else
- iRet = WriteDoc_Stage4();
-
- 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;
-}
-
-bool CPDF_Creator::SetFileVersion(int32_t fileVersion) {
- if (fileVersion < 10 || fileVersion > 17)
- return false;
-
- m_FileVersion = fileVersion;
- return true;
-}
-
-void CPDF_Creator::IncrementOffset(FX_FILESIZE inc) {
- pdfium::base::CheckedNumeric<FX_FILESIZE> size = m_Offset;
- size += inc;
- m_Offset = size.ValueOrDie();
-}
-
-void CPDF_Creator::RemoveSecurity() {
- m_pCryptoHandler.Reset();
- m_bSecurityChanged = true;
- m_pEncryptDict = nullptr;
-}