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
diff --git a/core/fpdfapi/page/cpdf_countedobject.h b/core/fpdfapi/page/cpdf_countedobject.h
index 97d710c..64f936c 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 1c99d52..af26182 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 @@
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 @@
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 @@
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 @@
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 @@
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 @@
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 @@
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();
}
diff --git a/core/fpdfapi/render/cpdf_docrenderdata.cpp b/core/fpdfapi/render/cpdf_docrenderdata.cpp
index c17b4d5..7ac0353 100644
--- a/core/fpdfapi/render/cpdf_docrenderdata.cpp
+++ b/core/fpdfapi/render/cpdf_docrenderdata.cpp
@@ -55,8 +55,8 @@
CPDF_CountedObject<CPDF_Type3Cache>* pCache;
auto it = m_Type3FaceMap.find(pFont);
if (it == m_Type3FaceMap.end()) {
- CPDF_Type3Cache* pType3 = new CPDF_Type3Cache(pFont);
- pCache = new CPDF_CountedObject<CPDF_Type3Cache>(pType3);
+ pCache = new CPDF_CountedObject<CPDF_Type3Cache>(
+ pdfium::MakeUnique<CPDF_Type3Cache>(pFont));
m_Type3FaceMap[pFont] = pCache;
} else {
pCache = it->second;
@@ -104,9 +104,10 @@
if (!pFuncs[0])
return nullptr;
}
- CPDF_TransferFunc* pTransfer = new CPDF_TransferFunc(m_pPDFDoc);
CPDF_CountedObject<CPDF_TransferFunc>* pTransferCounter =
- new CPDF_CountedObject<CPDF_TransferFunc>(pTransfer);
+ new CPDF_CountedObject<CPDF_TransferFunc>(
+ pdfium::MakeUnique<CPDF_TransferFunc>(m_pPDFDoc));
+ CPDF_TransferFunc* pTransfer = pTransferCounter->get();
m_TransferFuncMap[pObj] = pTransferCounter;
FX_FLOAT output[kMaxOutputs];
FXSYS_memset(output, 0, sizeof(output));
