DocumentParser.cpp 29 KB

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