| // 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/edit/cpdf_creator.h" |
| |
| #include <algorithm> |
| |
| #include "core/fpdfapi/edit/cpdf_encryptor.h" |
| #include "core/fpdfapi/edit/cpdf_flateencoder.h" |
| #include "core/fpdfapi/parser/cpdf_array.h" |
| #include "core/fpdfapi/parser/cpdf_crypto_handler.h" |
| #include "core/fpdfapi/parser/cpdf_document.h" |
| #include "core/fpdfapi/parser/cpdf_number.h" |
| #include "core/fpdfapi/parser/cpdf_parser.h" |
| #include "core/fpdfapi/parser/cpdf_security_handler.h" |
| #include "core/fpdfapi/parser/cpdf_string.h" |
| #include "core/fpdfapi/parser/fpdf_parser_decode.h" |
| #include "core/fxcrt/fx_extension.h" |
| #include "core/fxcrt/fx_random.h" |
| |
| namespace { |
| |
| const size_t kArchiveBufferSize = 32768; |
| |
| class CFX_FileBufferArchive : public IFX_ArchiveStream { |
| public: |
| explicit CFX_FileBufferArchive(const RetainPtr<IFX_WriteStream>& archive); |
| ~CFX_FileBufferArchive() override; |
| |
| bool WriteBlock(const void* pBuf, size_t size) override; |
| bool WriteByte(uint8_t byte) override; |
| bool WriteDWord(uint32_t i) override; |
| bool WriteString(const ByteStringView& str) override; |
| |
| FX_FILESIZE CurrentOffset() const override { return offset_; } |
| |
| private: |
| bool Flush(); |
| |
| FX_FILESIZE offset_; |
| size_t current_length_; |
| std::vector<uint8_t> buffer_; |
| RetainPtr<IFX_WriteStream> backing_file_; |
| }; |
| |
| CFX_FileBufferArchive::CFX_FileBufferArchive( |
| const RetainPtr<IFX_WriteStream>& file) |
| : offset_(0), |
| current_length_(0), |
| buffer_(kArchiveBufferSize), |
| backing_file_(file) { |
| ASSERT(file); |
| } |
| |
| CFX_FileBufferArchive::~CFX_FileBufferArchive() { |
| Flush(); |
| } |
| |
| bool CFX_FileBufferArchive::Flush() { |
| size_t nRemaining = current_length_; |
| current_length_ = 0; |
| if (!backing_file_) |
| return false; |
| if (!nRemaining) |
| return true; |
| return backing_file_->WriteBlock(buffer_.data(), nRemaining); |
| } |
| |
| bool CFX_FileBufferArchive::WriteBlock(const void* pBuf, size_t size) { |
| ASSERT(pBuf && size > 0); |
| |
| const uint8_t* buffer = reinterpret_cast<const uint8_t*>(pBuf); |
| size_t temp_size = size; |
| while (temp_size) { |
| size_t buf_size = std::min(kArchiveBufferSize - current_length_, temp_size); |
| memcpy(buffer_.data() + current_length_, buffer, buf_size); |
| |
| current_length_ += buf_size; |
| if (current_length_ == kArchiveBufferSize && !Flush()) |
| return false; |
| |
| temp_size -= buf_size; |
| buffer += buf_size; |
| } |
| |
| pdfium::base::CheckedNumeric<FX_FILESIZE> safe_offset = offset_; |
| safe_offset += size; |
| if (!safe_offset.IsValid()) |
| return false; |
| |
| offset_ = safe_offset.ValueOrDie(); |
| return true; |
| } |
| |
| bool CFX_FileBufferArchive::WriteByte(uint8_t byte) { |
| return WriteBlock(&byte, 1); |
| } |
| |
| bool CFX_FileBufferArchive::WriteDWord(uint32_t i) { |
| char buf[32]; |
| FXSYS_itoa(i, buf, 10); |
| return WriteBlock(buf, strlen(buf)); |
| } |
| |
| bool CFX_FileBufferArchive::WriteString(const ByteStringView& str) { |
| return WriteBlock(str.raw_str(), str.GetLength()); |
| } |
| |
| 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; |
| } |
| |
| int32_t OutputIndex(IFX_ArchiveStream* archive, FX_FILESIZE offset) { |
| if (!archive->WriteByte(static_cast<uint8_t>(offset >> 24)) || |
| !archive->WriteByte(static_cast<uint8_t>(offset >> 16)) || |
| !archive->WriteByte(static_cast<uint8_t>(offset >> 8)) || |
| !archive->WriteByte(static_cast<uint8_t>(offset)) || |
| !archive->WriteByte(0)) { |
| return -1; |
| } |
| return 0; |
| } |
| |
| } // namespace |
| |
| CPDF_Creator::CPDF_Creator(CPDF_Document* pDoc, |
| const RetainPtr<IFX_WriteStream>& archive) |
| : m_pDocument(pDoc), |
| m_pParser(pDoc->GetParser()), |
| m_bSecurityChanged(false), |
| m_pEncryptDict(m_pParser ? m_pParser->GetEncryptDict() : nullptr), |
| m_dwEncryptObjNum(0), |
| m_pSecurityHandler(m_pParser ? m_pParser->GetSecurityHandler() : nullptr), |
| m_pMetadata(nullptr), |
| m_dwLastObjNum(m_pDocument->GetLastObjNum()), |
| m_Archive(pdfium::MakeUnique<CFX_FileBufferArchive>(archive)), |
| 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() {} |
| |
| bool 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.GetData(), |
| encoder.GetSize()); |
| if (static_cast<uint32_t>(encoder.GetDict()->GetIntegerFor("Length")) != |
| encryptor.GetSize()) { |
| encoder.CloneDict(); |
| encoder.GetDict()->SetNewFor<CPDF_Number>( |
| "Length", static_cast<int>(encryptor.GetSize())); |
| } |
| |
| if (!WriteDirectObj(objnum, encoder.GetDict(), true) || |
| !m_Archive->WriteString("stream\r\n")) { |
| return false; |
| } |
| |
| // Allow for empty streams. |
| if (encryptor.GetSize() > 0 && |
| !m_Archive->WriteBlock(encryptor.GetData(), encryptor.GetSize())) { |
| return false; |
| } |
| |
| if (!m_Archive->WriteString("\r\nendstream")) |
| return false; |
| |
| return true; |
| } |
| |
| bool CPDF_Creator::WriteIndirectObj(uint32_t objnum, const CPDF_Object* pObj) { |
| if (!m_Archive->WriteDWord(objnum) || !m_Archive->WriteString(" 0 obj\r\n")) |
| return false; |
| |
| if (pObj->IsStream()) { |
| CPDF_CryptoHandler* pHandler = |
| pObj != m_pMetadata ? GetCryptoHandler() : nullptr; |
| if (!WriteStream(pObj, objnum, pHandler)) |
| return false; |
| } else if (!WriteDirectObj(objnum, pObj, true)) { |
| return false; |
| } |
| |
| return m_Archive->WriteString("\r\nendobj\r\n"); |
| } |
| |
| bool 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_Archive.get())) |
| return false; |
| break; |
| |
| case CPDF_Object::STRING: { |
| ByteString str = pObj->GetString(); |
| bool bHex = pObj->AsString()->IsHex(); |
| if (!GetCryptoHandler() || !bEncrypt) { |
| if (!pObj->WriteTo(m_Archive.get())) |
| return false; |
| break; |
| } |
| CPDF_Encryptor encryptor(GetCryptoHandler(), objnum, |
| (uint8_t*)str.c_str(), str.GetLength()); |
| ByteString content = PDF_EncodeString( |
| ByteString(encryptor.GetData(), encryptor.GetSize()), bHex); |
| if (!m_Archive->WriteString(content.AsStringView())) |
| return false; |
| break; |
| } |
| case CPDF_Object::STREAM: { |
| CPDF_FlateEncoder encoder(const_cast<CPDF_Stream*>(pObj->AsStream()), |
| true); |
| CPDF_Encryptor encryptor(GetCryptoHandler(), objnum, encoder.GetData(), |
| encoder.GetSize()); |
| if (static_cast<uint32_t>(encoder.GetDict()->GetIntegerFor("Length")) != |
| encryptor.GetSize()) { |
| encoder.CloneDict(); |
| encoder.GetDict()->SetNewFor<CPDF_Number>( |
| "Length", static_cast<int>(encryptor.GetSize())); |
| } |
| if (!WriteDirectObj(objnum, encoder.GetDict(), true) || |
| !m_Archive->WriteString("stream\r\n") || |
| !m_Archive->WriteBlock(encryptor.GetData(), encryptor.GetSize()) || |
| !m_Archive->WriteString("\r\nendstream")) { |
| return false; |
| } |
| |
| break; |
| } |
| case CPDF_Object::ARRAY: { |
| if (!m_Archive->WriteString("[")) |
| return false; |
| |
| 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_Archive->WriteString(" ") || |
| !m_Archive->WriteDWord(pElement->GetObjNum()) || |
| !m_Archive->WriteString(" 0 R")) { |
| return false; |
| } |
| } else if (!WriteDirectObj(objnum, pElement, true)) { |
| return false; |
| } |
| } |
| if (!m_Archive->WriteString("]")) |
| return false; |
| break; |
| } |
| case CPDF_Object::DICTIONARY: { |
| if (!GetCryptoHandler() || pObj == m_pEncryptDict) { |
| if (!pObj->WriteTo(m_Archive.get())) |
| return false; |
| break; |
| } |
| |
| if (!m_Archive->WriteString("<<")) |
| return false; |
| |
| const CPDF_Dictionary* p = pObj->AsDictionary(); |
| bool bSignDict = p->IsSignatureDict(); |
| for (const auto& it : *p) { |
| bool bSignValue = false; |
| const ByteString& key = it.first; |
| CPDF_Object* pValue = it.second.get(); |
| if (!m_Archive->WriteString("/") || |
| !m_Archive->WriteString(PDF_NameEncode(key).AsStringView())) { |
| return false; |
| } |
| |
| if (bSignDict && key == "Contents") |
| bSignValue = true; |
| if (!pValue->IsInline()) { |
| if (!m_Archive->WriteString(" ") || |
| !m_Archive->WriteDWord(pValue->GetObjNum()) || |
| !m_Archive->WriteString(" 0 R ")) { |
| return false; |
| } |
| } else if (!WriteDirectObj(objnum, pValue, !bSignValue)) { |
| return false; |
| } |
| } |
| if (!m_Archive->WriteString(">>")) |
| return false; |
| break; |
| } |
| } |
| return true; |
| } |
| |
| bool CPDF_Creator::WriteOldIndirectObject(uint32_t objnum) { |
| if (m_pParser->IsObjectFreeOrNull(objnum)) |
| return true; |
| |
| m_ObjectOffsets[objnum] = m_Archive->CurrentOffset(); |
| |
| bool bExistInMap = !!m_pDocument->GetIndirectObject(objnum); |
| CPDF_Object* pObj = m_pDocument->GetOrParseIndirectObject(objnum); |
| if (!pObj) { |
| m_ObjectOffsets.erase(objnum); |
| return true; |
| } |
| if (!WriteIndirectObj(pObj->GetObjNum(), pObj)) |
| return false; |
| if (!bExistInMap) |
| m_pDocument->DeleteIndirectObject(objnum); |
| return true; |
| } |
| |
| bool CPDF_Creator::WriteOldObjs() { |
| uint32_t nLastObjNum = m_pParser->GetLastObjNum(); |
| if (!m_pParser->IsValidObjectNumber(nLastObjNum)) |
| return true; |
| |
| for (uint32_t objnum = m_CurObjNum; objnum <= nLastObjNum; ++objnum) { |
| if (!WriteOldIndirectObject(objnum)) |
| return false; |
| } |
| return true; |
| } |
| |
| bool CPDF_Creator::WriteNewObjs() { |
| for (size_t i = m_CurObjNum; i < m_NewObjNumArray.size(); ++i) { |
| uint32_t objnum = m_NewObjNumArray[i]; |
| CPDF_Object* pObj = m_pDocument->GetIndirectObject(objnum); |
| if (!pObj) |
| continue; |
| |
| m_ObjectOffsets[objnum] = m_Archive->CurrentOffset(); |
| if (!WriteIndirectObj(pObj->GetObjNum(), pObj)) |
| return false; |
| } |
| return true; |
| } |
| |
| void CPDF_Creator::InitOldObjNumOffsets() { |
| if (!m_pParser) |
| return; |
| |
| 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; |
| |
| uint32_t 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; |
| if (IsIncremental() || |
| pair.second->GetObjNum() == CPDF_Object::kInvalidObjNum) { |
| continue; |
| } |
| if (m_pParser && m_pParser->IsValidObjectNumber(objnum) && |
| !m_pParser->IsObjectFree(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 && IsOriginal()) |
| m_dwFlags &= ~FPDFCREATE_INCREMENTAL; |
| |
| const CPDF_Dictionary* pDict = m_pDocument->GetRoot(); |
| m_pMetadata = pDict ? pDict->GetDirectObjectFor("Metadata") : nullptr; |
| m_iStage = 10; |
| } |
| if (m_iStage == 10) { |
| if (!IsIncremental()) { |
| if (!m_Archive->WriteString("%PDF-1.")) |
| return -1; |
| |
| int32_t version = 7; |
| if (m_FileVersion) |
| version = m_FileVersion; |
| else if (m_pParser) |
| version = m_pParser->GetFileVersion(); |
| |
| if (!m_Archive->WriteDWord(version % 10) || |
| !m_Archive->WriteString("\r\n%\xA1\xB3\xC5\xD7\r\n")) { |
| return -1; |
| } |
| |
| InitOldObjNumOffsets(); |
| m_iStage = 20; |
| } else { |
| m_SavedOffset = m_pParser->GetFileAccess()->GetSize(); |
| m_iStage = 15; |
| } |
| } |
| if (m_iStage == 15) { |
| if (IsOriginal() && m_SavedOffset > 0) { |
| 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_Archive->CurrentOffset() - src_size, |
| block_size)) { |
| return -1; |
| } |
| if (!m_Archive->WriteBlock(buffer.data(), block_size)) |
| return -1; |
| |
| src_size -= block_size; |
| } |
| } |
| if (IsOriginal() && m_pParser->GetLastXRefOffset() == 0) { |
| InitOldObjNumOffsets(); |
| |
| for (uint32_t num = 0; num <= m_pParser->GetLastObjNum(); ++num) { |
| if (m_pParser->IsObjectFreeOrNull(num)) |
| continue; |
| |
| m_ObjectOffsets[num] = m_pParser->GetObjectPositionOrZero(num); |
| } |
| } |
| 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) { |
| if (!WriteOldObjs()) |
| return -1; |
| |
| m_iStage = 25; |
| } |
| if (m_iStage == 25) { |
| m_CurObjNum = 0; |
| m_iStage = 26; |
| } |
| if (m_iStage == 26) { |
| if (!WriteNewObjs()) |
| return -1; |
| |
| m_iStage = 27; |
| } |
| if (m_iStage == 27) { |
| if (m_pEncryptDict && m_pEncryptDict->IsInline()) { |
| m_dwLastObjNum += 1; |
| FX_FILESIZE saveOffset = m_Archive->CurrentOffset(); |
| if (!WriteIndirectObj(m_dwLastObjNum, m_pEncryptDict.Get())) |
| 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_Archive->CurrentOffset(); |
| if (!IsIncremental() || !m_pParser->IsXRefStream()) { |
| if (!IsIncremental() || m_pParser->GetLastXRefOffset() == 0) { |
| ByteString str; |
| str = pdfium::ContainsKey(m_ObjectOffsets, 1) |
| ? "xref\r\n" |
| : "xref\r\n0 1\r\n0000000000 65535 f\r\n"; |
| if (!m_Archive->WriteString(str.AsStringView())) |
| return -1; |
| |
| m_CurObjNum = 1; |
| m_iStage = 81; |
| } else { |
| if (!m_Archive->WriteString("xref\r\n")) |
| return -1; |
| |
| m_CurObjNum = 0; |
| m_iStage = 82; |
| } |
| } else { |
| m_iStage = 90; |
| } |
| } |
| if (m_iStage == 81) { |
| 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_Archive->WriteBlock(str.c_str(), str.GetLength())) |
| return -1; |
| |
| while (i < j) { |
| str.Format("%010d 00000 n\r\n", m_ObjectOffsets[i++]); |
| if (!m_Archive->WriteBlock(str.c_str(), str.GetLength())) |
| return -1; |
| } |
| if (i > dwLastObjNum) |
| break; |
| } |
| m_iStage = 90; |
| } |
| if (m_iStage == 82) { |
| 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_Archive->WriteBlock(str.c_str(), str.GetLength())) |
| return -1; |
| |
| while (i < j) { |
| objnum = m_NewObjNumArray[i++]; |
| str.Format("%010d 00000 n\r\n", m_ObjectOffsets[objnum]); |
| if (!m_Archive->WriteBlock(str.c_str(), str.GetLength())) |
| return -1; |
| } |
| } |
| m_iStage = 90; |
| } |
| return m_iStage; |
| } |
| |
| int32_t CPDF_Creator::WriteDoc_Stage4() { |
| ASSERT(m_iStage >= 90); |
| |
| bool bXRefStream = IsIncremental() && m_pParser->IsXRefStream(); |
| if (!bXRefStream) { |
| if (!m_Archive->WriteString("trailer\r\n<<")) |
| return -1; |
| } else { |
| if (!m_Archive->WriteDWord(m_pDocument->GetLastObjNum() + 1) || |
| !m_Archive->WriteString(" 0 obj <<")) { |
| return -1; |
| } |
| } |
| |
| if (m_pParser) { |
| CPDF_Dictionary* p = m_pParser->GetTrailer(); |
| for (const auto& it : *p) { |
| const 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 == "ID") { |
| continue; |
| } |
| if (!m_Archive->WriteString(("/")) || |
| !m_Archive->WriteString(PDF_NameEncode(key).AsStringView())) { |
| return -1; |
| } |
| if (!pValue->IsInline()) { |
| if (!m_Archive->WriteString(" ") || |
| !m_Archive->WriteDWord(pValue->GetObjNum()) || |
| !m_Archive->WriteString(" 0 R ")) { |
| return -1; |
| } |
| } else if (!pValue->WriteTo(m_Archive.get())) { |
| return -1; |
| } |
| } |
| } else { |
| if (!m_Archive->WriteString("\r\n/Root ") || |
| !m_Archive->WriteDWord(m_pDocument->GetRoot()->GetObjNum()) || |
| !m_Archive->WriteString(" 0 R\r\n")) { |
| return -1; |
| } |
| if (m_pDocument->GetInfo()) { |
| if (!m_Archive->WriteString("/Info ") || |
| !m_Archive->WriteDWord(m_pDocument->GetInfo()->GetObjNum()) || |
| !m_Archive->WriteString(" 0 R\r\n")) { |
| return -1; |
| } |
| } |
| } |
| if (m_pEncryptDict) { |
| if (!m_Archive->WriteString("/Encrypt")) |
| return -1; |
| |
| uint32_t dwObjNum = m_pEncryptDict->GetObjNum(); |
| if (dwObjNum == 0) |
| dwObjNum = m_pDocument->GetLastObjNum() + 1; |
| if (!m_Archive->WriteString(" ") || !m_Archive->WriteDWord(dwObjNum) || |
| !m_Archive->WriteString(" 0 R ")) { |
| return -1; |
| } |
| } |
| |
| if (!m_Archive->WriteString("/Size ") || |
| !m_Archive->WriteDWord(m_dwLastObjNum + (bXRefStream ? 2 : 1))) { |
| return -1; |
| } |
| if (IsIncremental()) { |
| FX_FILESIZE prev = m_pParser->GetLastXRefOffset(); |
| if (prev) { |
| if (!m_Archive->WriteString("/Prev ")) |
| return -1; |
| |
| char offset_buf[20]; |
| memset(offset_buf, 0, sizeof(offset_buf)); |
| FXSYS_i64toa(prev, offset_buf, 10); |
| if (!m_Archive->WriteBlock(offset_buf, strlen(offset_buf))) |
| return -1; |
| } |
| } |
| if (m_pIDArray) { |
| if (!m_Archive->WriteString(("/ID")) || |
| !m_pIDArray->WriteTo(m_Archive.get())) { |
| return -1; |
| } |
| } |
| if (!bXRefStream) { |
| if (!m_Archive->WriteString(">>")) |
| return -1; |
| } else { |
| if (!m_Archive->WriteString("/W[0 4 1]/Index[")) |
| 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_Archive->WriteDWord(i) || !m_Archive->WriteString(" 1 ")) |
| return -1; |
| } |
| if (!m_Archive->WriteString("]/Length ") || |
| !m_Archive->WriteDWord(m_dwLastObjNum * 5) || |
| !m_Archive->WriteString(">>stream\r\n")) { |
| return -1; |
| } |
| for (i = 0; i < m_dwLastObjNum; i++) { |
| auto it = m_ObjectOffsets.find(i); |
| if (it == m_ObjectOffsets.end()) |
| continue; |
| OutputIndex(m_Archive.get(), it->second); |
| } |
| } else { |
| size_t count = m_NewObjNumArray.size(); |
| size_t i = 0; |
| for (i = 0; i < count; i++) { |
| if (!m_Archive->WriteDWord(m_NewObjNumArray[i]) || |
| !m_Archive->WriteString(" 1 ")) { |
| return -1; |
| } |
| } |
| if (!m_Archive->WriteString("]/Length ") || |
| !m_Archive->WriteDWord(count * 5) || |
| !m_Archive->WriteString(">>stream\r\n")) { |
| return -1; |
| } |
| for (i = 0; i < count; ++i) |
| OutputIndex(m_Archive.get(), m_ObjectOffsets[m_NewObjNumArray[i]]); |
| } |
| if (!m_Archive->WriteString("\r\nendstream")) |
| return -1; |
| } |
| |
| if (!m_Archive->WriteString("\r\nstartxref\r\n")) |
| return -1; |
| |
| char offset_buf[20]; |
| memset(offset_buf, 0, sizeof(offset_buf)); |
| FXSYS_i64toa(m_XrefStart, offset_buf, 10); |
| if (!m_Archive->WriteBlock(offset_buf, strlen(offset_buf)) || |
| !m_Archive->WriteString("\r\n%%EOF\r\n")) { |
| return -1; |
| } |
| |
| m_iStage = 100; |
| return m_iStage; |
| } |
| |
| bool CPDF_Creator::Create(uint32_t flags) { |
| m_dwFlags = flags; |
| m_iStage = 0; |
| m_dwLastObjNum = m_pDocument->GetLastObjNum(); |
| m_ObjectOffsets.clear(); |
| m_NewObjNumArray.clear(); |
| |
| InitID(); |
| return Continue() > -1; |
| } |
| |
| void CPDF_Creator::InitID() { |
| const CPDF_Array* pOldIDArray = m_pParser ? m_pParser->GetIDArray() : nullptr; |
| |
| bool idArrayPreExisting = !!m_pIDArray; |
| if (!idArrayPreExisting) { |
| 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); |
| ByteString bsBuffer(buffer.data(), buffer.size()); |
| m_pIDArray->AddNew<CPDF_String>(bsBuffer, true); |
| } |
| } |
| |
| 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); |
| 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 && !idArrayPreExisting) { |
| if (m_pEncryptDict->GetStringFor("Filter") == "Standard") { |
| ByteString user_pass = m_pParser->GetPassword(); |
| m_pSecurityHandler = pdfium::MakeUnique<CPDF_SecurityHandler>(); |
| m_pSecurityHandler->OnCreate(m_pEncryptDict.Get(), m_pIDArray.get(), |
| user_pass); |
| 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; |
| 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::RemoveSecurity() { |
| m_pSecurityHandler.Reset(); |
| m_bSecurityChanged = true; |
| m_pEncryptDict = nullptr; |
| } |
| |
| CPDF_CryptoHandler* CPDF_Creator::GetCryptoHandler() { |
| return m_pSecurityHandler ? m_pSecurityHandler->GetCryptoHandler() : nullptr; |
| } |
