DocumentParser.cpp 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742
  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 object_count = dict->get_value("Size").get<int>();
  244. Vector<Tuple<int, int>> subsection_indices;
  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. subsection_indices.append({ index_array->at(i).get<int>(), index_array->at(i + 1).get<int>() - 1 });
  251. } else {
  252. subsection_indices.append({ 0, object_count - 1 });
  253. }
  254. auto stream = TRY(parse_stream(dict));
  255. auto table = adopt_ref(*new XRefTable());
  256. auto field_to_long = [](Span<const u8> 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 < object_count; ++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. auto indices = subsection_indices[subsection_index];
  280. if (entry_index >= indices.get<1>()) {
  281. table->add_section({ indices.get<0>(), indices.get<1>(), entries });
  282. entries.clear();
  283. subsection_index++;
  284. }
  285. }
  286. m_trailer = dict;
  287. return table;
  288. }
  289. PDFErrorOr<NonnullRefPtr<XRefTable>> DocumentParser::parse_xref_table()
  290. {
  291. if (!m_reader.matches("xref")) {
  292. // Since version 1.5, there may be a cross-reference stream instead
  293. return parse_xref_stream();
  294. }
  295. m_reader.move_by(4);
  296. if (!m_reader.consume_eol())
  297. return error("Expected newline after \"xref\"");
  298. auto table = adopt_ref(*new XRefTable());
  299. do {
  300. if (m_reader.matches("trailer"))
  301. return table;
  302. Vector<XRefEntry> entries;
  303. auto starting_index_value = TRY(parse_number());
  304. auto starting_index = starting_index_value.get<int>();
  305. auto object_count_value = TRY(parse_number());
  306. auto object_count = object_count_value.get<int>();
  307. for (int i = 0; i < object_count; i++) {
  308. auto offset_string = String(m_reader.bytes().slice(m_reader.offset(), 10));
  309. m_reader.move_by(10);
  310. if (!m_reader.consume(' '))
  311. return error("Malformed xref entry");
  312. auto generation_string = String(m_reader.bytes().slice(m_reader.offset(), 5));
  313. m_reader.move_by(5);
  314. if (!m_reader.consume(' '))
  315. return error("Malformed xref entry");
  316. auto letter = m_reader.read();
  317. if (letter != 'n' && letter != 'f')
  318. return error("Malformed xref entry");
  319. // The line ending sequence can be one of the following:
  320. // SP CR, SP LF, or CR LF
  321. if (m_reader.matches(' ')) {
  322. m_reader.consume();
  323. auto ch = m_reader.consume();
  324. if (ch != '\r' && ch != '\n')
  325. return error("Malformed xref entry");
  326. } else {
  327. if (!m_reader.matches("\r\n"))
  328. return error("Malformed xref entry");
  329. m_reader.move_by(2);
  330. }
  331. auto offset = strtol(offset_string.characters(), nullptr, 10);
  332. auto generation = strtol(generation_string.characters(), nullptr, 10);
  333. entries.append({ offset, static_cast<u16>(generation), letter == 'n' });
  334. }
  335. table->add_section({ starting_index, object_count, entries });
  336. } while (m_reader.matches_number());
  337. return table;
  338. }
  339. PDFErrorOr<NonnullRefPtr<DictObject>> DocumentParser::parse_file_trailer()
  340. {
  341. while (m_reader.matches_eol())
  342. m_reader.consume_eol();
  343. if (!m_reader.matches("trailer"))
  344. return error("Expected \"trailer\" keyword");
  345. m_reader.move_by(7);
  346. m_reader.consume_whitespace();
  347. auto dict = TRY(parse_dict());
  348. if (!m_reader.matches("startxref"))
  349. return error("Expected \"startxref\"");
  350. m_reader.move_by(9);
  351. m_reader.consume_whitespace();
  352. m_reader.move_until([&](auto) { return m_reader.matches_eol(); });
  353. VERIFY(m_reader.consume_eol());
  354. if (!m_reader.matches("%%EOF"))
  355. return error("Expected \"%%EOF\"");
  356. m_reader.move_by(5);
  357. m_reader.consume_whitespace();
  358. return dict;
  359. }
  360. PDFErrorOr<Value> DocumentParser::parse_compressed_object_with_index(u32 index)
  361. {
  362. auto object_stream_index = m_xref_table->object_stream_for_object(index);
  363. auto stream_offset = m_xref_table->byte_offset_for_object(object_stream_index);
  364. m_reader.move_to(stream_offset);
  365. auto first_number = TRY(parse_number());
  366. auto second_number = TRY(parse_number());
  367. if (first_number.get<int>() != object_stream_index)
  368. return error("Mismatching object stream index");
  369. if (second_number.get<int>() != 0)
  370. return error("Non-zero object stream generation number");
  371. if (!m_reader.matches("obj"))
  372. return error("Malformed object stream");
  373. m_reader.move_by(3);
  374. if (m_reader.matches_eol())
  375. m_reader.consume_eol();
  376. auto dict = TRY(parse_dict());
  377. auto type = TRY(dict->get_name(m_document, CommonNames::Type))->name();
  378. if (type != "ObjStm")
  379. return error("Invalid object stream type");
  380. auto object_count = dict->get_value("N").get_u32();
  381. auto first_object_offset = dict->get_value("First").get_u32();
  382. auto stream = TRY(parse_stream(dict));
  383. Parser stream_parser(m_document, stream->bytes());
  384. for (u32 i = 0; i < object_count; ++i) {
  385. auto object_number = TRY(stream_parser.parse_number());
  386. auto object_offset = TRY(stream_parser.parse_number());
  387. if (object_number.get_u32() == index) {
  388. stream_parser.move_to(first_object_offset + object_offset.get_u32());
  389. break;
  390. }
  391. }
  392. return TRY(stream_parser.parse_value());
  393. }
  394. PDFErrorOr<DocumentParser::PageOffsetHintTable> DocumentParser::parse_page_offset_hint_table(ReadonlyBytes hint_stream_bytes)
  395. {
  396. if (hint_stream_bytes.size() < sizeof(PageOffsetHintTable))
  397. return error("Hint stream is too small");
  398. size_t offset = 0;
  399. auto read_u32 = [&] {
  400. u32 data = reinterpret_cast<const u32*>(hint_stream_bytes.data() + offset)[0];
  401. offset += 4;
  402. return AK::convert_between_host_and_big_endian(data);
  403. };
  404. auto read_u16 = [&] {
  405. u16 data = reinterpret_cast<const u16*>(hint_stream_bytes.data() + offset)[0];
  406. offset += 2;
  407. return AK::convert_between_host_and_big_endian(data);
  408. };
  409. PageOffsetHintTable hint_table {
  410. read_u32(),
  411. read_u32(),
  412. read_u16(),
  413. read_u32(),
  414. read_u16(),
  415. read_u32(),
  416. read_u16(),
  417. read_u32(),
  418. read_u16(),
  419. read_u16(),
  420. read_u16(),
  421. read_u16(),
  422. read_u16(),
  423. };
  424. // Verify that all of the bits_required_for_xyz fields are <= 32, since all of the numeric
  425. // fields in PageOffsetHintTableEntry are u32
  426. VERIFY(hint_table.bits_required_for_object_number <= 32);
  427. VERIFY(hint_table.bits_required_for_page_length <= 32);
  428. VERIFY(hint_table.bits_required_for_content_stream_offsets <= 32);
  429. VERIFY(hint_table.bits_required_for_content_stream_length <= 32);
  430. VERIFY(hint_table.bits_required_for_number_of_shared_obj_refs <= 32);
  431. VERIFY(hint_table.bits_required_for_greatest_shared_obj_identifier <= 32);
  432. VERIFY(hint_table.bits_required_for_fraction_numerator <= 32);
  433. return hint_table;
  434. }
  435. Vector<DocumentParser::PageOffsetHintTableEntry> DocumentParser::parse_all_page_offset_hint_table_entries(PageOffsetHintTable const& hint_table, ReadonlyBytes hint_stream_bytes)
  436. {
  437. InputMemoryStream input_stream(hint_stream_bytes);
  438. input_stream.seek(sizeof(PageOffsetHintTable));
  439. InputBitStream bit_stream(input_stream);
  440. auto number_of_pages = m_linearization_dictionary.value().number_of_pages;
  441. Vector<PageOffsetHintTableEntry> entries;
  442. for (size_t i = 0; i < number_of_pages; i++)
  443. entries.append(PageOffsetHintTableEntry {});
  444. auto bits_required_for_object_number = hint_table.bits_required_for_object_number;
  445. auto bits_required_for_page_length = hint_table.bits_required_for_page_length;
  446. auto bits_required_for_content_stream_offsets = hint_table.bits_required_for_content_stream_offsets;
  447. auto bits_required_for_content_stream_length = hint_table.bits_required_for_content_stream_length;
  448. auto bits_required_for_number_of_shared_obj_refs = hint_table.bits_required_for_number_of_shared_obj_refs;
  449. auto bits_required_for_greatest_shared_obj_identifier = hint_table.bits_required_for_greatest_shared_obj_identifier;
  450. auto bits_required_for_fraction_numerator = hint_table.bits_required_for_fraction_numerator;
  451. auto parse_int_entry = [&](u32 PageOffsetHintTableEntry::*field, u32 bit_size) {
  452. if (bit_size <= 0)
  453. return;
  454. for (int i = 0; i < number_of_pages; i++) {
  455. auto& entry = entries[i];
  456. entry.*field = bit_stream.read_bits(bit_size);
  457. }
  458. };
  459. auto parse_vector_entry = [&](Vector<u32> PageOffsetHintTableEntry::*field, u32 bit_size) {
  460. if (bit_size <= 0)
  461. return;
  462. for (int page = 1; page < number_of_pages; page++) {
  463. auto number_of_shared_objects = entries[page].number_of_shared_objects;
  464. Vector<u32> items;
  465. items.ensure_capacity(number_of_shared_objects);
  466. for (size_t i = 0; i < number_of_shared_objects; i++)
  467. items.unchecked_append(bit_stream.read_bits(bit_size));
  468. entries[page].*field = move(items);
  469. }
  470. };
  471. parse_int_entry(&PageOffsetHintTableEntry::objects_in_page_number, bits_required_for_object_number);
  472. parse_int_entry(&PageOffsetHintTableEntry::page_length_number, bits_required_for_page_length);
  473. parse_int_entry(&PageOffsetHintTableEntry::number_of_shared_objects, bits_required_for_number_of_shared_obj_refs);
  474. parse_vector_entry(&PageOffsetHintTableEntry::shared_object_identifiers, bits_required_for_greatest_shared_obj_identifier);
  475. parse_vector_entry(&PageOffsetHintTableEntry::shared_object_location_numerators, bits_required_for_fraction_numerator);
  476. parse_int_entry(&PageOffsetHintTableEntry::page_content_stream_offset_number, bits_required_for_content_stream_offsets);
  477. parse_int_entry(&PageOffsetHintTableEntry::page_content_stream_length_number, bits_required_for_content_stream_length);
  478. return entries;
  479. }
  480. bool DocumentParser::navigate_to_before_eof_marker()
  481. {
  482. m_reader.set_reading_backwards();
  483. while (!m_reader.done()) {
  484. m_reader.move_until([&](auto) { return m_reader.matches_eol(); });
  485. if (m_reader.done())
  486. return false;
  487. m_reader.consume_eol();
  488. if (!m_reader.matches("%%EOF"))
  489. continue;
  490. m_reader.move_by(5);
  491. if (!m_reader.matches_eol())
  492. continue;
  493. m_reader.consume_eol();
  494. return true;
  495. }
  496. return false;
  497. }
  498. bool DocumentParser::navigate_to_after_startxref()
  499. {
  500. m_reader.set_reading_backwards();
  501. while (!m_reader.done()) {
  502. m_reader.move_until([&](auto) { return m_reader.matches_eol(); });
  503. auto offset = m_reader.offset() + 1;
  504. m_reader.consume_eol();
  505. if (!m_reader.matches("startxref"))
  506. continue;
  507. m_reader.move_by(9);
  508. if (!m_reader.matches_eol())
  509. continue;
  510. m_reader.move_to(offset);
  511. return true;
  512. }
  513. return false;
  514. }
  515. PDFErrorOr<RefPtr<DictObject>> DocumentParser::conditionally_parse_page_tree_node(u32 object_index)
  516. {
  517. auto dict_value = TRY(parse_object_with_index(object_index));
  518. auto dict_object = dict_value.get<NonnullRefPtr<Object>>();
  519. if (!dict_object->is<DictObject>())
  520. return error(String::formatted("Invalid page tree with xref index {}", object_index));
  521. auto dict = dict_object->cast<DictObject>();
  522. if (!dict->contains_any_of(CommonNames::Type, CommonNames::Parent, CommonNames::Kids, CommonNames::Count))
  523. // This is a page, not a page tree node
  524. return RefPtr<DictObject> {};
  525. if (!dict->contains(CommonNames::Type))
  526. return RefPtr<DictObject> {};
  527. auto type_object = TRY(dict->get_object(m_document, CommonNames::Type));
  528. if (!type_object->is<NameObject>())
  529. return RefPtr<DictObject> {};
  530. auto type_name = type_object->cast<NameObject>();
  531. if (type_name->name() != CommonNames::Pages)
  532. return RefPtr<DictObject> {};
  533. return dict;
  534. }
  535. }
  536. namespace AK {
  537. template<>
  538. struct Formatter<PDF::DocumentParser::LinearizationDictionary> : Formatter<StringView> {
  539. ErrorOr<void> format(FormatBuilder& format_builder, PDF::DocumentParser::LinearizationDictionary const& dict)
  540. {
  541. StringBuilder builder;
  542. builder.append("{\n"sv);
  543. builder.appendff(" length_of_file={}\n", dict.length_of_file);
  544. builder.appendff(" primary_hint_stream_offset={}\n", dict.primary_hint_stream_offset);
  545. builder.appendff(" primary_hint_stream_length={}\n", dict.primary_hint_stream_length);
  546. builder.appendff(" overflow_hint_stream_offset={}\n", dict.overflow_hint_stream_offset);
  547. builder.appendff(" overflow_hint_stream_length={}\n", dict.overflow_hint_stream_length);
  548. builder.appendff(" first_page_object_number={}\n", dict.first_page_object_number);
  549. builder.appendff(" offset_of_first_page_end={}\n", dict.offset_of_first_page_end);
  550. builder.appendff(" number_of_pages={}\n", dict.number_of_pages);
  551. builder.appendff(" offset_of_main_xref_table={}\n", dict.offset_of_main_xref_table);
  552. builder.appendff(" first_page={}\n", dict.first_page);
  553. builder.append('}');
  554. return Formatter<StringView>::format(format_builder, builder.to_string());
  555. }
  556. };
  557. template<>
  558. struct Formatter<PDF::DocumentParser::PageOffsetHintTable> : Formatter<StringView> {
  559. ErrorOr<void> format(FormatBuilder& format_builder, PDF::DocumentParser::PageOffsetHintTable const& table)
  560. {
  561. StringBuilder builder;
  562. builder.append("{\n"sv);
  563. builder.appendff(" least_number_of_objects_in_a_page={}\n", table.least_number_of_objects_in_a_page);
  564. builder.appendff(" location_of_first_page_object={}\n", table.location_of_first_page_object);
  565. builder.appendff(" bits_required_for_object_number={}\n", table.bits_required_for_object_number);
  566. builder.appendff(" least_length_of_a_page={}\n", table.least_length_of_a_page);
  567. builder.appendff(" bits_required_for_page_length={}\n", table.bits_required_for_page_length);
  568. builder.appendff(" least_offset_of_any_content_stream={}\n", table.least_offset_of_any_content_stream);
  569. builder.appendff(" bits_required_for_content_stream_offsets={}\n", table.bits_required_for_content_stream_offsets);
  570. builder.appendff(" least_content_stream_length={}\n", table.least_content_stream_length);
  571. builder.appendff(" bits_required_for_content_stream_length={}\n", table.bits_required_for_content_stream_length);
  572. builder.appendff(" bits_required_for_number_of_shared_obj_refs={}\n", table.bits_required_for_number_of_shared_obj_refs);
  573. builder.appendff(" bits_required_for_greatest_shared_obj_identifier={}\n", table.bits_required_for_greatest_shared_obj_identifier);
  574. builder.appendff(" bits_required_for_fraction_numerator={}\n", table.bits_required_for_fraction_numerator);
  575. builder.appendff(" shared_object_reference_fraction_denominator={}\n", table.shared_object_reference_fraction_denominator);
  576. builder.append('}');
  577. return Formatter<StringView>::format(format_builder, builder.to_string());
  578. }
  579. };
  580. template<>
  581. struct Formatter<PDF::DocumentParser::PageOffsetHintTableEntry> : Formatter<StringView> {
  582. ErrorOr<void> format(FormatBuilder& format_builder, PDF::DocumentParser::PageOffsetHintTableEntry const& entry)
  583. {
  584. StringBuilder builder;
  585. builder.append("{\n"sv);
  586. builder.appendff(" objects_in_page_number={}\n", entry.objects_in_page_number);
  587. builder.appendff(" page_length_number={}\n", entry.page_length_number);
  588. builder.appendff(" number_of_shared_objects={}\n", entry.number_of_shared_objects);
  589. builder.append(" shared_object_identifiers=["sv);
  590. for (auto& identifier : entry.shared_object_identifiers)
  591. builder.appendff(" {}", identifier);
  592. builder.append(" ]\n"sv);
  593. builder.append(" shared_object_location_numerators=["sv);
  594. for (auto& numerator : entry.shared_object_location_numerators)
  595. builder.appendff(" {}", numerator);
  596. builder.append(" ]\n"sv);
  597. builder.appendff(" page_content_stream_offset_number={}\n", entry.page_content_stream_offset_number);
  598. builder.appendff(" page_content_stream_length_number={}\n", entry.page_content_stream_length_number);
  599. builder.append('}');
  600. return Formatter<StringView>::format(format_builder, builder.to_string());
  601. }
  602. };
  603. }