DocumentParser.cpp 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716
  1. /*
  2. * Copyright (c) 2021-2022, Matthew Olsson <mattco@serenityos.org>
  3. *
  4. * SPDX-License-Identifier: BSD-2-Clause
  5. */
  6. #include <AK/BitStream.h>
  7. #include <AK/MemoryStream.h>
  8. #include <AK/Tuple.h>
  9. #include <LibPDF/CommonNames.h>
  10. #include <LibPDF/Document.h>
  11. #include <LibPDF/DocumentParser.h>
  12. #include <LibPDF/ObjectDerivatives.h>
  13. namespace PDF {
  14. DocumentParser::DocumentParser(Document* document, ReadonlyBytes bytes)
  15. : Parser(document, bytes)
  16. {
  17. }
  18. PDFErrorOr<void> DocumentParser::initialize()
  19. {
  20. TRY(parse_header());
  21. auto const linearization_result = TRY(initialize_linearization_dict());
  22. if (linearization_result == LinearizationResult::NotLinearized)
  23. return initialize_non_linearized_xref_table();
  24. bool is_linearized = m_linearization_dictionary.has_value();
  25. if (is_linearized) {
  26. // The file may have been linearized at one point, but could have been updated afterwards,
  27. // which means it is no longer a linearized PDF file.
  28. is_linearized = m_linearization_dictionary.value().length_of_file == m_reader.bytes().size();
  29. if (!is_linearized) {
  30. // FIXME: The file shouldn't be treated as linearized, yet the xref tables are still
  31. // split. This might take some tweaking to ensure correct behavior, which can be
  32. // implemented later.
  33. TODO();
  34. }
  35. }
  36. if (is_linearized)
  37. return initialize_linearized_xref_table();
  38. return initialize_non_linearized_xref_table();
  39. }
  40. PDFErrorOr<Value> DocumentParser::parse_object_with_index(u32 index)
  41. {
  42. VERIFY(m_xref_table->has_object(index));
  43. auto byte_offset = m_xref_table->byte_offset_for_object(index);
  44. m_reader.move_to(byte_offset);
  45. auto indirect_value = TRY(parse_indirect_value());
  46. VERIFY(indirect_value->index() == index);
  47. return indirect_value->value();
  48. }
  49. PDFErrorOr<void> DocumentParser::parse_header()
  50. {
  51. // FIXME: Do something with the version?
  52. m_reader.set_reading_forwards();
  53. if (m_reader.remaining() == 0)
  54. return error("Empty PDF document");
  55. m_reader.move_to(0);
  56. if (m_reader.remaining() < 8 || !m_reader.matches("%PDF-"))
  57. return error("Not a PDF document");
  58. m_reader.move_by(5);
  59. char major_ver = m_reader.read();
  60. if (major_ver != '1' && major_ver != '2')
  61. return error(String::formatted("Unknown major version \"{}\"", major_ver));
  62. if (m_reader.read() != '.')
  63. return error("Malformed PDF version");
  64. char minor_ver = m_reader.read();
  65. if (minor_ver < '0' || minor_ver > '7')
  66. return error(String::formatted("Unknown minor version \"{}\"", minor_ver));
  67. m_reader.consume_eol();
  68. // Parse optional high-byte comment, which signifies a binary file
  69. // FIXME: Do something with this?
  70. auto comment = parse_comment();
  71. if (!comment.is_empty()) {
  72. auto binary = comment.length() >= 4;
  73. if (binary) {
  74. for (size_t i = 0; i < comment.length() && binary; i++)
  75. binary = static_cast<u8>(comment[i]) > 128;
  76. }
  77. }
  78. return {};
  79. }
  80. PDFErrorOr<DocumentParser::LinearizationResult> DocumentParser::initialize_linearization_dict()
  81. {
  82. // parse_header() is called immediately before this, so we are at the right location
  83. auto indirect_value = Value(*TRY(parse_indirect_value()));
  84. auto dict_value = TRY(m_document->resolve(indirect_value));
  85. if (!dict_value.has<NonnullRefPtr<Object>>())
  86. return error("Expected linearization object to be a dictionary");
  87. auto dict_object = dict_value.get<NonnullRefPtr<Object>>();
  88. if (!dict_object->is<DictObject>())
  89. return LinearizationResult::NotLinearized;
  90. auto dict = dict_object->cast<DictObject>();
  91. if (!dict->contains(CommonNames::Linearized))
  92. return LinearizationResult::NotLinearized;
  93. if (!dict->contains(CommonNames::L, CommonNames::H, CommonNames::O, CommonNames::E, CommonNames::N, CommonNames::T))
  94. return error("Malformed linearization dictionary");
  95. auto length_of_file = dict->get_value(CommonNames::L);
  96. auto hint_table = dict->get_value(CommonNames::H);
  97. auto first_page_object_number = dict->get_value(CommonNames::O);
  98. auto offset_of_first_page_end = dict->get_value(CommonNames::E);
  99. auto number_of_pages = dict->get_value(CommonNames::N);
  100. auto offset_of_main_xref_table = dict->get_value(CommonNames::T);
  101. auto first_page = dict->get(CommonNames::P).value_or({});
  102. // Validation
  103. if (!length_of_file.has_u32()
  104. || !hint_table.has<NonnullRefPtr<Object>>()
  105. || !first_page_object_number.has_u32()
  106. || !number_of_pages.has_u16()
  107. || !offset_of_main_xref_table.has_u32()
  108. || (!first_page.has<Empty>() && !first_page.has_u32())) {
  109. return error("Malformed linearization dictionary parameters");
  110. }
  111. auto hint_table_array = hint_table.get<NonnullRefPtr<Object>>()->cast<ArrayObject>();
  112. auto hint_table_size = hint_table_array->size();
  113. if (hint_table_size != 2 && hint_table_size != 4)
  114. return error("Expected hint table to be of length 2 or 4");
  115. auto primary_hint_stream_offset = hint_table_array->at(0);
  116. auto primary_hint_stream_length = hint_table_array->at(1);
  117. Value overflow_hint_stream_offset;
  118. Value overflow_hint_stream_length;
  119. if (hint_table_size == 4) {
  120. overflow_hint_stream_offset = hint_table_array->at(2);
  121. overflow_hint_stream_length = hint_table_array->at(3);
  122. }
  123. if (!primary_hint_stream_offset.has_u32()
  124. || !primary_hint_stream_length.has_u32()
  125. || (!overflow_hint_stream_offset.has<Empty>() && !overflow_hint_stream_offset.has_u32())
  126. || (!overflow_hint_stream_length.has<Empty>() && !overflow_hint_stream_length.has_u32())) {
  127. return error("Malformed hint stream");
  128. }
  129. m_linearization_dictionary = LinearizationDictionary {
  130. length_of_file.get_u32(),
  131. primary_hint_stream_offset.get_u32(),
  132. primary_hint_stream_length.get_u32(),
  133. overflow_hint_stream_offset.has<Empty>() ? NumericLimits<u32>::max() : overflow_hint_stream_offset.get_u32(),
  134. overflow_hint_stream_length.has<Empty>() ? NumericLimits<u32>::max() : overflow_hint_stream_length.get_u32(),
  135. first_page_object_number.get_u32(),
  136. offset_of_first_page_end.get_u32(),
  137. number_of_pages.get_u16(),
  138. offset_of_main_xref_table.get_u32(),
  139. first_page.has<Empty>() ? NumericLimits<u32>::max() : first_page.get_u32(),
  140. };
  141. return LinearizationResult::Linearized;
  142. }
  143. PDFErrorOr<void> DocumentParser::initialize_linearized_xref_table()
  144. {
  145. // The linearization parameter dictionary has just been parsed, and the xref table
  146. // comes immediately after it. We are in the correct spot.
  147. m_xref_table = TRY(parse_xref_table());
  148. if (!m_trailer)
  149. m_trailer = TRY(parse_file_trailer());
  150. // Also parse the main xref table and merge into the first-page xref table. Note
  151. // that we don't use the main xref table offset from the linearization dict because
  152. // for some reason, it specified the offset of the whitespace after the object
  153. // index start and length? So it's much easier to do it this way.
  154. auto main_xref_table_offset = m_trailer->get_value(CommonNames::Prev).to_int();
  155. m_reader.move_to(main_xref_table_offset);
  156. auto main_xref_table = TRY(parse_xref_table());
  157. TRY(m_xref_table->merge(move(*main_xref_table)));
  158. return {};
  159. }
  160. PDFErrorOr<void> DocumentParser::initialize_hint_tables()
  161. {
  162. auto linearization_dict = m_linearization_dictionary.value();
  163. auto primary_offset = linearization_dict.primary_hint_stream_offset;
  164. auto overflow_offset = linearization_dict.overflow_hint_stream_offset;
  165. auto parse_hint_table = [&](size_t offset) -> RefPtr<StreamObject> {
  166. m_reader.move_to(offset);
  167. auto stream_indirect_value = parse_indirect_value();
  168. if (stream_indirect_value.is_error())
  169. return {};
  170. auto stream_value = stream_indirect_value.value()->value();
  171. if (!stream_value.has<NonnullRefPtr<Object>>())
  172. return {};
  173. auto stream_object = stream_value.get<NonnullRefPtr<Object>>();
  174. if (!stream_object->is<StreamObject>())
  175. return {};
  176. return stream_object->cast<StreamObject>();
  177. };
  178. auto primary_hint_stream = parse_hint_table(primary_offset);
  179. if (!primary_hint_stream)
  180. return error("Invalid primary hint stream");
  181. RefPtr<StreamObject> overflow_hint_stream;
  182. if (overflow_offset != NumericLimits<u32>::max())
  183. overflow_hint_stream = parse_hint_table(overflow_offset);
  184. ByteBuffer possible_merged_stream_buffer;
  185. ReadonlyBytes hint_stream_bytes;
  186. if (overflow_hint_stream) {
  187. auto primary_size = primary_hint_stream->bytes().size();
  188. auto overflow_size = overflow_hint_stream->bytes().size();
  189. auto total_size = primary_size + overflow_size;
  190. auto buffer_result = ByteBuffer::create_uninitialized(total_size);
  191. if (buffer_result.is_error())
  192. return Error { Error::Type::Internal, "Failed to allocate hint stream buffer" };
  193. possible_merged_stream_buffer = buffer_result.release_value();
  194. MUST(possible_merged_stream_buffer.try_append(primary_hint_stream->bytes()));
  195. MUST(possible_merged_stream_buffer.try_append(overflow_hint_stream->bytes()));
  196. hint_stream_bytes = possible_merged_stream_buffer.bytes();
  197. } else {
  198. hint_stream_bytes = primary_hint_stream->bytes();
  199. }
  200. auto hint_table = TRY(parse_page_offset_hint_table(hint_stream_bytes));
  201. auto hint_table_entries = parse_all_page_offset_hint_table_entries(hint_table, hint_stream_bytes);
  202. // FIXME: Do something with the hint tables
  203. return {};
  204. }
  205. PDFErrorOr<void> DocumentParser::initialize_non_linearized_xref_table()
  206. {
  207. m_reader.move_to(m_reader.bytes().size() - 1);
  208. if (!navigate_to_before_eof_marker())
  209. return error("No EOF marker");
  210. if (!navigate_to_after_startxref())
  211. return error("No xref");
  212. m_reader.set_reading_forwards();
  213. auto xref_offset_value = parse_number();
  214. if (xref_offset_value.is_error() || !xref_offset_value.value().has<int>())
  215. return error("Invalid xref offset");
  216. auto xref_offset = xref_offset_value.value().get<int>();
  217. m_reader.move_to(xref_offset);
  218. m_xref_table = TRY(parse_xref_table());
  219. if (!m_trailer)
  220. m_trailer = TRY(parse_file_trailer());
  221. return {};
  222. }
  223. PDFErrorOr<NonnullRefPtr<XRefTable>> DocumentParser::parse_xref_stream()
  224. {
  225. auto first_number = TRY(parse_number());
  226. auto second_number = TRY(parse_number());
  227. if (!m_reader.matches("obj"))
  228. return error("Malformed xref object");
  229. m_reader.move_by(3);
  230. if (m_reader.matches_eol())
  231. m_reader.consume_eol();
  232. auto dict = TRY(parse_dict());
  233. auto type = TRY(dict->get_name(m_document, CommonNames::Type))->name();
  234. if (type != "XRef")
  235. return error("Malformed xref dictionary");
  236. auto field_sizes = TRY(dict->get_array(m_document, "W"));
  237. if (field_sizes->size() != 3)
  238. return error("Malformed xref dictionary");
  239. auto object_count = dict->get_value("Size").get<int>();
  240. Vector<Tuple<int, int>> subsection_indices;
  241. if (dict->contains(CommonNames::Index)) {
  242. auto index_array = TRY(dict->get_array(m_document, CommonNames::Index));
  243. if (index_array->size() % 2 != 0)
  244. return error("Malformed xref dictionary");
  245. for (size_t i = 0; i < index_array->size(); i += 2)
  246. subsection_indices.append({ index_array->at(i).get<int>(), index_array->at(i + 1).get<int>() - 1 });
  247. } else {
  248. subsection_indices.append({ 0, object_count - 1 });
  249. }
  250. auto stream = TRY(parse_stream(dict));
  251. auto table = adopt_ref(*new XRefTable());
  252. auto field_to_long = [](Span<const u8> field) -> long {
  253. long value = 0;
  254. const u8 max = (field.size() - 1) * 8;
  255. for (size_t i = 0; i < field.size(); ++i) {
  256. value |= static_cast<long>(field[i]) << (max - (i * 8));
  257. }
  258. return value;
  259. };
  260. size_t byte_index = 0;
  261. size_t subsection_index = 0;
  262. Vector<XRefEntry> entries;
  263. for (int entry_index = 0; entry_index < object_count; ++entry_index) {
  264. Array<long, 3> fields;
  265. for (size_t field_index = 0; field_index < 3; ++field_index) {
  266. auto field_size = field_sizes->at(field_index).get_u32();
  267. auto field = stream->bytes().slice(byte_index, field_size);
  268. fields[field_index] = field_to_long(field);
  269. byte_index += field_size;
  270. }
  271. u8 type = fields[0];
  272. if (!field_sizes->at(0).get_u32())
  273. type = 1;
  274. entries.append({ fields[1], static_cast<u16>(fields[2]), type != 0, type == 2 });
  275. auto indices = subsection_indices[subsection_index];
  276. if (entry_index >= indices.get<1>()) {
  277. table->add_section({ indices.get<0>(), indices.get<1>(), entries });
  278. entries.clear();
  279. subsection_index++;
  280. }
  281. }
  282. m_trailer = dict;
  283. return table;
  284. }
  285. PDFErrorOr<NonnullRefPtr<XRefTable>> DocumentParser::parse_xref_table()
  286. {
  287. if (!m_reader.matches("xref")) {
  288. // Since version 1.5, there may be a cross-reference stream instead
  289. return parse_xref_stream();
  290. }
  291. m_reader.move_by(4);
  292. if (!m_reader.consume_eol())
  293. return error("Expected newline after \"xref\"");
  294. auto table = adopt_ref(*new XRefTable());
  295. do {
  296. if (m_reader.matches("trailer"))
  297. return table;
  298. Vector<XRefEntry> entries;
  299. auto starting_index_value = TRY(parse_number());
  300. auto starting_index = starting_index_value.get<int>();
  301. auto object_count_value = TRY(parse_number());
  302. auto object_count = object_count_value.get<int>();
  303. for (int i = 0; i < object_count; i++) {
  304. auto offset_string = String(m_reader.bytes().slice(m_reader.offset(), 10));
  305. m_reader.move_by(10);
  306. if (!m_reader.consume(' '))
  307. return error("Malformed xref entry");
  308. auto generation_string = String(m_reader.bytes().slice(m_reader.offset(), 5));
  309. m_reader.move_by(5);
  310. if (!m_reader.consume(' '))
  311. return error("Malformed xref entry");
  312. auto letter = m_reader.read();
  313. if (letter != 'n' && letter != 'f')
  314. return error("Malformed xref entry");
  315. // The line ending sequence can be one of the following:
  316. // SP CR, SP LF, or CR LF
  317. if (m_reader.matches(' ')) {
  318. m_reader.consume();
  319. auto ch = m_reader.consume();
  320. if (ch != '\r' && ch != '\n')
  321. return error("Malformed xref entry");
  322. } else {
  323. if (!m_reader.matches("\r\n"))
  324. return error("Malformed xref entry");
  325. m_reader.move_by(2);
  326. }
  327. auto offset = strtol(offset_string.characters(), nullptr, 10);
  328. auto generation = strtol(generation_string.characters(), nullptr, 10);
  329. entries.append({ offset, static_cast<u16>(generation), letter == 'n' });
  330. }
  331. table->add_section({ starting_index, object_count, entries });
  332. } while (m_reader.matches_number());
  333. return table;
  334. }
  335. PDFErrorOr<NonnullRefPtr<DictObject>> DocumentParser::parse_file_trailer()
  336. {
  337. while (m_reader.matches_eol())
  338. m_reader.consume_eol();
  339. if (!m_reader.matches("trailer"))
  340. return error("Expected \"trailer\" keyword");
  341. m_reader.move_by(7);
  342. m_reader.consume_whitespace();
  343. auto dict = TRY(parse_dict());
  344. if (!m_reader.matches("startxref"))
  345. return error("Expected \"startxref\"");
  346. m_reader.move_by(9);
  347. m_reader.consume_whitespace();
  348. m_reader.move_until([&](auto) { return m_reader.matches_eol(); });
  349. VERIFY(m_reader.consume_eol());
  350. if (!m_reader.matches("%%EOF"))
  351. return error("Expected \"%%EOF\"");
  352. m_reader.move_by(5);
  353. m_reader.consume_whitespace();
  354. return dict;
  355. }
  356. PDFErrorOr<DocumentParser::PageOffsetHintTable> DocumentParser::parse_page_offset_hint_table(ReadonlyBytes hint_stream_bytes)
  357. {
  358. if (hint_stream_bytes.size() < sizeof(PageOffsetHintTable))
  359. return error("Hint stream is too small");
  360. size_t offset = 0;
  361. auto read_u32 = [&] {
  362. u32 data = reinterpret_cast<const u32*>(hint_stream_bytes.data() + offset)[0];
  363. offset += 4;
  364. return AK::convert_between_host_and_big_endian(data);
  365. };
  366. auto read_u16 = [&] {
  367. u16 data = reinterpret_cast<const u16*>(hint_stream_bytes.data() + offset)[0];
  368. offset += 2;
  369. return AK::convert_between_host_and_big_endian(data);
  370. };
  371. PageOffsetHintTable hint_table {
  372. read_u32(),
  373. read_u32(),
  374. read_u16(),
  375. read_u32(),
  376. read_u16(),
  377. read_u32(),
  378. read_u16(),
  379. read_u32(),
  380. read_u16(),
  381. read_u16(),
  382. read_u16(),
  383. read_u16(),
  384. read_u16(),
  385. };
  386. // Verify that all of the bits_required_for_xyz fields are <= 32, since all of the numeric
  387. // fields in PageOffsetHintTableEntry are u32
  388. VERIFY(hint_table.bits_required_for_object_number <= 32);
  389. VERIFY(hint_table.bits_required_for_page_length <= 32);
  390. VERIFY(hint_table.bits_required_for_content_stream_offsets <= 32);
  391. VERIFY(hint_table.bits_required_for_content_stream_length <= 32);
  392. VERIFY(hint_table.bits_required_for_number_of_shared_obj_refs <= 32);
  393. VERIFY(hint_table.bits_required_for_greatest_shared_obj_identifier <= 32);
  394. VERIFY(hint_table.bits_required_for_fraction_numerator <= 32);
  395. return hint_table;
  396. }
  397. Vector<DocumentParser::PageOffsetHintTableEntry> DocumentParser::parse_all_page_offset_hint_table_entries(PageOffsetHintTable const& hint_table, ReadonlyBytes hint_stream_bytes)
  398. {
  399. InputMemoryStream input_stream(hint_stream_bytes);
  400. input_stream.seek(sizeof(PageOffsetHintTable));
  401. InputBitStream bit_stream(input_stream);
  402. auto number_of_pages = m_linearization_dictionary.value().number_of_pages;
  403. Vector<PageOffsetHintTableEntry> entries;
  404. for (size_t i = 0; i < number_of_pages; i++)
  405. entries.append(PageOffsetHintTableEntry {});
  406. auto bits_required_for_object_number = hint_table.bits_required_for_object_number;
  407. auto bits_required_for_page_length = hint_table.bits_required_for_page_length;
  408. auto bits_required_for_content_stream_offsets = hint_table.bits_required_for_content_stream_offsets;
  409. auto bits_required_for_content_stream_length = hint_table.bits_required_for_content_stream_length;
  410. auto bits_required_for_number_of_shared_obj_refs = hint_table.bits_required_for_number_of_shared_obj_refs;
  411. auto bits_required_for_greatest_shared_obj_identifier = hint_table.bits_required_for_greatest_shared_obj_identifier;
  412. auto bits_required_for_fraction_numerator = hint_table.bits_required_for_fraction_numerator;
  413. auto parse_int_entry = [&](u32 PageOffsetHintTableEntry::*field, u32 bit_size) {
  414. if (bit_size <= 0)
  415. return;
  416. for (int i = 0; i < number_of_pages; i++) {
  417. auto& entry = entries[i];
  418. entry.*field = bit_stream.read_bits(bit_size);
  419. }
  420. };
  421. auto parse_vector_entry = [&](Vector<u32> PageOffsetHintTableEntry::*field, u32 bit_size) {
  422. if (bit_size <= 0)
  423. return;
  424. for (int page = 1; page < number_of_pages; page++) {
  425. auto number_of_shared_objects = entries[page].number_of_shared_objects;
  426. Vector<u32> items;
  427. items.ensure_capacity(number_of_shared_objects);
  428. for (size_t i = 0; i < number_of_shared_objects; i++)
  429. items.unchecked_append(bit_stream.read_bits(bit_size));
  430. entries[page].*field = move(items);
  431. }
  432. };
  433. parse_int_entry(&PageOffsetHintTableEntry::objects_in_page_number, bits_required_for_object_number);
  434. parse_int_entry(&PageOffsetHintTableEntry::page_length_number, bits_required_for_page_length);
  435. parse_int_entry(&PageOffsetHintTableEntry::number_of_shared_objects, bits_required_for_number_of_shared_obj_refs);
  436. parse_vector_entry(&PageOffsetHintTableEntry::shared_object_identifiers, bits_required_for_greatest_shared_obj_identifier);
  437. parse_vector_entry(&PageOffsetHintTableEntry::shared_object_location_numerators, bits_required_for_fraction_numerator);
  438. parse_int_entry(&PageOffsetHintTableEntry::page_content_stream_offset_number, bits_required_for_content_stream_offsets);
  439. parse_int_entry(&PageOffsetHintTableEntry::page_content_stream_length_number, bits_required_for_content_stream_length);
  440. return entries;
  441. }
  442. bool DocumentParser::navigate_to_before_eof_marker()
  443. {
  444. m_reader.set_reading_backwards();
  445. while (!m_reader.done()) {
  446. m_reader.move_until([&](auto) { return m_reader.matches_eol(); });
  447. if (m_reader.done())
  448. return false;
  449. m_reader.consume_eol();
  450. if (!m_reader.matches("%%EOF"))
  451. continue;
  452. m_reader.move_by(5);
  453. if (!m_reader.matches_eol())
  454. continue;
  455. m_reader.consume_eol();
  456. return true;
  457. }
  458. return false;
  459. }
  460. bool DocumentParser::navigate_to_after_startxref()
  461. {
  462. m_reader.set_reading_backwards();
  463. while (!m_reader.done()) {
  464. m_reader.move_until([&](auto) { return m_reader.matches_eol(); });
  465. auto offset = m_reader.offset() + 1;
  466. m_reader.consume_eol();
  467. if (!m_reader.matches("startxref"))
  468. continue;
  469. m_reader.move_by(9);
  470. if (!m_reader.matches_eol())
  471. continue;
  472. m_reader.move_to(offset);
  473. return true;
  474. }
  475. return false;
  476. }
  477. PDFErrorOr<RefPtr<DictObject>> DocumentParser::conditionally_parse_page_tree_node(u32 object_index)
  478. {
  479. VERIFY(m_xref_table->has_object(object_index));
  480. auto byte_offset = m_xref_table->byte_offset_for_object(object_index);
  481. m_reader.move_to(byte_offset);
  482. TRY(parse_number());
  483. TRY(parse_number());
  484. if (!m_reader.matches("obj"))
  485. return error(String::formatted("Invalid page tree offset {}", object_index));
  486. m_reader.move_by(3);
  487. m_reader.consume_whitespace();
  488. VERIFY(m_reader.consume('<') && m_reader.consume('<'));
  489. m_reader.consume_whitespace();
  490. HashMap<FlyString, Value> map;
  491. while (true) {
  492. if (m_reader.matches(">>"))
  493. break;
  494. auto name = TRY(parse_name());
  495. auto name_string = name->name();
  496. if (!name_string.is_one_of(CommonNames::Type, CommonNames::Parent, CommonNames::Kids, CommonNames::Count)) {
  497. // This is a page, not a page tree node
  498. return RefPtr<DictObject> {};
  499. }
  500. auto value = TRY(parse_value());
  501. if (name_string == CommonNames::Type) {
  502. if (!value.has<NonnullRefPtr<Object>>())
  503. return RefPtr<DictObject> {};
  504. auto type_object = value.get<NonnullRefPtr<Object>>();
  505. if (!type_object->is<NameObject>())
  506. return RefPtr<DictObject> {};
  507. auto type_name = type_object->cast<NameObject>();
  508. if (type_name->name() != CommonNames::Pages)
  509. return RefPtr<DictObject> {};
  510. }
  511. map.set(name->name(), value);
  512. }
  513. VERIFY(m_reader.consume('>') && m_reader.consume('>'));
  514. m_reader.consume_whitespace();
  515. return make_object<DictObject>(map);
  516. }
  517. }
  518. namespace AK {
  519. template<>
  520. struct Formatter<PDF::DocumentParser::LinearizationDictionary> : Formatter<StringView> {
  521. ErrorOr<void> format(FormatBuilder& format_builder, PDF::DocumentParser::LinearizationDictionary const& dict)
  522. {
  523. StringBuilder builder;
  524. builder.append("{\n"sv);
  525. builder.appendff(" length_of_file={}\n", dict.length_of_file);
  526. builder.appendff(" primary_hint_stream_offset={}\n", dict.primary_hint_stream_offset);
  527. builder.appendff(" primary_hint_stream_length={}\n", dict.primary_hint_stream_length);
  528. builder.appendff(" overflow_hint_stream_offset={}\n", dict.overflow_hint_stream_offset);
  529. builder.appendff(" overflow_hint_stream_length={}\n", dict.overflow_hint_stream_length);
  530. builder.appendff(" first_page_object_number={}\n", dict.first_page_object_number);
  531. builder.appendff(" offset_of_first_page_end={}\n", dict.offset_of_first_page_end);
  532. builder.appendff(" number_of_pages={}\n", dict.number_of_pages);
  533. builder.appendff(" offset_of_main_xref_table={}\n", dict.offset_of_main_xref_table);
  534. builder.appendff(" first_page={}\n", dict.first_page);
  535. builder.append('}');
  536. return Formatter<StringView>::format(format_builder, builder.to_string());
  537. }
  538. };
  539. template<>
  540. struct Formatter<PDF::DocumentParser::PageOffsetHintTable> : Formatter<StringView> {
  541. ErrorOr<void> format(FormatBuilder& format_builder, PDF::DocumentParser::PageOffsetHintTable const& table)
  542. {
  543. StringBuilder builder;
  544. builder.append("{\n"sv);
  545. builder.appendff(" least_number_of_objects_in_a_page={}\n", table.least_number_of_objects_in_a_page);
  546. builder.appendff(" location_of_first_page_object={}\n", table.location_of_first_page_object);
  547. builder.appendff(" bits_required_for_object_number={}\n", table.bits_required_for_object_number);
  548. builder.appendff(" least_length_of_a_page={}\n", table.least_length_of_a_page);
  549. builder.appendff(" bits_required_for_page_length={}\n", table.bits_required_for_page_length);
  550. builder.appendff(" least_offset_of_any_content_stream={}\n", table.least_offset_of_any_content_stream);
  551. builder.appendff(" bits_required_for_content_stream_offsets={}\n", table.bits_required_for_content_stream_offsets);
  552. builder.appendff(" least_content_stream_length={}\n", table.least_content_stream_length);
  553. builder.appendff(" bits_required_for_content_stream_length={}\n", table.bits_required_for_content_stream_length);
  554. builder.appendff(" bits_required_for_number_of_shared_obj_refs={}\n", table.bits_required_for_number_of_shared_obj_refs);
  555. builder.appendff(" bits_required_for_greatest_shared_obj_identifier={}\n", table.bits_required_for_greatest_shared_obj_identifier);
  556. builder.appendff(" bits_required_for_fraction_numerator={}\n", table.bits_required_for_fraction_numerator);
  557. builder.appendff(" shared_object_reference_fraction_denominator={}\n", table.shared_object_reference_fraction_denominator);
  558. builder.append('}');
  559. return Formatter<StringView>::format(format_builder, builder.to_string());
  560. }
  561. };
  562. template<>
  563. struct Formatter<PDF::DocumentParser::PageOffsetHintTableEntry> : Formatter<StringView> {
  564. ErrorOr<void> format(FormatBuilder& format_builder, PDF::DocumentParser::PageOffsetHintTableEntry const& entry)
  565. {
  566. StringBuilder builder;
  567. builder.append("{\n"sv);
  568. builder.appendff(" objects_in_page_number={}\n", entry.objects_in_page_number);
  569. builder.appendff(" page_length_number={}\n", entry.page_length_number);
  570. builder.appendff(" number_of_shared_objects={}\n", entry.number_of_shared_objects);
  571. builder.append(" shared_object_identifiers=["sv);
  572. for (auto& identifier : entry.shared_object_identifiers)
  573. builder.appendff(" {}", identifier);
  574. builder.append(" ]\n"sv);
  575. builder.append(" shared_object_location_numerators=["sv);
  576. for (auto& numerator : entry.shared_object_location_numerators)
  577. builder.appendff(" {}", numerator);
  578. builder.append(" ]\n"sv);
  579. builder.appendff(" page_content_stream_offset_number={}\n", entry.page_content_stream_offset_number);
  580. builder.appendff(" page_content_stream_length_number={}\n", entry.page_content_stream_length_number);
  581. builder.append('}');
  582. return Formatter<StringView>::format(format_builder, builder.to_string());
  583. }
  584. };
  585. }