diff options
author | thestig <thestig@chromium.org> | 2016-05-19 15:36:36 -0700 |
---|---|---|
committer | Commit bot <commit-bot@chromium.org> | 2016-05-19 15:36:36 -0700 |
commit | db9e49889d4129bbe96abdbce7dc3662e97a9df8 (patch) | |
tree | 5edcf8cd16fd7d89a6a1750f4f02e9b4af870c7f /core/fxcrt/fx_basic_maps.cpp | |
parent | 80f25a5a8135933a405349ffc798d13273b3d690 (diff) | |
download | pdfium-db9e49889d4129bbe96abdbce7dc3662e97a9df8.tar.xz |
Fix a potential nullptr deref in CFX_MapPtrToPtr.
And also in CFX_PtrList.
BUG=596528
Review-Url: https://codereview.chromium.org/1991123002
Diffstat (limited to 'core/fxcrt/fx_basic_maps.cpp')
-rw-r--r-- | core/fxcrt/fx_basic_maps.cpp | 25 |
1 files changed, 14 insertions, 11 deletions
diff --git a/core/fxcrt/fx_basic_maps.cpp b/core/fxcrt/fx_basic_maps.cpp index 660306b11a..dda8167bdb 100644 --- a/core/fxcrt/fx_basic_maps.cpp +++ b/core/fxcrt/fx_basic_maps.cpp @@ -8,22 +8,26 @@ #include "core/fxcrt/plex.h" CFX_MapPtrToPtr::CFX_MapPtrToPtr(int nBlockSize) - : m_pHashTable(NULL), + : m_pHashTable(nullptr), m_nHashTableSize(17), m_nCount(0), - m_pFreeList(NULL), - m_pBlocks(NULL), + m_pFreeList(nullptr), + m_pBlocks(nullptr), m_nBlockSize(nBlockSize) { ASSERT(m_nBlockSize > 0); } + void CFX_MapPtrToPtr::RemoveAll() { FX_Free(m_pHashTable); - m_pHashTable = NULL; + m_pHashTable = nullptr; m_nCount = 0; - m_pFreeList = NULL; - m_pBlocks->FreeDataChain(); - m_pBlocks = NULL; + m_pFreeList = nullptr; + if (m_pBlocks) { + m_pBlocks->FreeDataChain(); + m_pBlocks = nullptr; + } } + CFX_MapPtrToPtr::~CFX_MapPtrToPtr() { RemoveAll(); ASSERT(m_nCount == 0); @@ -63,14 +67,13 @@ FX_BOOL CFX_MapPtrToPtr::Lookup(void* key, void*& rValue) const { rValue = pAssoc->value; return TRUE; } + void* CFX_MapPtrToPtr::GetValueAt(void* key) const { uint32_t nHash; CAssoc* pAssoc = GetAssocAt(key, nHash); - if (!pAssoc) { - return NULL; - } - return pAssoc->value; + return pAssoc ? pAssoc->value : nullptr; } + void*& CFX_MapPtrToPtr::operator[](void* key) { uint32_t nHash; CAssoc* pAssoc; |