RGBCube/serenity
1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-10-31 16:52:43 +00:00
Code Issues Activity Actions

LibPDF: Propagate errors in Parser and Document

Browse source
This commit is contained in:
Matthew Olsson 2022-03-05 17:30:55 -07:00 • committed by Andreas Kling
parent 7e1c823725
commit 73cf8205b4
16 changed files with 472 additions and 420 deletions
Download patch file Download diff file Expand all files Collapse all files

8
Meta/Lagom/Fuzzers/FuzzPDF.cpp
View file

@ -10,12 +10,12 @@
extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
{ {
ReadonlyBytes bytes { data, size }; ReadonlyBytes bytes { data, size };
auto doc = PDF::Document::create(bytes);
if (doc) { if (auto maybe_document = PDF::Document::create(bytes); !maybe_document.is_error()) {
auto pages = doc->get_page_count(); auto document = maybe_document.release_value();
auto pages = document->get_page_count();
for (size_t i = 0; i < pages; ++i) { for (size_t i = 0; i < pages; ++i) {
(void)doc->get_page(i); (void)document->get_page(i);
} }
} }

13
Tests/LibPDF/TestPDF.cpp
View file

@ -15,33 +15,36 @@ TEST_CASE(linearized_pdf)
{ {
auto file = Core::MappedFile::map("linearized.pdf").release_value(); auto file = Core::MappedFile::map("linearized.pdf").release_value();
auto document = PDF::Document::create(file->bytes()); auto document = PDF::Document::create(file->bytes());
EXPECT_EQ(document->get_page_count(), 1U); EXPECT(!document.is_error());
EXPECT_EQ(document.value()->get_page_count(), 1U);
} }
TEST_CASE(non_linearized_pdf) TEST_CASE(non_linearized_pdf)
{ {
auto file = Core::MappedFile::map("non-linearized.pdf").release_value(); auto file = Core::MappedFile::map("non-linearized.pdf").release_value();
auto document = PDF::Document::create(file->bytes()); auto document = PDF::Document::create(file->bytes());
EXPECT_EQ(document->get_page_count(), 1U); EXPECT(!document.is_error());
EXPECT_EQ(document.value()->get_page_count(), 1U);
} }
TEST_CASE(complex_pdf) TEST_CASE(complex_pdf)
{ {
auto file = Core::MappedFile::map("complex.pdf").release_value(); auto file = Core::MappedFile::map("complex.pdf").release_value();
auto document = PDF::Document::create(file->bytes()); auto document = PDF::Document::create(file->bytes());
EXPECT_EQ(document->get_page_count(), 3U); EXPECT(!document.is_error());
EXPECT_EQ(document.value()->get_page_count(), 3U);
} }
TEST_CASE(empty_file_issue_10702) TEST_CASE(empty_file_issue_10702)
{ {
AK::ReadonlyBytes empty; AK::ReadonlyBytes empty;
auto document = PDF::Document::create(empty); auto document = PDF::Document::create(empty);
EXPECT(document.is_null()); EXPECT(document.is_error());
} }
TEST_CASE(truncated_pdf_header_issue_10717) TEST_CASE(truncated_pdf_header_issue_10717)
{ {
AK::String string { "%PDF-2.11%" }; AK::String string { "%PDF-2.11%" };
auto document = PDF::Document::create(string.bytes()); auto document = PDF::Document::create(string.bytes());
EXPECT(document.is_null()); EXPECT(document.is_error());
} }

3
Userland/Applications/PDFViewer/PDFViewer.cpp
View file

@ -63,7 +63,8 @@ RefPtr<Gfx::Bitmap> PDFViewer::get_rendered_page(u32 index)
if (existing_rendered_page.has_value() && existing_rendered_page.value().rotation == m_rotations) if (existing_rendered_page.has_value() && existing_rendered_page.value().rotation == m_rotations)
return existing_rendered_page.value().bitmap; return existing_rendered_page.value().bitmap;
auto rendered_page = render_page(m_document->get_page(index)); // FIXME: Propogate errors in the Renderer
auto rendered_page = render_page(MUST(m_document->get_page(index)));
rendered_page_map.set(m_zoom_level, { rendered_page, m_rotations }); rendered_page_map.set(m_zoom_level, { rendered_page, m_rotations });
return rendered_page; return rendered_page;
} }

11
Userland/Applications/PDFViewer/PDFViewerWidget.cpp
View file

