From a620372c9e1d0ff825bcbe6620fe7e1b63c769c2 Mon Sep 17 00:00:00 2001 From: Lei Zhang Date: Mon, 1 Feb 2016 14:40:26 -0800 Subject: Remove the m_ObjVersion member variable from CPDF_Parser. Use the gennum field in the m_ObjectInfo map instead. BUG=pdfium:111 R=weili@chromium.org Review URL: https://codereview.chromium.org/1654303002 . --- core/include/fpdfapi/fpdf_parser.h | 6 +----- core/src/fpdfapi/fpdf_parser/fpdf_parser_parser.cpp | 16 ++++++++-------- .../fpdf_parser/fpdf_parser_parser_unittest.cpp | 18 ++---------------- 3 files changed, 11 insertions(+), 29 deletions(-) diff --git a/core/include/fpdfapi/fpdf_parser.h b/core/include/fpdfapi/fpdf_parser.h index 57120ec467..ffc29ef3ae 100644 --- a/core/include/fpdfapi/fpdf_parser.h +++ b/core/include/fpdfapi/fpdf_parser.h @@ -430,18 +430,14 @@ class CPDF_Parser { CFX_ByteString m_Password; struct ObjectInfo { - ObjectInfo() : pos(0), type(0) {} + ObjectInfo() : pos(0), type(0), gennum(0) {} FX_FILESIZE pos; uint8_t type; -// TODO(thestig): Use field below in place of |m_ObjVersion| -#if 0 uint16_t gennum; -#endif }; std::map m_ObjectInfo; - CFX_WordArray m_ObjVersion; CFX_FileSizeArray m_SortedOffset; CFX_ArrayTemplate m_Trailers; FX_BOOL m_bVersionUpdated; diff --git a/core/src/fpdfapi/fpdf_parser/fpdf_parser_parser.cpp b/core/src/fpdfapi/fpdf_parser/fpdf_parser_parser.cpp index 5cc117798c..3696bb13db 100644 --- a/core/src/fpdfapi/fpdf_parser/fpdf_parser_parser.cpp +++ b/core/src/fpdfapi/fpdf_parser/fpdf_parser_parser.cpp @@ -131,7 +131,9 @@ uint8_t CPDF_Parser::GetObjectType(FX_DWORD objnum) const { } uint16_t CPDF_Parser::GetObjectGenNum(FX_DWORD objnum) const { - return m_ObjVersion[objnum]; + ASSERT(IsValidObjectNumber(objnum)); + auto it = m_ObjectInfo.find(objnum); + return it != m_ObjectInfo.end() ? it->second.gennum : 0; } bool CPDF_Parser::IsObjectFreeOrNull(FX_DWORD objnum) const { @@ -179,7 +181,6 @@ void CPDF_Parser::CloseParser() { m_SortedOffset.RemoveAll(); m_ObjectInfo.clear(); - m_ObjVersion.RemoveAll(); int32_t iLen = m_Trailers.GetSize(); for (int32_t i = 0; i < iLen; ++i) { if (CPDF_Dictionary* trailer = m_Trailers.GetAt(i)) @@ -486,7 +487,7 @@ FX_BOOL CPDF_Parser::LoadLinearizedCrossRefV4(FX_FILESIZE pos, if (version >= 1) { m_bVersionUpdated = TRUE; } - m_ObjVersion.SetAtGrow(objnum, version); + m_ObjectInfo[objnum].gennum = version; if (m_ObjectInfo[objnum].pos < m_Syntax.m_FileLen) { void* pResult = FXSYS_bsearch( &m_ObjectInfo[objnum].pos, m_SortedOffset.GetData(), @@ -568,7 +569,7 @@ bool CPDF_Parser::LoadCrossRefV4(FX_FILESIZE pos, if (version >= 1) { m_bVersionUpdated = TRUE; } - m_ObjVersion.SetAtGrow(objnum, version); + m_ObjectInfo[objnum].gennum = version; if (m_ObjectInfo[objnum].pos < m_Syntax.m_FileLen && !FindPosInOffsets(m_ObjectInfo[objnum].pos)) { m_SortedOffset.Add(m_ObjectInfo[objnum].pos); @@ -606,7 +607,6 @@ FX_BOOL CPDF_Parser::LoadAllCrossRefV5(FX_FILESIZE xrefpos) { FX_BOOL CPDF_Parser::RebuildCrossRef() { m_ObjectInfo.clear(); m_SortedOffset.RemoveAll(); - m_ObjVersion.RemoveAll(); if (m_pTrailer) { m_pTrailer->Release(); m_pTrailer = NULL; @@ -811,9 +811,9 @@ FX_BOOL CPDF_Parser::RebuildCrossRef() { if (!m_ObjectInfo.empty() && IsValidObjectNumber(objnum) && m_ObjectInfo[objnum].pos) { if (pObject) { - FX_DWORD oldgen = m_ObjVersion.GetAt(objnum); + FX_DWORD oldgen = GetObjectGenNum(objnum); m_ObjectInfo[objnum].pos = obj_pos; - m_ObjVersion.SetAt(objnum, (int16_t)gennum); + m_ObjectInfo[objnum].gennum = gennum; if (oldgen != gennum) { m_bVersionUpdated = TRUE; } @@ -821,7 +821,7 @@ FX_BOOL CPDF_Parser::RebuildCrossRef() { } else { m_ObjectInfo[objnum].pos = obj_pos; m_ObjectInfo[objnum].type = 1; - m_ObjVersion.SetAtGrow(objnum, (int16_t)gennum); + m_ObjectInfo[objnum].gennum = gennum; } if (pObject) { pObject->Release(); diff --git a/core/src/fpdfapi/fpdf_parser/fpdf_parser_parser_unittest.cpp b/core/src/fpdfapi/fpdf_parser/fpdf_parser_parser_unittest.cpp index b39eb96785..77780b9dcf 100644 --- a/core/src/fpdfapi/fpdf_parser/fpdf_parser_parser_unittest.cpp +++ b/core/src/fpdfapi/fpdf_parser/fpdf_parser_parser_unittest.cpp @@ -7,20 +7,6 @@ #include "testing/gtest/include/gtest/gtest.h" #include "testing/utils/path_service.h" -// Functions to help test an array's content against expected results. -template -bool CompareArray(const CFX_ArrayTemplate& array1, - const TYPE* array2, - size_t size) { - if (array1.GetSize() != size) - return false; - - for (int i = 0; i < size; ++i) - if (array1.GetAt(i) != array2[i]) - return false; - return true; -} - // Provide a way to read test data from a buffer instead of a file. class CFX_TestBufferRead : public IFX_FileRead { public: @@ -257,8 +243,8 @@ TEST(fpdf_parser_parser, RebuildCrossRefCorrectly) { const FX_WORD versions[] = {0, 0, 2, 4, 6, 8, 0}; for (size_t i = 0; i < FX_ArraySize(offsets); ++i) EXPECT_EQ(offsets[i], parser.m_ObjectInfo[i].pos); - ASSERT_TRUE( - CompareArray(parser.m_ObjVersion, versions, FX_ArraySize(versions))); + for (size_t i = 0; i < FX_ArraySize(versions); ++i) + EXPECT_EQ(versions[i], parser.m_ObjectInfo[i].gennum); } TEST(fpdf_parser_parser, RebuildCrossRefFailed) { -- cgit v1.2.3