From 7c39bf7b87f871ccc50b66066c3bfb1883f66010 Mon Sep 17 00:00:00 2001 From: Lei Zhang Date: Mon, 15 Oct 2018 16:56:47 +0000 Subject: Make FlateEncode()'s buffer out parameter a unique_ptr. Change-Id: I31a5abb3c8dd31826098493f51effbd5afe109d1 Reviewed-on: https://pdfium-review.googlesource.com/c/41856 Reviewed-by: Tom Sepez Commit-Queue: Lei Zhang --- core/fxcodec/codec/ccodec_flatemodule.cpp | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) (limited to 'core/fxcodec/codec/ccodec_flatemodule.cpp') diff --git a/core/fxcodec/codec/ccodec_flatemodule.cpp b/core/fxcodec/codec/ccodec_flatemodule.cpp index f35b9581be..f447ec567f 100644 --- a/core/fxcodec/codec/ccodec_flatemodule.cpp +++ b/core/fxcodec/codec/ccodec_flatemodule.cpp @@ -834,14 +834,15 @@ uint32_t CCodec_FlateModule::FlateOrLZWDecode( return ret ? offset : FX_INVALID_OFFSET; } -bool CCodec_FlateModule::Encode(const uint8_t* src_buf, - uint32_t src_size, - uint8_t** dest_buf, - uint32_t* dest_size) { +bool CCodec_FlateModule::Encode( + const uint8_t* src_buf, + uint32_t src_size, + std::unique_ptr* dest_buf, + uint32_t* dest_size) { *dest_size = src_size + src_size / 1000 + 12; - *dest_buf = FX_Alloc(uint8_t, *dest_size); + dest_buf->reset(FX_Alloc(uint8_t, *dest_size)); unsigned long temp_size = *dest_size; - if (!FlateCompress(*dest_buf, &temp_size, src_buf, src_size)) + if (!FlateCompress(dest_buf->get(), &temp_size, src_buf, src_size)) return false; *dest_size = (uint32_t)temp_size; -- cgit v1.2.3