blob: 28c907a101a27a85cb664611447c61e4298d9869 [file] [log] [blame]
// Copyright 2015 PDFium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <limits>
#include <memory>
#include <string>
#include <vector>
#include "core/fpdfapi/parser/cpdf_linearized_header.h"
#include "core/fpdfapi/parser/cpdf_object.h"
#include "core/fpdfapi/parser/cpdf_parser.h"
#include "core/fpdfapi/parser/cpdf_syntax_parser.h"
#include "core/fxcrt/cfx_readonlymemorystream.h"
#include "core/fxcrt/fx_extension.h"
#include "core/fxcrt/fx_stream.h"
#include "core/fxcrt/retain_ptr.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "testing/utils/path_service.h"
namespace {
CPDF_CrossRefTable::ObjectInfo GetObjInfo(const CPDF_Parser& parser,
uint32_t obj_num) {
const auto* info = parser.GetCrossRefTable()->GetObjectInfo(obj_num);
return info ? *info : CPDF_CrossRefTable::ObjectInfo();
}
} // namespace
// A wrapper class to help test member functions of CPDF_Parser.
class CPDF_TestParser final : public CPDF_Parser {
public:
CPDF_TestParser() {}
~CPDF_TestParser() {}
// Setup reading from a file and initial states.
bool InitTestFromFile(const char* path) {
RetainPtr<IFX_SeekableReadStream> pFileAccess =
IFX_SeekableReadStream::CreateFromFilename(path);
if (!pFileAccess)
return false;
// For the test file, the header is set at the beginning.
m_pSyntax = pdfium::MakeUnique<CPDF_SyntaxParser>(pFileAccess);
return true;
}
// Setup reading from a buffer and initial states.
bool InitTestFromBufferWithOffset(pdfium::span<const uint8_t> buffer,
FX_FILESIZE header_offset) {
m_pSyntax = CPDF_SyntaxParser::CreateForTesting(
pdfium::MakeRetain<CFX_ReadOnlyMemoryStream>(buffer), header_offset);
return true;
}
bool InitTestFromBuffer(pdfium::span<const uint8_t> buffer) {
return InitTestFromBufferWithOffset(buffer, 0 /*header_offset*/);
}
private:
// Add test cases here as private friend so that protected members in
// CPDF_Parser can be accessed by test cases.
// Need to access RebuildCrossRef.
FRIEND_TEST(cpdf_parser, RebuildCrossRefCorrectly);
FRIEND_TEST(cpdf_parser, RebuildCrossRefFailed);
// Need to access LoadCrossRefV4.
FRIEND_TEST(cpdf_parser, LoadCrossRefV4);
};
TEST(cpdf_parser, RebuildCrossRefCorrectly) {
CPDF_TestParser parser;
std::string test_file;
ASSERT_TRUE(PathService::GetTestFilePath("parser_rebuildxref_correct.pdf",
&test_file));
ASSERT_TRUE(parser.InitTestFromFile(test_file.c_str())) << test_file;
ASSERT_TRUE(parser.RebuildCrossRef());
const FX_FILESIZE offsets[] = {0, 15, 61, 154, 296, 374, 450};
const uint16_t versions[] = {0, 0, 2, 4, 6, 8, 0};
for (size_t i = 0; i < FX_ArraySize(offsets); ++i)
EXPECT_EQ(offsets[i], GetObjInfo(parser, i).pos);
for (size_t i = 0; i < FX_ArraySize(versions); ++i)
EXPECT_EQ(versions[i], GetObjInfo(parser, i).gennum);
}
TEST(cpdf_parser, RebuildCrossRefFailed) {
CPDF_TestParser parser;
std::string test_file;
ASSERT_TRUE(PathService::GetTestFilePath(
"parser_rebuildxref_error_notrailer.pdf", &test_file));
ASSERT_TRUE(parser.InitTestFromFile(test_file.c_str())) << test_file;
ASSERT_FALSE(parser.RebuildCrossRef());
}
TEST(cpdf_parser, LoadCrossRefV4) {
{
const unsigned char xref_table[] =
"xref \n"
"0 6 \n"
"0000000003 65535 f \n"
"0000000017 00000 n \n"
"0000000081 00000 n \n"
"0000000000 00007 f \n"
"0000000331 00000 n \n"
"0000000409 00000 n \n"
"trail"; // Needed to end cross ref table reading.
CPDF_TestParser parser;
ASSERT_TRUE(parser.InitTestFromBuffer(xref_table));
ASSERT_TRUE(parser.LoadCrossRefV4(0, false));
const FX_FILESIZE offsets[] = {0, 17, 81, 0, 331, 409};
const CPDF_TestParser::ObjectType types[] = {
CPDF_TestParser::ObjectType::kFree,
CPDF_TestParser::ObjectType::kNotCompressed,
CPDF_TestParser::ObjectType::kNotCompressed,
CPDF_TestParser::ObjectType::kFree,
CPDF_TestParser::ObjectType::kNotCompressed,
CPDF_TestParser::ObjectType::kNotCompressed};
for (size_t i = 0; i < FX_ArraySize(offsets); ++i) {
EXPECT_EQ(offsets[i], GetObjInfo(parser, i).pos);
EXPECT_EQ(types[i], GetObjInfo(parser, i).type);
}
}
{
const unsigned char xref_table[] =
"xref \n"
"0 1 \n"
"0000000000 65535 f \n"
"3 1 \n"
"0000025325 00000 n \n"
"8 2 \n"
"0000025518 00002 n \n"
"0000025635 00000 n \n"
"12 1 \n"
"0000025777 00000 n \n"
"trail"; // Needed to end cross ref table reading.
CPDF_TestParser parser;
ASSERT_TRUE(parser.InitTestFromBuffer(xref_table));
ASSERT_TRUE(parser.LoadCrossRefV4(0, false));
const FX_FILESIZE offsets[] = {0, 0, 0, 25325, 0, 0, 0,
0, 25518, 25635, 0, 0, 25777};
const CPDF_TestParser::ObjectType types[] = {
CPDF_TestParser::ObjectType::kFree,
CPDF_TestParser::ObjectType::kFree,
CPDF_TestParser::ObjectType::kFree,
CPDF_TestParser::ObjectType::kNotCompressed,
CPDF_TestParser::ObjectType::kFree,
CPDF_TestParser::ObjectType::kFree,
CPDF_TestParser::ObjectType::kFree,
CPDF_TestParser::ObjectType::kFree,
CPDF_TestParser::ObjectType::kNotCompressed,
CPDF_TestParser::ObjectType::kNotCompressed,
CPDF_TestParser::ObjectType::kFree,
CPDF_TestParser::ObjectType::kFree,
CPDF_TestParser::ObjectType::kNotCompressed};
for (size_t i = 0; i < FX_ArraySize(offsets); ++i) {
EXPECT_EQ(offsets[i], GetObjInfo(parser, i).pos);
EXPECT_EQ(types[i], GetObjInfo(parser, i).type);
}
}
{
const unsigned char xref_table[] =
"xref \n"
"0 1 \n"
"0000000000 65535 f \n"
"3 1 \n"
"0000025325 00000 n \n"
"8 2 \n"
"0000000000 65535 f \n"
"0000025635 00000 n \n"
"12 1 \n"
"0000025777 00000 n \n"
"trail"; // Needed to end cross ref table reading.
CPDF_TestParser parser;
ASSERT_TRUE(parser.InitTestFromBuffer(xref_table));
ASSERT_TRUE(parser.LoadCrossRefV4(0, false));
const FX_FILESIZE offsets[] = {0, 0, 0, 25325, 0, 0, 0,
0, 0, 25635, 0, 0, 25777};
const CPDF_TestParser::ObjectType types[] = {
CPDF_TestParser::ObjectType::kFree,
CPDF_TestParser::ObjectType::kFree,
CPDF_TestParser::ObjectType::kFree,
CPDF_TestParser::ObjectType::kNotCompressed,
CPDF_TestParser::ObjectType::kFree,
CPDF_TestParser::ObjectType::kFree,
CPDF_TestParser::ObjectType::kFree,
CPDF_TestParser::ObjectType::kFree,
CPDF_TestParser::ObjectType::kFree,
CPDF_TestParser::ObjectType::kNotCompressed,
CPDF_TestParser::ObjectType::kFree,
CPDF_TestParser::ObjectType::kFree,
CPDF_TestParser::ObjectType::kNotCompressed};
for (size_t i = 0; i < FX_ArraySize(offsets); ++i) {
EXPECT_EQ(offsets[i], GetObjInfo(parser, i).pos);
EXPECT_EQ(types[i], GetObjInfo(parser, i).type);
}
}
{
const unsigned char xref_table[] =
"xref \n"
"0 7 \n"
"0000000002 65535 f \n"
"0000000023 00000 n \n"
"0000000003 65535 f \n"
"0000000004 65535 f \n"
"0000000000 65535 f \n"
"0000000045 00000 n \n"
"0000000179 00000 n \n"
"trail"; // Needed to end cross ref table reading.
CPDF_TestParser parser;
ASSERT_TRUE(parser.InitTestFromBuffer(xref_table));
ASSERT_TRUE(parser.LoadCrossRefV4(0, false));
const FX_FILESIZE offsets[] = {0, 23, 0, 0, 0, 45, 179};
const CPDF_TestParser::ObjectType types[] = {
CPDF_TestParser::ObjectType::kFree,
CPDF_TestParser::ObjectType::kNotCompressed,
CPDF_TestParser::ObjectType::kFree,
CPDF_TestParser::ObjectType::kFree,
CPDF_TestParser::ObjectType::kFree,
CPDF_TestParser::ObjectType::kNotCompressed,
CPDF_TestParser::ObjectType::kNotCompressed};
for (size_t i = 0; i < FX_ArraySize(offsets); ++i) {
EXPECT_EQ(offsets[i], GetObjInfo(parser, i).pos);
EXPECT_EQ(types[i], GetObjInfo(parser, i).type);
}
}
}
TEST(cpdf_parser, ParseStartXRef) {
CPDF_TestParser parser;
std::string test_file;
ASSERT_TRUE(
PathService::GetTestFilePath("annotation_stamp_with_ap.pdf", &test_file));
ASSERT_TRUE(parser.InitTestFromFile(test_file.c_str())) << test_file;
EXPECT_EQ(100940, parser.ParseStartXRef());
std::unique_ptr<CPDF_Object> cross_ref_v5_obj =
parser.ParseIndirectObjectAt(100940, 0);
ASSERT_TRUE(cross_ref_v5_obj);
EXPECT_EQ(75u, cross_ref_v5_obj->GetObjNum());
}
TEST(cpdf_parser, ParseStartXRefWithHeaderOffset) {
static constexpr FX_FILESIZE kTestHeaderOffset = 765;
std::string test_file;
ASSERT_TRUE(
PathService::GetTestFilePath("annotation_stamp_with_ap.pdf", &test_file));
RetainPtr<IFX_SeekableReadStream> pFileAccess =
IFX_SeekableReadStream::CreateFromFilename(test_file.c_str());
ASSERT_TRUE(pFileAccess);
std::vector<unsigned char> data(pFileAccess->GetSize() + kTestHeaderOffset);
ASSERT_TRUE(pFileAccess->ReadBlock(&data.front() + kTestHeaderOffset, 0,
pFileAccess->GetSize()));
CPDF_TestParser parser;
parser.InitTestFromBufferWithOffset(data, kTestHeaderOffset);
EXPECT_EQ(100940, parser.ParseStartXRef());
std::unique_ptr<CPDF_Object> cross_ref_v5_obj =
parser.ParseIndirectObjectAt(100940, 0);
ASSERT_TRUE(cross_ref_v5_obj);
EXPECT_EQ(75u, cross_ref_v5_obj->GetObjNum());
}
TEST(cpdf_parser, ParseLinearizedWithHeaderOffset) {
static constexpr FX_FILESIZE kTestHeaderOffset = 765;
std::string test_file;
ASSERT_TRUE(PathService::GetTestFilePath("linearized.pdf", &test_file));
RetainPtr<IFX_SeekableReadStream> pFileAccess =
IFX_SeekableReadStream::CreateFromFilename(test_file.c_str());
ASSERT_TRUE(pFileAccess);
std::vector<unsigned char> data(pFileAccess->GetSize() + kTestHeaderOffset);
ASSERT_TRUE(pFileAccess->ReadBlock(&data.front() + kTestHeaderOffset, 0,
pFileAccess->GetSize()));
CPDF_TestParser parser;
parser.InitTestFromBufferWithOffset(data, kTestHeaderOffset);
EXPECT_TRUE(parser.ParseLinearizedHeader());
}