diff options
Diffstat (limited to 'core')
-rw-r--r-- | core/fpdfapi/edit/cpdf_flateencoder.cpp | 2 | ||||
-rw-r--r-- | core/fpdfapi/edit/cpdf_flateencoder.h | 4 | ||||
-rw-r--r-- | core/fpdfapi/parser/cpdf_syntax_parser.cpp | 67 |
3 files changed, 12 insertions, 61 deletions
diff --git a/core/fpdfapi/edit/cpdf_flateencoder.cpp b/core/fpdfapi/edit/cpdf_flateencoder.cpp index 573c141ff2..a290da6dac 100644 --- a/core/fpdfapi/edit/cpdf_flateencoder.cpp +++ b/core/fpdfapi/edit/cpdf_flateencoder.cpp @@ -15,7 +15,7 @@ CPDF_FlateEncoder::CPDF_FlateEncoder(const CPDF_Stream* pStream, bool bFlateEncode) - : m_pAcc(pdfium::MakeRetain<CPDF_StreamAcc>(pStream)), m_dwSize(0) { + : m_dwSize(0), m_pAcc(pdfium::MakeRetain<CPDF_StreamAcc>(pStream)) { m_pAcc->LoadAllDataRaw(); bool bHasFilter = pStream && pStream->HasFilter(); diff --git a/core/fpdfapi/edit/cpdf_flateencoder.h b/core/fpdfapi/edit/cpdf_flateencoder.h index 14ca7cec16..05633f6814 100644 --- a/core/fpdfapi/edit/cpdf_flateencoder.h +++ b/core/fpdfapi/edit/cpdf_flateencoder.h @@ -34,14 +34,14 @@ class CPDF_FlateEncoder { } private: - RetainPtr<CPDF_StreamAcc> m_pAcc; - uint32_t m_dwSize; MaybeOwned<uint8_t, FxFreeDeleter> m_pData; // Only one of these two pointers is valid at any time. UnownedPtr<const CPDF_Dictionary> m_pDict; std::unique_ptr<CPDF_Dictionary> m_pClonedDict; + + RetainPtr<CPDF_StreamAcc> m_pAcc; }; #endif // CORE_FPDFAPI_EDIT_CPDF_FLATEENCODER_H_ diff --git a/core/fpdfapi/parser/cpdf_syntax_parser.cpp b/core/fpdfapi/parser/cpdf_syntax_parser.cpp index f8f36aee11..ec5ac70a29 100644 --- a/core/fpdfapi/parser/cpdf_syntax_parser.cpp +++ b/core/fpdfapi/parser/cpdf_syntax_parser.cpp @@ -34,37 +34,6 @@ namespace { enum class ReadStatus { Normal, Backslash, Octal, FinishOctal, CarriageReturn }; -class ReadableSubStream : public IFX_SeekableReadStream { - public: - ReadableSubStream(const RetainPtr<IFX_SeekableReadStream>& pFileRead, - FX_FILESIZE part_offset, - FX_FILESIZE part_size) - : m_pFileRead(pFileRead), - m_PartOffset(part_offset), - m_PartSize(part_size) {} - - ~ReadableSubStream() override = default; - - // IFX_SeekableReadStream overrides: - bool ReadBlock(void* buffer, FX_FILESIZE offset, size_t size) override { - FX_SAFE_FILESIZE safe_end = offset; - safe_end += size; - // Check that requested range is valid, to prevent calling of ReadBlock - // of original m_pFileRead with incorrect params. - if (!safe_end.IsValid() || safe_end.ValueOrDie() > m_PartSize) - return false; - - return m_pFileRead->ReadBlock(buffer, m_PartOffset + offset, size); - } - - FX_FILESIZE GetSize() override { return m_PartSize; } - - private: - RetainPtr<IFX_SeekableReadStream> m_pFileRead; - FX_FILESIZE m_PartOffset; - FX_FILESIZE m_PartSize; -}; - } // namespace // static @@ -608,6 +577,7 @@ std::unique_ptr<CPDF_Stream> CPDF_SyntaxParser::ReadStream( ToNextLine(); const FX_FILESIZE streamStartPos = GetPos(); + std::unique_ptr<uint8_t, FxFreeDeleter> pData; if (len > 0) { FX_SAFE_FILESIZE pos = GetPos(); pos += len; @@ -615,18 +585,12 @@ std::unique_ptr<CPDF_Stream> CPDF_SyntaxParser::ReadStream( len = -1; } - RetainPtr<IFX_SeekableReadStream> data; if (len > 0) { - // Check data availability first to allow the Validator to request data + pData.reset(FX_Alloc(uint8_t, len)); + // We should try read data first to allow the Validator to request data // smoothly, without jumps. - if (!GetValidator()->CheckDataRangeAndRequestIfUnavailable( - m_HeaderOffset + GetPos(), len)) { + if (!ReadBlock(pData.get(), len)) return nullptr; - } - - data = pdfium::MakeRetain<ReadableSubStream>( - GetValidator(), m_HeaderOffset + GetPos(), len); - SetPos(GetPos() + len); } const ByteStringView kEndStreamStr("endstream"); @@ -647,7 +611,7 @@ std::unique_ptr<CPDF_Stream> CPDF_SyntaxParser::ReadStream( // specified length, it signals the end of stream. if (memcmp(m_WordBuffer, kEndStreamStr.raw_str(), kEndStreamStr.GetLength()) != 0) { - data.Reset(); + pData.reset(); len = -1; SetPos(streamStartPos); } @@ -664,27 +628,14 @@ std::unique_ptr<CPDF_Stream> CPDF_SyntaxParser::ReadStream( ASSERT(len >= 0); if (len > 0) { SetPos(streamStartPos); - // Check data availability first to allow the Validator to request data - // smoothly, without jumps. - if (!GetValidator()->CheckDataRangeAndRequestIfUnavailable( - m_HeaderOffset + GetPos(), len)) { + pData.reset(FX_Alloc(uint8_t, len)); + if (!ReadBlock(pData.get(), len)) return nullptr; - } - - data = pdfium::MakeRetain<ReadableSubStream>( - GetValidator(), m_HeaderOffset + GetPos(), len); - SetPos(GetPos() + len); } } - auto pStream = pdfium::MakeUnique<CPDF_Stream>(); - if (data) { - pStream->InitStreamFromFile(data, std::move(pDict)); - } else { - DCHECK(!len); - // Empty stream - pStream->InitStream(nullptr, 0, std::move(pDict)); - } + auto pStream = + pdfium::MakeUnique<CPDF_Stream>(std::move(pData), len, std::move(pDict)); const FX_FILESIZE end_stream_offset = GetPos(); memset(m_WordBuffer, 0, kEndObjStr.GetLength() + 1); GetNextWordInternal(nullptr); |