Take buffers from the partition in CPDF_Creator Change-Id: Ide3a994940f2580f740c7deea586a74cea1c57c3 Reviewed-on: https://pdfium-review.googlesource.com/c/pdfium/+/62751 Reviewed-by: Lei Zhang <thestig@chromium.org> Commit-Queue: Tom Sepez <tsepez@chromium.org>
diff --git a/core/fpdfapi/edit/cpdf_creator.cpp b/core/fpdfapi/edit/cpdf_creator.cpp index 00936a3..71ef379 100644 --- a/core/fpdfapi/edit/cpdf_creator.cpp +++ b/core/fpdfapi/edit/cpdf_creator.cpp
@@ -21,6 +21,7 @@ #include "core/fpdfapi/parser/cpdf_syntax_parser.h" #include "core/fpdfapi/parser/fpdf_parser_utility.h" #include "core/fxcrt/fx_extension.h" +#include "core/fxcrt/fx_memory_wrappers.h" #include "core/fxcrt/fx_random.h" #include "third_party/base/ptr_util.h" #include "third_party/base/stl_util.h" @@ -47,7 +48,7 @@ FX_FILESIZE offset_; size_t current_length_; - std::vector<uint8_t> buffer_; + std::vector<uint8_t, FxAllocAllocator<uint8_t>> buffer_; RetainPtr<IFX_RetainableWriteStream> backing_file_; }; @@ -259,7 +260,7 @@ if (m_iStage == Stage::kWriteIncremental15) { if (m_IsOriginal && m_SavedOffset > 0) { static constexpr FX_FILESIZE kBufferSize = 4096; - std::vector<uint8_t> buffer(kBufferSize); + std::vector<uint8_t, FxAllocAllocator<uint8_t>> buffer(kBufferSize); FX_FILESIZE src_size = m_SavedOffset; m_pParser->GetSyntax()->SetPos(0); while (src_size) {