From c064d54cd169390039eedae422bbb132f771d20a Mon Sep 17 00:00:00 2001
From: Nicolas Pena <npm@chromium.org>
Date: Wed, 8 Nov 2017 15:49:51 +0000
Subject: Use MaybeOwned in CPDF_ContentParser
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Change-Id: I7e5daeafd70ef29bcc9bdebbae004fa4b946b83d
Reviewed-on: https://pdfium-review.googlesource.com/18019
Reviewed-by: Tom Sepez <tsepez@chromium.org>
Commit-Queue: Nicolás Peña Moreno <npm@chromium.org>
---
 core/fpdfapi/page/cpdf_contentparser.cpp | 19 ++++++++-----------
 core/fpdfapi/page/cpdf_contentparser.h   |  3 ++-
 2 files changed, 10 insertions(+), 12 deletions(-)

(limited to 'core/fpdfapi/page')

diff --git a/core/fpdfapi/page/cpdf_contentparser.cpp b/core/fpdfapi/page/cpdf_contentparser.cpp
index 0571405908..30701422d2 100644
--- a/core/fpdfapi/page/cpdf_contentparser.cpp
+++ b/core/fpdfapi/page/cpdf_contentparser.cpp
@@ -103,15 +103,11 @@ CPDF_ContentParser::CPDF_ContentParser(CPDF_Form* pForm,
   m_pSingleStream =
       pdfium::MakeRetain<CPDF_StreamAcc>(pForm->m_pFormStream.Get());
   m_pSingleStream->LoadAllData(false);
-  m_pData = const_cast<uint8_t*>(m_pSingleStream->GetData());
+  m_pData.Reset(m_pSingleStream->GetData());
   m_Size = m_pSingleStream->GetSize();
 }
 
-CPDF_ContentParser::~CPDF_ContentParser() {
-  // TODO(thestig): Switch |m_pData| to MaybeOwned?
-  if (!m_pSingleStream)
-    FX_Free(m_pData);
-}
+CPDF_ContentParser::~CPDF_ContentParser() {}
 
 bool CPDF_ContentParser::Continue(IFX_PauseIndicator* pPause) {
   if (m_bIsDone)
@@ -131,16 +127,17 @@ bool CPDF_ContentParser::Continue(IFX_PauseIndicator* pPause) {
             return false;
           }
           m_Size = safeSize.ValueOrDie();
-          m_pData = FX_Alloc(uint8_t, m_Size);
+          m_pData.Reset(std::unique_ptr<uint8_t, FxFreeDeleter>(
+              FX_Alloc(uint8_t, m_Size)));
           uint32_t pos = 0;
           for (const auto& stream : m_StreamArray) {
-            memcpy(m_pData + pos, stream->GetData(), stream->GetSize());
+            memcpy(m_pData.Get() + pos, stream->GetData(), stream->GetSize());
             pos += stream->GetSize();
-            m_pData[pos++] = ' ';
+            m_pData.Get()[pos++] = ' ';
           }
           m_StreamArray.clear();
         } else {
-          m_pData = const_cast<uint8_t*>(m_pSingleStream->GetData());
+          m_pData.Reset(m_pSingleStream->GetData());
           m_Size = m_pSingleStream->GetSize();
         }
         m_InternalStage = STAGE_PARSE;
@@ -170,7 +167,7 @@ bool CPDF_ContentParser::Continue(IFX_PauseIndicator* pPause) {
         m_InternalStage = STAGE_CHECKCLIP;
       } else {
         m_CurrentOffset +=
-            m_pParser->Parse(m_pData + m_CurrentOffset,
+            m_pParser->Parse(m_pData.Get() + m_CurrentOffset,
                              m_Size - m_CurrentOffset, PARSE_STEP_LIMIT);
       }
     }
diff --git a/core/fpdfapi/page/cpdf_contentparser.h b/core/fpdfapi/page/cpdf_contentparser.h
index c4fcb9060b..c65da6da55 100644
--- a/core/fpdfapi/page/cpdf_contentparser.h
+++ b/core/fpdfapi/page/cpdf_contentparser.h
@@ -14,6 +14,7 @@
 #include "core/fpdfapi/page/cpdf_pageobjectholder.h"
 #include "core/fpdfapi/page/cpdf_streamcontentparser.h"
 #include "core/fpdfapi/parser/cpdf_stream_acc.h"
+#include "core/fxcrt/maybe_owned.h"
 #include "core/fxcrt/unowned_ptr.h"
 
 class CPDF_AllStates;
@@ -52,7 +53,7 @@ class CPDF_ContentParser {
   uint32_t m_nStreams = 0;
   RetainPtr<CPDF_StreamAcc> m_pSingleStream;
   std::vector<RetainPtr<CPDF_StreamAcc>> m_StreamArray;
-  uint8_t* m_pData = nullptr;
+  MaybeOwned<uint8_t, FxFreeDeleter> m_pData;
   uint32_t m_Size = 0;
   uint32_t m_CurrentOffset = 0;
 
-- 
cgit v1.2.3