| OLD | NEW |
| 1 // Copyright 2016 PDFium Authors. All rights reserved. | 1 // Copyright 2016 PDFium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "core/fpdfapi/fpdf_parser/cpdf_boolean.h" | 5 #include "core/fpdfapi/fpdf_parser/cpdf_boolean.h" |
| 6 #include "core/fpdfapi/fpdf_parser/cpdf_null.h" | 6 #include "core/fpdfapi/fpdf_parser/cpdf_null.h" |
| 7 #include "core/fpdfapi/fpdf_parser/include/cpdf_array.h" | 7 #include "core/fpdfapi/fpdf_parser/include/cpdf_array.h" |
| 8 #include "core/fpdfapi/fpdf_parser/include/cpdf_dictionary.h" | 8 #include "core/fpdfapi/fpdf_parser/include/cpdf_dictionary.h" |
| 9 #include "core/fpdfapi/fpdf_parser/include/cpdf_name.h" | 9 #include "core/fpdfapi/fpdf_parser/include/cpdf_name.h" |
| 10 #include "core/fpdfapi/fpdf_parser/include/cpdf_number.h" | 10 #include "core/fpdfapi/fpdf_parser/include/cpdf_number.h" |
| (...skipping 76 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 87 stream_obj, null_obj}; | 87 stream_obj, null_obj}; |
| 88 m_DirectObjTypes = { | 88 m_DirectObjTypes = { |
| 89 CPDF_Object::BOOLEAN, CPDF_Object::BOOLEAN, CPDF_Object::NUMBER, | 89 CPDF_Object::BOOLEAN, CPDF_Object::BOOLEAN, CPDF_Object::NUMBER, |
| 90 CPDF_Object::NUMBER, CPDF_Object::STRING, CPDF_Object::STRING, | 90 CPDF_Object::NUMBER, CPDF_Object::STRING, CPDF_Object::STRING, |
| 91 CPDF_Object::NAME, CPDF_Object::ARRAY, CPDF_Object::DICTIONARY, | 91 CPDF_Object::NAME, CPDF_Object::ARRAY, CPDF_Object::DICTIONARY, |
| 92 CPDF_Object::STREAM, CPDF_Object::NULLOBJ}; | 92 CPDF_Object::STREAM, CPDF_Object::NULLOBJ}; |
| 93 for (size_t i = 0; i < FX_ArraySize(objs); ++i) | 93 for (size_t i = 0; i < FX_ArraySize(objs); ++i) |
| 94 m_DirectObjs.emplace_back(objs[i]); | 94 m_DirectObjs.emplace_back(objs[i]); |
| 95 | 95 |
| 96 // Indirect references to indirect objects. | 96 // Indirect references to indirect objects. |
| 97 m_ObjHolder.reset(new CPDF_IndirectObjectHolder(nullptr)); | 97 m_ObjHolder.reset(new CPDF_IndirectObjectHolder()); |
| 98 m_IndirectObjs = {boolean_true_obj, number_int_obj, str_spec_obj, name_obj, | 98 m_IndirectObjs = {boolean_true_obj, number_int_obj, str_spec_obj, name_obj, |
| 99 m_ArrayObj, m_DictObj, stream_obj}; | 99 m_ArrayObj, m_DictObj, stream_obj}; |
| 100 for (size_t i = 0; i < m_IndirectObjs.size(); ++i) { | 100 for (size_t i = 0; i < m_IndirectObjs.size(); ++i) { |
| 101 m_ObjHolder->AddIndirectObject(m_IndirectObjs[i]); | 101 m_ObjHolder->AddIndirectObject(m_IndirectObjs[i]); |
| 102 m_RefObjs.emplace_back(new CPDF_Reference( | 102 m_RefObjs.emplace_back(new CPDF_Reference( |
| 103 m_ObjHolder.get(), m_IndirectObjs[i]->GetObjNum())); | 103 m_ObjHolder.get(), m_IndirectObjs[i]->GetObjNum())); |
| 104 } | 104 } |
| 105 } | 105 } |
| 106 | 106 |
| 107 bool Equal(CPDF_Object* obj1, CPDF_Object* obj2) { | 107 bool Equal(CPDF_Object* obj1, CPDF_Object* obj2) { |
| (...skipping 597 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 705 for (size_t i = 0; i < FX_ArraySize(vals); ++i) { | 705 for (size_t i = 0; i < FX_ArraySize(vals); ++i) { |
| 706 EXPECT_EQ(CPDF_Object::STRING, string_array->GetObjectAt(i)->GetType()); | 706 EXPECT_EQ(CPDF_Object::STRING, string_array->GetObjectAt(i)->GetType()); |
| 707 EXPECT_STREQ(vals[i], string_array->GetObjectAt(i)->GetString().c_str()); | 707 EXPECT_STREQ(vals[i], string_array->GetObjectAt(i)->GetString().c_str()); |
| 708 EXPECT_EQ(CPDF_Object::NAME, name_array->GetObjectAt(i)->GetType()); | 708 EXPECT_EQ(CPDF_Object::NAME, name_array->GetObjectAt(i)->GetType()); |
| 709 EXPECT_STREQ(vals[i], name_array->GetObjectAt(i)->GetString().c_str()); | 709 EXPECT_STREQ(vals[i], name_array->GetObjectAt(i)->GetString().c_str()); |
| 710 } | 710 } |
| 711 } | 711 } |
| 712 | 712 |
| 713 TEST(PDFArrayTest, AddReferenceAndGetObjectAt) { | 713 TEST(PDFArrayTest, AddReferenceAndGetObjectAt) { |
| 714 std::unique_ptr<CPDF_IndirectObjectHolder> holder( | 714 std::unique_ptr<CPDF_IndirectObjectHolder> holder( |
| 715 new CPDF_IndirectObjectHolder(nullptr)); | 715 new CPDF_IndirectObjectHolder()); |
| 716 CPDF_Boolean* boolean_obj = new CPDF_Boolean(true); | 716 CPDF_Boolean* boolean_obj = new CPDF_Boolean(true); |
| 717 CPDF_Number* int_obj = new CPDF_Number(-1234); | 717 CPDF_Number* int_obj = new CPDF_Number(-1234); |
| 718 CPDF_Number* float_obj = new CPDF_Number(2345.089f); | 718 CPDF_Number* float_obj = new CPDF_Number(2345.089f); |
| 719 CPDF_String* str_obj = new CPDF_String("Adsfdsf 343434 %&&*\n", false); | 719 CPDF_String* str_obj = new CPDF_String("Adsfdsf 343434 %&&*\n", false); |
| 720 CPDF_Name* name_obj = new CPDF_Name("Title:"); | 720 CPDF_Name* name_obj = new CPDF_Name("Title:"); |
| 721 CPDF_Null* null_obj = new CPDF_Null(); | 721 CPDF_Null* null_obj = new CPDF_Null(); |
| 722 CPDF_Object* indirect_objs[] = {boolean_obj, int_obj, float_obj, | 722 CPDF_Object* indirect_objs[] = {boolean_obj, int_obj, float_obj, |
| 723 str_obj, name_obj, null_obj}; | 723 str_obj, name_obj, null_obj}; |
| 724 unsigned int obj_nums[] = {2, 4, 7, 2345, 799887, 1}; | 724 unsigned int obj_nums[] = {2, 4, 7, 2345, 799887, 1}; |
| 725 ScopedArray arr(new CPDF_Array); | 725 ScopedArray arr(new CPDF_Array); |
| (...skipping 36 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 762 ASSERT_TRUE(cloned_array); | 762 ASSERT_TRUE(cloned_array); |
| 763 EXPECT_EQ(1u, cloned_array->GetCount()); | 763 EXPECT_EQ(1u, cloned_array->GetCount()); |
| 764 CPDF_Object* cloned_dict = cloned_array->GetObjectAt(0); | 764 CPDF_Object* cloned_dict = cloned_array->GetObjectAt(0); |
| 765 ASSERT_TRUE(cloned_dict); | 765 ASSERT_TRUE(cloned_dict); |
| 766 ASSERT_TRUE(cloned_dict->IsDictionary()); | 766 ASSERT_TRUE(cloned_dict->IsDictionary()); |
| 767 // Recursively referenced object is not cloned. | 767 // Recursively referenced object is not cloned. |
| 768 EXPECT_EQ(nullptr, cloned_dict->AsDictionary()->GetObjectBy("arr")); | 768 EXPECT_EQ(nullptr, cloned_dict->AsDictionary()->GetObjectBy("arr")); |
| 769 } | 769 } |
| 770 { | 770 { |
| 771 std::unique_ptr<CPDF_IndirectObjectHolder> m_ObjHolder( | 771 std::unique_ptr<CPDF_IndirectObjectHolder> m_ObjHolder( |
| 772 new CPDF_IndirectObjectHolder(nullptr)); | 772 new CPDF_IndirectObjectHolder()); |
| 773 // Create an object with a reference loop. | 773 // Create an object with a reference loop. |
| 774 CPDF_Dictionary* dict_obj = new CPDF_Dictionary; | 774 CPDF_Dictionary* dict_obj = new CPDF_Dictionary; |
| 775 CPDF_Array* arr_obj = new CPDF_Array; | 775 CPDF_Array* arr_obj = new CPDF_Array; |
| 776 m_ObjHolder->AddIndirectObject(dict_obj); | 776 m_ObjHolder->AddIndirectObject(dict_obj); |
| 777 EXPECT_EQ(1u, dict_obj->GetObjNum()); | 777 EXPECT_EQ(1u, dict_obj->GetObjNum()); |
| 778 dict_obj->SetAt("arr", arr_obj); | 778 dict_obj->SetAt("arr", arr_obj); |
| 779 arr_obj->InsertAt(0, dict_obj, m_ObjHolder.get()); | 779 arr_obj->InsertAt(0, dict_obj, m_ObjHolder.get()); |
| 780 CPDF_Object* elem0 = arr_obj->GetObjectAt(0); | 780 CPDF_Object* elem0 = arr_obj->GetObjectAt(0); |
| 781 ASSERT_TRUE(elem0); | 781 ASSERT_TRUE(elem0); |
| 782 ASSERT_TRUE(elem0->IsReference()); | 782 ASSERT_TRUE(elem0->IsReference()); |
| 783 EXPECT_EQ(1u, elem0->AsReference()->GetRefObjNum()); | 783 EXPECT_EQ(1u, elem0->AsReference()->GetRefObjNum()); |
| 784 EXPECT_EQ(dict_obj, elem0->AsReference()->GetDirect()); | 784 EXPECT_EQ(dict_obj, elem0->AsReference()->GetDirect()); |
| 785 | 785 |
| 786 // Clone this object to see whether stack overflow will be triggered. | 786 // Clone this object to see whether stack overflow will be triggered. |
| 787 ScopedDict cloned_dict(ToDictionary(dict_obj->CloneDirectObject())); | 787 ScopedDict cloned_dict(ToDictionary(dict_obj->CloneDirectObject())); |
| 788 // Cloned object should be the same as the original. | 788 // Cloned object should be the same as the original. |
| 789 ASSERT_TRUE(cloned_dict); | 789 ASSERT_TRUE(cloned_dict); |
| 790 CPDF_Object* cloned_arr = cloned_dict->GetObjectBy("arr"); | 790 CPDF_Object* cloned_arr = cloned_dict->GetObjectBy("arr"); |
| 791 ASSERT_TRUE(cloned_arr); | 791 ASSERT_TRUE(cloned_arr); |
| 792 ASSERT_TRUE(cloned_arr->IsArray()); | 792 ASSERT_TRUE(cloned_arr->IsArray()); |
| 793 EXPECT_EQ(1u, cloned_arr->AsArray()->GetCount()); | 793 EXPECT_EQ(1u, cloned_arr->AsArray()->GetCount()); |
| 794 // Recursively referenced object is not cloned. | 794 // Recursively referenced object is not cloned. |
| 795 EXPECT_EQ(nullptr, cloned_arr->AsArray()->GetObjectAt(0)); | 795 EXPECT_EQ(nullptr, cloned_arr->AsArray()->GetObjectAt(0)); |
| 796 } | 796 } |
| 797 } | 797 } |
| OLD | NEW |