Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 1 | // Copyright 2016 PDFium Authors. All rights reserved. |
| 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
| 4 | |
dsinclair | 488b7ad | 2016-10-04 11:55:50 -0700 | [diff] [blame] | 5 | #include "core/fpdfapi/parser/cpdf_array.h" |
| 6 | #include "core/fpdfapi/parser/cpdf_boolean.h" |
| 7 | #include "core/fpdfapi/parser/cpdf_dictionary.h" |
| 8 | #include "core/fpdfapi/parser/cpdf_name.h" |
| 9 | #include "core/fpdfapi/parser/cpdf_null.h" |
| 10 | #include "core/fpdfapi/parser/cpdf_number.h" |
| 11 | #include "core/fpdfapi/parser/cpdf_reference.h" |
| 12 | #include "core/fpdfapi/parser/cpdf_stream.h" |
| 13 | #include "core/fpdfapi/parser/cpdf_string.h" |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 14 | |
| 15 | #include <memory> |
Dan Sinclair | 3ebd121 | 2016-03-09 09:59:23 -0500 | [diff] [blame] | 16 | #include <string> |
tsepez | 0e606b5 | 2016-11-18 16:22:41 -0800 | [diff] [blame] | 17 | #include <utility> |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 18 | #include <vector> |
| 19 | |
dsinclair | 488b7ad | 2016-10-04 11:55:50 -0700 | [diff] [blame] | 20 | #include "core/fpdfapi/parser/cpdf_indirect_object_holder.h" |
dsinclair | a52ab74 | 2016-09-29 13:59:29 -0700 | [diff] [blame] | 21 | #include "core/fxcrt/fx_basic.h" |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 22 | #include "testing/gtest/include/gtest/gtest.h" |
| 23 | |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 24 | namespace { |
| 25 | |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 26 | void TestArrayAccessors(const CPDF_Array* arr, |
| 27 | size_t index, |
| 28 | const char* str_val, |
| 29 | const char* const_str_val, |
| 30 | int int_val, |
| 31 | float float_val, |
| 32 | CPDF_Array* arr_val, |
| 33 | CPDF_Dictionary* dict_val, |
| 34 | CPDF_Stream* stream_val) { |
| 35 | EXPECT_STREQ(str_val, arr->GetStringAt(index).c_str()); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 36 | EXPECT_EQ(int_val, arr->GetIntegerAt(index)); |
| 37 | EXPECT_EQ(float_val, arr->GetNumberAt(index)); |
| 38 | EXPECT_EQ(float_val, arr->GetFloatAt(index)); |
| 39 | EXPECT_EQ(arr_val, arr->GetArrayAt(index)); |
| 40 | EXPECT_EQ(dict_val, arr->GetDictAt(index)); |
| 41 | EXPECT_EQ(stream_val, arr->GetStreamAt(index)); |
| 42 | } |
| 43 | |
| 44 | } // namespace |
| 45 | |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 46 | class PDFObjectsTest : public testing::Test { |
| 47 | public: |
| 48 | void SetUp() override { |
| 49 | // Initialize different kinds of objects. |
| 50 | // Boolean objects. |
| 51 | CPDF_Boolean* boolean_false_obj = new CPDF_Boolean(false); |
| 52 | CPDF_Boolean* boolean_true_obj = new CPDF_Boolean(true); |
| 53 | // Number objects. |
| 54 | CPDF_Number* number_int_obj = new CPDF_Number(1245); |
| 55 | CPDF_Number* number_float_obj = new CPDF_Number(9.00345f); |
| 56 | // String objects. |
tsepez | 0e606b5 | 2016-11-18 16:22:41 -0800 | [diff] [blame] | 57 | CPDF_String* str_reg_obj = new CPDF_String(nullptr, L"A simple test"); |
| 58 | CPDF_String* str_spec_obj = new CPDF_String(nullptr, L"\t\n"); |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 59 | // Name object. |
tsepez | 5913a6c | 2016-11-16 17:31:18 -0800 | [diff] [blame] | 60 | CPDF_Name* name_obj = new CPDF_Name(nullptr, "space"); |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 61 | // Array object. |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 62 | m_ArrayObj = new CPDF_Array; |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 63 | m_ArrayObj->InsertNewAt<CPDF_Number>(0, 8902); |
| 64 | m_ArrayObj->InsertNewAt<CPDF_Name>(1, "address"); |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 65 | // Dictionary object. |
tsepez | cd5bca4 | 2016-09-30 10:45:06 -0700 | [diff] [blame] | 66 | m_DictObj = new CPDF_Dictionary(); |
tsepez | 0e606b5 | 2016-11-18 16:22:41 -0800 | [diff] [blame] | 67 | m_DictObj->SetNewFor<CPDF_Boolean>("bool", false); |
| 68 | m_DictObj->SetNewFor<CPDF_Number>("num", 0.23f); |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 69 | // Stream object. |
| 70 | const char content[] = "abcdefghijklmnopqrstuvwxyz"; |
| 71 | size_t buf_len = FX_ArraySize(content); |
tsepez | 47fb8c0 | 2016-12-15 13:51:34 -0800 | [diff] [blame] | 72 | std::unique_ptr<uint8_t, FxFreeDeleter> buf(FX_Alloc(uint8_t, buf_len)); |
| 73 | memcpy(buf.get(), content, buf_len); |
tsepez | 9e05ee1 | 2016-11-21 13:19:10 -0800 | [diff] [blame] | 74 | auto pNewDict = pdfium::MakeUnique<CPDF_Dictionary>(); |
| 75 | m_StreamDictObj = pNewDict.get(); |
tsepez | 0e606b5 | 2016-11-18 16:22:41 -0800 | [diff] [blame] | 76 | m_StreamDictObj->SetNewFor<CPDF_String>("key1", L" test dict"); |
| 77 | m_StreamDictObj->SetNewFor<CPDF_Number>("key2", -1); |
tsepez | 9e05ee1 | 2016-11-21 13:19:10 -0800 | [diff] [blame] | 78 | CPDF_Stream* stream_obj = |
tsepez | 47fb8c0 | 2016-12-15 13:51:34 -0800 | [diff] [blame] | 79 | new CPDF_Stream(std::move(buf), buf_len, std::move(pNewDict)); |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 80 | // Null Object. |
| 81 | CPDF_Null* null_obj = new CPDF_Null; |
| 82 | // All direct objects. |
| 83 | CPDF_Object* objs[] = {boolean_false_obj, boolean_true_obj, number_int_obj, |
| 84 | number_float_obj, str_reg_obj, str_spec_obj, |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 85 | name_obj, m_ArrayObj, m_DictObj, |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 86 | stream_obj, null_obj}; |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 87 | m_DirectObjTypes = { |
| 88 | CPDF_Object::BOOLEAN, CPDF_Object::BOOLEAN, CPDF_Object::NUMBER, |
| 89 | CPDF_Object::NUMBER, CPDF_Object::STRING, CPDF_Object::STRING, |
| 90 | CPDF_Object::NAME, CPDF_Object::ARRAY, CPDF_Object::DICTIONARY, |
| 91 | CPDF_Object::STREAM, CPDF_Object::NULLOBJ}; |
| 92 | for (size_t i = 0; i < FX_ArraySize(objs); ++i) |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 93 | m_DirectObjs.emplace_back(objs[i]); |
| 94 | |
| 95 | // Indirect references to indirect objects. |
tsepez | 70c4afd | 2016-11-15 11:33:44 -0800 | [diff] [blame] | 96 | m_ObjHolder = pdfium::MakeUnique<CPDF_IndirectObjectHolder>(); |
| 97 | m_IndirectObjs = {m_ObjHolder->AddIndirectObject(boolean_true_obj->Clone()), |
| 98 | m_ObjHolder->AddIndirectObject(number_int_obj->Clone()), |
| 99 | m_ObjHolder->AddIndirectObject(str_spec_obj->Clone()), |
| 100 | m_ObjHolder->AddIndirectObject(name_obj->Clone()), |
| 101 | m_ObjHolder->AddIndirectObject(m_ArrayObj->Clone()), |
| 102 | m_ObjHolder->AddIndirectObject(m_DictObj->Clone()), |
| 103 | m_ObjHolder->AddIndirectObject(stream_obj->Clone())}; |
| 104 | for (CPDF_Object* pObj : m_IndirectObjs) { |
| 105 | m_RefObjs.emplace_back( |
| 106 | new CPDF_Reference(m_ObjHolder.get(), pObj->GetObjNum())); |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 107 | } |
| 108 | } |
| 109 | |
tsepez | 836f7d5 | 2016-10-10 14:31:05 -0700 | [diff] [blame] | 110 | bool Equal(const CPDF_Object* obj1, const CPDF_Object* obj2) { |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 111 | if (obj1 == obj2) |
| 112 | return true; |
| 113 | if (!obj1 || !obj2 || obj1->GetType() != obj2->GetType()) |
| 114 | return false; |
| 115 | switch (obj1->GetType()) { |
| 116 | case CPDF_Object::BOOLEAN: |
| 117 | return obj1->GetInteger() == obj2->GetInteger(); |
| 118 | case CPDF_Object::NUMBER: |
| 119 | return obj1->AsNumber()->IsInteger() == obj2->AsNumber()->IsInteger() && |
| 120 | obj1->GetInteger() == obj2->GetInteger(); |
| 121 | case CPDF_Object::STRING: |
| 122 | case CPDF_Object::NAME: |
| 123 | return obj1->GetString() == obj2->GetString(); |
| 124 | case CPDF_Object::ARRAY: { |
| 125 | const CPDF_Array* array1 = obj1->AsArray(); |
| 126 | const CPDF_Array* array2 = obj2->AsArray(); |
| 127 | if (array1->GetCount() != array2->GetCount()) |
| 128 | return false; |
| 129 | for (size_t i = 0; i < array1->GetCount(); ++i) { |
tsepez | bd56755 | 2016-03-29 14:51:50 -0700 | [diff] [blame] | 130 | if (!Equal(array1->GetObjectAt(i), array2->GetObjectAt(i))) |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 131 | return false; |
| 132 | } |
| 133 | return true; |
| 134 | } |
| 135 | case CPDF_Object::DICTIONARY: { |
| 136 | const CPDF_Dictionary* dict1 = obj1->AsDictionary(); |
| 137 | const CPDF_Dictionary* dict2 = obj2->AsDictionary(); |
| 138 | if (dict1->GetCount() != dict2->GetCount()) |
| 139 | return false; |
| 140 | for (CPDF_Dictionary::const_iterator it = dict1->begin(); |
| 141 | it != dict1->end(); ++it) { |
tsepez | 0e606b5 | 2016-11-18 16:22:41 -0800 | [diff] [blame] | 142 | if (!Equal(it->second.get(), dict2->GetObjectFor(it->first))) |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 143 | return false; |
| 144 | } |
| 145 | return true; |
| 146 | } |
| 147 | case CPDF_Object::NULLOBJ: |
| 148 | return true; |
| 149 | case CPDF_Object::STREAM: { |
| 150 | const CPDF_Stream* stream1 = obj1->AsStream(); |
| 151 | const CPDF_Stream* stream2 = obj2->AsStream(); |
| 152 | if (!stream1->GetDict() && !stream2->GetDict()) |
| 153 | return true; |
| 154 | // Compare dictionaries. |
| 155 | if (!Equal(stream1->GetDict(), stream2->GetDict())) |
| 156 | return false; |
| 157 | // Compare sizes. |
| 158 | if (stream1->GetRawSize() != stream2->GetRawSize()) |
| 159 | return false; |
| 160 | // Compare contents. |
| 161 | // Since this function is used for testing Clone(), only memory based |
| 162 | // streams need to be handled. |
| 163 | if (!stream1->IsMemoryBased() || !stream2->IsMemoryBased()) |
| 164 | return false; |
Dan Sinclair | 1c5d0b4 | 2017-04-03 15:05:11 -0400 | [diff] [blame^] | 165 | return memcmp(stream1->GetRawData(), stream2->GetRawData(), |
| 166 | stream1->GetRawSize()) == 0; |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 167 | } |
| 168 | case CPDF_Object::REFERENCE: |
| 169 | return obj1->AsReference()->GetRefObjNum() == |
| 170 | obj2->AsReference()->GetRefObjNum(); |
| 171 | } |
| 172 | return false; |
| 173 | } |
| 174 | |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 175 | protected: |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 176 | // m_ObjHolder needs to be declared first and destructed last since it also |
| 177 | // refers to some objects in m_DirectObjs. |
| 178 | std::unique_ptr<CPDF_IndirectObjectHolder> m_ObjHolder; |
tsepez | 33fdebc | 2016-11-04 11:38:40 -0700 | [diff] [blame] | 179 | std::vector<std::unique_ptr<CPDF_Object>> m_DirectObjs; |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 180 | std::vector<int> m_DirectObjTypes; |
tsepez | 33fdebc | 2016-11-04 11:38:40 -0700 | [diff] [blame] | 181 | std::vector<std::unique_ptr<CPDF_Object>> m_RefObjs; |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 182 | CPDF_Dictionary* m_DictObj; |
| 183 | CPDF_Dictionary* m_StreamDictObj; |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 184 | CPDF_Array* m_ArrayObj; |
| 185 | std::vector<CPDF_Object*> m_IndirectObjs; |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 186 | }; |
| 187 | |
| 188 | TEST_F(PDFObjectsTest, GetString) { |
thestig | b8bf55f | 2016-05-21 21:08:05 -0700 | [diff] [blame] | 189 | const char* const direct_obj_results[] = { |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 190 | "false", "true", "1245", "9.00345", "A simple test", "\t\n", "space", |
| 191 | "", "", "", ""}; |
| 192 | // Check for direct objects. |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 193 | for (size_t i = 0; i < m_DirectObjs.size(); ++i) |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 194 | EXPECT_STREQ(direct_obj_results[i], m_DirectObjs[i]->GetString().c_str()); |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 195 | |
| 196 | // Check indirect references. |
thestig | b8bf55f | 2016-05-21 21:08:05 -0700 | [diff] [blame] | 197 | const char* const indirect_obj_results[] = {"true", "1245", "\t\n", "space", |
| 198 | "", "", ""}; |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 199 | for (size_t i = 0; i < m_RefObjs.size(); ++i) { |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 200 | EXPECT_STREQ(indirect_obj_results[i], m_RefObjs[i]->GetString().c_str()); |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 201 | } |
| 202 | } |
| 203 | |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 204 | TEST_F(PDFObjectsTest, GetUnicodeText) { |
thestig | b8bf55f | 2016-05-21 21:08:05 -0700 | [diff] [blame] | 205 | const wchar_t* const direct_obj_results[] = { |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 206 | L"", L"", L"", L"", L"A simple test", |
| 207 | L"\t\n", L"space", L"", L"", L"abcdefghijklmnopqrstuvwxyz", |
| 208 | L""}; |
| 209 | // Check for direct objects. |
dsinclair | 685bb88 | 2016-04-20 07:32:39 -0700 | [diff] [blame] | 210 | for (size_t i = 0; i < m_DirectObjs.size(); ++i) { |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 211 | EXPECT_STREQ(direct_obj_results[i], |
| 212 | m_DirectObjs[i]->GetUnicodeText().c_str()); |
dsinclair | 685bb88 | 2016-04-20 07:32:39 -0700 | [diff] [blame] | 213 | } |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 214 | |
| 215 | // Check indirect references. |
| 216 | for (const auto& it : m_RefObjs) |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 217 | EXPECT_STREQ(L"", it->GetUnicodeText().c_str()); |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 218 | } |
| 219 | |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 220 | TEST_F(PDFObjectsTest, GetNumber) { |
Dan Sinclair | 05df075 | 2017-03-14 14:43:42 -0400 | [diff] [blame] | 221 | const float direct_obj_results[] = {0, 0, 1245, 9.00345f, 0, 0, |
| 222 | 0, 0, 0, 0, 0}; |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 223 | // Check for direct objects. |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 224 | for (size_t i = 0; i < m_DirectObjs.size(); ++i) |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 225 | EXPECT_EQ(direct_obj_results[i], m_DirectObjs[i]->GetNumber()); |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 226 | |
| 227 | // Check indirect references. |
Dan Sinclair | 05df075 | 2017-03-14 14:43:42 -0400 | [diff] [blame] | 228 | const float indirect_obj_results[] = {0, 1245, 0, 0, 0, 0, 0}; |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 229 | for (size_t i = 0; i < m_RefObjs.size(); ++i) |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 230 | EXPECT_EQ(indirect_obj_results[i], m_RefObjs[i]->GetNumber()); |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 231 | } |
| 232 | |
| 233 | TEST_F(PDFObjectsTest, GetInteger) { |
| 234 | const int direct_obj_results[] = {0, 1, 1245, 9, 0, 0, 0, 0, 0, 0, 0}; |
| 235 | // Check for direct objects. |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 236 | for (size_t i = 0; i < m_DirectObjs.size(); ++i) |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 237 | EXPECT_EQ(direct_obj_results[i], m_DirectObjs[i]->GetInteger()); |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 238 | |
| 239 | // Check indirect references. |
| 240 | const int indirect_obj_results[] = {1, 1245, 0, 0, 0, 0, 0}; |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 241 | for (size_t i = 0; i < m_RefObjs.size(); ++i) |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 242 | EXPECT_EQ(indirect_obj_results[i], m_RefObjs[i]->GetInteger()); |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 243 | } |
| 244 | |
| 245 | TEST_F(PDFObjectsTest, GetDict) { |
thestig | b8bf55f | 2016-05-21 21:08:05 -0700 | [diff] [blame] | 246 | const CPDF_Dictionary* const direct_obj_results[] = { |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 247 | nullptr, nullptr, nullptr, nullptr, nullptr, nullptr, |
| 248 | nullptr, nullptr, m_DictObj, m_StreamDictObj, nullptr}; |
| 249 | // Check for direct objects. |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 250 | for (size_t i = 0; i < m_DirectObjs.size(); ++i) |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 251 | EXPECT_EQ(direct_obj_results[i], m_DirectObjs[i]->GetDict()); |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 252 | |
| 253 | // Check indirect references. |
thestig | b8bf55f | 2016-05-21 21:08:05 -0700 | [diff] [blame] | 254 | const CPDF_Dictionary* const indirect_obj_results[] = { |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 255 | nullptr, nullptr, nullptr, nullptr, nullptr, m_DictObj, m_StreamDictObj}; |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 256 | for (size_t i = 0; i < m_RefObjs.size(); ++i) |
tsepez | 836f7d5 | 2016-10-10 14:31:05 -0700 | [diff] [blame] | 257 | EXPECT_TRUE(Equal(indirect_obj_results[i], m_RefObjs[i]->GetDict())); |
Lei Zhang | 9429368 | 2016-01-27 18:27:56 -0800 | [diff] [blame] | 258 | } |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 259 | |
| 260 | TEST_F(PDFObjectsTest, GetArray) { |
thestig | b8bf55f | 2016-05-21 21:08:05 -0700 | [diff] [blame] | 261 | const CPDF_Array* const direct_obj_results[] = { |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 262 | nullptr, nullptr, nullptr, nullptr, nullptr, nullptr, |
| 263 | nullptr, m_ArrayObj, nullptr, nullptr, nullptr}; |
| 264 | // Check for direct objects. |
| 265 | for (size_t i = 0; i < m_DirectObjs.size(); ++i) |
thestig | b8bf55f | 2016-05-21 21:08:05 -0700 | [diff] [blame] | 266 | EXPECT_EQ(direct_obj_results[i], m_DirectObjs[i]->AsArray()); |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 267 | |
| 268 | // Check indirect references. |
| 269 | for (const auto& it : m_RefObjs) |
thestig | b8bf55f | 2016-05-21 21:08:05 -0700 | [diff] [blame] | 270 | EXPECT_EQ(nullptr, it->AsArray()); |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 271 | } |
| 272 | |
| 273 | TEST_F(PDFObjectsTest, Clone) { |
| 274 | // Check for direct objects. |
| 275 | for (size_t i = 0; i < m_DirectObjs.size(); ++i) { |
tsepez | 335cf09 | 2016-11-09 13:28:26 -0800 | [diff] [blame] | 276 | std::unique_ptr<CPDF_Object> obj = m_DirectObjs[i]->Clone(); |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 277 | EXPECT_TRUE(Equal(m_DirectObjs[i].get(), obj.get())); |
| 278 | } |
| 279 | |
| 280 | // Check indirect references. |
| 281 | for (const auto& it : m_RefObjs) { |
tsepez | 335cf09 | 2016-11-09 13:28:26 -0800 | [diff] [blame] | 282 | std::unique_ptr<CPDF_Object> obj = it->Clone(); |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 283 | EXPECT_TRUE(Equal(it.get(), obj.get())); |
| 284 | } |
| 285 | } |
| 286 | |
| 287 | TEST_F(PDFObjectsTest, GetType) { |
| 288 | // Check for direct objects. |
| 289 | for (size_t i = 0; i < m_DirectObjs.size(); ++i) |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 290 | EXPECT_EQ(m_DirectObjTypes[i], m_DirectObjs[i]->GetType()); |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 291 | |
| 292 | // Check indirect references. |
| 293 | for (const auto& it : m_RefObjs) |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 294 | EXPECT_EQ(CPDF_Object::REFERENCE, it->GetType()); |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 295 | } |
| 296 | |
| 297 | TEST_F(PDFObjectsTest, GetDirect) { |
| 298 | // Check for direct objects. |
| 299 | for (size_t i = 0; i < m_DirectObjs.size(); ++i) |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 300 | EXPECT_EQ(m_DirectObjs[i].get(), m_DirectObjs[i]->GetDirect()); |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 301 | |
| 302 | // Check indirect references. |
| 303 | for (size_t i = 0; i < m_RefObjs.size(); ++i) |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 304 | EXPECT_EQ(m_IndirectObjs[i], m_RefObjs[i]->GetDirect()); |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 305 | } |
| 306 | |
| 307 | TEST_F(PDFObjectsTest, SetString) { |
| 308 | // Check for direct objects. |
| 309 | const char* const set_values[] = {"true", "fake", "3.125f", "097", |
| 310 | "changed", "", "NewName"}; |
thestig | b8bf55f | 2016-05-21 21:08:05 -0700 | [diff] [blame] | 311 | const char* const expected[] = {"true", "false", "3.125", "97", |
| 312 | "changed", "", "NewName"}; |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 313 | for (size_t i = 0; i < FX_ArraySize(set_values); ++i) { |
| 314 | m_DirectObjs[i]->SetString(set_values[i]); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 315 | EXPECT_STREQ(expected[i], m_DirectObjs[i]->GetString().c_str()); |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 316 | } |
| 317 | } |
| 318 | |
| 319 | TEST_F(PDFObjectsTest, IsTypeAndAsType) { |
| 320 | // Check for direct objects. |
| 321 | for (size_t i = 0; i < m_DirectObjs.size(); ++i) { |
| 322 | if (m_DirectObjTypes[i] == CPDF_Object::ARRAY) { |
| 323 | EXPECT_TRUE(m_DirectObjs[i]->IsArray()); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 324 | EXPECT_EQ(m_DirectObjs[i].get(), m_DirectObjs[i]->AsArray()); |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 325 | } else { |
| 326 | EXPECT_FALSE(m_DirectObjs[i]->IsArray()); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 327 | EXPECT_EQ(nullptr, m_DirectObjs[i]->AsArray()); |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 328 | } |
| 329 | |
| 330 | if (m_DirectObjTypes[i] == CPDF_Object::BOOLEAN) { |
| 331 | EXPECT_TRUE(m_DirectObjs[i]->IsBoolean()); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 332 | EXPECT_EQ(m_DirectObjs[i].get(), m_DirectObjs[i]->AsBoolean()); |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 333 | } else { |
| 334 | EXPECT_FALSE(m_DirectObjs[i]->IsBoolean()); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 335 | EXPECT_EQ(nullptr, m_DirectObjs[i]->AsBoolean()); |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 336 | } |
| 337 | |
| 338 | if (m_DirectObjTypes[i] == CPDF_Object::NAME) { |
| 339 | EXPECT_TRUE(m_DirectObjs[i]->IsName()); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 340 | EXPECT_EQ(m_DirectObjs[i].get(), m_DirectObjs[i]->AsName()); |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 341 | } else { |
| 342 | EXPECT_FALSE(m_DirectObjs[i]->IsName()); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 343 | EXPECT_EQ(nullptr, m_DirectObjs[i]->AsName()); |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 344 | } |
| 345 | |
| 346 | if (m_DirectObjTypes[i] == CPDF_Object::NUMBER) { |
| 347 | EXPECT_TRUE(m_DirectObjs[i]->IsNumber()); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 348 | EXPECT_EQ(m_DirectObjs[i].get(), m_DirectObjs[i]->AsNumber()); |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 349 | } else { |
| 350 | EXPECT_FALSE(m_DirectObjs[i]->IsNumber()); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 351 | EXPECT_EQ(nullptr, m_DirectObjs[i]->AsNumber()); |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 352 | } |
| 353 | |
| 354 | if (m_DirectObjTypes[i] == CPDF_Object::STRING) { |
| 355 | EXPECT_TRUE(m_DirectObjs[i]->IsString()); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 356 | EXPECT_EQ(m_DirectObjs[i].get(), m_DirectObjs[i]->AsString()); |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 357 | } else { |
| 358 | EXPECT_FALSE(m_DirectObjs[i]->IsString()); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 359 | EXPECT_EQ(nullptr, m_DirectObjs[i]->AsString()); |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 360 | } |
| 361 | |
| 362 | if (m_DirectObjTypes[i] == CPDF_Object::DICTIONARY) { |
| 363 | EXPECT_TRUE(m_DirectObjs[i]->IsDictionary()); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 364 | EXPECT_EQ(m_DirectObjs[i].get(), m_DirectObjs[i]->AsDictionary()); |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 365 | } else { |
| 366 | EXPECT_FALSE(m_DirectObjs[i]->IsDictionary()); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 367 | EXPECT_EQ(nullptr, m_DirectObjs[i]->AsDictionary()); |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 368 | } |
| 369 | |
| 370 | if (m_DirectObjTypes[i] == CPDF_Object::STREAM) { |
| 371 | EXPECT_TRUE(m_DirectObjs[i]->IsStream()); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 372 | EXPECT_EQ(m_DirectObjs[i].get(), m_DirectObjs[i]->AsStream()); |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 373 | } else { |
| 374 | EXPECT_FALSE(m_DirectObjs[i]->IsStream()); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 375 | EXPECT_EQ(nullptr, m_DirectObjs[i]->AsStream()); |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 376 | } |
| 377 | |
| 378 | EXPECT_FALSE(m_DirectObjs[i]->IsReference()); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 379 | EXPECT_EQ(nullptr, m_DirectObjs[i]->AsReference()); |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 380 | } |
| 381 | // Check indirect references. |
| 382 | for (size_t i = 0; i < m_RefObjs.size(); ++i) { |
| 383 | EXPECT_TRUE(m_RefObjs[i]->IsReference()); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 384 | EXPECT_EQ(m_RefObjs[i].get(), m_RefObjs[i]->AsReference()); |
| 385 | } |
| 386 | } |
| 387 | |
| 388 | TEST(PDFArrayTest, GetMatrix) { |
| 389 | float elems[][6] = {{0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f}, |
| 390 | {1, 2, 3, 4, 5, 6}, |
| 391 | {2.3f, 4.05f, 3, -2, -3, 0.0f}, |
| 392 | {0.05f, 0.1f, 0.56f, 0.67f, 1.34f, 99.9f}}; |
| 393 | for (size_t i = 0; i < FX_ArraySize(elems); ++i) { |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 394 | auto arr = pdfium::MakeUnique<CPDF_Array>(); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 395 | CFX_Matrix matrix(elems[i][0], elems[i][1], elems[i][2], elems[i][3], |
| 396 | elems[i][4], elems[i][5]); |
| 397 | for (size_t j = 0; j < 6; ++j) |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 398 | arr->AddNew<CPDF_Number>(elems[i][j]); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 399 | CFX_Matrix arr_matrix = arr->GetMatrix(); |
Dan Sinclair | 687a79c | 2017-02-09 14:08:53 -0500 | [diff] [blame] | 400 | EXPECT_EQ(matrix.a, arr_matrix.a); |
| 401 | EXPECT_EQ(matrix.b, arr_matrix.b); |
| 402 | EXPECT_EQ(matrix.c, arr_matrix.c); |
| 403 | EXPECT_EQ(matrix.d, arr_matrix.d); |
| 404 | EXPECT_EQ(matrix.e, arr_matrix.e); |
| 405 | EXPECT_EQ(matrix.f, arr_matrix.f); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 406 | } |
| 407 | } |
| 408 | |
| 409 | TEST(PDFArrayTest, GetRect) { |
| 410 | float elems[][4] = {{0.0f, 0.0f, 0.0f, 0.0f}, |
| 411 | {1, 2, 5, 6}, |
| 412 | {2.3f, 4.05f, -3, 0.0f}, |
| 413 | {0.05f, 0.1f, 1.34f, 99.9f}}; |
| 414 | for (size_t i = 0; i < FX_ArraySize(elems); ++i) { |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 415 | auto arr = pdfium::MakeUnique<CPDF_Array>(); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 416 | CFX_FloatRect rect(elems[i]); |
| 417 | for (size_t j = 0; j < 4; ++j) |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 418 | arr->AddNew<CPDF_Number>(elems[i][j]); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 419 | CFX_FloatRect arr_rect = arr->GetRect(); |
| 420 | EXPECT_EQ(rect.left, arr_rect.left); |
| 421 | EXPECT_EQ(rect.right, arr_rect.right); |
| 422 | EXPECT_EQ(rect.bottom, arr_rect.bottom); |
| 423 | EXPECT_EQ(rect.top, arr_rect.top); |
| 424 | } |
| 425 | } |
| 426 | |
| 427 | TEST(PDFArrayTest, GetTypeAt) { |
| 428 | { |
| 429 | // Boolean array. |
thestig | b8bf55f | 2016-05-21 21:08:05 -0700 | [diff] [blame] | 430 | const bool vals[] = {true, false, false, true, true}; |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 431 | auto arr = pdfium::MakeUnique<CPDF_Array>(); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 432 | for (size_t i = 0; i < FX_ArraySize(vals); ++i) |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 433 | arr->InsertNewAt<CPDF_Boolean>(i, vals[i]); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 434 | for (size_t i = 0; i < FX_ArraySize(vals); ++i) { |
| 435 | TestArrayAccessors(arr.get(), i, // Array and index. |
| 436 | vals[i] ? "true" : "false", // String value. |
| 437 | nullptr, // Const string value. |
| 438 | vals[i] ? 1 : 0, // Integer value. |
| 439 | 0, // Float value. |
| 440 | nullptr, // Array value. |
| 441 | nullptr, // Dictionary value. |
| 442 | nullptr); // Stream value. |
| 443 | } |
| 444 | } |
| 445 | { |
| 446 | // Integer array. |
thestig | b8bf55f | 2016-05-21 21:08:05 -0700 | [diff] [blame] | 447 | const int vals[] = {10, 0, -345, 2089345456, -1000000000, 567, 93658767}; |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 448 | auto arr = pdfium::MakeUnique<CPDF_Array>(); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 449 | for (size_t i = 0; i < FX_ArraySize(vals); ++i) |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 450 | arr->InsertNewAt<CPDF_Number>(i, vals[i]); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 451 | for (size_t i = 0; i < FX_ArraySize(vals); ++i) { |
| 452 | char buf[33]; |
| 453 | TestArrayAccessors(arr.get(), i, // Array and index. |
| 454 | FXSYS_itoa(vals[i], buf, 10), // String value. |
| 455 | nullptr, // Const string value. |
| 456 | vals[i], // Integer value. |
| 457 | vals[i], // Float value. |
| 458 | nullptr, // Array value. |
| 459 | nullptr, // Dictionary value. |
| 460 | nullptr); // Stream value. |
| 461 | } |
| 462 | } |
| 463 | { |
| 464 | // Float array. |
thestig | b8bf55f | 2016-05-21 21:08:05 -0700 | [diff] [blame] | 465 | const float vals[] = {0.0f, 0, 10, 10.0f, 0.0345f, |
| 466 | 897.34f, -2.5f, -1.0f, -345.0f, -0.0f}; |
| 467 | const char* const expected_str[] = { |
| 468 | "0", "0", "10", "10", "0.0345", "897.34", "-2.5", "-1", "-345", "0"}; |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 469 | auto arr = pdfium::MakeUnique<CPDF_Array>(); |
| 470 | for (size_t i = 0; i < FX_ArraySize(vals); ++i) |
| 471 | arr->InsertNewAt<CPDF_Number>(i, vals[i]); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 472 | for (size_t i = 0; i < FX_ArraySize(vals); ++i) { |
| 473 | TestArrayAccessors(arr.get(), i, // Array and index. |
| 474 | expected_str[i], // String value. |
| 475 | nullptr, // Const string value. |
| 476 | vals[i], // Integer value. |
| 477 | vals[i], // Float value. |
| 478 | nullptr, // Array value. |
| 479 | nullptr, // Dictionary value. |
| 480 | nullptr); // Stream value. |
| 481 | } |
| 482 | } |
| 483 | { |
| 484 | // String and name array |
| 485 | const char* const vals[] = {"this", "adsde$%^", "\r\t", "\"012", |
| 486 | ".", "EYREW", "It is a joke :)"}; |
Dan Sinclair | 0bb1333 | 2017-03-30 16:12:02 -0400 | [diff] [blame] | 487 | auto string_array = pdfium::MakeUnique<CPDF_Array>(); |
| 488 | auto name_array = pdfium::MakeUnique<CPDF_Array>(); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 489 | for (size_t i = 0; i < FX_ArraySize(vals); ++i) { |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 490 | string_array->InsertNewAt<CPDF_String>(i, vals[i], false); |
| 491 | name_array->InsertNewAt<CPDF_Name>(i, vals[i]); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 492 | } |
| 493 | for (size_t i = 0; i < FX_ArraySize(vals); ++i) { |
| 494 | TestArrayAccessors(string_array.get(), i, // Array and index. |
| 495 | vals[i], // String value. |
| 496 | vals[i], // Const string value. |
| 497 | 0, // Integer value. |
| 498 | 0, // Float value. |
| 499 | nullptr, // Array value. |
| 500 | nullptr, // Dictionary value. |
| 501 | nullptr); // Stream value. |
| 502 | TestArrayAccessors(name_array.get(), i, // Array and index. |
| 503 | vals[i], // String value. |
| 504 | vals[i], // Const string value. |
| 505 | 0, // Integer value. |
| 506 | 0, // Float value. |
| 507 | nullptr, // Array value. |
| 508 | nullptr, // Dictionary value. |
| 509 | nullptr); // Stream value. |
| 510 | } |
| 511 | } |
| 512 | { |
| 513 | // Null element array. |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 514 | auto arr = pdfium::MakeUnique<CPDF_Array>(); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 515 | for (size_t i = 0; i < 3; ++i) |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 516 | arr->InsertNewAt<CPDF_Null>(i); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 517 | for (size_t i = 0; i < 3; ++i) { |
| 518 | TestArrayAccessors(arr.get(), i, // Array and index. |
| 519 | "", // String value. |
| 520 | nullptr, // Const string value. |
| 521 | 0, // Integer value. |
| 522 | 0, // Float value. |
| 523 | nullptr, // Array value. |
| 524 | nullptr, // Dictionary value. |
| 525 | nullptr); // Stream value. |
| 526 | } |
| 527 | } |
| 528 | { |
| 529 | // Array of array. |
| 530 | CPDF_Array* vals[3]; |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 531 | auto arr = pdfium::MakeUnique<CPDF_Array>(); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 532 | for (size_t i = 0; i < 3; ++i) { |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 533 | vals[i] = arr->AddNew<CPDF_Array>(); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 534 | for (size_t j = 0; j < 3; ++j) { |
| 535 | int value = j + 100; |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 536 | vals[i]->InsertNewAt<CPDF_Number>(i, value); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 537 | } |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 538 | } |
| 539 | for (size_t i = 0; i < 3; ++i) { |
| 540 | TestArrayAccessors(arr.get(), i, // Array and index. |
| 541 | "", // String value. |
| 542 | nullptr, // Const string value. |
| 543 | 0, // Integer value. |
| 544 | 0, // Float value. |
| 545 | vals[i], // Array value. |
| 546 | nullptr, // Dictionary value. |
| 547 | nullptr); // Stream value. |
| 548 | } |
| 549 | } |
| 550 | { |
| 551 | // Dictionary array. |
| 552 | CPDF_Dictionary* vals[3]; |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 553 | auto arr = pdfium::MakeUnique<CPDF_Array>(); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 554 | for (size_t i = 0; i < 3; ++i) { |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 555 | vals[i] = arr->AddNew<CPDF_Dictionary>(); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 556 | for (size_t j = 0; j < 3; ++j) { |
| 557 | std::string key("key"); |
| 558 | char buf[33]; |
| 559 | key.append(FXSYS_itoa(j, buf, 10)); |
| 560 | int value = j + 200; |
tsepez | 0e606b5 | 2016-11-18 16:22:41 -0800 | [diff] [blame] | 561 | vals[i]->SetNewFor<CPDF_Number>(key.c_str(), value); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 562 | } |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 563 | } |
| 564 | for (size_t i = 0; i < 3; ++i) { |
| 565 | TestArrayAccessors(arr.get(), i, // Array and index. |
| 566 | "", // String value. |
| 567 | nullptr, // Const string value. |
| 568 | 0, // Integer value. |
| 569 | 0, // Float value. |
| 570 | nullptr, // Array value. |
| 571 | vals[i], // Dictionary value. |
| 572 | nullptr); // Stream value. |
| 573 | } |
| 574 | } |
| 575 | { |
| 576 | // Stream array. |
| 577 | CPDF_Dictionary* vals[3]; |
| 578 | CPDF_Stream* stream_vals[3]; |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 579 | auto arr = pdfium::MakeUnique<CPDF_Array>(); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 580 | for (size_t i = 0; i < 3; ++i) { |
tsepez | cd5bca4 | 2016-09-30 10:45:06 -0700 | [diff] [blame] | 581 | vals[i] = new CPDF_Dictionary(); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 582 | for (size_t j = 0; j < 3; ++j) { |
| 583 | std::string key("key"); |
| 584 | char buf[33]; |
| 585 | key.append(FXSYS_itoa(j, buf, 10)); |
| 586 | int value = j + 200; |
tsepez | 0e606b5 | 2016-11-18 16:22:41 -0800 | [diff] [blame] | 587 | vals[i]->SetNewFor<CPDF_Number>(key.c_str(), value); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 588 | } |
| 589 | uint8_t content[] = "content: this is a stream"; |
| 590 | size_t data_size = FX_ArraySize(content); |
tsepez | 47fb8c0 | 2016-12-15 13:51:34 -0800 | [diff] [blame] | 591 | std::unique_ptr<uint8_t, FxFreeDeleter> data( |
| 592 | FX_Alloc(uint8_t, data_size)); |
| 593 | memcpy(data.get(), content, data_size); |
| 594 | stream_vals[i] = arr->AddNew<CPDF_Stream>(std::move(data), data_size, |
tsepez | 9e05ee1 | 2016-11-21 13:19:10 -0800 | [diff] [blame] | 595 | pdfium::WrapUnique(vals[i])); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 596 | } |
| 597 | for (size_t i = 0; i < 3; ++i) { |
| 598 | TestArrayAccessors(arr.get(), i, // Array and index. |
| 599 | "", // String value. |
| 600 | nullptr, // Const string value. |
| 601 | 0, // Integer value. |
| 602 | 0, // Float value. |
| 603 | nullptr, // Array value. |
| 604 | vals[i], // Dictionary value. |
| 605 | stream_vals[i]); // Stream value. |
| 606 | } |
| 607 | } |
| 608 | { |
| 609 | // Mixed array. |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 610 | auto arr = pdfium::MakeUnique<CPDF_Array>(); |
| 611 | arr->InsertNewAt<CPDF_Boolean>(0, true); |
| 612 | arr->InsertNewAt<CPDF_Boolean>(1, false); |
| 613 | arr->InsertNewAt<CPDF_Number>(2, 0); |
| 614 | arr->InsertNewAt<CPDF_Number>(3, -1234); |
| 615 | arr->InsertNewAt<CPDF_Number>(4, 2345.0f); |
| 616 | arr->InsertNewAt<CPDF_Number>(5, 0.05f); |
| 617 | arr->InsertNewAt<CPDF_String>(6, "", false); |
| 618 | arr->InsertNewAt<CPDF_String>(7, "It is a test!", false); |
| 619 | arr->InsertNewAt<CPDF_Name>(8, "NAME"); |
| 620 | arr->InsertNewAt<CPDF_Name>(9, "test"); |
| 621 | arr->InsertNewAt<CPDF_Null>(10); |
| 622 | |
| 623 | CPDF_Array* arr_val = arr->InsertNewAt<CPDF_Array>(11); |
| 624 | arr_val->AddNew<CPDF_Number>(1); |
| 625 | arr_val->AddNew<CPDF_Number>(2); |
| 626 | |
| 627 | CPDF_Dictionary* dict_val = arr->InsertNewAt<CPDF_Dictionary>(12); |
tsepez | 0e606b5 | 2016-11-18 16:22:41 -0800 | [diff] [blame] | 628 | dict_val->SetNewFor<CPDF_String>("key1", "Linda", false); |
| 629 | dict_val->SetNewFor<CPDF_String>("key2", "Zoe", false); |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 630 | |
tsepez | cd5bca4 | 2016-09-30 10:45:06 -0700 | [diff] [blame] | 631 | CPDF_Dictionary* stream_dict = new CPDF_Dictionary(); |
tsepez | 0e606b5 | 2016-11-18 16:22:41 -0800 | [diff] [blame] | 632 | stream_dict->SetNewFor<CPDF_String>("key1", "John", false); |
| 633 | stream_dict->SetNewFor<CPDF_String>("key2", "King", false); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 634 | uint8_t data[] = "A stream for test"; |
| 635 | // The data buffer will be owned by stream object, so it needs to be |
| 636 | // dynamically allocated. |
| 637 | size_t buf_size = sizeof(data); |
tsepez | 47fb8c0 | 2016-12-15 13:51:34 -0800 | [diff] [blame] | 638 | std::unique_ptr<uint8_t, FxFreeDeleter> buf(FX_Alloc(uint8_t, buf_size)); |
| 639 | memcpy(buf.get(), data, buf_size); |
tsepez | 9e05ee1 | 2016-11-21 13:19:10 -0800 | [diff] [blame] | 640 | CPDF_Stream* stream_val = arr->InsertNewAt<CPDF_Stream>( |
tsepez | 47fb8c0 | 2016-12-15 13:51:34 -0800 | [diff] [blame] | 641 | 13, std::move(buf), buf_size, pdfium::WrapUnique(stream_dict)); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 642 | const char* const expected_str[] = { |
| 643 | "true", "false", "0", "-1234", "2345", "0.05", "", |
| 644 | "It is a test!", "NAME", "test", "", "", "", ""}; |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 645 | const int expected_int[] = {1, 0, 0, -1234, 2345, 0, 0, |
| 646 | 0, 0, 0, 0, 0, 0, 0}; |
| 647 | const float expected_float[] = {0, 0, 0, -1234, 2345, 0.05f, 0, |
| 648 | 0, 0, 0, 0, 0, 0, 0}; |
| 649 | for (size_t i = 0; i < arr->GetCount(); ++i) { |
| 650 | EXPECT_STREQ(expected_str[i], arr->GetStringAt(i).c_str()); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 651 | EXPECT_EQ(expected_int[i], arr->GetIntegerAt(i)); |
| 652 | EXPECT_EQ(expected_float[i], arr->GetNumberAt(i)); |
| 653 | EXPECT_EQ(expected_float[i], arr->GetFloatAt(i)); |
| 654 | if (i == 11) |
| 655 | EXPECT_EQ(arr_val, arr->GetArrayAt(i)); |
| 656 | else |
| 657 | EXPECT_EQ(nullptr, arr->GetArrayAt(i)); |
| 658 | if (i == 13) { |
| 659 | EXPECT_EQ(stream_dict, arr->GetDictAt(i)); |
| 660 | EXPECT_EQ(stream_val, arr->GetStreamAt(i)); |
| 661 | } else { |
| 662 | EXPECT_EQ(nullptr, arr->GetStreamAt(i)); |
| 663 | if (i == 12) |
| 664 | EXPECT_EQ(dict_val, arr->GetDictAt(i)); |
| 665 | else |
| 666 | EXPECT_EQ(nullptr, arr->GetDictAt(i)); |
| 667 | } |
| 668 | } |
| 669 | } |
| 670 | } |
| 671 | |
| 672 | TEST(PDFArrayTest, AddNumber) { |
| 673 | float vals[] = {1.0f, -1.0f, 0, 0.456734f, |
| 674 | 12345.54321f, 0.5f, 1000, 0.000045f}; |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 675 | auto arr = pdfium::MakeUnique<CPDF_Array>(); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 676 | for (size_t i = 0; i < FX_ArraySize(vals); ++i) |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 677 | arr->AddNew<CPDF_Number>(vals[i]); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 678 | for (size_t i = 0; i < FX_ArraySize(vals); ++i) { |
tsepez | bd56755 | 2016-03-29 14:51:50 -0700 | [diff] [blame] | 679 | EXPECT_EQ(CPDF_Object::NUMBER, arr->GetObjectAt(i)->GetType()); |
| 680 | EXPECT_EQ(vals[i], arr->GetObjectAt(i)->GetNumber()); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 681 | } |
| 682 | } |
| 683 | |
| 684 | TEST(PDFArrayTest, AddInteger) { |
| 685 | int vals[] = {0, 1, 934435456, 876, 10000, -1, -24354656, -100}; |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 686 | auto arr = pdfium::MakeUnique<CPDF_Array>(); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 687 | for (size_t i = 0; i < FX_ArraySize(vals); ++i) |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 688 | arr->AddNew<CPDF_Number>(vals[i]); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 689 | for (size_t i = 0; i < FX_ArraySize(vals); ++i) { |
tsepez | bd56755 | 2016-03-29 14:51:50 -0700 | [diff] [blame] | 690 | EXPECT_EQ(CPDF_Object::NUMBER, arr->GetObjectAt(i)->GetType()); |
| 691 | EXPECT_EQ(vals[i], arr->GetObjectAt(i)->GetNumber()); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 692 | } |
| 693 | } |
| 694 | |
| 695 | TEST(PDFArrayTest, AddStringAndName) { |
| 696 | const char* vals[] = {"", "a", "ehjhRIOYTTFdfcdnv", "122323", |
| 697 | "$#%^&**", " ", "This is a test.\r\n"}; |
Dan Sinclair | 0bb1333 | 2017-03-30 16:12:02 -0400 | [diff] [blame] | 698 | auto string_array = pdfium::MakeUnique<CPDF_Array>(); |
| 699 | auto name_array = pdfium::MakeUnique<CPDF_Array>(); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 700 | for (size_t i = 0; i < FX_ArraySize(vals); ++i) { |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 701 | string_array->AddNew<CPDF_String>(vals[i], false); |
| 702 | name_array->AddNew<CPDF_Name>(vals[i]); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 703 | } |
| 704 | for (size_t i = 0; i < FX_ArraySize(vals); ++i) { |
tsepez | bd56755 | 2016-03-29 14:51:50 -0700 | [diff] [blame] | 705 | EXPECT_EQ(CPDF_Object::STRING, string_array->GetObjectAt(i)->GetType()); |
| 706 | EXPECT_STREQ(vals[i], string_array->GetObjectAt(i)->GetString().c_str()); |
| 707 | EXPECT_EQ(CPDF_Object::NAME, name_array->GetObjectAt(i)->GetType()); |
| 708 | EXPECT_STREQ(vals[i], name_array->GetObjectAt(i)->GetString().c_str()); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 709 | } |
| 710 | } |
| 711 | |
tsepez | bd56755 | 2016-03-29 14:51:50 -0700 | [diff] [blame] | 712 | TEST(PDFArrayTest, AddReferenceAndGetObjectAt) { |
Dan Sinclair | 0bb1333 | 2017-03-30 16:12:02 -0400 | [diff] [blame] | 713 | auto holder = pdfium::MakeUnique<CPDF_IndirectObjectHolder>(); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 714 | CPDF_Boolean* boolean_obj = new CPDF_Boolean(true); |
| 715 | CPDF_Number* int_obj = new CPDF_Number(-1234); |
| 716 | CPDF_Number* float_obj = new CPDF_Number(2345.089f); |
tsepez | 5913a6c | 2016-11-16 17:31:18 -0800 | [diff] [blame] | 717 | CPDF_String* str_obj = |
| 718 | new CPDF_String(nullptr, "Adsfdsf 343434 %&&*\n", false); |
| 719 | CPDF_Name* name_obj = new CPDF_Name(nullptr, "Title:"); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 720 | CPDF_Null* null_obj = new CPDF_Null(); |
| 721 | CPDF_Object* indirect_objs[] = {boolean_obj, int_obj, float_obj, |
| 722 | str_obj, name_obj, null_obj}; |
| 723 | unsigned int obj_nums[] = {2, 4, 7, 2345, 799887, 1}; |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 724 | auto arr = pdfium::MakeUnique<CPDF_Array>(); |
Dan Sinclair | 0bb1333 | 2017-03-30 16:12:02 -0400 | [diff] [blame] | 725 | auto arr1 = pdfium::MakeUnique<CPDF_Array>(); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 726 | // Create two arrays of references by different AddReference() APIs. |
| 727 | for (size_t i = 0; i < FX_ArraySize(indirect_objs); ++i) { |
tsepez | 931d087 | 2016-11-04 16:00:03 -0700 | [diff] [blame] | 728 | holder->ReplaceIndirectObjectIfHigherGeneration( |
| 729 | obj_nums[i], pdfium::WrapUnique<CPDF_Object>(indirect_objs[i])); |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 730 | arr->AddNew<CPDF_Reference>(holder.get(), obj_nums[i]); |
| 731 | arr1->AddNew<CPDF_Reference>(holder.get(), indirect_objs[i]->GetObjNum()); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 732 | } |
| 733 | // Check indirect objects. |
| 734 | for (size_t i = 0; i < FX_ArraySize(obj_nums); ++i) |
dsinclair | 03bd7c7 | 2016-08-23 20:13:41 -0700 | [diff] [blame] | 735 | EXPECT_EQ(indirect_objs[i], holder->GetOrParseIndirectObject(obj_nums[i])); |
Wei Li | bb53a5d | 2016-02-03 10:04:19 -0800 | [diff] [blame] | 736 | // Check arrays. |
| 737 | EXPECT_EQ(arr->GetCount(), arr1->GetCount()); |
| 738 | for (size_t i = 0; i < arr->GetCount(); ++i) { |
tsepez | bd56755 | 2016-03-29 14:51:50 -0700 | [diff] [blame] | 739 | EXPECT_EQ(CPDF_Object::REFERENCE, arr->GetObjectAt(i)->GetType()); |
| 740 | EXPECT_EQ(indirect_objs[i], arr->GetObjectAt(i)->GetDirect()); |
| 741 | EXPECT_EQ(indirect_objs[i], arr->GetDirectObjectAt(i)); |
| 742 | EXPECT_EQ(CPDF_Object::REFERENCE, arr1->GetObjectAt(i)->GetType()); |
| 743 | EXPECT_EQ(indirect_objs[i], arr1->GetObjectAt(i)->GetDirect()); |
| 744 | EXPECT_EQ(indirect_objs[i], arr1->GetDirectObjectAt(i)); |
Wei Li | 8f95227 | 2016-01-29 18:04:42 -0800 | [diff] [blame] | 745 | } |
| 746 | } |
weili | a470b5e | 2016-08-23 22:08:37 -0700 | [diff] [blame] | 747 | |
thestig | 22b176d | 2016-08-25 11:14:21 -0700 | [diff] [blame] | 748 | TEST(PDFArrayTest, CloneDirectObject) { |
| 749 | CPDF_IndirectObjectHolder objects_holder; |
Dan Sinclair | 0bb1333 | 2017-03-30 16:12:02 -0400 | [diff] [blame] | 750 | auto array = pdfium::MakeUnique<CPDF_Array>(); |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 751 | array->AddNew<CPDF_Reference>(&objects_holder, 1234); |
thestig | 22b176d | 2016-08-25 11:14:21 -0700 | [diff] [blame] | 752 | ASSERT_EQ(1U, array->GetCount()); |
| 753 | CPDF_Object* obj = array->GetObjectAt(0); |
| 754 | ASSERT_TRUE(obj); |
| 755 | EXPECT_TRUE(obj->IsReference()); |
| 756 | |
tsepez | 335cf09 | 2016-11-09 13:28:26 -0800 | [diff] [blame] | 757 | std::unique_ptr<CPDF_Object> cloned_array_object = array->CloneDirectObject(); |
thestig | 22b176d | 2016-08-25 11:14:21 -0700 | [diff] [blame] | 758 | ASSERT_TRUE(cloned_array_object); |
| 759 | ASSERT_TRUE(cloned_array_object->IsArray()); |
| 760 | |
tsepez | 335cf09 | 2016-11-09 13:28:26 -0800 | [diff] [blame] | 761 | std::unique_ptr<CPDF_Array> cloned_array = |
| 762 | ToArray(std::move(cloned_array_object)); |
Wei Li | 6bdd824 | 2017-03-23 09:45:04 -0700 | [diff] [blame] | 763 | ASSERT_EQ(0U, cloned_array->GetCount()); |
thestig | 22b176d | 2016-08-25 11:14:21 -0700 | [diff] [blame] | 764 | CPDF_Object* cloned_obj = cloned_array->GetObjectAt(0); |
| 765 | EXPECT_FALSE(cloned_obj); |
| 766 | } |
| 767 | |
tsepez | 381fc83 | 2016-10-10 14:06:44 -0700 | [diff] [blame] | 768 | TEST(PDFArrayTest, ConvertIndirect) { |
| 769 | CPDF_IndirectObjectHolder objects_holder; |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 770 | auto array = pdfium::MakeUnique<CPDF_Array>(); |
| 771 | CPDF_Object* pObj = array->AddNew<CPDF_Number>(42); |
tsepez | 381fc83 | 2016-10-10 14:06:44 -0700 | [diff] [blame] | 772 | array->ConvertToIndirectObjectAt(0, &objects_holder); |
| 773 | CPDF_Object* pRef = array->GetObjectAt(0); |
| 774 | CPDF_Object* pNum = array->GetDirectObjectAt(0); |
| 775 | EXPECT_TRUE(pRef->IsReference()); |
| 776 | EXPECT_TRUE(pNum->IsNumber()); |
| 777 | EXPECT_NE(pObj, pRef); |
| 778 | EXPECT_EQ(pObj, pNum); |
| 779 | EXPECT_EQ(42, array->GetIntegerAt(0)); |
| 780 | } |
| 781 | |
thestig | 22b176d | 2016-08-25 11:14:21 -0700 | [diff] [blame] | 782 | TEST(PDFDictionaryTest, CloneDirectObject) { |
| 783 | CPDF_IndirectObjectHolder objects_holder; |
Dan Sinclair | 0bb1333 | 2017-03-30 16:12:02 -0400 | [diff] [blame] | 784 | auto dict = pdfium::MakeUnique<CPDF_Dictionary>(); |
tsepez | 0e606b5 | 2016-11-18 16:22:41 -0800 | [diff] [blame] | 785 | dict->SetNewFor<CPDF_Reference>("foo", &objects_holder, 1234); |
thestig | 22b176d | 2016-08-25 11:14:21 -0700 | [diff] [blame] | 786 | ASSERT_EQ(1U, dict->GetCount()); |
dsinclair | 38fd844 | 2016-09-15 10:15:32 -0700 | [diff] [blame] | 787 | CPDF_Object* obj = dict->GetObjectFor("foo"); |
thestig | 22b176d | 2016-08-25 11:14:21 -0700 | [diff] [blame] | 788 | ASSERT_TRUE(obj); |
| 789 | EXPECT_TRUE(obj->IsReference()); |
| 790 | |
tsepez | 335cf09 | 2016-11-09 13:28:26 -0800 | [diff] [blame] | 791 | std::unique_ptr<CPDF_Object> cloned_dict_object = dict->CloneDirectObject(); |
thestig | 22b176d | 2016-08-25 11:14:21 -0700 | [diff] [blame] | 792 | ASSERT_TRUE(cloned_dict_object); |
| 793 | ASSERT_TRUE(cloned_dict_object->IsDictionary()); |
| 794 | |
tsepez | 335cf09 | 2016-11-09 13:28:26 -0800 | [diff] [blame] | 795 | std::unique_ptr<CPDF_Dictionary> cloned_dict = |
| 796 | ToDictionary(std::move(cloned_dict_object)); |
Wei Li | 6bdd824 | 2017-03-23 09:45:04 -0700 | [diff] [blame] | 797 | ASSERT_EQ(0U, cloned_dict->GetCount()); |
dsinclair | 38fd844 | 2016-09-15 10:15:32 -0700 | [diff] [blame] | 798 | CPDF_Object* cloned_obj = cloned_dict->GetObjectFor("foo"); |
thestig | 22b176d | 2016-08-25 11:14:21 -0700 | [diff] [blame] | 799 | EXPECT_FALSE(cloned_obj); |
| 800 | } |
| 801 | |
weili | a470b5e | 2016-08-23 22:08:37 -0700 | [diff] [blame] | 802 | TEST(PDFObjectTest, CloneCheckLoop) { |
| 803 | { |
tsepez | 0e606b5 | 2016-11-18 16:22:41 -0800 | [diff] [blame] | 804 | // Create a dictionary/array pair with a reference loop. It takes |
| 805 | // some work to do this nowadays, in particular we need the |
| 806 | // anti-pattern pdfium::WrapUnique(arr.get()). |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 807 | auto arr_obj = pdfium::MakeUnique<CPDF_Array>(); |
| 808 | CPDF_Dictionary* dict_obj = arr_obj->InsertNewAt<CPDF_Dictionary>(0); |
tsepez | 0e606b5 | 2016-11-18 16:22:41 -0800 | [diff] [blame] | 809 | dict_obj->SetFor("arr", pdfium::WrapUnique(arr_obj.get())); |
weili | a470b5e | 2016-08-23 22:08:37 -0700 | [diff] [blame] | 810 | // Clone this object to see whether stack overflow will be triggered. |
tsepez | 335cf09 | 2016-11-09 13:28:26 -0800 | [diff] [blame] | 811 | std::unique_ptr<CPDF_Array> cloned_array = ToArray(arr_obj->Clone()); |
weili | a470b5e | 2016-08-23 22:08:37 -0700 | [diff] [blame] | 812 | // Cloned object should be the same as the original. |
| 813 | ASSERT_TRUE(cloned_array); |
| 814 | EXPECT_EQ(1u, cloned_array->GetCount()); |
| 815 | CPDF_Object* cloned_dict = cloned_array->GetObjectAt(0); |
| 816 | ASSERT_TRUE(cloned_dict); |
| 817 | ASSERT_TRUE(cloned_dict->IsDictionary()); |
| 818 | // Recursively referenced object is not cloned. |
dsinclair | 38fd844 | 2016-09-15 10:15:32 -0700 | [diff] [blame] | 819 | EXPECT_EQ(nullptr, cloned_dict->AsDictionary()->GetObjectFor("arr")); |
weili | a470b5e | 2016-08-23 22:08:37 -0700 | [diff] [blame] | 820 | } |
| 821 | { |
tsepez | 9e05ee1 | 2016-11-21 13:19:10 -0800 | [diff] [blame] | 822 | // Create a dictionary/stream pair with a reference loop. It takes |
| 823 | // some work to do this nowadays, in particular we need the |
| 824 | // anti-pattern pdfium::WrapUnique(dict.get()). |
tsepez | 0e606b5 | 2016-11-18 16:22:41 -0800 | [diff] [blame] | 825 | auto dict_obj = pdfium::MakeUnique<CPDF_Dictionary>(); |
tsepez | 9e05ee1 | 2016-11-21 13:19:10 -0800 | [diff] [blame] | 826 | CPDF_Stream* stream_obj = dict_obj->SetNewFor<CPDF_Stream>( |
| 827 | "stream", nullptr, 0, pdfium::WrapUnique(dict_obj.get())); |
tsepez | 836f7d5 | 2016-10-10 14:31:05 -0700 | [diff] [blame] | 828 | // Clone this object to see whether stack overflow will be triggered. |
tsepez | 335cf09 | 2016-11-09 13:28:26 -0800 | [diff] [blame] | 829 | std::unique_ptr<CPDF_Stream> cloned_stream = ToStream(stream_obj->Clone()); |
tsepez | 836f7d5 | 2016-10-10 14:31:05 -0700 | [diff] [blame] | 830 | // Cloned object should be the same as the original. |
| 831 | ASSERT_TRUE(cloned_stream); |
| 832 | CPDF_Object* cloned_dict = cloned_stream->GetDict(); |
| 833 | ASSERT_TRUE(cloned_dict); |
| 834 | ASSERT_TRUE(cloned_dict->IsDictionary()); |
| 835 | // Recursively referenced object is not cloned. |
| 836 | EXPECT_EQ(nullptr, cloned_dict->AsDictionary()->GetObjectFor("stream")); |
| 837 | } |
| 838 | { |
thestig | 22b176d | 2016-08-25 11:14:21 -0700 | [diff] [blame] | 839 | CPDF_IndirectObjectHolder objects_holder; |
weili | a470b5e | 2016-08-23 22:08:37 -0700 | [diff] [blame] | 840 | // Create an object with a reference loop. |
tsepez | 70c4afd | 2016-11-15 11:33:44 -0800 | [diff] [blame] | 841 | CPDF_Dictionary* dict_obj = objects_holder.NewIndirect<CPDF_Dictionary>(); |
| 842 | std::unique_ptr<CPDF_Array> arr_obj = pdfium::MakeUnique<CPDF_Array>(); |
tsepez | 8a3aa45 | 2016-11-16 12:26:06 -0800 | [diff] [blame] | 843 | arr_obj->InsertNewAt<CPDF_Reference>(0, &objects_holder, |
| 844 | dict_obj->GetObjNum()); |
weili | a470b5e | 2016-08-23 22:08:37 -0700 | [diff] [blame] | 845 | CPDF_Object* elem0 = arr_obj->GetObjectAt(0); |
tsepez | 0e606b5 | 2016-11-18 16:22:41 -0800 | [diff] [blame] | 846 | dict_obj->SetFor("arr", std::move(arr_obj)); |
tsepez | 70c4afd | 2016-11-15 11:33:44 -0800 | [diff] [blame] | 847 | EXPECT_EQ(1u, dict_obj->GetObjNum()); |
weili | a470b5e | 2016-08-23 22:08:37 -0700 | [diff] [blame] | 848 | ASSERT_TRUE(elem0); |
| 849 | ASSERT_TRUE(elem0->IsReference()); |
| 850 | EXPECT_EQ(1u, elem0->AsReference()->GetRefObjNum()); |
| 851 | EXPECT_EQ(dict_obj, elem0->AsReference()->GetDirect()); |
| 852 | |
| 853 | // Clone this object to see whether stack overflow will be triggered. |
tsepez | 335cf09 | 2016-11-09 13:28:26 -0800 | [diff] [blame] | 854 | std::unique_ptr<CPDF_Dictionary> cloned_dict = |
| 855 | ToDictionary(dict_obj->CloneDirectObject()); |
weili | a470b5e | 2016-08-23 22:08:37 -0700 | [diff] [blame] | 856 | // Cloned object should be the same as the original. |
| 857 | ASSERT_TRUE(cloned_dict); |
dsinclair | 38fd844 | 2016-09-15 10:15:32 -0700 | [diff] [blame] | 858 | CPDF_Object* cloned_arr = cloned_dict->GetObjectFor("arr"); |
weili | a470b5e | 2016-08-23 22:08:37 -0700 | [diff] [blame] | 859 | ASSERT_TRUE(cloned_arr); |
| 860 | ASSERT_TRUE(cloned_arr->IsArray()); |
Wei Li | 6bdd824 | 2017-03-23 09:45:04 -0700 | [diff] [blame] | 861 | EXPECT_EQ(0U, cloned_arr->AsArray()->GetCount()); |
weili | a470b5e | 2016-08-23 22:08:37 -0700 | [diff] [blame] | 862 | // Recursively referenced object is not cloned. |
| 863 | EXPECT_EQ(nullptr, cloned_arr->AsArray()->GetObjectAt(0)); |
| 864 | } |
| 865 | } |
tsepez | 381fc83 | 2016-10-10 14:06:44 -0700 | [diff] [blame] | 866 | |
| 867 | TEST(PDFDictionaryTest, ConvertIndirect) { |
| 868 | CPDF_IndirectObjectHolder objects_holder; |
Dan Sinclair | 0bb1333 | 2017-03-30 16:12:02 -0400 | [diff] [blame] | 869 | auto dict = pdfium::MakeUnique<CPDF_Dictionary>(); |
tsepez | 0e606b5 | 2016-11-18 16:22:41 -0800 | [diff] [blame] | 870 | CPDF_Object* pObj = dict->SetNewFor<CPDF_Number>("clams", 42); |
tsepez | 381fc83 | 2016-10-10 14:06:44 -0700 | [diff] [blame] | 871 | dict->ConvertToIndirectObjectFor("clams", &objects_holder); |
| 872 | CPDF_Object* pRef = dict->GetObjectFor("clams"); |
| 873 | CPDF_Object* pNum = dict->GetDirectObjectFor("clams"); |
| 874 | EXPECT_TRUE(pRef->IsReference()); |
| 875 | EXPECT_TRUE(pNum->IsNumber()); |
| 876 | EXPECT_NE(pObj, pRef); |
| 877 | EXPECT_EQ(pObj, pNum); |
| 878 | EXPECT_EQ(42, dict->GetIntegerFor("clams")); |
| 879 | } |