blob: 472a3a986f5d688d88994ccebf6489fd74a86b3b [file] [log] [blame]
// Copyright 2016 The PDFium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <iterator>
#include "public/fpdf_structtree.h"
#include "testing/embedder_test.h"
#include "testing/fx_string_testhelpers.h"
#include "third_party/abseil-cpp/absl/types/optional.h"
class FPDFStructTreeEmbedderTest : public EmbedderTest {};
TEST_F(FPDFStructTreeEmbedderTest, GetAltText) {
ASSERT_TRUE(OpenDocument("tagged_alt_text.pdf"));
FPDF_PAGE page = LoadPage(0);
ASSERT_TRUE(page);
{
ScopedFPDFStructTree struct_tree(FPDF_StructTree_GetForPage(page));
ASSERT_TRUE(struct_tree);
ASSERT_EQ(1, FPDF_StructTree_CountChildren(struct_tree.get()));
FPDF_STRUCTELEMENT element =
FPDF_StructTree_GetChildAtIndex(struct_tree.get(), -1);
EXPECT_FALSE(element);
element = FPDF_StructTree_GetChildAtIndex(struct_tree.get(), 1);
EXPECT_FALSE(element);
element = FPDF_StructTree_GetChildAtIndex(struct_tree.get(), 0);
ASSERT_TRUE(element);
EXPECT_EQ(-1, FPDF_StructElement_GetMarkedContentID(element));
EXPECT_EQ(0U, FPDF_StructElement_GetAltText(element, nullptr, 0));
ASSERT_EQ(1, FPDF_StructElement_CountChildren(element));
FPDF_STRUCTELEMENT child_element =
FPDF_StructElement_GetChildAtIndex(element, -1);
EXPECT_FALSE(child_element);
child_element = FPDF_StructElement_GetChildAtIndex(element, 1);
EXPECT_FALSE(child_element);
child_element = FPDF_StructElement_GetChildAtIndex(element, 0);
ASSERT_TRUE(child_element);
EXPECT_EQ(-1, FPDF_StructElement_GetMarkedContentID(child_element));
EXPECT_EQ(0U, FPDF_StructElement_GetAltText(child_element, nullptr, 0));
ASSERT_EQ(1, FPDF_StructElement_CountChildren(child_element));
FPDF_STRUCTELEMENT gchild_element =
FPDF_StructElement_GetChildAtIndex(child_element, -1);
EXPECT_FALSE(gchild_element);
gchild_element = FPDF_StructElement_GetChildAtIndex(child_element, 1);
EXPECT_FALSE(gchild_element);
gchild_element = FPDF_StructElement_GetChildAtIndex(child_element, 0);
ASSERT_TRUE(gchild_element);
EXPECT_EQ(-1, FPDF_StructElement_GetMarkedContentID(gchild_element));
ASSERT_EQ(24U, FPDF_StructElement_GetAltText(gchild_element, nullptr, 0));
unsigned short buffer[12];
memset(buffer, 0, sizeof(buffer));
// Deliberately pass in a small buffer size to make sure |buffer| remains
// untouched.
ASSERT_EQ(24U, FPDF_StructElement_GetAltText(gchild_element, buffer, 1));
for (size_t i = 0; i < std::size(buffer); ++i)
EXPECT_EQ(0U, buffer[i]);
EXPECT_EQ(-1, FPDF_StructElement_GetMarkedContentID(gchild_element));
ASSERT_EQ(24U, FPDF_StructElement_GetAltText(gchild_element, buffer,
sizeof(buffer)));
EXPECT_EQ(L"Black Image", GetPlatformWString(buffer));
ASSERT_EQ(1, FPDF_StructElement_CountChildren(gchild_element));
FPDF_STRUCTELEMENT ggchild_element =
FPDF_StructElement_GetChildAtIndex(gchild_element, 0);
EXPECT_FALSE(ggchild_element);
}
UnloadPage(page);
}
TEST_F(FPDFStructTreeEmbedderTest, GetActualText) {
ASSERT_TRUE(OpenDocument("tagged_actual_text.pdf"));
FPDF_PAGE page = LoadPage(0);
ASSERT_TRUE(page);
{
ScopedFPDFStructTree struct_tree(FPDF_StructTree_GetForPage(page));
ASSERT_TRUE(struct_tree);
ASSERT_EQ(1, FPDF_StructTree_CountChildren(struct_tree.get()));
EXPECT_EQ(0U, FPDF_StructElement_GetActualText(nullptr, nullptr, 0));
FPDF_STRUCTELEMENT element =
FPDF_StructTree_GetChildAtIndex(struct_tree.get(), 0);
ASSERT_TRUE(element);
EXPECT_EQ(0U, FPDF_StructElement_GetActualText(element, nullptr, 0));
ASSERT_EQ(1, FPDF_StructElement_CountChildren(element));
FPDF_STRUCTELEMENT child_element =
FPDF_StructElement_GetChildAtIndex(element, 0);
ASSERT_TRUE(child_element);
EXPECT_EQ(0U, FPDF_StructElement_GetActualText(child_element, nullptr, 0));
ASSERT_EQ(1, FPDF_StructElement_CountChildren(child_element));
FPDF_STRUCTELEMENT gchild_element =
FPDF_StructElement_GetChildAtIndex(child_element, 0);
ASSERT_TRUE(gchild_element);
ASSERT_EQ(24U,
FPDF_StructElement_GetActualText(gchild_element, nullptr, 0));
unsigned short buffer[12] = {};
// Deliberately pass in a small buffer size to make sure |buffer| remains
// untouched.
ASSERT_EQ(24U, FPDF_StructElement_GetActualText(gchild_element, buffer, 1));
for (size_t i = 0; i < std::size(buffer); ++i)
EXPECT_EQ(0U, buffer[i]);
ASSERT_EQ(24U, FPDF_StructElement_GetActualText(gchild_element, buffer,
sizeof(buffer)));
EXPECT_EQ(L"Actual Text", GetPlatformWString(buffer));
}
UnloadPage(page);
}
TEST_F(FPDFStructTreeEmbedderTest, GetStringAttribute) {
ASSERT_TRUE(OpenDocument("tagged_table.pdf"));
FPDF_PAGE page = LoadPage(0);
ASSERT_TRUE(page);
{
ScopedFPDFStructTree struct_tree(FPDF_StructTree_GetForPage(page));
ASSERT_TRUE(struct_tree);
ASSERT_EQ(1, FPDF_StructTree_CountChildren(struct_tree.get()));
FPDF_STRUCTELEMENT document =
FPDF_StructTree_GetChildAtIndex(struct_tree.get(), 0);
ASSERT_TRUE(document);
constexpr int kBufLen = 100;
uint16_t buffer[kBufLen] = {0};
EXPECT_EQ(18U, FPDF_StructElement_GetType(document, buffer, kBufLen));
EXPECT_EQ("Document", GetPlatformString(buffer));
ASSERT_EQ(1, FPDF_StructElement_CountChildren(document));
FPDF_STRUCTELEMENT table = FPDF_StructElement_GetChildAtIndex(document, 0);
ASSERT_TRUE(table);
EXPECT_EQ(12U, FPDF_StructElement_GetType(table, buffer, kBufLen));
EXPECT_EQ("Table", GetPlatformString(buffer));
// The table should have an attribute "Summary" set to the empty string.
EXPECT_EQ(2U, FPDF_StructElement_GetStringAttribute(table, "Summary",
buffer, kBufLen));
ASSERT_EQ(2, FPDF_StructElement_CountChildren(table));
FPDF_STRUCTELEMENT row = FPDF_StructElement_GetChildAtIndex(table, 0);
ASSERT_TRUE(row);
ASSERT_EQ(2, FPDF_StructElement_CountChildren(row));
FPDF_STRUCTELEMENT header_cell = FPDF_StructElement_GetChildAtIndex(row, 0);
ASSERT_TRUE(header_cell);
EXPECT_EQ(6U, FPDF_StructElement_GetType(header_cell, buffer, kBufLen));
EXPECT_EQ("TH", GetPlatformString(buffer));
// The header should have an attribute "Scope" with a scope of "Row".
EXPECT_EQ(8U, FPDF_StructElement_GetStringAttribute(header_cell, "Scope",
buffer, kBufLen));
EXPECT_EQ("Row", GetPlatformString(buffer));
// The header has an attribute "ColSpan", but it's not a string so it
// returns null.
EXPECT_EQ(0U, FPDF_StructElement_GetStringAttribute(header_cell, "ColSpan",
buffer, kBufLen));
// An unsupported attribute should return 0.
EXPECT_EQ(0U, FPDF_StructElement_GetStringAttribute(header_cell, "Other",
buffer, kBufLen));
// A null struct element should not crash.
EXPECT_EQ(0U, FPDF_StructElement_GetStringAttribute(nullptr, "Other",
buffer, kBufLen));
}
UnloadPage(page);
}
TEST_F(FPDFStructTreeEmbedderTest, GetStringAttributeBadStructElement) {
ASSERT_TRUE(OpenDocument("tagged_table_bad_elem.pdf"));
FPDF_PAGE page = LoadPage(0);
ASSERT_TRUE(page);
{
ScopedFPDFStructTree struct_tree(FPDF_StructTree_GetForPage(page));
ASSERT_TRUE(struct_tree);
ASSERT_EQ(1, FPDF_StructTree_CountChildren(struct_tree.get()));
FPDF_STRUCTELEMENT document =
FPDF_StructTree_GetChildAtIndex(struct_tree.get(), 0);
ASSERT_TRUE(document);
constexpr int kBufLen = 100;
uint16_t buffer[kBufLen] = {0};
EXPECT_EQ(18U, FPDF_StructElement_GetType(document, buffer, kBufLen));
EXPECT_EQ("Document", GetPlatformString(buffer));
// The table can be retrieved, even though it does not have /Type.
ASSERT_EQ(1, FPDF_StructElement_CountChildren(document));
FPDF_STRUCTELEMENT table = FPDF_StructElement_GetChildAtIndex(document, 0);
ASSERT_TRUE(table);
EXPECT_EQ(12U, FPDF_StructElement_GetType(table, buffer, kBufLen));
EXPECT_EQ("Table", GetPlatformString(buffer));
// The table entry cannot be retrieved, as the element is malformed.
EXPECT_EQ(0U, FPDF_StructElement_GetStringAttribute(table, "Summary",
buffer, kBufLen));
// The row can be retrieved, even though it had an invalid /Type.
ASSERT_EQ(1, FPDF_StructElement_CountChildren(table));
FPDF_STRUCTELEMENT row = FPDF_StructElement_GetChildAtIndex(table, 0);
EXPECT_TRUE(row);
}
UnloadPage(page);
}
TEST_F(FPDFStructTreeEmbedderTest, GetID) {
ASSERT_TRUE(OpenDocument("tagged_table.pdf"));
FPDF_PAGE page = LoadPage(0);
ASSERT_TRUE(page);
{
ScopedFPDFStructTree struct_tree(FPDF_StructTree_GetForPage(page));
ASSERT_TRUE(struct_tree);
ASSERT_EQ(1, FPDF_StructTree_CountChildren(struct_tree.get()));
FPDF_STRUCTELEMENT document =
FPDF_StructTree_GetChildAtIndex(struct_tree.get(), 0);
ASSERT_TRUE(document);
constexpr int kBufLen = 100;
uint16_t buffer[kBufLen] = {0};
EXPECT_EQ(18U, FPDF_StructElement_GetType(document, buffer, kBufLen));
EXPECT_EQ("Document", GetPlatformString(buffer));
// The document has no ID.
EXPECT_EQ(0U, FPDF_StructElement_GetID(document, buffer, kBufLen));
ASSERT_EQ(1, FPDF_StructElement_CountChildren(document));
FPDF_STRUCTELEMENT table = FPDF_StructElement_GetChildAtIndex(document, 0);
ASSERT_TRUE(table);
EXPECT_EQ(12U, FPDF_StructElement_GetType(table, buffer, kBufLen));
EXPECT_EQ("Table", GetPlatformString(buffer));
// The table has an ID.
EXPECT_EQ(14U, FPDF_StructElement_GetID(table, buffer, kBufLen));
EXPECT_EQ("node12", GetPlatformString(buffer));
// The first child of the table is a row, which has an empty ID.
// It returns 2U, the length of an empty string, instead of 0U,
// representing null.
ASSERT_EQ(2, FPDF_StructElement_CountChildren(table));
FPDF_STRUCTELEMENT row = FPDF_StructElement_GetChildAtIndex(table, 0);
ASSERT_TRUE(row);
EXPECT_EQ(2U, FPDF_StructElement_GetID(row, buffer, kBufLen));
}
UnloadPage(page);
}
TEST_F(FPDFStructTreeEmbedderTest, GetLang) {
ASSERT_TRUE(OpenDocument("tagged_table.pdf"));
FPDF_PAGE page = LoadPage(0);
ASSERT_TRUE(page);
{
ScopedFPDFStructTree struct_tree(FPDF_StructTree_GetForPage(page));
ASSERT_TRUE(struct_tree);
ASSERT_EQ(1, FPDF_StructTree_CountChildren(struct_tree.get()));
FPDF_STRUCTELEMENT document =
FPDF_StructTree_GetChildAtIndex(struct_tree.get(), 0);
ASSERT_TRUE(document);
constexpr int kBufLen = 100;
uint16_t buffer[kBufLen] = {0};
EXPECT_EQ(18U, FPDF_StructElement_GetType(document, buffer, kBufLen));
EXPECT_EQ("Document", GetPlatformString(buffer));
// Nullptr test
EXPECT_EQ(0U, FPDF_StructElement_GetLang(nullptr, buffer, kBufLen));
// The document has a language.
EXPECT_EQ(12U, FPDF_StructElement_GetLang(document, buffer, kBufLen));
EXPECT_EQ("en-US", GetPlatformString(buffer));
ASSERT_EQ(1, FPDF_StructElement_CountChildren(document));
FPDF_STRUCTELEMENT table = FPDF_StructElement_GetChildAtIndex(document, 0);
ASSERT_TRUE(table);
// The first child is a table, with a language.
EXPECT_EQ(12U, FPDF_StructElement_GetType(table, buffer, kBufLen));
EXPECT_EQ("Table", GetPlatformString(buffer));
EXPECT_EQ(6U, FPDF_StructElement_GetLang(table, buffer, kBufLen));
EXPECT_EQ("hu", GetPlatformString(buffer));
// The first child of the table is a row, which doesn't have a
// language explicitly set on it.
ASSERT_EQ(2, FPDF_StructElement_CountChildren(table));
FPDF_STRUCTELEMENT row = FPDF_StructElement_GetChildAtIndex(table, 0);
ASSERT_TRUE(row);
EXPECT_EQ(0U, FPDF_StructElement_GetLang(row, buffer, kBufLen));
}
UnloadPage(page);
}
// See also FPDFEditEmbedderTest.TraverseMarkedContentID, which traverses the
// marked contents using FPDFPageObj_GetMark() and related API.
TEST_F(FPDFStructTreeEmbedderTest, GetMarkedContentID) {
ASSERT_TRUE(OpenDocument("marked_content_id.pdf"));
FPDF_PAGE page = LoadPage(0);
ASSERT_TRUE(page);
{
ScopedFPDFStructTree struct_tree(FPDF_StructTree_GetForPage(page));
ASSERT_TRUE(struct_tree);
ASSERT_EQ(1, FPDF_StructTree_CountChildren(struct_tree.get()));
FPDF_STRUCTELEMENT element =
FPDF_StructTree_GetChildAtIndex(struct_tree.get(), 0);
EXPECT_EQ(0, FPDF_StructElement_GetMarkedContentID(element));
}
UnloadPage(page);
}
TEST_F(FPDFStructTreeEmbedderTest, GetMarkedContentIdAtIndex) {
ASSERT_TRUE(OpenDocument("tagged_marked_content.pdf"));
FPDF_PAGE page = LoadPage(0);
ASSERT_TRUE(page);
{
ScopedFPDFStructTree struct_tree(FPDF_StructTree_GetForPage(page));
ASSERT_TRUE(struct_tree);
ASSERT_EQ(4, FPDF_StructTree_CountChildren(struct_tree.get()));
// K is an integer MCID
FPDF_STRUCTELEMENT child1 =
FPDF_StructTree_GetChildAtIndex(struct_tree.get(), 0);
ASSERT_TRUE(child1);
// Legacy API
EXPECT_EQ(0, FPDF_StructElement_GetMarkedContentID(child1));
// K is a dict containing MCR object reference
FPDF_STRUCTELEMENT child2 =
FPDF_StructTree_GetChildAtIndex(struct_tree.get(), 1);
ASSERT_TRUE(child2);
// K is an array containing dict MCR object reference and integer MCID
FPDF_STRUCTELEMENT child3 =
FPDF_StructTree_GetChildAtIndex(struct_tree.get(), 2);
ASSERT_TRUE(child3);
// K does not exist
FPDF_STRUCTELEMENT child4 =
FPDF_StructTree_GetChildAtIndex(struct_tree.get(), 3);
ASSERT_TRUE(child4);
// New APIs
EXPECT_EQ(-1, FPDF_StructElement_GetMarkedContentIdCount(nullptr));
EXPECT_EQ(-1, FPDF_StructElement_GetMarkedContentIdAtIndex(nullptr, 0));
EXPECT_EQ(-1, FPDF_StructElement_GetMarkedContentIdAtIndex(child1, -1));
EXPECT_EQ(-1, FPDF_StructElement_GetMarkedContentIdAtIndex(child1, 1));
EXPECT_EQ(1, FPDF_StructElement_GetMarkedContentIdCount(child1));
EXPECT_EQ(0, FPDF_StructElement_GetMarkedContentIdAtIndex(child1, 0));
EXPECT_EQ(1, FPDF_StructElement_GetMarkedContentIdCount(child2));
EXPECT_EQ(1, FPDF_StructElement_GetMarkedContentIdAtIndex(child2, 0));
EXPECT_EQ(2, FPDF_StructElement_GetMarkedContentIdCount(child3));
EXPECT_EQ(2, FPDF_StructElement_GetMarkedContentIdAtIndex(child3, 0));
EXPECT_EQ(3, FPDF_StructElement_GetMarkedContentIdAtIndex(child3, 1));
EXPECT_EQ(-1, FPDF_StructElement_GetMarkedContentIdCount(child4));
EXPECT_EQ(-1, FPDF_StructElement_GetMarkedContentIdAtIndex(child4, 0));
}
UnloadPage(page);
}
TEST_F(FPDFStructTreeEmbedderTest, GetType) {
ASSERT_TRUE(OpenDocument("tagged_alt_text.pdf"));
FPDF_PAGE page = LoadPage(0);
ASSERT_TRUE(page);
{
ScopedFPDFStructTree struct_tree(FPDF_StructTree_GetForPage(page));
ASSERT_TRUE(struct_tree);
ASSERT_EQ(1, FPDF_StructTree_CountChildren(struct_tree.get()));
FPDF_STRUCTELEMENT element =
FPDF_StructTree_GetChildAtIndex(struct_tree.get(), 0);
ASSERT_TRUE(element);
// test nullptr inputs
unsigned short buffer[12];
ASSERT_EQ(0U, FPDF_StructElement_GetType(nullptr, buffer, sizeof(buffer)));
ASSERT_EQ(0U, FPDF_StructElement_GetType(nullptr, nullptr, 0));
ASSERT_EQ(18U, FPDF_StructElement_GetType(element, nullptr, 0));
memset(buffer, 0, sizeof(buffer));
// Deliberately pass in a small buffer size to make sure |buffer| remains
// untouched.
ASSERT_EQ(18U, FPDF_StructElement_GetType(element, buffer, 1));
for (size_t i = 0; i < std::size(buffer); ++i)
EXPECT_EQ(0U, buffer[i]);
ASSERT_EQ(18U, FPDF_StructElement_GetType(element, buffer, sizeof(buffer)));
EXPECT_EQ(L"Document", GetPlatformWString(buffer));
}
UnloadPage(page);
}
TEST_F(FPDFStructTreeEmbedderTest, GetObjType) {
ASSERT_TRUE(OpenDocument("tagged_table_bad_elem.pdf"));
FPDF_PAGE page = LoadPage(0);
ASSERT_TRUE(page);
{
ScopedFPDFStructTree struct_tree(FPDF_StructTree_GetForPage(page));
ASSERT_TRUE(struct_tree);
ASSERT_EQ(1, FPDF_StructTree_CountChildren(struct_tree.get()));
FPDF_STRUCTELEMENT child =
FPDF_StructTree_GetChildAtIndex(struct_tree.get(), 0);
ASSERT_TRUE(child);
// test nullptr inputs
unsigned short buffer[28] = {};
ASSERT_EQ(0U,
FPDF_StructElement_GetObjType(nullptr, buffer, sizeof(buffer)));
ASSERT_EQ(0U, FPDF_StructElement_GetObjType(nullptr, nullptr, 0));
ASSERT_EQ(22U, FPDF_StructElement_GetObjType(child, nullptr, 0));
// Deliberately pass in a small buffer size to make sure `buffer` remains
// untouched.
ASSERT_EQ(22U, FPDF_StructElement_GetObjType(child, buffer, 1));
for (size_t i = 0; i < std::size(buffer); ++i)
EXPECT_EQ(0U, buffer[i]);
ASSERT_EQ(22U,
FPDF_StructElement_GetObjType(child, buffer, sizeof(buffer)));
EXPECT_EQ(L"StructElem", GetPlatformWString(buffer));
ASSERT_EQ(1, FPDF_StructElement_CountChildren(child));
FPDF_STRUCTELEMENT gchild = FPDF_StructElement_GetChildAtIndex(child, 0);
memset(buffer, 0, sizeof(buffer));
// Missing /Type in `gchild`
ASSERT_EQ(0U,
FPDF_StructElement_GetObjType(gchild, buffer, sizeof(buffer)));
// Buffer is untouched.
for (size_t i = 0; i < std::size(buffer); ++i)
EXPECT_EQ(0U, buffer[i]);
ASSERT_EQ(1, FPDF_StructElement_CountChildren(gchild));
FPDF_STRUCTELEMENT ggchild = FPDF_StructElement_GetChildAtIndex(gchild, 0);
ASSERT_EQ(28U,
FPDF_StructElement_GetObjType(ggchild, buffer, sizeof(buffer)));
// Reading bad elem also works.
EXPECT_EQ(L"NotStructElem", GetPlatformWString(buffer));
}
UnloadPage(page);
}
TEST_F(FPDFStructTreeEmbedderTest, GetParent) {
ASSERT_TRUE(OpenDocument("tagged_alt_text.pdf"));
FPDF_PAGE page = LoadPage(0);
ASSERT_TRUE(page);
{
ScopedFPDFStructTree struct_tree(FPDF_StructTree_GetForPage(page));
ASSERT_TRUE(struct_tree);
ASSERT_EQ(1, FPDF_StructTree_CountChildren(struct_tree.get()));
FPDF_STRUCTELEMENT parent =
FPDF_StructTree_GetChildAtIndex(struct_tree.get(), 0);
ASSERT_TRUE(parent);
ASSERT_EQ(1, FPDF_StructElement_CountChildren(parent));
FPDF_STRUCTELEMENT child = FPDF_StructElement_GetChildAtIndex(parent, 0);
ASSERT_TRUE(child);
// test nullptr inputs
ASSERT_EQ(nullptr, FPDF_StructElement_GetParent(nullptr));
ASSERT_EQ(parent, FPDF_StructElement_GetParent(child));
// The parent of `parent` is StructTreeRoot and no longer a StructElement.
// We currently handle this case by returning a nullptr.
ASSERT_EQ(nullptr, FPDF_StructElement_GetParent(parent));
}
UnloadPage(page);
}
TEST_F(FPDFStructTreeEmbedderTest, GetTitle) {
ASSERT_TRUE(OpenDocument("tagged_alt_text.pdf"));
FPDF_PAGE page = LoadPage(0);
ASSERT_TRUE(page);
{
ScopedFPDFStructTree struct_tree(FPDF_StructTree_GetForPage(page));
ASSERT_TRUE(struct_tree);
ASSERT_EQ(1, FPDF_StructTree_CountChildren(struct_tree.get()));
FPDF_STRUCTELEMENT element =
FPDF_StructTree_GetChildAtIndex(struct_tree.get(), 0);
ASSERT_TRUE(element);
// test nullptr inputs
unsigned short buffer[13];
ASSERT_EQ(0U, FPDF_StructElement_GetTitle(nullptr, buffer, sizeof(buffer)));
ASSERT_EQ(0U, FPDF_StructElement_GetTitle(nullptr, nullptr, 0));
ASSERT_EQ(20U, FPDF_StructElement_GetTitle(element, nullptr, 0));
memset(buffer, 0, sizeof(buffer));
// Deliberately pass in a small buffer size to make sure |buffer| remains
// untouched.
ASSERT_EQ(20U, FPDF_StructElement_GetTitle(element, buffer, 1));
for (size_t i = 0; i < std::size(buffer); ++i)
EXPECT_EQ(0U, buffer[i]);
ASSERT_EQ(20U,
FPDF_StructElement_GetTitle(element, buffer, sizeof(buffer)));
EXPECT_EQ(L"TitleText", GetPlatformWString(buffer));
ASSERT_EQ(1, FPDF_StructElement_CountChildren(element));
FPDF_STRUCTELEMENT child_element =
FPDF_StructElement_GetChildAtIndex(element, 0);
ASSERT_TRUE(element);
ASSERT_EQ(26U, FPDF_StructElement_GetTitle(child_element, buffer,
sizeof(buffer)));
EXPECT_EQ(L"symbol: 100k", GetPlatformWString(buffer));
}
UnloadPage(page);
}
TEST_F(FPDFStructTreeEmbedderTest, GetAttributes) {
ASSERT_TRUE(OpenDocument("tagged_table.pdf"));
FPDF_PAGE page = LoadPage(0);
ASSERT_TRUE(page);
{
ScopedFPDFStructTree struct_tree(FPDF_StructTree_GetForPage(page));
ASSERT_TRUE(struct_tree);
ASSERT_EQ(1, FPDF_StructTree_CountChildren(struct_tree.get()));
FPDF_STRUCTELEMENT document =
FPDF_StructTree_GetChildAtIndex(struct_tree.get(), 0);
ASSERT_TRUE(document);
ASSERT_EQ(1, FPDF_StructElement_CountChildren(document));
ASSERT_EQ(-1, FPDF_StructElement_GetAttributeCount(document));
FPDF_STRUCTELEMENT table = FPDF_StructElement_GetChildAtIndex(document, 0);
ASSERT_TRUE(table);
ASSERT_EQ(2, FPDF_StructElement_CountChildren(table));
{
FPDF_STRUCTELEMENT tr = FPDF_StructElement_GetChildAtIndex(table, 0);
ASSERT_TRUE(tr);
ASSERT_EQ(2, FPDF_StructElement_CountChildren(tr));
FPDF_STRUCTELEMENT th = FPDF_StructElement_GetChildAtIndex(tr, 0);
ASSERT_TRUE(th);
ASSERT_EQ(2, FPDF_StructElement_GetAttributeCount(th));
// nullptr test
ASSERT_EQ(nullptr, FPDF_StructElement_GetAttributeAtIndex(document, 0));
ASSERT_EQ(nullptr, FPDF_StructElement_GetAttributeAtIndex(document, -1));
ASSERT_EQ(nullptr, FPDF_StructElement_GetAttributeAtIndex(th, 2));
FPDF_STRUCTELEMENT_ATTR attr =
FPDF_StructElement_GetAttributeAtIndex(th, 1);
ASSERT_TRUE(attr);
ASSERT_EQ(2, FPDF_StructElement_Attr_GetCount(attr));
ASSERT_FALSE(
FPDF_StructElement_Attr_GetName(attr, 1, nullptr, 0U, nullptr));
char buffer[8] = {};
unsigned long out_len = ULONG_MAX;
// Deliberately pass in a small buffer size to make sure `buffer` remains
// untouched.
ASSERT_TRUE(
FPDF_StructElement_Attr_GetName(attr, 1, buffer, 1, &out_len));
EXPECT_EQ(2U, out_len);
for (size_t i = 0; i < std::size(buffer); ++i)
EXPECT_EQ(0, buffer[i]);
ASSERT_TRUE(FPDF_StructElement_Attr_GetName(attr, 1, buffer,
sizeof(buffer), &out_len));
EXPECT_EQ(2U, out_len);
EXPECT_STREQ("O", buffer);
EXPECT_EQ(FPDF_OBJECT_NAME,
FPDF_StructElement_Attr_GetType(attr, buffer));
unsigned short str_val[12] = {};
ASSERT_TRUE(FPDF_StructElement_Attr_GetStringValue(
attr, buffer, str_val, sizeof(str_val), &out_len));
EXPECT_EQ(12U, out_len);
EXPECT_EQ(L"Table", GetPlatformWString(str_val));
memset(buffer, 0, sizeof(buffer));
ASSERT_TRUE(FPDF_StructElement_Attr_GetName(attr, 0, buffer,
sizeof(buffer), &out_len));
EXPECT_EQ(8U, out_len);
EXPECT_STREQ("ColSpan", buffer);
EXPECT_EQ(FPDF_OBJECT_NUMBER,
FPDF_StructElement_Attr_GetType(attr, buffer));
float num_val;
ASSERT_TRUE(
FPDF_StructElement_Attr_GetNumberValue(attr, buffer, &num_val));
EXPECT_FLOAT_EQ(2.0f, num_val);
}
{
FPDF_STRUCTELEMENT tr = FPDF_StructElement_GetChildAtIndex(table, 1);
ASSERT_TRUE(tr);
ASSERT_EQ(1, FPDF_StructElement_GetAttributeCount(tr));
// nullptr when index out of range
ASSERT_EQ(nullptr, FPDF_StructElement_GetAttributeAtIndex(tr, 1));
ASSERT_EQ(2, FPDF_StructElement_CountChildren(tr));
FPDF_STRUCTELEMENT td = FPDF_StructElement_GetChildAtIndex(tr, 1);
ASSERT_TRUE(td);
{
ASSERT_EQ(1, FPDF_StructElement_GetAttributeCount(td));
FPDF_STRUCTELEMENT_ATTR attr =
FPDF_StructElement_GetAttributeAtIndex(td, 0);
ASSERT_TRUE(attr);
ASSERT_EQ(3, FPDF_StructElement_Attr_GetCount(attr));
// Test string and blob type
{
char buffer[16] = {};
unsigned long out_len = ULONG_MAX;
ASSERT_TRUE(FPDF_StructElement_Attr_GetName(
attr, 0, buffer, sizeof(buffer), &out_len));
EXPECT_EQ(8U, out_len);
EXPECT_STREQ("ColProp", buffer);
EXPECT_EQ(FPDF_OBJECT_STRING,
FPDF_StructElement_Attr_GetType(attr, buffer));
unsigned short str_val[12] = {};
ASSERT_TRUE(FPDF_StructElement_Attr_GetStringValue(
attr, buffer, str_val, sizeof(str_val), &out_len));
EXPECT_EQ(8U, out_len);
EXPECT_EQ(L"Sum", GetPlatformWString(str_val));
char blob_val[3] = {};
ASSERT_TRUE(FPDF_StructElement_Attr_GetBlobValue(
attr, buffer, blob_val, sizeof(blob_val), &out_len));
EXPECT_EQ(3U, out_len);
EXPECT_EQ('S', blob_val[0]);
EXPECT_EQ('u', blob_val[1]);
EXPECT_EQ('m', blob_val[2]);
}
// Test boolean type
{
char buffer[16] = {};
unsigned long out_len = ULONG_MAX;
ASSERT_TRUE(FPDF_StructElement_Attr_GetName(
attr, 1, buffer, sizeof(buffer), &out_len));
EXPECT_EQ(7U, out_len);
EXPECT_STREQ("CurUSD", buffer);
EXPECT_EQ(FPDF_OBJECT_BOOLEAN,
FPDF_StructElement_Attr_GetType(attr, buffer));
FPDF_BOOL val;
ASSERT_TRUE(
FPDF_StructElement_Attr_GetBooleanValue(attr, buffer, &val));
EXPECT_TRUE(val);
}
}
}
}
UnloadPage(page);
}
TEST_F(FPDFStructTreeEmbedderTest, GetStructTreeForNestedTaggedPDF) {
ASSERT_TRUE(OpenDocument("tagged_nested.pdf"));
FPDF_PAGE page = LoadPage(0);
ASSERT_TRUE(page);
{
// This call should not crash. https://crbug.com/pdfium/1480
ScopedFPDFStructTree struct_tree(FPDF_StructTree_GetForPage(page));
ASSERT_TRUE(struct_tree);
}
UnloadPage(page);
}
TEST_F(FPDFStructTreeEmbedderTest, MarkedContentReferenceAndObjectReference) {
ASSERT_TRUE(OpenDocument("tagged_mcr_objr.pdf"));
FPDF_PAGE page = LoadPage(0);
ASSERT_TRUE(page);
{
ScopedFPDFStructTree struct_tree(FPDF_StructTree_GetForPage(page));
ASSERT_TRUE(struct_tree);
ASSERT_EQ(1, FPDF_StructTree_CountChildren(struct_tree.get()));
FPDF_STRUCTELEMENT object8 =
FPDF_StructTree_GetChildAtIndex(struct_tree.get(), 0);
ASSERT_TRUE(object8);
unsigned short buffer[12];
ASSERT_EQ(18U, FPDF_StructElement_GetType(object8, buffer, sizeof(buffer)));
EXPECT_EQ(L"Document", GetPlatformWString(buffer));
EXPECT_EQ(-1, FPDF_StructElement_GetMarkedContentID(object8));
ASSERT_EQ(2, FPDF_StructElement_CountChildren(object8));
// First branch. 10 -> 12 -> 13 -> Inline dict.
FPDF_STRUCTELEMENT object10 =
FPDF_StructElement_GetChildAtIndex(object8, 0);
ASSERT_TRUE(object10);
ASSERT_EQ(20U,
FPDF_StructElement_GetType(object10, buffer, sizeof(buffer)));
EXPECT_EQ(L"NonStruct", GetPlatformWString(buffer));
EXPECT_EQ(-1, FPDF_StructElement_GetMarkedContentID(object10));
ASSERT_EQ(1, FPDF_StructElement_CountChildren(object10));
FPDF_STRUCTELEMENT object12 =
FPDF_StructElement_GetChildAtIndex(object10, 0);
ASSERT_TRUE(object12);
ASSERT_EQ(4U, FPDF_StructElement_GetType(object12, buffer, sizeof(buffer)));
EXPECT_EQ(L"P", GetPlatformWString(buffer));
EXPECT_EQ(-1, FPDF_StructElement_GetMarkedContentID(object12));
ASSERT_EQ(1, FPDF_StructElement_CountChildren(object12));
FPDF_STRUCTELEMENT object13 =
FPDF_StructElement_GetChildAtIndex(object12, 0);
ASSERT_TRUE(object13);
ASSERT_EQ(20U,
FPDF_StructElement_GetType(object13, buffer, sizeof(buffer)));
EXPECT_EQ(L"NonStruct", GetPlatformWString(buffer));
EXPECT_EQ(-1, FPDF_StructElement_GetMarkedContentID(object13));
ASSERT_EQ(1, FPDF_StructElement_CountChildren(object13));
// TODO(crbug.com/pdfium/672): Fetch this child element.
EXPECT_FALSE(FPDF_StructElement_GetChildAtIndex(object13, 0));
// Second branch. 11 -> 14 -> Inline dict.
// -> 15 -> Inline dict.
FPDF_STRUCTELEMENT object11 =
FPDF_StructElement_GetChildAtIndex(object8, 1);
ASSERT_TRUE(object11);
ASSERT_EQ(4U, FPDF_StructElement_GetType(object11, buffer, sizeof(buffer)));
EXPECT_EQ(L"P", GetPlatformWString(buffer));
EXPECT_EQ(-1, FPDF_StructElement_GetMarkedContentID(object11));
ASSERT_EQ(1, FPDF_StructElement_CountChildren(object11));
FPDF_STRUCTELEMENT object14 =
FPDF_StructElement_GetChildAtIndex(object11, 0);
ASSERT_TRUE(object14);
ASSERT_EQ(20U,
FPDF_StructElement_GetType(object14, buffer, sizeof(buffer)));
EXPECT_EQ(L"NonStruct", GetPlatformWString(buffer));
EXPECT_EQ(-1, FPDF_StructElement_GetMarkedContentID(object14));
ASSERT_EQ(2, FPDF_StructElement_CountChildren(object14));
// TODO(crbug.com/pdfium/672): Object 15 should be at index 1.
EXPECT_FALSE(FPDF_StructElement_GetChildAtIndex(object14, 1));
FPDF_STRUCTELEMENT object15 =
FPDF_StructElement_GetChildAtIndex(object14, 0);
ASSERT_TRUE(object15);
ASSERT_EQ(20U,
FPDF_StructElement_GetType(object15, buffer, sizeof(buffer)));
EXPECT_EQ(L"NonStruct", GetPlatformWString(buffer));
EXPECT_EQ(-1, FPDF_StructElement_GetMarkedContentID(object15));
ASSERT_EQ(1, FPDF_StructElement_CountChildren(object15));
// TODO(crbug.com/pdfium/672): Fetch this child element.
EXPECT_FALSE(FPDF_StructElement_GetChildAtIndex(object15, 0));
}
UnloadPage(page);
}
TEST_F(FPDFStructTreeEmbedderTest, Bug1768) {
ASSERT_TRUE(OpenDocument("bug_1768.pdf"));
FPDF_PAGE page = LoadPage(0);
ASSERT_TRUE(page);
{
ScopedFPDFStructTree struct_tree(FPDF_StructTree_GetForPage(page));
ASSERT_TRUE(struct_tree);
ASSERT_EQ(1, FPDF_StructTree_CountChildren(struct_tree.get()));
// TODO(crbug.com/pdfium/1768): Fetch this child element. Then consider
// writing more of the test to make sure other elements in the tree can be
// fetched correctly as well.
EXPECT_FALSE(FPDF_StructTree_GetChildAtIndex(struct_tree.get(), 0));
}
UnloadPage(page);
}
TEST_F(FPDFStructTreeEmbedderTest, Bug1296920) {
ASSERT_TRUE(OpenDocument("bug_1296920.pdf"));
FPDF_PAGE page = LoadPage(0);
ASSERT_TRUE(page);
{
ScopedFPDFStructTree struct_tree(FPDF_StructTree_GetForPage(page));
ASSERT_TRUE(struct_tree);
ASSERT_EQ(1, FPDF_StructTree_CountChildren(struct_tree.get()));
// Destroying this tree should not crash.
}
UnloadPage(page);
}