summaryrefslogtreecommitdiff
path: root/core/fpdfapi/page
diff options
context:
space:
mode:
authortsepez <tsepez@chromium.org>2016-11-22 13:01:40 -0800
committerCommit bot <commit-bot@chromium.org>2016-11-22 13:01:40 -0800
commit6136ec6347a5858a85912a805ea41126863558cd (patch)
tree9aee627585a525844e9861fee6012509bbc39a14 /core/fpdfapi/page
parent687fbde2e4ee13637cb3fd9b1fec39a436ef73d7 (diff)
downloadpdfium-6136ec6347a5858a85912a805ea41126863558cd.tar.xz
Ensure CPDF_CountedObjects only made from owned references.
Deletion of said object is still inflicted on the callers. But that's an issue for another day. Review-Url: https://codereview.chromium.org/2523743004
Diffstat (limited to 'core/fpdfapi/page')
-rw-r--r--core/fpdfapi/page/cpdf_countedobject.h9
-rw-r--r--core/fpdfapi/page/cpdf_docpagedata.cpp51
2 files changed, 33 insertions, 27 deletions
diff --git a/core/fpdfapi/page/cpdf_countedobject.h b/core/fpdfapi/page/cpdf_countedobject.h
index 97d710cf76..64f936c52c 100644
--- a/core/fpdfapi/page/cpdf_countedobject.h
+++ b/core/fpdfapi/page/cpdf_countedobject.h
@@ -7,6 +7,8 @@
#ifndef CORE_FPDFAPI_PAGE_CPDF_COUNTEDOBJECT_H_
#define CORE_FPDFAPI_PAGE_CPDF_COUNTEDOBJECT_H_
+#include <memory>
+
#include "core/fpdfapi/page/cpdf_colorspace.h"
#include "core/fpdfapi/page/cpdf_pattern.h"
#include "core/fxcrt/fx_system.h"
@@ -14,10 +16,11 @@
template <class T>
class CPDF_CountedObject {
public:
- explicit CPDF_CountedObject(T* ptr) : m_nCount(1), m_pObj(ptr) {}
- void reset(T* ptr) { // CAUTION: tosses prior ref counts.
+ explicit CPDF_CountedObject(std::unique_ptr<T> ptr)
+ : m_nCount(1), m_pObj(ptr.release()) {}
+ void reset(std::unique_ptr<T> ptr) { // CAUTION: tosses prior ref counts.
m_nCount = 1;
- m_pObj = ptr;
+ m_pObj = ptr.release();
}
void clear() { // Now you're all weak ptrs ...
// Guard against accidental re-entry.
diff --git a/core/fpdfapi/page/cpdf_docpagedata.cpp b/core/fpdfapi/page/cpdf_docpagedata.cpp
index 1c99d523b0..af2618286e 100644
--- a/core/fpdfapi/page/cpdf_docpagedata.cpp
+++ b/core/fpdfapi/page/cpdf_docpagedata.cpp
@@ -9,6 +9,7 @@
#include <algorithm>
#include <memory>
#include <set>
+#include <utility>
#include "core/fdrm/crypto/fx_crypt.h"
#include "core/fpdfapi/cpdf_modulemgr.h"
@@ -143,9 +144,9 @@ CPDF_Font* CPDF_DocPageData::GetFont(CPDF_Dictionary* pFontDict) {
return nullptr;
if (pFontData) {
- pFontData->reset(pFont.release());
+ pFontData->reset(std::move(pFont));
} else {
- pFontData = new CPDF_CountedFont(pFont.release());
+ pFontData = new CPDF_CountedFont(std::move(pFont));
m_FontMap[pFontDict] = pFontData;
}
return pFontData->AddRef();
@@ -190,7 +191,7 @@ CPDF_Font* CPDF_DocPageData::GetStandardFont(const CFX_ByteString& fontName,
if (!pFont)
return nullptr;
- CPDF_CountedFont* fontData = new CPDF_CountedFont(pFont.release());
+ CPDF_CountedFont* fontData = new CPDF_CountedFont(std::move(pFont));
m_FontMap[pDict] = fontData;
return fontData->AddRef();
}
@@ -293,11 +294,11 @@ CPDF_ColorSpace* CPDF_DocPageData::GetColorSpaceImpl(
if (!pCS)
return nullptr;
- if (!csData) {
- csData = new CPDF_CountedColorSpace(pCS.release());
- m_ColorSpaceMap[pCSObj] = csData;
+ if (csData) {
+ csData->reset(std::move(pCS));
} else {
- csData->reset(pCS.release());
+ csData = new CPDF_CountedColorSpace(std::move(pCS));
+ m_ColorSpaceMap[pCSObj] = csData;
}
return csData->AddRef();
}
@@ -348,29 +349,31 @@ CPDF_Pattern* CPDF_DocPageData::GetPattern(CPDF_Object* pPatternObj,
return ptData->AddRef();
}
}
- CPDF_Pattern* pPattern = nullptr;
+ std::unique_ptr<CPDF_Pattern> pPattern;
if (bShading) {
- pPattern = new CPDF_ShadingPattern(m_pPDFDoc, pPatternObj, true, matrix);
+ pPattern = pdfium::MakeUnique<CPDF_ShadingPattern>(m_pPDFDoc, pPatternObj,
+ true, matrix);
} else {
CPDF_Dictionary* pDict = pPatternObj ? pPatternObj->GetDict() : nullptr;
if (pDict) {
int type = pDict->GetIntegerFor("PatternType");
if (type == CPDF_Pattern::TILING) {
- pPattern = new CPDF_TilingPattern(m_pPDFDoc, pPatternObj, matrix);
+ pPattern = pdfium::MakeUnique<CPDF_TilingPattern>(m_pPDFDoc,
+ pPatternObj, matrix);
} else if (type == CPDF_Pattern::SHADING) {
- pPattern =
- new CPDF_ShadingPattern(m_pPDFDoc, pPatternObj, false, matrix);
+ pPattern = pdfium::MakeUnique<CPDF_ShadingPattern>(
+ m_pPDFDoc, pPatternObj, false, matrix);
}
}
}
if (!pPattern)
return nullptr;
- if (!ptData) {
- ptData = new CPDF_CountedPattern(pPattern);
- m_PatternMap[pPatternObj] = ptData;
+ if (ptData) {
+ ptData->reset(std::move(pPattern));
} else {
- ptData->reset(pPattern);
+ ptData = new CPDF_CountedPattern(std::move(pPattern));
+ m_PatternMap[pPatternObj] = ptData;
}
return ptData->AddRef();
}
@@ -401,8 +404,8 @@ CPDF_Image* CPDF_DocPageData::GetImage(uint32_t dwStreamObjNum) {
if (it != m_ImageMap.end())
return it->second->AddRef();
- CPDF_CountedImage* pCountedImage =
- new CPDF_CountedImage(new CPDF_Image(m_pPDFDoc, dwStreamObjNum));
+ CPDF_CountedImage* pCountedImage = new CPDF_CountedImage(
+ pdfium::MakeUnique<CPDF_Image>(m_pPDFDoc, dwStreamObjNum));
m_ImageMap[dwStreamObjNum] = pCountedImage;
return pCountedImage->AddRef();
}
@@ -447,9 +450,8 @@ CPDF_IccProfile* CPDF_DocPageData::GetIccProfile(
if (it_copied_stream != m_IccProfileMap.end())
return it_copied_stream->second->AddRef();
}
- CPDF_IccProfile* pProfile =
- new CPDF_IccProfile(stream.GetData(), stream.GetSize());
- CPDF_CountedIccProfile* ipData = new CPDF_CountedIccProfile(pProfile);
+ CPDF_CountedIccProfile* ipData = new CPDF_CountedIccProfile(
+ pdfium::MakeUnique<CPDF_IccProfile>(stream.GetData(), stream.GetSize()));
m_IccProfileMap[pIccProfileStream] = ipData;
m_HashProfileMap[bsDigest] = pIccProfileStream;
return ipData->AddRef();
@@ -488,10 +490,11 @@ CPDF_StreamAcc* CPDF_DocPageData::GetFontFileStreamAcc(
pFontDict->GetIntegerFor("Length3");
org_size = std::max(org_size, 0);
- CPDF_StreamAcc* pFontFile = new CPDF_StreamAcc;
- pFontFile->LoadAllData(pFontStream, false, org_size);
+ auto pFontAcc = pdfium::MakeUnique<CPDF_StreamAcc>();
+ pFontAcc->LoadAllData(pFontStream, false, org_size);
- CPDF_CountedStreamAcc* pCountedFont = new CPDF_CountedStreamAcc(pFontFile);
+ CPDF_CountedStreamAcc* pCountedFont =
+ new CPDF_CountedStreamAcc(std::move(pFontAcc));
m_FontFileMap[pFontStream] = pCountedFont;
return pCountedFont->AddRef();
}