@ -1,5 +1,5 @@
/* /*
* Copyright (c) 2021, Matthew Olsson <mattco@serenityos.org> * Copyright (c) 2021-2022, Matthew Olsson <mattco@serenityos.org>
* Copyright (c) 2021, Mustafa Quraish <mustafa@serenityos.org> * Copyright (c) 2021, Mustafa Quraish <mustafa@serenityos.org>
* *
* SPDX-License-Identifier: BSD-2-Clause * SPDX-License-Identifier: BSD-2-Clause
@ -151,12 +151,15 @@ void PDFViewerWidget::open_file(Core::File& file)
window()->set_title(String::formatted("{} - PDF Viewer", file.filename())); window()->set_title(String::formatted("{} - PDF Viewer", file.filename()));
m_buffer = file.read_all(); m_buffer = file.read_all();
auto document = PDF::Document::create(m_buffer); auto maybe_document = PDF::Document::create(m_buffer);
if (!document) { if (maybe_document.is_error()) {
GUI::MessageBox::show_error(nullptr, String::formatted("Couldn't load PDF: {}", file.filename())); auto error = maybe_document.release_error();
GUI::MessageBox::show_error(nullptr, String::formatted("Couldn't load PDF {}:\n{}", file.filename(), error.message()));
return; return;
} }
auto document = maybe_document.release_value();
m_viewer->set_document(document); m_viewer->set_document(document);
m_total_page_label->set_text(String::formatted("of {}", document->get_page_count())); m_total_page_label->set_text(String::formatted("of {}", document->get_page_count()));

15
Userland/Libraries/LibPDF/ColorSpace.cpp
View file

@ -1,5 +1,5 @@
/* /*
* Copyright (c) 2021, Matthew Olsson <mattco@serenityos.org> * Copyright (c) 2021-2022, Matthew Olsson <mattco@serenityos.org>
* *
* SPDX-License-Identifier: BSD-2-Clause * SPDX-License-Identifier: BSD-2-Clause
*/ */
@ -60,14 +60,15 @@ RefPtr<CalRGBColorSpace> CalRGBColorSpace::create(RefPtr<Document> document, Vec
return {}; return {};
auto param = parameters[0]; auto param = parameters[0];
if (!param.has<NonnullRefPtr<Object>>() || !param.get<NonnullRefPtr<Object>>()->is_dict()) if (!param.has<NonnullRefPtr<Object>>() || !param.get<NonnullRefPtr<Object>>()->is<DictObject>())
return {}; return {};
auto dict = object_cast<DictObject>(param.get<NonnullRefPtr<Object>>()); auto dict = param.get<NonnullRefPtr<Object>>()->cast<DictObject>();
if (!dict->contains(CommonNames::WhitePoint)) if (!dict->contains(CommonNames::WhitePoint))
return {}; return {};
auto white_point_array = dict->get_array(document, CommonNames::WhitePoint); // FIXME: Propagate errors
auto white_point_array = MUST(dict->get_array(document, CommonNames::WhitePoint));
if (white_point_array->size() != 3) if (white_point_array->size() != 3)
return {}; return {};
@ -81,7 +82,7 @@ RefPtr<CalRGBColorSpace> CalRGBColorSpace::create(RefPtr<Document> document, Vec
return {}; return {};
if (dict->contains(CommonNames::BlackPoint)) { if (dict->contains(CommonNames::BlackPoint)) {
auto black_point_array = dict->get_array(document, CommonNames::BlackPoint); auto black_point_array = MUST(dict->get_array(document, CommonNames::BlackPoint));
if (black_point_array->size() == 3) { if (black_point_array->size() == 3) {
color_space->m_blackpoint[0] = black_point_array->at(0).to_float(); color_space->m_blackpoint[0] = black_point_array->at(0).to_float();
color_space->m_blackpoint[1] = black_point_array->at(1).to_float(); color_space->m_blackpoint[1] = black_point_array->at(1).to_float();
@ -90,7 +91,7 @@ RefPtr<CalRGBColorSpace> CalRGBColorSpace::create(RefPtr<Document> document, Vec
} }
if (dict->contains(CommonNames::Gamma)) { if (dict->contains(CommonNames::Gamma)) {
auto gamma_array = dict->get_array(document, CommonNames::Gamma); auto gamma_array = MUST(dict->get_array(document, CommonNames::Gamma));
if (gamma_array->size() == 3) { if (gamma_array->size() == 3) {
color_space->m_gamma[0] = gamma_array->at(0).to_float(); color_space->m_gamma[0] = gamma_array->at(0).to_float();
color_space->m_gamma[1] = gamma_array->at(1).to_float(); color_space->m_gamma[1] = gamma_array->at(1).to_float();
@ -99,7 +100,7 @@ RefPtr<CalRGBColorSpace> CalRGBColorSpace::create(RefPtr<Document> document, Vec
} }
if (dict->contains(CommonNames::Matrix)) { if (dict->contains(CommonNames::Matrix)) {
auto matrix_array = dict->get_array(document, CommonNames::Matrix); auto matrix_array = MUST(dict->get_array(document, CommonNames::Matrix));
if (matrix_array->size() == 3) { if (matrix_array->size() == 3) {
color_space->m_matrix[0] = matrix_array->at(0).to_float(); color_space->m_matrix[0] = matrix_array->at(0).to_float();
color_space->m_matrix[1] = matrix_array->at(1).to_float(); color_space->m_matrix[1] = matrix_array->at(1).to_float();

97
Userland/Libraries/LibPDF/Document.cpp
View file

@ -1,5 +1,5 @@
/* /*
* Copyright (c) 2021, Matthew Olsson <mattco@serenityos.org> * Copyright (c) 2021-2022, Matthew Olsson <mattco@serenityos.org>
* *
* SPDX-License-Identifier: BSD-2-Clause * SPDX-License-Identifier: BSD-2-Clause
*/ */
@ -34,17 +34,16 @@ String OutlineItem::to_string(int indent) const
return builder.to_string(); return builder.to_string();
} }
RefPtr<Document> Document::create(ReadonlyBytes bytes) PDFErrorOr<NonnullRefPtr<Document>> Document::create(ReadonlyBytes bytes)
{ {
auto parser = adopt_ref(*new Parser({}, bytes)); auto parser = adopt_ref(*new Parser({}, bytes));
auto document = adopt_ref(*new Document(parser)); auto document = adopt_ref(*new Document(parser));
if (!parser->initialize()) TRY(parser->initialize());
return {};
document->m_catalog = parser->trailer()->get_dict(document, CommonNames::Root); document->m_catalog = TRY(parser->trailer()->get_dict(document, CommonNames::Root));
document->build_page_tree(); TRY(document->build_page_tree());
document->build_outline(); TRY(document->build_outline());
return document; return document;
} }
@ -55,13 +54,13 @@ Document::Document(NonnullRefPtr<Parser> const& parser)
m_parser->set_document(this); m_parser->set_document(this);
} }
Value Document::get_or_load_value(u32 index) PDFErrorOr<Value> Document::get_or_load_value(u32 index)
{ {
auto value = get_value(index); auto value = get_value(index);
if (!value.has<Empty>()) // FIXME: Use Optional instead? if (!value.has<Empty>()) // FIXME: Use Optional instead?
return value; return value;
auto object = m_parser->parse_object_with_index(index); auto object = TRY(m_parser->parse_object_with_index(index));
m_values.set(index, object); m_values.set(index, object);
return object; return object;
} }
@ -78,7 +77,7 @@ u32 Document::get_page_count() const
return m_page_object_indices.size(); return m_page_object_indices.size();
} }
Page Document::get_page(u32 index) PDFErrorOr<Page> Document::get_page(u32 index)
{ {
VERIFY(index < m_page_object_indices.size()); VERIFY(index < m_page_object_indices.size());
@ -87,17 +86,18 @@ Page Document::get_page(u32 index)
return cached_page.value(); return cached_page.value();
auto page_object_index = m_page_object_indices[index]; auto page_object_index = m_page_object_indices[index];
auto raw_page_object = resolve_to<DictObject>(get_or_load_value(page_object_index)); auto page_object = TRY(get_or_load_value(page_object_index));
auto raw_page_object = TRY(resolve_to<DictObject>(page_object));
if (!raw_page_object->contains(CommonNames::Resources)) { if (!raw_page_object->contains(CommonNames::Resources)) {
// This page inherits its resource dictionary // This page inherits its resource dictionary
TODO(); TODO();
} }
auto resources = raw_page_object->get_dict(this, CommonNames::Resources); auto resources = TRY(raw_page_object->get_dict(this, CommonNames::Resources));
auto contents = raw_page_object->get_object(this, CommonNames::Contents); auto contents = TRY(raw_page_object->get_object(this, CommonNames::Contents));
auto media_box_array = raw_page_object->get_array(this, CommonNames::MediaBox); auto media_box_array = TRY(raw_page_object->get_array(this, CommonNames::MediaBox));
auto media_box = Rectangle { auto media_box = Rectangle {
media_box_array->at(0).to_float(), media_box_array->at(0).to_float(),
media_box_array->at(1).to_float(), media_box_array->at(1).to_float(),
@ -107,7 +107,7 @@ Page Document::get_page(u32 index)
auto crop_box = media_box; auto crop_box = media_box;
if (raw_page_object->contains(CommonNames::CropBox)) { if (raw_page_object->contains(CommonNames::CropBox)) {
auto crop_box_array = raw_page_object->get_array(this, CommonNames::CropBox); auto crop_box_array = TRY(raw_page_object->get_array(this, CommonNames::CropBox));
crop_box = Rectangle { crop_box = Rectangle {
crop_box_array->at(0).to_float(), crop_box_array->at(0).to_float(),
crop_box_array->at(1).to_float(), crop_box_array->at(1).to_float(),
@ -131,7 +131,7 @@ Page Document::get_page(u32 index)
return page; return page;
} }
Value Document::resolve(Value const& value) PDFErrorOr<Value> Document::resolve(Value const& value)
{ {
if (value.has<Reference>()) { if (value.has<Reference>()) {
// FIXME: Surely indirect PDF objects can't contain another indirect PDF object, // FIXME: Surely indirect PDF objects can't contain another indirect PDF object,
@ -145,26 +145,21 @@ Value Document::resolve(Value const& value)
auto& obj = value.get<NonnullRefPtr<Object>>(); auto& obj = value.get<NonnullRefPtr<Object>>();
if (obj->is_indirect_value()) if (obj->is<IndirectValue>())
return static_ptr_cast<IndirectValue>(obj)->value(); return static_ptr_cast<IndirectValue>(obj)->value();
return value; return value;
} }
bool Document::build_page_tree() PDFErrorOr<void> Document::build_page_tree()
{ {
if (!m_catalog->contains(CommonNames::Pages)) auto page_tree = TRY(m_catalog->get_dict(this, CommonNames::Pages));
return false;
auto page_tree = m_catalog->get_dict(this, CommonNames::Pages);
return add_page_tree_node_to_page_tree(page_tree); return add_page_tree_node_to_page_tree(page_tree);
} }
bool Document::add_page_tree_node_to_page_tree(NonnullRefPtr<DictObject> const& page_tree) PDFErrorOr<void> Document::add_page_tree_node_to_page_tree(NonnullRefPtr<DictObject> const& page_tree)
{ {
if (!page_tree->contains(CommonNames::Kids) || !page_tree->contains(CommonNames::Count)) auto kids_array = TRY(page_tree->get_array(this, CommonNames::Kids));
return false;
auto kids_array = page_tree->get_array(this, CommonNames::Kids);
auto page_count = page_tree->get(CommonNames::Count).value().get<int>(); auto page_count = page_tree->get(CommonNames::Count).value().get<int>();
if (static_cast<size_t>(page_count) != kids_array->elements().size()) { if (static_cast<size_t>(page_count) != kids_array->elements().size()) {
@ -173,13 +168,9 @@ bool Document::add_page_tree_node_to_page_tree(NonnullRefPtr<DictObject> const&
for (auto& value : *kids_array) { for (auto& value : *kids_array) {
auto reference_index = value.as_ref_index(); auto reference_index = value.as_ref_index();
bool ok; auto maybe_page_tree_node = TRY(m_parser->conditionally_parse_page_tree_node(reference_index));
auto maybe_page_tree_node = m_parser->conditionally_parse_page_tree_node(reference_index, ok);
if (!ok)
return false;
if (maybe_page_tree_node) { if (maybe_page_tree_node) {
if (!add_page_tree_node_to_page_tree(maybe_page_tree_node.release_nonnull())) TRY(add_page_tree_node_to_page_tree(maybe_page_tree_node.release_nonnull()));
return false;
} else { } else {
m_page_object_indices.append(reference_index); m_page_object_indices.append(reference_index);
} }
@ -190,33 +181,35 @@ bool Document::add_page_tree_node_to_page_tree(NonnullRefPtr<DictObject> const&
m_page_object_indices.append(value.as_ref_index()); m_page_object_indices.append(value.as_ref_index());
} }
return true; return {};
} }
void Document::build_outline() PDFErrorOr<void> Document::build_outline()
{ {
if (!m_catalog->contains(CommonNames::Outlines)) if (!m_catalog->contains(CommonNames::Outlines))
return; return {};
auto outline_dict = m_catalog->get_dict(this, CommonNames::Outlines); auto outline_dict = TRY(m_catalog->get_dict(this, CommonNames::Outlines));
if (!outline_dict->contains(CommonNames::First)) if (!outline_dict->contains(CommonNames::First))
return; return {};
if (!outline_dict->contains(CommonNames::Last)) if (!outline_dict->contains(CommonNames::Last))
return; return {};
auto first_ref = outline_dict->get_value(CommonNames::First); auto first_ref = outline_dict->get_value(CommonNames::First);
auto last_ref = outline_dict->get_value(CommonNames::Last); auto last_ref = outline_dict->get_value(CommonNames::Last);
auto children = build_outline_item_chain(first_ref, last_ref); auto children = TRY(build_outline_item_chain(first_ref, last_ref));
m_outline = adopt_ref(*new OutlineDict()); m_outline = adopt_ref(*new OutlineDict());
m_outline->children = move(children); m_outline->children = move(children);
if (outline_dict->contains(CommonNames::Count)) if (outline_dict->contains(CommonNames::Count))
m_outline->count = outline_dict->get_value(CommonNames::Count).get<int>(); m_outline->count = outline_dict->get_value(CommonNames::Count).get<int>();
return {};
} }
NonnullRefPtr<OutlineItem> Document::build_outline_item(NonnullRefPtr<DictObject> const& outline_item_dict) PDFErrorOr<NonnullRefPtr<OutlineItem>> Document::build_outline_item(NonnullRefPtr<DictObject> const& outline_item_dict)
{ {
auto outline_item = adopt_ref(*new OutlineItem {}); auto outline_item = adopt_ref(*new OutlineItem {});
@ -225,19 +218,20 @@ NonnullRefPtr<OutlineItem> Document::build_outline_item(NonnullRefPtr<DictObject
auto first_ref = outline_item_dict->get_value(CommonNames::First); auto first_ref = outline_item_dict->get_value(CommonNames::First);
auto last_ref = outline_item_dict->get_value(CommonNames::Last); auto last_ref = outline_item_dict->get_value(CommonNames::Last);
auto children = build_outline_item_chain(first_ref, last_ref); auto children = TRY(build_outline_item_chain(first_ref, last_ref));
outline_item->children = move(children); outline_item->children = move(children);
} }
outline_item->title = outline_item_dict->get_string(this, CommonNames::Title)->string(); outline_item->title = TRY(outline_item_dict->get_string(this, CommonNames::Title))->string();
if (outline_item_dict->contains(CommonNames::Count)) if (outline_item_dict->contains(CommonNames::Count))
outline_item->count = outline_item_dict->get_value(CommonNames::Count).get<int>(); outline_item->count = outline_item_dict->get_value(CommonNames::Count).get<int>();
if (outline_item_dict->contains(CommonNames::Dest)) { if (outline_item_dict->contains(CommonNames::Dest)) {
auto dest_arr = outline_item_dict->get_array(this, CommonNames::Dest); auto dest_arr = TRY(outline_item_dict->get_array(this, CommonNames::Dest));
dbgln("IS ARRAY = {}", dest_arr->is_array());
auto page_ref = dest_arr->at(0); auto page_ref = dest_arr->at(0);
auto type_name = dest_arr->get_name_at(this, 1)->name(); auto type_name = TRY(dest_arr->get_name_at(this, 1))->name();
Vector<float> parameters; Vector<float> parameters;
for (size_t i = 2; i < dest_arr->size(); i++) for (size_t i = 2; i < dest_arr->size(); i++)
@ -268,7 +262,7 @@ NonnullRefPtr<OutlineItem> Document::build_outline_item(NonnullRefPtr<DictObject
} }
if (outline_item_dict->contains(CommonNames::C)) { if (outline_item_dict->contains(CommonNames::C)) {
auto color_array = outline_item_dict->get_array(this, CommonNames::C); auto color_array = TRY(outline_item_dict->get_array(this, CommonNames::C));
auto r = static_cast<int>(255.0f * color_array->at(0).get<float>()); auto r = static_cast<int>(255.0f * color_array->at(0).get<float>());
auto g = static_cast<int>(255.0f * color_array->at(1).get<float>()); auto g = static_cast<int>(255.0f * color_array->at(1).get<float>());
auto b = static_cast<int>(255.0f * color_array->at(2).get<float>()); auto b = static_cast<int>(255.0f * color_array->at(2).get<float>());
@ -284,16 +278,16 @@ NonnullRefPtr<OutlineItem> Document::build_outline_item(NonnullRefPtr<DictObject
return outline_item; return outline_item;
} }
NonnullRefPtrVector<OutlineItem> Document::build_outline_item_chain(Value const& first_ref, Value const& last_ref) PDFErrorOr<NonnullRefPtrVector<OutlineItem>> Document::build_outline_item_chain(Value const& first_ref, Value const& last_ref)
{ {
VERIFY(first_ref.has<Reference>()); VERIFY(first_ref.has<Reference>());
VERIFY(last_ref.has<Reference>()); VERIFY(last_ref.has<Reference>());
NonnullRefPtrVector<OutlineItem> children; NonnullRefPtrVector<OutlineItem> children;
auto first_dict = object_cast<DictObject>( auto first_value = TRY(get_or_load_value(first_ref.as_ref_index())).get<NonnullRefPtr<Object>>();
get_or_load_value(first_ref.as_ref_index()).get<NonnullRefPtr<Object>>()); auto first_dict = first_value->cast<DictObject>();
auto first = build_outline_item(first_dict); auto first = TRY(build_outline_item(first_dict));
children.append(first); children.append(first);
auto current_child_dict = first_dict; auto current_child_dict = first_dict;
@ -302,8 +296,9 @@ NonnullRefPtrVector<OutlineItem> Document::build_outline_item_chain(Value const&
while (current_child_dict->contains(CommonNames::Next)) { while (current_child_dict->contains(CommonNames::Next)) {
auto next_child_dict_ref = current_child_dict->get_value(CommonNames::Next); auto next_child_dict_ref = current_child_dict->get_value(CommonNames::Next);
current_child_index = next_child_dict_ref.as_ref_index(); current_child_index = next_child_dict_ref.as_ref_index();
auto next_child_dict = object_cast<DictObject>(get_or_load_value(current_child_index).get<NonnullRefPtr<Object>>()); auto next_child_value = TRY(get_or_load_value(current_child_index)).get<NonnullRefPtr<Object>>();
auto next_child = build_outline_item(next_child_dict); auto next_child_dict = next_child_value->cast<DictObject>();
auto next_child = TRY(build_outline_item(next_child_dict));
children.append(next_child); children.append(next_child);
current_child_dict = move(next_child_dict); current_child_dict = move(next_child_dict);

35
Userland/Libraries/LibPDF/Document.h
View file

@ -1,5 +1,5 @@
/* /*
* Copyright (c) 2021, Matthew Olsson <mattco@serenityos.org> * Copyright (c) 2021-2022, Matthew Olsson <mattco@serenityos.org>
* *
* SPDX-License-Identifier: BSD-2-Clause * SPDX-License-Identifier: BSD-2-Clause
*/ */
@ -11,6 +11,7 @@
#include <AK/RefCounted.h> #include <AK/RefCounted.h>
#include <AK/Weakable.h> #include <AK/Weakable.h>
#include <LibGfx/Color.h> #include <LibGfx/Color.h>
#include <LibPDF/Error.h>
#include <LibPDF/ObjectDerivatives.h> #include <LibPDF/ObjectDerivatives.h>
#include <LibPDF/Parser.h> #include <LibPDF/Parser.h>
@ -75,17 +76,17 @@ class Document final
: public RefCounted<Document> : public RefCounted<Document>
, public Weakable<Document> { , public Weakable<Document> {
public: public:
static RefPtr<Document> create(ReadonlyBytes bytes); static PDFErrorOr<NonnullRefPtr<Document>> create(ReadonlyBytes bytes);
ALWAYS_INLINE RefPtr<OutlineDict> const& outline() const { return m_outline; } ALWAYS_INLINE RefPtr<OutlineDict> const& outline() const { return m_outline; }
[[nodiscard]] Value get_or_load_value(u32 index); [[nodiscard]] PDFErrorOr<Value> get_or_load_value(u32 index);
[[nodiscard]] u32 get_first_page_index() const; [[nodiscard]] u32 get_first_page_index() const;
[[nodiscard]] u32 get_page_count() const; [[nodiscard]] u32 get_page_count() const;
[[nodiscard]] Page get_page(u32 index); [[nodiscard]] PDFErrorOr<Page> get_page(u32 index);
ALWAYS_INLINE Value get_value(u32 index) const ALWAYS_INLINE Value get_value(u32 index) const
{ {
@ -95,23 +96,25 @@ public:
// Strips away the layer of indirection by turning indirect value // Strips away the layer of indirection by turning indirect value
// refs into the value they reference, and indirect values into // refs into the value they reference, and indirect values into
// the value being wrapped. // the value being wrapped.
Value resolve(Value const& value); PDFErrorOr<Value> resolve(Value const& value);
// Like resolve, but unwraps the Value into the given type. Accepts // Like resolve, but unwraps the Value into the given type. Accepts
// any object type, and the three primitive Value types. // any object type, and the three primitive Value types.
template<IsValueType T> template<IsValueType T>
UnwrappedValueType<T> resolve_to(Value const& value) PDFErrorOr<UnwrappedValueType<T>> resolve_to(Value const& value)
{ {
auto resolved = resolve(value); auto resolved = TRY(resolve(value));
if constexpr (IsSame<T, bool>) if constexpr (IsSame<T, bool>)
return resolved.get<bool>(); return resolved.get<bool>();
if constexpr (IsSame<T, int>) else if constexpr (IsSame<T, int>)
return resolved.get<int>(); return resolved.get<int>();
if constexpr (IsSame<T, float>) else if constexpr (IsSame<T, float>)
return resolved.get<float>(); return resolved.get<float>();
if constexpr (IsObject<T>) else if constexpr (IsSame<T, Object>)
return object_cast<T>(resolved.get<NonnullRefPtr<Object>>()); return resolved.get<NonnullRefPtr<Object>>();
else if constexpr (IsObject<T>)
return resolved.get<NonnullRefPtr<Object>>()->cast<T>();
VERIFY_NOT_REACHED(); VERIFY_NOT_REACHED();
} }
@ -125,12 +128,12 @@ private:
// parsing, as good PDF writers will layout the page tree in a balanced tree to // parsing, as good PDF writers will layout the page tree in a balanced tree to
// improve lookup time. This would reduce the initial overhead by not loading // improve lookup time. This would reduce the initial overhead by not loading
// every page tree node of, say, a 1000+ page PDF file. // every page tree node of, say, a 1000+ page PDF file.
bool build_page_tree(); PDFErrorOr<void> build_page_tree();
bool add_page_tree_node_to_page_tree(NonnullRefPtr<DictObject> const& page_tree); PDFErrorOr<void> add_page_tree_node_to_page_tree(NonnullRefPtr<DictObject> const& page_tree);
void build_outline(); PDFErrorOr<void> build_outline();
NonnullRefPtr<OutlineItem> build_outline_item(NonnullRefPtr<DictObject> const& outline_item_dict); PDFErrorOr<NonnullRefPtr<OutlineItem>> build_outline_item(NonnullRefPtr<DictObject> const& outline_item_dict);
NonnullRefPtrVector<OutlineItem> build_outline_item_chain(Value const& first_ref, Value const& last_ref); PDFErrorOr<NonnullRefPtrVector<OutlineItem>> build_outline_item_chain(Value const& first_ref, Value const& last_ref);
NonnullRefPtr<Parser> m_parser; NonnullRefPtr<Parser> m_parser;
RefPtr<DictObject> m_catalog; RefPtr<DictObject> m_catalog;

48
Userland/Libraries/LibPDF/Error.h Normal file
View file

@ -0,0 +1,48 @@
/*
* Copyright (c) 2022, Matthew Olsson <mattco@serenityos.org>
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#pragma once
#include <AK/String.h>
namespace PDF {
class Error {
public:
enum class Type {
Parse,
Internal,
MalformedPDF,
};
Error(Type type, String const& message)
: m_type(type)
{
switch (type) {
case Type::Parse:
m_message = String::formatted("Failed to parse PDF file: {}", message);
break;
case Type::Internal:
m_message = String::formatted("Internal error while processing PDF file: {}", message);
break;
case Type::MalformedPDF:
m_message = String::formatted("Malformed PDF file: {}", message);
break;
}
}
Type type() const { return m_type; }
String const& message() const { return m_message; }
private:
Type m_type;
String m_message;
};
template<typename T>
using PDFErrorOr = ErrorOr<T, Error>;
}

83
Userland/Libraries/LibPDF/Object.h
View file

@ -1,5 +1,5 @@
/* /*
* Copyright (c) 2021, Matthew Olsson <mattco@serenityos.org> * Copyright (c) 2021-2022, Matthew Olsson <mattco@serenityos.org>
* *
* SPDX-License-Identifier: BSD-2-Clause * SPDX-License-Identifier: BSD-2-Clause
*/ */
@ -11,9 +11,29 @@
#include <AK/Format.h> #include <AK/Format.h>
#include <AK/RefCounted.h> #include <AK/RefCounted.h>
#include <AK/SourceLocation.h> #include <AK/SourceLocation.h>
#include <LibPDF/Error.h>
#include <LibPDF/Forward.h> #include <LibPDF/Forward.h>
#include <LibPDF/Value.h> #include <LibPDF/Value.h>
#ifdef PDF_DEBUG
namespace {
template<PDF::IsObject T>
const char* object_name()
{
# define ENUMERATE_TYPE(class_name, snake_name) \
if constexpr (IsSame<PDF::class_name, T>) { \
return #class_name; \
}
ENUMERATE_OBJECT_TYPES(ENUMERATE_TYPE)
# undef ENUMERATE_TYPE
VERIFY_NOT_REACHED();
}
}
#endif
namespace PDF { namespace PDF {
class Object : public RefCounted<Object> { class Object : public RefCounted<Object> {
@ -23,40 +43,49 @@ public:
[[nodiscard]] ALWAYS_INLINE u32 generation_index() const { return m_generation_index; } [[nodiscard]] ALWAYS_INLINE u32 generation_index() const { return m_generation_index; }
ALWAYS_INLINE void set_generation_index(u32 generation_index) { m_generation_index = generation_index; } ALWAYS_INLINE void set_generation_index(u32 generation_index) { m_generation_index = generation_index; }
#define DEFINE_ID(_, name) \ template<IsObject T>
virtual bool is_##name() const { return false; } bool is() const requires(!IsSame<T, Object>)
ENUMERATE_OBJECT_TYPES(DEFINE_ID) {
#undef DEFINE_ID #define ENUMERATE_TYPE(class_name, snake_name) \
if constexpr (IsSame<class_name, T>) { \
return is_##snake_name(); \
}
ENUMERATE_OBJECT_TYPES(ENUMERATE_TYPE)
#undef ENUMERATE_TYPE
VERIFY_NOT_REACHED();
}
template<IsObject T>
[[nodiscard]] ALWAYS_INLINE NonnullRefPtr<T> cast(
#ifdef PDF_DEBUG
SourceLocation loc = SourceLocation::current()
#endif
) const requires(!IsSame<T, Object>)
{
#ifdef PDF_DEBUG
if (!is<T>()) {
dbgln("{} invalid cast from {} to {}", loc, type_name(), object_name<T>());
VERIFY_NOT_REACHED();
}
#endif
return NonnullRefPtr<T>(static_cast<T const&>(*this));
}
virtual const char* type_name() const = 0; virtual const char* type_name() const = 0;
virtual String to_string(int indent) const = 0; virtual String to_string(int indent) const = 0;
protected:
#define ENUMERATE_TYPE(_, name) \
virtual bool is_##name() const { return false; }
ENUMERATE_OBJECT_TYPES(ENUMERATE_TYPE)
#undef ENUMERATE_TYPE
private: private:
u32 m_generation_index { 0 }; u32 m_generation_index { 0 };
}; };
template<IsObject To, IsObject From>
[[nodiscard]] ALWAYS_INLINE static NonnullRefPtr<To> object_cast(NonnullRefPtr<From> obj
#ifdef PDF_DEBUG
,
SourceLocation loc = SourceLocation::current()
#endif
)
{
#ifdef PDF_DEBUG
# define ENUMERATE_TYPES(class_name, snake_name) \
if constexpr (IsSame<To, class_name>) { \
if (!obj->is_##snake_name()) { \
dbgln("{} invalid cast from type {} to type " #snake_name, loc, obj->type_name()); \
} \
}
ENUMERATE_OBJECT_TYPES(ENUMERATE_TYPES)
# undef ENUMERATE_TYPES
#endif
return static_ptr_cast<To>(obj);
}
} }
namespace AK { namespace AK {

15
Userland/Libraries/LibPDF/ObjectDerivatives.cpp
View file

@ -1,5 +1,5 @@
/* /*
* Copyright (c) 2021, Matthew Olsson <mattco@serenityos.org> * Copyright (c) 2021-2022, Matthew Olsson <mattco@serenityos.org>
* *
* SPDX-License-Identifier: BSD-2-Clause * SPDX-License-Identifier: BSD-2-Clause
*/ */
@ -10,23 +10,20 @@
namespace PDF { namespace PDF {
NonnullRefPtr<Object> ArrayObject::get_object_at(Document* document, size_t index) const PDFErrorOr<NonnullRefPtr<Object>> DictObject::get_object(Document* document, FlyString const& key) const
{
return document->resolve_to<Object>(m_elements[index]);
}
NonnullRefPtr<Object> DictObject::get_object(Document* document, FlyString const& key) const
{ {
return document->resolve_to<Object>(get_value(key)); return document->resolve_to<Object>(get_value(key));
} }
#define DEFINE_ACCESSORS(class_name, snake_name) \ #define DEFINE_ACCESSORS(class_name, snake_name) \
NonnullRefPtr<class_name> ArrayObject::get_##snake_name##_at(Document* document, size_t index) const \ PDFErrorOr<NonnullRefPtr<class_name>> ArrayObject::get_##snake_name##_at(Document* document, size_t index) const \
{ \ { \
if (index >= m_elements.size()) \
return Error { Error::Type::Internal, "Out of bounds array access" }; \
return document->resolve_to<class_name>(m_elements[index]); \ return document->resolve_to<class_name>(m_elements[index]); \
} \ } \
\ \
NonnullRefPtr<class_name> DictObject::get_##snake_name(Document* document, FlyString const& key) const \ PDFErrorOr<NonnullRefPtr<class_name>> DictObject::get_##snake_name(Document* document, FlyString const& key) const \
{ \ { \
return document->resolve_to<class_name>(get(key).value()); \ return document->resolve_to<class_name>(get(key).value()); \
} }

50
Userland/Libraries/LibPDF/ObjectDerivatives.h
View file

@ -1,5 +1,5 @@
/* /*
* Copyright (c) 2021, Matthew Olsson <mattco@serenityos.org> * Copyright (c) 2021-2022, Matthew Olsson <mattco@serenityos.org>
* Copyright (c) 2021, Ben Wiederhake <BenWiederhake.GitHub@gmx.de> * Copyright (c) 2021, Ben Wiederhake <BenWiederhake.GitHub@gmx.de>
* *
* SPDX-License-Identifier: BSD-2-Clause * SPDX-License-Identifier: BSD-2-Clause
@ -30,10 +30,12 @@ public:
[[nodiscard]] ALWAYS_INLINE String const& string() const { return m_string; } [[nodiscard]] ALWAYS_INLINE String const& string() const { return m_string; }
[[nodiscard]] ALWAYS_INLINE bool is_binary() const { return m_is_binary; } [[nodiscard]] ALWAYS_INLINE bool is_binary() const { return m_is_binary; }
ALWAYS_INLINE bool is_string() const override { return true; } const char* type_name() const override { return "string"; }
ALWAYS_INLINE const char* type_name() const override { return "string"; }
String to_string(int indent) const override; String to_string(int indent) const override;
protected:
bool is_string() const override { return true; }
private: private:
String m_string; String m_string;
bool m_is_binary; bool m_is_binary;
@ -50,10 +52,12 @@ public:
[[nodiscard]] ALWAYS_INLINE FlyString const& name() const { return m_name; } [[nodiscard]] ALWAYS_INLINE FlyString const& name() const { return m_name; }
ALWAYS_INLINE bool is_name() const override { return true; } const char* type_name() const override { return "name"; }
ALWAYS_INLINE const char* type_name() const override { return "name"; }
String to_string(int indent) const override; String to_string(int indent) const override;
protected:
bool is_name() const override { return true; }
private: private:
FlyString m_name; FlyString m_name;
}; };
@ -76,20 +80,20 @@ public:
ALWAYS_INLINE Value const& operator[](size_t index) const { return at(index); } ALWAYS_INLINE Value const& operator[](size_t index) const { return at(index); }
ALWAYS_INLINE Value const& at(size_t index) const { return m_elements[index]; } ALWAYS_INLINE Value const& at(size_t index) const { return m_elements[index]; }
NonnullRefPtr<Object> get_object_at(Document*, size_t index) const;
#define DEFINE_INDEXER(class_name, snake_name) \ #define DEFINE_INDEXER(class_name, snake_name) \
NonnullRefPtr<class_name> get_##snake_name##_at(Document*, size_t index) const; PDFErrorOr<NonnullRefPtr<class_name>> get_##snake_name##_at(Document*, size_t index) const;
ENUMERATE_OBJECT_TYPES(DEFINE_INDEXER) ENUMERATE_OBJECT_TYPES(DEFINE_INDEXER)
#undef DEFINE_INDEXER #undef DEFINE_INDEXER
ALWAYS_INLINE bool is_array() const override const char* type_name() const override
{ {
return true; return "array";
} }
ALWAYS_INLINE const char* type_name() const override { return "array"; }
String to_string(int indent) const override; String to_string(int indent) const override;
protected:
bool is_array() const override { return true; }
private: private:
Vector<Value> m_elements; Vector<Value> m_elements;
}; };
@ -117,20 +121,22 @@ public:
return value.value(); return value.value();
} }
NonnullRefPtr<Object> get_object(Document*, FlyString const& key) const; PDFErrorOr<NonnullRefPtr<Object>> get_object(Document*, FlyString const& key) const;
#define DEFINE_GETTER(class_name, snake_name) \ #define DEFINE_GETTER(class_name, snake_name) \
NonnullRefPtr<class_name> get_##snake_name(Document*, FlyString const& key) const; PDFErrorOr<NonnullRefPtr<class_name>> get_##snake_name(Document*, FlyString const& key) const;
ENUMERATE_OBJECT_TYPES(DEFINE_GETTER) ENUMERATE_OBJECT_TYPES(DEFINE_GETTER)
#undef DEFINE_GETTER #undef DEFINE_GETTER
ALWAYS_INLINE bool is_dict() const override const char* type_name() const override
{ {
return true; return "dict";
} }
ALWAYS_INLINE const char* type_name() const override { return "dict"; }
String to_string(int indent) const override; String to_string(int indent) const override;
protected:
bool is_dict() const override { return true; }
private: private:
HashMap<FlyString, Value> m_map; HashMap<FlyString, Value> m_map;
}; };
@ -147,10 +153,12 @@ public:
[[nodiscard]] ALWAYS_INLINE NonnullRefPtr<DictObject> dict() const { return m_dict; } [[nodiscard]] ALWAYS_INLINE NonnullRefPtr<DictObject> dict() const { return m_dict; }
[[nodiscard]] virtual ReadonlyBytes bytes() const = 0; [[nodiscard]] virtual ReadonlyBytes bytes() const = 0;
ALWAYS_INLINE bool is_stream() const override { return true; } const char* type_name() const override { return "stream"; }
ALWAYS_INLINE const char* type_name() const override { return "stream"; }
String to_string(int indent) const override; String to_string(int indent) const override;
protected:
bool is_stream() const override { return true; }
private: private:
NonnullRefPtr<DictObject> m_dict; NonnullRefPtr<DictObject> m_dict;
}; };
@ -201,10 +209,12 @@ public:
[[nodiscard]] ALWAYS_INLINE u32 index() const { return m_index; } [[nodiscard]] ALWAYS_INLINE u32 index() const { return m_index; }
[[nodiscard]] ALWAYS_INLINE Value const& value() const { return m_value; } [[nodiscard]] ALWAYS_INLINE Value const& value() const { return m_value; }
ALWAYS_INLINE bool is_indirect_value() const override { return true; } const char* type_name() const override { return "indirect_object"; }
ALWAYS_INLINE const char* type_name() const override { return "indirect_object"; }
String to_string(int indent) const override; String to_string(int indent) const override;
protected:
bool is_indirect_value() const override { return true; }
private: private:
u32 m_index; u32 m_index;
Value m_value; Value m_value;

374
Userland/Libraries/LibPDF/Parser.cpp
View file

@ -22,7 +22,7 @@ static NonnullRefPtr<T> make_object(Args... args) requires(IsBaseOf<Object, T>)
return adopt_ref(*new T(forward<Args>(args)...)); return adopt_ref(*new T(forward<Args>(args)...));
} }
Vector<Command> Parser::parse_graphics_commands(ReadonlyBytes bytes) PDFErrorOr<Vector<Command>> Parser::parse_graphics_commands(ReadonlyBytes bytes)
{ {
auto parser = adopt_ref(*new Parser(bytes)); auto parser = adopt_ref(*new Parser(bytes));
return parser->parse_graphics_commands(); return parser->parse_graphics_commands();
@ -43,16 +43,13 @@ void Parser::set_document(WeakPtr<Document> const& document)
m_document = document; m_document = document;
} }
bool Parser::initialize() PDFErrorOr<void> Parser::initialize()
{ {
if (!parse_header()) TRY(parse_header());
return {};
const auto result = initialize_linearization_dict(); const auto linearization_result = TRY(initialize_linearization_dict());
if (result == LinearizationResult::Error)
return {};
if (result == LinearizationResult::NotLinearized) if (linearization_result == LinearizationResult::NotLinearized)
return initialize_non_linearized_xref_table(); return initialize_non_linearized_xref_table();
bool is_linearized = m_linearization_dictionary.has_value(); bool is_linearized = m_linearization_dictionary.has_value();
@ -75,37 +72,40 @@ bool Parser::initialize()
return initialize_non_linearized_xref_table(); return initialize_non_linearized_xref_table();
} }
Value Parser::parse_object_with_index(u32 index) PDFErrorOr<Value> Parser::parse_object_with_index(u32 index)
{ {
VERIFY(m_xref_table->has_object(index)); VERIFY(m_xref_table->has_object(index));
auto byte_offset = m_xref_table->byte_offset_for_object(index); auto byte_offset = m_xref_table->byte_offset_for_object(index);
m_reader.move_to(byte_offset); m_reader.move_to(byte_offset);
auto indirect_value = parse_indirect_value(); auto indirect_value = TRY(parse_indirect_value());
VERIFY(indirect_value);
VERIFY(indirect_value->index() == index); VERIFY(indirect_value->index() == index);
return indirect_value->value(); return indirect_value->value();
} }
bool Parser::parse_header() PDFErrorOr<void> Parser::parse_header()
{ {
// FIXME: Do something with the version? // FIXME: Do something with the version?
m_reader.set_reading_forwards(); m_reader.set_reading_forwards();
if (m_reader.remaining() == 0) if (m_reader.remaining() == 0)
return false; return error("Empty PDF document");
m_reader.move_to(0); m_reader.move_to(0);
if (m_reader.remaining() < 8 || !m_reader.matches("%PDF-")) if (m_reader.remaining() < 8 || !m_reader.matches("%PDF-"))
return false; return error("Not a PDF document");
m_reader.move_by(5); m_reader.move_by(5);
char major_ver = m_reader.read(); char major_ver = m_reader.read();
if (major_ver != '1' && major_ver != '2') if (major_ver != '1' && major_ver != '2')
return false; return error(String::formatted("Unknown major version \"{}\"", major_ver));
if (m_reader.read() != '.') if (m_reader.read() != '.')
return false; return error("Malformed PDF version");
char minor_ver = m_reader.read(); char minor_ver = m_reader.read();
if (minor_ver < '0' || minor_ver > '7') if (minor_ver < '0' || minor_ver > '7')
return false; return error(String::formatted("Unknown minor version \"{}\"", minor_ver));
consume_eol(); consume_eol();
// Parse optional high-byte comment, which signifies a binary file // Parse optional high-byte comment, which signifies a binary file
@ -119,27 +119,28 @@ bool Parser::parse_header()
} }
} }
return true; return {};
} }
Parser::LinearizationResult Parser::initialize_linearization_dict() PDFErrorOr<Parser::LinearizationResult> Parser::initialize_linearization_dict()
{ {
// parse_header() is called immediately before this, so we are at the right location // parse_header() is called immediately before this, so we are at the right location
auto dict_value = m_document->resolve(parse_indirect_value()); auto indirect_value = Value(*TRY(parse_indirect_value()));
auto dict_value = TRY(m_document->resolve(indirect_value));
if (!dict_value.has<NonnullRefPtr<Object>>()) if (!dict_value.has<NonnullRefPtr<Object>>())
return LinearizationResult::Error; return error("Expected linearization object to be a dictionary");
auto dict_object = dict_value.get<NonnullRefPtr<Object>>(); auto dict_object = dict_value.get<NonnullRefPtr<Object>>();
if (!dict_object->is_dict()) if (!dict_object->is<DictObject>())
return LinearizationResult::NotLinearized; return LinearizationResult::NotLinearized;
auto dict = object_cast<DictObject>(dict_object); auto dict = dict_object->cast<DictObject>();
if (!dict->contains(CommonNames::Linearized)) if (!dict->contains(CommonNames::Linearized))
return LinearizationResult::NotLinearized; return LinearizationResult::NotLinearized;
if (!dict->contains(CommonNames::L, CommonNames::H, CommonNames::O, CommonNames::E, CommonNames::N, CommonNames::T)) if (!dict->contains(CommonNames::L, CommonNames::H, CommonNames::O, CommonNames::E, CommonNames::N, CommonNames::T))
return LinearizationResult::Error; return error("Malformed linearization dictionary");
auto length_of_file = dict->get_value(CommonNames::L); auto length_of_file = dict->get_value(CommonNames::L);
auto hint_table = dict->get_value(CommonNames::H); auto hint_table = dict->get_value(CommonNames::H);
@ -156,17 +157,13 @@ Parser::LinearizationResult Parser::initialize_linearization_dict()
|| !number_of_pages.has_u16() || !number_of_pages.has_u16()
|| !offset_of_main_xref_table.has_u32() || !offset_of_main_xref_table.has_u32()
|| (!first_page.has<Empty>() && !first_page.has_u32())) { || (!first_page.has<Empty>() && !first_page.has_u32())) {
return LinearizationResult::Error; return error("Malformed linearization dictionary parameters");
} }
auto hint_table_object = hint_table.get<NonnullRefPtr<Object>>(); auto hint_table_array = hint_table.get<NonnullRefPtr<Object>>()->cast<ArrayObject>();
if (!hint_table_object->is_array())
return LinearizationResult::Error;
auto hint_table_array = object_cast<ArrayObject>(hint_table_object);
auto hint_table_size = hint_table_array->size(); auto hint_table_size = hint_table_array->size();
if (hint_table_size != 2 && hint_table_size != 4) if (hint_table_size != 2 && hint_table_size != 4)
return LinearizationResult::Error; return error("Expected hint table to be of length 2 or 4");
auto primary_hint_stream_offset = hint_table_array->at(0); auto primary_hint_stream_offset = hint_table_array->at(0);
auto primary_hint_stream_length = hint_table_array->at(1); auto primary_hint_stream_length = hint_table_array->at(1);
@ -182,7 +179,7 @@ Parser::LinearizationResult Parser::initialize_linearization_dict()
|| !primary_hint_stream_length.has_u32() || !primary_hint_stream_length.has_u32()
|| (!overflow_hint_stream_offset.has<Empty>() && !overflow_hint_stream_offset.has_u32()) || (!overflow_hint_stream_offset.has<Empty>() && !overflow_hint_stream_offset.has_u32())
|| (!overflow_hint_stream_length.has<Empty>() && !overflow_hint_stream_length.has_u32())) { || (!overflow_hint_stream_length.has<Empty>() && !overflow_hint_stream_length.has_u32())) {
return LinearizationResult::Error; return error("Malformed hint stream");
} }
m_linearization_dictionary = LinearizationDictionary { m_linearization_dictionary = LinearizationDictionary {
@ -201,20 +198,12 @@ Parser::LinearizationResult Parser::initialize_linearization_dict()
return LinearizationResult::Linearized; return LinearizationResult::Linearized;
} }
bool Parser::initialize_linearized_xref_table() PDFErrorOr<void> Parser::initialize_linearized_xref_table()
{ {
// The linearization parameter dictionary has just been parsed, and the xref table // The linearization parameter dictionary has just been parsed, and the xref table
// comes immediately after it. We are in the correct spot. // comes immediately after it. We are in the correct spot.
if (!m_reader.matches("xref")) m_xref_table = TRY(parse_xref_table());
return false; m_trailer = TRY(parse_file_trailer());
m_xref_table = parse_xref_table();
if (!m_xref_table)
return false;
m_trailer = parse_file_trailer();
if (!m_trailer)
return false;
// Also parse the main xref table and merge into the first-page xref table. Note // Also parse the main xref table and merge into the first-page xref table. Note
// that we don't use the main xref table offset from the linearization dict because // that we don't use the main xref table offset from the linearization dict because
@ -222,14 +211,12 @@ bool Parser::initialize_linearized_xref_table()
// index start and length? So it's much easier to do it this way. // index start and length? So it's much easier to do it this way.
auto main_xref_table_offset = m_trailer->get_value(CommonNames::Prev).to_int(); auto main_xref_table_offset = m_trailer->get_value(CommonNames::Prev).to_int();
m_reader.move_to(main_xref_table_offset); m_reader.move_to(main_xref_table_offset);
auto main_xref_table = parse_xref_table(); auto main_xref_table = TRY(parse_xref_table());
if (!main_xref_table) TRY(m_xref_table->merge(move(*main_xref_table)));
return false; return {};
return m_xref_table->merge(move(*main_xref_table));
} }
bool Parser::initialize_hint_tables() PDFErrorOr<void> Parser::initialize_hint_tables()
{ {
auto linearization_dict = m_linearization_dictionary.value(); auto linearization_dict = m_linearization_dictionary.value();
auto primary_offset = linearization_dict.primary_hint_stream_offset; auto primary_offset = linearization_dict.primary_hint_stream_offset;
@ -238,23 +225,23 @@ bool Parser::initialize_hint_tables()
auto parse_hint_table = [&](size_t offset) -> RefPtr<StreamObject> { auto parse_hint_table = [&](size_t offset) -> RefPtr<StreamObject> {
m_reader.move_to(offset); m_reader.move_to(offset);
auto stream_indirect_value = parse_indirect_value(); auto stream_indirect_value = parse_indirect_value();
if (!stream_indirect_value) if (stream_indirect_value.is_error())
return {}; return {};
auto stream_value = stream_indirect_value->value(); auto stream_value = stream_indirect_value.value()->value();
if (!stream_value.has<NonnullRefPtr<Object>>()) if (!stream_value.has<NonnullRefPtr<Object>>())
return {}; return {};
auto stream_object = stream_value.get<NonnullRefPtr<Object>>(); auto stream_object = stream_value.get<NonnullRefPtr<Object>>();
if (!stream_object->is_stream()) if (!stream_object->is<StreamObject>())
return {}; return {};
return object_cast<StreamObject>(stream_object); return stream_object->cast<StreamObject>();
}; };
auto primary_hint_stream = parse_hint_table(primary_offset); auto primary_hint_stream = parse_hint_table(primary_offset);
if (!primary_hint_stream) if (!primary_hint_stream)
return false; return error("Invalid primary hint stream");
RefPtr<StreamObject> overflow_hint_stream; RefPtr<StreamObject> overflow_hint_stream;
if (overflow_offset != NumericLimits<u32>::max()) if (overflow_offset != NumericLimits<u32>::max())
@ -270,66 +257,49 @@ bool Parser::initialize_hint_tables()
auto buffer_result = ByteBuffer::create_uninitialized(total_size); auto buffer_result = ByteBuffer::create_uninitialized(total_size);
if (buffer_result.is_error()) if (buffer_result.is_error())
return false; return Error { Error::Type::Internal, "Failed to allocate hint stream buffer" };
possible_merged_stream_buffer = buffer_result.release_value(); possible_merged_stream_buffer = buffer_result.release_value();
auto ok = !possible_merged_stream_buffer.try_append(primary_hint_stream->bytes()).is_error(); MUST(possible_merged_stream_buffer.try_append(primary_hint_stream->bytes()));
ok = ok && !possible_merged_stream_buffer.try_append(overflow_hint_stream->bytes()).is_error(); MUST(possible_merged_stream_buffer.try_append(overflow_hint_stream->bytes()));
if (!ok)
return false;
hint_stream_bytes = possible_merged_stream_buffer.bytes(); hint_stream_bytes = possible_merged_stream_buffer.bytes();
} else { } else {
hint_stream_bytes = primary_hint_stream->bytes(); hint_stream_bytes = primary_hint_stream->bytes();
} }
auto hint_table = parse_page_offset_hint_table(hint_stream_bytes); auto hint_table = TRY(parse_page_offset_hint_table(hint_stream_bytes));
if (!hint_table.has_value()) auto hint_table_entries = parse_all_page_offset_hint_table_entries(hint_table, hint_stream_bytes);
return false;
dbgln("hint table: {}", hint_table.value()); // FIXME: Do something with the hint tables
return {};
auto hint_table_entries = parse_all_page_offset_hint_table_entries(hint_table.value(), hint_stream_bytes);
if (!hint_table_entries.has_value())
return false;
auto entries = hint_table_entries.value();
dbgln("hint table entries size: {}", entries.size());
for (auto& entry : entries)
dbgln("{}", entry);
return true;
} }
bool Parser::initialize_non_linearized_xref_table() PDFErrorOr<void> Parser::initialize_non_linearized_xref_table()
{ {
m_reader.move_to(m_reader.bytes().size() - 1); m_reader.move_to(m_reader.bytes().size() - 1);
if (!navigate_to_before_eof_marker()) if (!navigate_to_before_eof_marker())
return false; return error("No EOF marker");
if (!navigate_to_after_startxref()) if (!navigate_to_after_startxref())
return false; return error("No xref");
if (m_reader.done())
return false;
m_reader.set_reading_forwards(); m_reader.set_reading_forwards();
auto xref_offset_value = parse_number(); auto xref_offset_value = parse_number();
if (!xref_offset_value.has<int>()) if (xref_offset_value.is_error() || !xref_offset_value.value().has<int>())
return false; return error("Invalid xref offset");
auto xref_offset = xref_offset_value.get<int>(); auto xref_offset = xref_offset_value.value().get<int>();
m_reader.move_to(xref_offset); m_reader.move_to(xref_offset);
m_xref_table = parse_xref_table(); m_xref_table = TRY(parse_xref_table());
if (!m_xref_table) m_trailer = TRY(parse_file_trailer());
return false; return {};
m_trailer = parse_file_trailer();
return m_trailer;
} }
RefPtr<XRefTable> Parser::parse_xref_table() PDFErrorOr<NonnullRefPtr<XRefTable>> Parser::parse_xref_table()
{ {
if (!m_reader.matches("xref")) if (!m_reader.matches("xref"))
return {}; return error("Expected \"xref\"");
m_reader.move_by(4); m_reader.move_by(4);
if (!consume_eol()) if (!consume_eol())
return {}; return error("Expected newline after \"xref\"");
auto table = adopt_ref(*new XRefTable()); auto table = adopt_ref(*new XRefTable());
@ -339,25 +309,25 @@ RefPtr<XRefTable> Parser::parse_xref_table()
Vector<XRefEntry> entries; Vector<XRefEntry> entries;
auto starting_index_value = parse_number(); auto starting_index_value = TRY(parse_number());
auto starting_index = starting_index_value.get<int>(); auto starting_index = starting_index_value.get<int>();
auto object_count_value = parse_number(); auto object_count_value = TRY(parse_number());
auto object_count = object_count_value.get<int>(); auto object_count = object_count_value.get<int>();
for (int i = 0; i < object_count; i++) { for (int i = 0; i < object_count; i++) {
auto offset_string = String(m_reader.bytes().slice(m_reader.offset(), 10)); auto offset_string = String(m_reader.bytes().slice(m_reader.offset(), 10));
m_reader.move_by(10); m_reader.move_by(10);
if (!consume(' ')) if (!consume(' '))
return {}; return error("Malformed xref entry");
auto generation_string = String(m_reader.bytes().slice(m_reader.offset(), 5)); auto generation_string = String(m_reader.bytes().slice(m_reader.offset(), 5));
m_reader.move_by(5); m_reader.move_by(5);
if (!consume(' ')) if (!consume(' '))
return {}; return error("Malformed xref entry");
auto letter = m_reader.read(); auto letter = m_reader.read();
if (letter != 'n' && letter != 'f') if (letter != 'n' && letter != 'f')
return {}; return error("Malformed xref entry");
// The line ending sequence can be one of the following: // The line ending sequence can be one of the following:
// SP CR, SP LF, or CR LF // SP CR, SP LF, or CR LF
@ -365,10 +335,10 @@ RefPtr<XRefTable> Parser::parse_xref_table()
consume(); consume();
auto ch = consume(); auto ch = consume();
if (ch != '\r' && ch != '\n') if (ch != '\r' && ch != '\n')
return {}; return error("Malformed xref entry");
} else { } else {
if (!m_reader.matches("\r\n")) if (!m_reader.matches("\r\n"))
return {}; return error("Malformed xref entry");
m_reader.move_by(2); m_reader.move_by(2);
} }
@ -382,35 +352,33 @@ RefPtr<XRefTable> Parser::parse_xref_table()
} }
} }
RefPtr<DictObject> Parser::parse_file_trailer() PDFErrorOr<NonnullRefPtr<DictObject>> Parser::parse_file_trailer()
{ {
if (!m_reader.matches("trailer")) if (!m_reader.matches("trailer"))
return {}; return error("Expected \"trailer\" keyword");
m_reader.move_by(7); m_reader.move_by(7);
consume_whitespace(); consume_whitespace();
auto dict = parse_dict(); auto dict = TRY(parse_dict());
if (!dict)
return {};
if (!m_reader.matches("startxref")) if (!m_reader.matches("startxref"))
return {}; return error("Expected \"startxref\"");
m_reader.move_by(9); m_reader.move_by(9);
consume_whitespace(); consume_whitespace();
m_reader.move_until([&](auto) { return matches_eol(); }); m_reader.move_until([&](auto) { return matches_eol(); });
VERIFY(consume_eol()); VERIFY(consume_eol());
if (!m_reader.matches("%%EOF")) if (!m_reader.matches("%%EOF"))
return {}; return error("Expected \"%%EOF\"");
m_reader.move_by(5); m_reader.move_by(5);
consume_whitespace(); consume_whitespace();
return dict; return dict;
} }
Optional<Parser::PageOffsetHintTable> Parser::parse_page_offset_hint_table(ReadonlyBytes hint_stream_bytes) PDFErrorOr<Parser::PageOffsetHintTable> Parser::parse_page_offset_hint_table(ReadonlyBytes hint_stream_bytes)
{ {
if (hint_stream_bytes.size() < sizeof(PageOffsetHintTable)) if (hint_stream_bytes.size() < sizeof(PageOffsetHintTable))
return {}; return error("Hint stream is too small");
size_t offset = 0; size_t offset = 0;
@ -455,12 +423,10 @@ Optional<Parser::PageOffsetHintTable> Parser::parse_page_offset_hint_table(Reado
return hint_table; return hint_table;
} }
Optional<Vector<Parser::PageOffsetHintTableEntry>> Parser::parse_all_page_offset_hint_table_entries(PageOffsetHintTable const& hint_table, ReadonlyBytes hint_stream_bytes) Vector<Parser::PageOffsetHintTableEntry> Parser::parse_all_page_offset_hint_table_entries(PageOffsetHintTable const& hint_table, ReadonlyBytes hint_stream_bytes)
{ {
InputMemoryStream input_stream(hint_stream_bytes); InputMemoryStream input_stream(hint_stream_bytes);
input_stream.seek(sizeof(PageOffsetHintTable)); input_stream.seek(sizeof(PageOffsetHintTable));
if (input_stream.has_any_error())
return {};
InputBitStream bit_stream(input_stream); InputBitStream bit_stream(input_stream);
@ -576,7 +542,7 @@ String Parser::parse_comment()
return str; return str;
} }
Value Parser::parse_value() PDFErrorOr<Value> Parser::parse_value()
{ {
parse_comment(); parse_comment();
@ -602,14 +568,12 @@ Value Parser::parse_value()
return parse_possible_indirect_value_or_ref(); return parse_possible_indirect_value_or_ref();
if (m_reader.matches('/')) if (m_reader.matches('/'))
return parse_name(); return MUST(parse_name());
if (m_reader.matches("<<")) { if (m_reader.matches("<<")) {
auto dict = parse_dict(); auto dict = TRY(parse_dict());
if (!dict)
return {};
if (m_reader.matches("stream")) if (m_reader.matches("stream"))
return parse_stream(dict.release_nonnull()); return TRY(parse_stream(dict));
return dict; return dict;
} }
@ -617,21 +581,20 @@ Value Parser::parse_value()
return parse_string(); return parse_string();
if (m_reader.matches('[')) if (m_reader.matches('['))
return parse_array(); return TRY(parse_array());
dbgln("tried to parse value, but found char {} ({}) at offset {}", m_reader.peek(), static_cast<u8>(m_reader.peek()), m_reader.offset()); return error(String::formatted("Unexpected char \"{}\"", m_reader.peek()));
VERIFY_NOT_REACHED();
} }
Value Parser::parse_possible_indirect_value_or_ref() PDFErrorOr<Value> Parser::parse_possible_indirect_value_or_ref()
{ {
auto first_number = parse_number(); auto first_number = TRY(parse_number());
if (!first_number.has<int>() || !matches_number()) if (!matches_number())
return first_number; return first_number;
m_reader.save(); m_reader.save();
auto second_number = parse_number(); auto second_number = parse_number();
if (!second_number.has<int>()) { if (second_number.is_error()) {
m_reader.load(); m_reader.load();
return first_number; return first_number;
} }
@ -640,28 +603,28 @@ Value Parser::parse_possible_indirect_value_or_ref()
m_reader.discard(); m_reader.discard();
consume(); consume();
consume_whitespace(); consume_whitespace();
return Value(Reference(first_number.get<int>(), second_number.get<int>())); return Value(Reference(first_number.get<int>(), second_number.value().get<int>()));
} }
if (m_reader.matches("obj")) { if (m_reader.matches("obj")) {
m_reader.discard(); m_reader.discard();
return parse_indirect_value(first_number.get<int>(), second_number.get<int>()); return TRY(parse_indirect_value(first_number.get<int>(), second_number.value().get<int>()));
} }
m_reader.load(); m_reader.load();
return first_number; return first_number;
} }
RefPtr<IndirectValue> Parser::parse_indirect_value(int index, int generation) PDFErrorOr<NonnullRefPtr<IndirectValue>> Parser::parse_indirect_value(int index, int generation)
{ {
if (!m_reader.matches("obj")) if (!m_reader.matches("obj"))
return {}; return error("Expected \"obj\" at beginning of indirect value");
m_reader.move_by(3); m_reader.move_by(3);
if (matches_eol()) if (matches_eol())
consume_eol(); consume_eol();
auto value = parse_value(); auto value = TRY(parse_value());
if (!m_reader.matches("endobj")) if (!m_reader.matches("endobj"))
return {}; return error("Expected \"endobj\" at end of indirect value");
consume(6); consume(6);
consume_whitespace(); consume_whitespace();
@ -669,21 +632,18 @@ RefPtr<IndirectValue> Parser::parse_indirect_value(int index, int generation)
return make_object<IndirectValue>(index, generation, value); return make_object<IndirectValue>(index, generation, value);
} }
RefPtr<IndirectValue> Parser::parse_indirect_value() PDFErrorOr<NonnullRefPtr<IndirectValue>> Parser::parse_indirect_value()
{ {
auto first_number = parse_number(); auto first_number = TRY(parse_number());
if (!first_number.has<int>()) auto second_number = TRY(parse_number());
return {};
auto second_number = parse_number();
if (!second_number.has<int>())
return {};
return parse_indirect_value(first_number.get<int>(), second_number.get<int>()); return parse_indirect_value(first_number.get<int>(), second_number.get<int>());
} }
Value Parser::parse_number() PDFErrorOr<Value> Parser::parse_number()
{ {
size_t start_offset = m_reader.offset(); size_t start_offset = m_reader.offset();
bool is_float = false; bool is_float = false;
bool consumed_digit = false;
if (m_reader.matches('+') || m_reader.matches('-')) if (m_reader.matches('+') || m_reader.matches('-'))
consume(); consume();
@ -696,11 +656,15 @@ Value Parser::parse_number()
consume(); consume();
} else if (isdigit(m_reader.peek())) { } else if (isdigit(m_reader.peek())) {
consume(); consume();
consumed_digit = true;
} else { } else {
break; break;
} }
} }
if (!consumed_digit)
return error("Invalid number");
consume_whitespace(); consume_whitespace();
auto string = String(m_reader.bytes().slice(start_offset, m_reader.offset() - start_offset)); auto string = String(m_reader.bytes().slice(start_offset, m_reader.offset() - start_offset));
@ -712,10 +676,11 @@ Value Parser::parse_number()
return Value(static_cast<int>(f)); return Value(static_cast<int>(f));
} }
RefPtr<NameObject> Parser::parse_name() PDFErrorOr<NonnullRefPtr<NameObject>> Parser::parse_name()
{ {
if (!consume('/')) if (!consume('/'))
return {}; return error("Expected Name object to start with \"/\"");
StringBuilder builder; StringBuilder builder;
while (true) { while (true) {
@ -726,8 +691,7 @@ RefPtr<NameObject> Parser::parse_name()
int hex_value = 0; int hex_value = 0;
for (int i = 0; i < 2; i++) { for (int i = 0; i < 2; i++) {
auto ch = consume(); auto ch = consume();
if (!isxdigit(ch)) VERIFY(isxdigit(ch));
return {};
hex_value *= 16; hex_value *= 16;
if (ch <= '9') { if (ch <= '9') {
hex_value += ch - '0'; hex_value += ch - '0';
@ -747,7 +711,7 @@ RefPtr<NameObject> Parser::parse_name()
return make_object<NameObject>(builder.to_string()); return make_object<NameObject>(builder.to_string());
} }
RefPtr<StringObject> Parser::parse_string() NonnullRefPtr<StringObject> Parser::parse_string()
{ {
ScopeGuard guard([&] { consume_whitespace(); }); ScopeGuard guard([&] { consume_whitespace(); });
@ -762,8 +726,7 @@ RefPtr<StringObject> Parser::parse_string()
is_binary_string = true; is_binary_string = true;
} }
if (string.is_null()) VERIFY(!string.is_null());
return {};
if (string.bytes().starts_with(Array<u8, 2> { 0xfe, 0xff })) { if (string.bytes().starts_with(Array<u8, 2> { 0xfe, 0xff })) {
// The string is encoded in UTF16-BE // The string is encoded in UTF16-BE
@ -779,8 +742,7 @@ RefPtr<StringObject> Parser::parse_string()
String Parser::parse_literal_string() String Parser::parse_literal_string()
{ {
if (!consume('(')) VERIFY(consume('('));
return {};
StringBuilder builder; StringBuilder builder;
auto opened_parens = 0; auto opened_parens = 0;
@ -853,16 +815,13 @@ String Parser::parse_literal_string()
} }
} }
if (opened_parens != 0)
return {};
return builder.to_string(); return builder.to_string();
} }
String Parser::parse_hex_string() String Parser::parse_hex_string()
{ {
if (!consume('<')) VERIFY(consume('<'));
return {};
StringBuilder builder; StringBuilder builder;
while (true) { while (true) {
@ -883,8 +842,7 @@ String Parser::parse_hex_string()
return builder.to_string(); return builder.to_string();
} }
if (!isxdigit(ch)) VERIFY(isxdigit(ch));
return {};
hex_value *= 16; hex_value *= 16;
if (ch <= '9') { if (ch <= '9') {
@ -899,122 +857,101 @@ String Parser::parse_hex_string()
} }
} }
RefPtr<ArrayObject> Parser::parse_array() PDFErrorOr<NonnullRefPtr<ArrayObject>> Parser::parse_array()
{ {
if (!consume('[')) if (!consume('['))
return {}; return error("Expected array to start with \"[\"");
consume_whitespace(); consume_whitespace();
Vector<Value> values; Vector<Value> values;
while (!m_reader.matches(']')) { while (!m_reader.matches(']'))
auto value = parse_value(); values.append(TRY(parse_value()));
if (value.has<Empty>())
return {};
values.append(value);
}
if (!consume(']')) VERIFY(consume(']'));
return {};
consume_whitespace(); consume_whitespace();
return make_object<ArrayObject>(values); return make_object<ArrayObject>(values);
} }
RefPtr<DictObject> Parser::parse_dict() PDFErrorOr<NonnullRefPtr<DictObject>> Parser::parse_dict()
{ {
if (!consume('<') || !consume('<')) if (!consume('<') || !consume('<'))
return {}; return error("Expected dict to start with \"<<\"");
consume_whitespace(); consume_whitespace();
HashMap<FlyString, Value> map; HashMap<FlyString, Value> map;
while (true) { while (!m_reader.done()) {
if (m_reader.matches(">>")) if (m_reader.matches(">>"))
break; break;
auto name = parse_name(); auto name = TRY(parse_name())->name();
if (!name) auto value = TRY(parse_value());
return {}; map.set(name, value);
auto value = parse_value();
if (value.has<Empty>())
return {};
map.set(name->name(), value);
} }
if (!consume('>') || !consume('>')) if (!consume('>') || !consume('>'))
return {}; return error("Expected dict to end with \">>\"");
consume_whitespace(); consume_whitespace();
return make_object<DictObject>(map); return make_object<DictObject>(map);
} }
RefPtr<DictObject> Parser::conditionally_parse_page_tree_node(u32 object_index, bool& ok) PDFErrorOr<RefPtr<DictObject>> Parser::conditionally_parse_page_tree_node(u32 object_index)
{ {
ok = true;
VERIFY(m_xref_table->has_object(object_index)); VERIFY(m_xref_table->has_object(object_index));
auto byte_offset = m_xref_table->byte_offset_for_object(object_index); auto byte_offset = m_xref_table->byte_offset_for_object(object_index);
m_reader.move_to(byte_offset); m_reader.move_to(byte_offset);
parse_number(); TRY(parse_number());
parse_number(); TRY(parse_number());
if (!m_reader.matches("obj")) { if (!m_reader.matches("obj"))
ok = false; return error(String::formatted("Invalid page tree offset {}", object_index));
return {};
}
m_reader.move_by(3); m_reader.move_by(3);
consume_whitespace(); consume_whitespace();
if (!consume('<') || !consume('<')) VERIFY(consume('<') && consume('<'));
return {};
consume_whitespace(); consume_whitespace();
HashMap<FlyString, Value> map; HashMap<FlyString, Value> map;
while (true) { while (true) {
if (m_reader.matches(">>")) if (m_reader.matches(">>"))
break; break;
auto name = parse_name(); auto name = TRY(parse_name());
if (!name) {
ok = false;
return {};
}
auto name_string = name->name(); auto name_string = name->name();
if (!name_string.is_one_of(CommonNames::Type, CommonNames::Parent, CommonNames::Kids, CommonNames::Count)) { if (!name_string.is_one_of(CommonNames::Type, CommonNames::Parent, CommonNames::Kids, CommonNames::Count)) {
// This is a page, not a page tree node // This is a page, not a page tree node
return {}; return RefPtr<DictObject> {};
}
auto value = parse_value();
if (value.has<Empty>()) {
ok = false;
return {};
} }
auto value = TRY(parse_value());
if (name_string == CommonNames::Type) { if (name_string == CommonNames::Type) {
if (!value.has<NonnullRefPtr<Object>>()) if (!value.has<NonnullRefPtr<Object>>())
return {}; return RefPtr<DictObject> {};
auto type_object = value.get<NonnullRefPtr<Object>>(); auto type_object = value.get<NonnullRefPtr<Object>>();
if (!type_object->is_name()) if (!type_object->is<NameObject>())
return {}; return RefPtr<DictObject> {};
auto type_name = object_cast<NameObject>(type_object); auto type_name = type_object->cast<NameObject>();
if (type_name->name() != CommonNames::Pages) if (type_name->name() != CommonNames::Pages)
return {}; return RefPtr<DictObject> {};
} }
map.set(name->name(), value); map.set(name->name(), value);
} }
if (!consume('>') || !consume('>')) VERIFY(consume('>') && consume('>'));
return {};
consume_whitespace(); consume_whitespace();
return make_object<DictObject>(map); return make_object<DictObject>(map);
} }
RefPtr<StreamObject> Parser::parse_stream(NonnullRefPtr<DictObject> dict) PDFErrorOr<NonnullRefPtr<StreamObject>> Parser::parse_stream(NonnullRefPtr<DictObject> dict)
{ {
if (!m_reader.matches("stream")) if (!m_reader.matches("stream"))
return {}; return error("Expected stream to start with \"stream\"");
m_reader.move_by(6); m_reader.move_by(6);
if (!consume_eol()) if (!consume_eol())
return {}; return error("Expected \"stream\" to be followed by a newline");
ReadonlyBytes bytes; ReadonlyBytes bytes;
@ -1022,7 +959,7 @@ RefPtr<StreamObject> Parser::parse_stream(NonnullRefPtr<DictObject> dict)
if (maybe_length.has_value() && (!maybe_length->has<Reference>() || m_xref_table)) { if (maybe_length.has_value() && (!maybe_length->has<Reference>() || m_xref_table)) {
// The PDF writer has kindly provided us with the direct length of the stream // The PDF writer has kindly provided us with the direct length of the stream
m_reader.save(); m_reader.save();
auto length = m_document->resolve_to<int>(maybe_length.value()); auto length = TRY(m_document->resolve_to<int>(maybe_length.value()));
m_reader.load(); m_reader.load();
bytes = m_reader.bytes().slice(m_reader.offset(), length); bytes = m_reader.bytes().slice(m_reader.offset(), length);
m_reader.move_by(length); m_reader.move_by(length);
@ -1047,11 +984,11 @@ RefPtr<StreamObject> Parser::parse_stream(NonnullRefPtr<DictObject> dict)
consume_whitespace(); consume_whitespace();
if (dict->contains(CommonNames::Filter)) { if (dict->contains(CommonNames::Filter)) {
auto filter_type = dict->get_name(m_document, CommonNames::Filter)->name(); auto filter_type = MUST(dict->get_name(m_document, CommonNames::Filter))->name();
auto maybe_bytes = Filter::decode(bytes, filter_type); auto maybe_bytes = Filter::decode(bytes, filter_type);
if (maybe_bytes.is_error()) { if (maybe_bytes.is_error()) {
warnln("Failed to decode filter: {}", maybe_bytes.error().string_literal()); warnln("Failed to decode filter: {}", maybe_bytes.error().string_literal());
return {}; return error(String::formatted("Failed to decode filter {}", maybe_bytes.error().string_literal()));
} }
return make_object<EncodedStreamObject>(dict, move(maybe_bytes.value())); return make_object<EncodedStreamObject>(dict, move(maybe_bytes.value()));
} }
@ -1059,7 +996,7 @@ RefPtr<StreamObject> Parser::parse_stream(NonnullRefPtr<DictObject> dict)
return make_object<PlainTextStreamObject>(dict, bytes); return make_object<PlainTextStreamObject>(dict, bytes);
} }
Vector<Command> Parser::parse_graphics_commands() PDFErrorOr<Vector<Command>> Parser::parse_graphics_commands()
{ {
Vector<Command> commands; Vector<Command> commands;
Vector<Value> command_args; Vector<Value> command_args;
@ -1088,7 +1025,7 @@ Vector<Command> Parser::parse_graphics_commands()
continue; continue;
} }
command_args.append(parse_value()); command_args.append(TRY(parse_value()));
} }
return commands; return commands;
@ -1161,6 +1098,21 @@ bool Parser::consume(char ch)
return consume() == ch; return consume() == ch;
} }
Error Parser::error(
String const& message
#ifdef PDF_DEBUG
,
SourceLocation loc
#endif
) const
{
#ifdef PDF_DEBUG
dbgln("\033[31m{} Parser error at offset {}: {}\033[0m", loc, m_reader.offset(), message);
#endif
return Error { Error::Type::Parse, message };
}
} }
namespace AK { namespace AK {

62
Userland/Libraries/LibPDF/Parser.h
View file

@ -7,6 +7,7 @@
#pragma once #pragma once
#include <AK/NonnullRefPtrVector.h> #include <AK/NonnullRefPtrVector.h>
#include <AK/SourceLocation.h>
#include <AK/WeakPtr.h> #include <AK/WeakPtr.h>
#include <LibPDF/Command.h> #include <LibPDF/Command.h>
#include <LibPDF/Object.h> #include <LibPDF/Object.h>
@ -20,12 +21,11 @@ class Document;
class Parser final : public RefCounted<Parser> { class Parser final : public RefCounted<Parser> {
public: public:
enum class LinearizationResult { enum class LinearizationResult {
Error,
NotLinearized, NotLinearized,
Linearized, Linearized,
}; };
static Vector<Command> parse_graphics_commands(ReadonlyBytes); static PDFErrorOr<Vector<Command>> parse_graphics_commands(ReadonlyBytes);
Parser(Badge<Document>, ReadonlyBytes); Parser(Badge<Document>, ReadonlyBytes);
@ -33,15 +33,13 @@ public:
void set_document(WeakPtr<Document> const&); void set_document(WeakPtr<Document> const&);
// Parses the header and initializes the xref table and trailer // Parses the header and initializes the xref table and trailer
bool initialize(); PDFErrorOr<void> initialize();
Value parse_object_with_index(u32 index); PDFErrorOr<Value> parse_object_with_index(u32 index);
// Specialized version of parse_dict which aborts early if the dict being parsed // Specialized version of parse_dict which aborts early if the dict being parsed
// is not a page object. A null RefPtr return indicates that the dict at this index // is not a page object
// is not a page tree node, whereas ok == false indicates a malformed PDF file and PDFErrorOr<RefPtr<DictObject>> conditionally_parse_page_tree_node(u32 object_index);
// should cause an abort of the current operation.
RefPtr<DictObject> conditionally_parse_page_tree_node(u32 object_index, bool& ok);
private: private:
struct LinearizationDictionary { struct LinearizationDictionary {
@ -89,35 +87,35 @@ private:
explicit Parser(ReadonlyBytes); explicit Parser(ReadonlyBytes);
bool parse_header(); PDFErrorOr<void> parse_header();
LinearizationResult initialize_linearization_dict(); PDFErrorOr<LinearizationResult> initialize_linearization_dict();
bool initialize_linearized_xref_table(); PDFErrorOr<void> initialize_linearized_xref_table();
bool initialize_non_linearized_xref_table(); PDFErrorOr<void> initialize_non_linearized_xref_table();
bool initialize_hint_tables(); PDFErrorOr<void> initialize_hint_tables();
Optional<PageOffsetHintTable> parse_page_offset_hint_table(ReadonlyBytes hint_stream_bytes); PDFErrorOr<PageOffsetHintTable> parse_page_offset_hint_table(ReadonlyBytes hint_stream_bytes);
Optional<Vector<PageOffsetHintTableEntry>> parse_all_page_offset_hint_table_entries(PageOffsetHintTable const&, ReadonlyBytes hint_stream_bytes); Vector<PageOffsetHintTableEntry> parse_all_page_offset_hint_table_entries(PageOffsetHintTable const&, ReadonlyBytes hint_stream_bytes);
RefPtr<XRefTable> parse_xref_table(); PDFErrorOr<NonnullRefPtr<XRefTable>> parse_xref_table();
RefPtr<DictObject> parse_file_trailer(); PDFErrorOr<NonnullRefPtr<DictObject>> parse_file_trailer();
bool navigate_to_before_eof_marker(); bool navigate_to_before_eof_marker();
bool navigate_to_after_startxref(); bool navigate_to_after_startxref();
String parse_comment(); String parse_comment();
Value parse_value(); PDFErrorOr<Value> parse_value();
Value parse_possible_indirect_value_or_ref(); PDFErrorOr<Value> parse_possible_indirect_value_or_ref();
RefPtr<IndirectValue> parse_indirect_value(int index, int generation); PDFErrorOr<NonnullRefPtr<IndirectValue>> parse_indirect_value(int index, int generation);
RefPtr<IndirectValue> parse_indirect_value(); PDFErrorOr<NonnullRefPtr<IndirectValue>> parse_indirect_value();
Value parse_number(); PDFErrorOr<Value> parse_number();
RefPtr<NameObject> parse_name(); PDFErrorOr<NonnullRefPtr<NameObject>> parse_name();
RefPtr<StringObject> parse_string(); NonnullRefPtr<StringObject> parse_string();
String parse_literal_string(); String parse_literal_string();
String parse_hex_string(); String parse_hex_string();
RefPtr<ArrayObject> parse_array(); PDFErrorOr<NonnullRefPtr<ArrayObject>> parse_array();
RefPtr<DictObject> parse_dict(); PDFErrorOr<NonnullRefPtr<DictObject>> parse_dict();
RefPtr<StreamObject> parse_stream(NonnullRefPtr<DictObject> dict); PDFErrorOr<NonnullRefPtr<StreamObject>> parse_stream(NonnullRefPtr<DictObject> dict);
Vector<Command> parse_graphics_commands(); PDFErrorOr<Vector<Command>> parse_graphics_commands();
bool matches_eol() const; bool matches_eol() const;
bool matches_whitespace() const; bool matches_whitespace() const;
@ -131,6 +129,14 @@ private:
void consume(int amount); void consume(int amount);
bool consume(char); bool consume(char);
Error error(
String const& message
#ifdef PDF_DEBUG
,
SourceLocation loc = SourceLocation::current()
#endif
) const;
Reader m_reader; Reader m_reader;
WeakPtr<Document> m_document; WeakPtr<Document> m_document;
RefPtr<XRefTable> m_xref_table; RefPtr<XRefTable> m_xref_table;

47
Userland/Libraries/LibPDF/Renderer.cpp
View file

@ -1,5 +1,5 @@
/* /*
* Copyright (c) 2021, Matthew Olsson <mattco@serenityos.org> * Copyright (c) 2021-2022, Matthew Olsson <mattco@serenityos.org>
* *
* SPDX-License-Identifier: BSD-2-Clause * SPDX-License-Identifier: BSD-2-Clause
*/ */
@ -65,19 +65,18 @@ void Renderer::render()
// as one stream or multiple? // as one stream or multiple?
ByteBuffer byte_buffer; ByteBuffer byte_buffer;
if (m_page.contents->is_array()) { if (m_page.contents->is<ArrayObject>()) {
auto contents = object_cast<ArrayObject>(m_page.contents); auto contents = m_page.contents->cast<ArrayObject>();
for (auto& ref : *contents) { for (auto& ref : *contents) {
auto bytes = m_document->resolve_to<StreamObject>(ref)->bytes(); auto bytes = MUST(m_document->resolve_to<StreamObject>(ref))->bytes();
byte_buffer.append(bytes.data(), bytes.size()); byte_buffer.append(bytes.data(), bytes.size());
} }
} else { } else {
VERIFY(m_page.contents->is_stream()); auto bytes = m_page.contents->cast<StreamObject>()->bytes();
auto bytes = object_cast<StreamObject>(m_page.contents)->bytes();
byte_buffer.append(bytes.data(), bytes.size()); byte_buffer.append(bytes.data(), bytes.size());
} }
auto commands = Parser::parse_graphics_commands(byte_buffer); auto commands = MUST(Parser::parse_graphics_commands(byte_buffer));
for (auto& command : commands) for (auto& command : commands)
handle_command(command); handle_command(command);
@ -147,7 +146,7 @@ RENDERER_HANDLER(set_miter_limit)
RENDERER_HANDLER(set_dash_pattern) RENDERER_HANDLER(set_dash_pattern)
{ {
auto dash_array = m_document->resolve_to<ArrayObject>(args[0]); auto dash_array = MUST(m_document->resolve_to<ArrayObject>(args[0]));
Vector<int> pattern; Vector<int> pattern;
for (auto& element : *dash_array) for (auto& element : *dash_array)
pattern.append(element.get<int>()); pattern.append(element.get<int>());
@ -160,9 +159,9 @@ RENDERER_TODO(set_flatness_tolerance);
RENDERER_HANDLER(set_graphics_state_from_dict) RENDERER_HANDLER(set_graphics_state_from_dict)
{ {
VERIFY(m_page.resources->contains(CommonNames::ExtGState)); VERIFY(m_page.resources->contains(CommonNames::ExtGState));
auto dict_name = m_document->resolve_to<NameObject>(args[0])->name(); auto dict_name = MUST(m_document->resolve_to<NameObject>(args[0]))->name();
auto ext_gstate_dict = m_page.resources->get_dict(m_document, CommonNames::ExtGState); auto ext_gstate_dict = MUST(m_page.resources->get_dict(m_document, CommonNames::ExtGState));
auto target_dict = ext_gstate_dict->get_dict(m_document, dict_name); auto target_dict = MUST(ext_gstate_dict->get_dict(m_document, dict_name));
set_graphics_state_from_dict(target_dict); set_graphics_state_from_dict(target_dict);
} }
@ -303,14 +302,14 @@ RENDERER_HANDLER(text_set_leading)
RENDERER_HANDLER(text_set_font) RENDERER_HANDLER(text_set_font)
{ {
auto target_font_name = m_document->resolve_to<NameObject>(args[0])->name(); auto target_font_name = MUST(m_document->resolve_to<NameObject>(args[0]))->name();
auto fonts_dictionary = m_page.resources->get_dict(m_document, CommonNames::Font); auto fonts_dictionary = MUST(m_page.resources->get_dict(m_document, CommonNames::Font));
auto font_dictionary = fonts_dictionary->get_dict(m_document, target_font_name); auto font_dictionary = MUST(fonts_dictionary->get_dict(m_document, target_font_name));
// FIXME: We do not yet have the standard 14 fonts, as some of them are not open fonts, // FIXME: We do not yet have the standard 14 fonts, as some of them are not open fonts,
// so we just use LiberationSerif for everything // so we just use LiberationSerif for everything
auto font_name = font_dictionary->get_name(m_document, CommonNames::BaseFont)->name().to_lowercase(); auto font_name = MUST(font_dictionary->get_name(m_document, CommonNames::BaseFont))->name().to_lowercase();
auto font_view = font_name.view(); auto font_view = font_name.view();
bool is_bold = font_view.contains("bold"); bool is_bold = font_view.contains("bold");
bool is_italic = font_view.contains("italic"); bool is_italic = font_view.contains("italic");
@ -380,7 +379,7 @@ RENDERER_HANDLER(text_next_line)
RENDERER_HANDLER(text_show_string) RENDERER_HANDLER(text_show_string)
{ {
auto text = m_document->resolve_to<StringObject>(args[0])->string(); auto text = MUST(m_document->resolve_to<StringObject>(args[0]))->string();
show_text(text); show_text(text);
} }
@ -394,7 +393,7 @@ RENDERER_TODO(text_next_line_show_string_set_spacing);
RENDERER_HANDLER(text_show_string_array) RENDERER_HANDLER(text_show_string_array)
{ {
auto elements = m_document->resolve_to<ArrayObject>(args[0])->elements(); auto elements = MUST(m_document->resolve_to<ArrayObject>(args[0]))->elements();
float next_shift = 0.0f; float next_shift = 0.0f;
for (auto& element : elements) { for (auto& element : elements) {
@ -403,9 +402,7 @@ RENDERER_HANDLER(text_show_string_array)
} else if (element.has<float>()) { } else if (element.has<float>()) {
next_shift = element.get<float>(); next_shift = element.get<float>();
} else { } else {
auto& obj = element.get<NonnullRefPtr<Object>>(); auto str = element.get<NonnullRefPtr<Object>>()->cast<StringObject>()->string();
VERIFY(obj->is_string());
auto str = object_cast<StringObject>(obj)->string();
show_text(str, next_shift); show_text(str, next_shift);
} }
} }
@ -523,7 +520,7 @@ void Renderer::set_graphics_state_from_dict(NonnullRefPtr<DictObject> dict)
handle_set_miter_limit({ dict->get_value(CommonNames::ML) }); handle_set_miter_limit({ dict->get_value(CommonNames::ML) });
if (dict->contains(CommonNames::D)) if (dict->contains(CommonNames::D))
handle_set_dash_pattern(dict->get_array(m_document, CommonNames::D)->elements()); handle_set_dash_pattern(MUST(dict->get_array(m_document, CommonNames::D))->elements());
if (dict->contains(CommonNames::FL)) if (dict->contains(CommonNames::FL))
handle_set_flatness_tolerance({ dict->get_value(CommonNames::FL) }); handle_set_flatness_tolerance({ dict->get_value(CommonNames::FL) });
@ -569,7 +566,7 @@ void Renderer::show_text(String const& string, float shift)
RefPtr<ColorSpace> Renderer::get_color_space(Value const& value) RefPtr<ColorSpace> Renderer::get_color_space(Value const& value)
{ {
auto name = object_cast<NameObject>(value.get<NonnullRefPtr<Object>>())->name(); auto name = value.get<NonnullRefPtr<Object>>()->cast<NameObject>()->name();
// Simple color spaces with no parameters, which can be specified directly // Simple color spaces with no parameters, which can be specified directly
if (name == CommonNames::DeviceGray) if (name == CommonNames::DeviceGray)
@ -583,12 +580,12 @@ RefPtr<ColorSpace> Renderer::get_color_space(Value const& value)
// The color space is a complex color space with parameters that resides in // The color space is a complex color space with parameters that resides in
// the resource dictionary // the resource dictionary
auto color_space_resource_dict = m_page.resources->get_dict(m_document, CommonNames::ColorSpace); auto color_space_resource_dict = MUST(m_page.resources->get_dict(m_document, CommonNames::ColorSpace));
if (!color_space_resource_dict->contains(name)) if (!color_space_resource_dict->contains(name))
TODO(); TODO();
auto color_space_array = color_space_resource_dict->get_array(m_document, name); auto color_space_array = MUST(color_space_resource_dict->get_array(m_document, name));
name = color_space_array->get_name_at(m_document, 0)->name(); name = MUST(color_space_array->get_name_at(m_document, 0))->name();
Vector<Value> parameters; Vector<Value> parameters;
parameters.ensure_capacity(color_space_array->size() - 1); parameters.ensure_capacity(color_space_array->size() - 1);

9
Userland/Libraries/LibPDF/Value.h
View file

@ -1,5 +1,5 @@
/* /*
* Copyright (c) 2021, Matthew Olsson <mattco@serenityos.org> * Copyright (c) 2021-2022, Matthew Olsson <mattco@serenityos.org>
* *
* SPDX-License-Identifier: BSD-2-Clause * SPDX-License-Identifier: BSD-2-Clause
*/ */
@ -28,6 +28,13 @@ public:
set<NonnullRefPtr<Object>>(*refptr); set<NonnullRefPtr<Object>>(*refptr);
} }
template<IsObject T>
Value(NonnullRefPtr<T> const& refptr) requires(!IsSame<Object, T>)
: Variant(nullptr)
{
set<NonnullRefPtr<Object>>(*refptr);
}
[[nodiscard]] String to_string(int indent = 0) const; [[nodiscard]] String to_string(int indent = 0) const;
[[nodiscard]] ALWAYS_INLINE bool has_number() const { return has<int>() || has<float>(); } [[nodiscard]] ALWAYS_INLINE bool has_number() const { return has<int>() || has<float>(); }

8
Userland/Libraries/LibPDF/XRefTable.h
View file

@ -1,5 +1,5 @@
/* /*
* Copyright (c) 2021, Matthew Olsson <mattco@serenityos.org> * Copyright (c) 2021-2022, Matthew Olsson <mattco@serenityos.org>
* *
* SPDX-License-Identifier: BSD-2-Clause * SPDX-License-Identifier: BSD-2-Clause
*/ */
@ -29,7 +29,7 @@ struct XRefSection {
class XRefTable final : public RefCounted<XRefTable> { class XRefTable final : public RefCounted<XRefTable> {
public: public:
bool merge(XRefTable&& other) PDFErrorOr<void> merge(XRefTable&& other)
{ {
auto this_size = m_entries.size(); auto this_size = m_entries.size();
auto other_size = other.m_entries.size(); auto other_size = other.m_entries.size();
@ -48,11 +48,11 @@ public:
m_entries[i] = other_entry; m_entries[i] = other_entry;
} else if (other_entry.byte_offset != invalid_byte_offset) { } else if (other_entry.byte_offset != invalid_byte_offset) {
// Both xref tables have an entry for the same object index // Both xref tables have an entry for the same object index
return false; return Error { Error::Type::Parse, "Conflicting xref entry during merge" };
} }
} }
return true; return {};
} }
void add_section(XRefSection const& section) void add_section(XRefSection const& section)