5#include "core/fpdfapi/parser/cpdf_object_avail.h"
10#include "core/fpdfapi/parser/cpdf_array.h"
11#include "core/fpdfapi/parser/cpdf_dictionary.h"
12#include "core/fpdfapi/parser/cpdf_indirect_object_holder.h"
13#include "core/fpdfapi/parser/cpdf_read_validator.h"
14#include "core/fpdfapi/parser/cpdf_reference.h"
15#include "core/fpdfapi/parser/cpdf_string.h"
16#include "core/fxcrt/fx_stream.h"
17#include "testing/gtest/include/gtest/gtest.h"
18#include "testing/invalid_seekable_read_stream.h"
19#include "third_party/base/check.h"
20#include "third_party/base/notreached.h"
34 ~TestReadValidator()
override =
default;
39 enum class ObjectState {
43 TestHolder() : validator_(pdfium::MakeRetain<TestReadValidator>()) {}
44 ~TestHolder()
override =
default;
48 auto it = objects_data_.find(objnum);
49 if (it == objects_data_.end())
52 ObjectData& obj_data = it->second;
53 if (obj_data.state == ObjectState::Unavailable) {
54 validator_->SimulateReadError();
57 return obj_data.object;
62 void AddObject(uint32_t objnum,
65 ObjectData object_data;
66 object_data.object =
std::move(object);
67 object_data.state = state;
68 DCHECK(objects_data_.find(objnum) == objects_data_.end());
69 objects_data_[objnum] = std::move(object_data);
72 void SetObjectState(uint32_t objnum, ObjectState state) {
73 auto it = objects_data_.find(objnum);
74 DCHECK(it != objects_data_.end());
75 ObjectData& obj_data = it->second;
76 obj_data.state = state;
80 auto it = objects_data_.find(objnum);
81 if (it == objects_data_.end())
83 return it->second.object.Get();
89 ObjectState state = ObjectState::Unavailable;
91 std::map<uint32_t, ObjectData> objects_data_;
98 ~CPDF_ObjectAvailFailOnExclude()
override =
default;
99 bool ExcludeObject(
const CPDF_Object* object)
const override {
100 NOTREACHED_NORETURN();
107 ~CPDF_ObjectAvailExcludeArray()
override =
default;
108 bool ExcludeObject(
const CPDF_Object* object)
const override {
116 ~CPDF_ObjectAvailExcludeTypeKey()
override =
default;
117 bool ExcludeObject(
const CPDF_Object* object)
const override {
122 return object->IsDictionary() &&
123 object->GetDict()->GetByteStringFor(
"Type") ==
"Exclude me";
131 holder.AddObject(1, pdfium::MakeRetain<CPDF_String>(
nullptr,
"string",
false),
132 TestHolder::ObjectState::Unavailable);
135 holder.SetObjectState(1, TestHolder::ObjectState::Available);
141 holder.AddObject(1, pdfium::MakeRetain<CPDF_Reference>(&holder, 2),
142 TestHolder::ObjectState::Unavailable);
143 holder.AddObject(2, pdfium::MakeRetain<CPDF_String>(
nullptr,
"string",
false),
144 TestHolder::ObjectState::Unavailable);
148 holder.SetObjectState(1, TestHolder::ObjectState::Available);
151 holder.SetObjectState(2, TestHolder::ObjectState::Available);
157 holder.AddObject(1, pdfium::MakeRetain<CPDF_Reference>(&holder, 2),
158 TestHolder::ObjectState::Unavailable);
159 holder.AddObject(2, pdfium::MakeRetain<CPDF_Reference>(&holder, 3),
160 TestHolder::ObjectState::Unavailable);
161 holder.AddObject(3, pdfium::MakeRetain<CPDF_Reference>(&holder, 1),
162 TestHolder::ObjectState::Unavailable);
167 holder.SetObjectState(1, TestHolder::ObjectState::Available);
170 holder.SetObjectState(2, TestHolder::ObjectState::Available);
173 holder.SetObjectState(3, TestHolder::ObjectState::Available);
179 holder.AddObject(1, pdfium::MakeRetain<CPDF_Dictionary>(),
180 TestHolder::ObjectState::Unavailable);
181 holder.AddObject(2, pdfium::MakeRetain<CPDF_Dictionary>(),
182 TestHolder::ObjectState::Unavailable);
184 holder.GetTestObject(2)->GetMutableDict()->SetNewFor<CPDF_Reference>(
185 "Parent", &holder, 1);
190 holder.SetObjectState(2, TestHolder::ObjectState::Available);
197 const uint32_t kDepth = 100;
198 for (uint32_t i = 1; i < kDepth; ++i) {
199 holder.AddObject(i, pdfium::MakeRetain<CPDF_Dictionary>(),
200 TestHolder::ObjectState::Unavailable);
202 holder.GetTestObject(i)->GetMutableDict()->SetNewFor<CPDF_Reference>(
203 "Child", &holder, i + 1);
206 holder.AddObject(kDepth, pdfium::MakeRetain<CPDF_Dictionary>(),
207 TestHolder::ObjectState::Unavailable);
210 for (uint32_t i = 1; i <= kDepth; ++i) {
212 holder.SetObjectState(i, TestHolder::ObjectState::Available);
219 holder.AddObject(1, pdfium::MakeRetain<CPDF_Dictionary>(),
220 TestHolder::ObjectState::Available);
221 CPDF_ObjectAvailFailOnExclude avail(holder.GetValidator(), &holder, 1);
227 holder.AddObject(1, pdfium::MakeRetain<CPDF_Reference>(&holder, 2),
228 TestHolder::ObjectState::Available);
229 holder.AddObject(2, pdfium::MakeRetain<CPDF_Dictionary>(),
230 TestHolder::ObjectState::Available);
231 CPDF_ObjectAvailFailOnExclude avail(holder.GetValidator(), &holder, 1);
237 holder.AddObject(1, pdfium::MakeRetain<CPDF_Dictionary>(),
238 TestHolder::ObjectState::Available);
239 holder.GetTestObject(1)->GetMutableDict()->SetNewFor<CPDF_Reference>(
240 "ArrayRef", &holder, 2);
241 holder.AddObject(2, pdfium::MakeRetain<CPDF_Array>(),
242 TestHolder::ObjectState::Available);
249 pdfium::MakeRetain<CPDF_String>(
nullptr,
"Not available string",
false),
250 TestHolder::ObjectState::Unavailable);
251 CPDF_ObjectAvailExcludeArray avail(holder.GetValidator(), &holder, 1);
257 holder.AddObject(1, pdfium::MakeRetain<CPDF_Dictionary>(),
258 TestHolder::ObjectState::Available);
259 holder.GetTestObject(1)->GetMutableDict()->SetNewFor<CPDF_Reference>(
260 "DictRef", &holder, 2);
261 holder.AddObject(2, pdfium::MakeRetain<CPDF_Dictionary>(),
262 TestHolder::ObjectState::Available);
264 holder.GetTestObject(2)->GetMutableDict()->SetNewFor<CPDF_Reference>(
268 3, pdfium::MakeRetain<CPDF_String>(
nullptr,
"Exclude me",
false),
269 TestHolder::ObjectState::Unavailable);
271 holder.GetTestObject(2)->GetMutableDict()->SetNewFor<CPDF_Reference>(
272 "OtherData", &holder, 4);
277 pdfium::MakeRetain<CPDF_String>(
nullptr,
"Not available string",
false),
278 TestHolder::ObjectState::Unavailable);
280 CPDF_ObjectAvailExcludeTypeKey avail(holder.GetValidator(), &holder, 1);
285 holder.SetObjectState(3, TestHolder::ObjectState::Available);
294 holder.AddObject(1, pdfium::MakeRetain<CPDF_Dictionary>(),
295 TestHolder::ObjectState::Available);
296 holder.GetTestObject(1)->GetMutableDict()->SetNewFor<CPDF_Reference>(
297 "NotExistsObjRef", &holder, 2);
306 holder.AddObject(1, pdfium::MakeRetain<CPDF_String>(
nullptr,
"string",
false),
307 TestHolder::ObjectState::Unavailable);
312 holder.SetObjectState(1, TestHolder::ObjectState::Available);
318 holder.AddObject(1, pdfium::MakeRetain<CPDF_Dictionary>(),
319 TestHolder::ObjectState::Available);
321 holder.GetTestObject(1)->GetMutableDict()->SetNewFor<CPDF_Reference>(
324 holder.GetTestObject(1)->GetMutableDict()->SetNewFor<CPDF_Dictionary>(
327 root->SetNewFor<CPDF_Reference>(
"Self", &holder, 1);
328 holder.AddObject(2, pdfium::MakeRetain<CPDF_String>(
nullptr,
"Data",
false),
329 TestHolder::ObjectState::Unavailable);
334 holder.SetObjectState(2, TestHolder::ObjectState::Available);
CPDF_DataAvail::DocAvailStatus CheckAvail()
virtual CPDF_Array * AsMutableArray()
bool ReadBlockAtOffset(pdfium::span< uint8_t > buffer, FX_FILESIZE offset) override
TEST(FXCRYPT, MD5GenerateEmtpyData)
#define CONSTRUCT_VIA_MAKE_RETAIN