// Copyright 2016 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 <string>
|
|
#include "public/fpdf_edit.h"
|
#include "public/fpdf_ppo.h"
|
#include "public/fpdf_save.h"
|
#include "public/fpdfview.h"
|
#include "testing/embedder_test.h"
|
#include "testing/gtest/include/gtest/gtest.h"
|
#include "testing/test_support.h"
|
|
namespace {
|
|
class FPDFPPOEmbeddertest : public EmbedderTest {};
|
|
int FakeBlockWriter(FPDF_FILEWRITE* pThis,
|
const void* pData,
|
unsigned long size) {
|
return size;
|
}
|
|
} // namespace
|
|
TEST_F(FPDFPPOEmbeddertest, NoViewerPreferences) {
|
EXPECT_TRUE(OpenDocument("hello_world.pdf"));
|
|
FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
|
EXPECT_TRUE(output_doc);
|
EXPECT_FALSE(FPDF_CopyViewerPreferences(output_doc, document()));
|
FPDF_CloseDocument(output_doc);
|
}
|
|
TEST_F(FPDFPPOEmbeddertest, ViewerPreferences) {
|
EXPECT_TRUE(OpenDocument("viewer_ref.pdf"));
|
|
FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
|
EXPECT_TRUE(output_doc);
|
EXPECT_TRUE(FPDF_CopyViewerPreferences(output_doc, document()));
|
FPDF_CloseDocument(output_doc);
|
}
|
|
TEST_F(FPDFPPOEmbeddertest, ImportPages) {
|
ASSERT_TRUE(OpenDocument("viewer_ref.pdf"));
|
|
FPDF_PAGE page = LoadPage(0);
|
EXPECT_TRUE(page);
|
|
FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
|
ASSERT_TRUE(output_doc);
|
EXPECT_TRUE(FPDF_CopyViewerPreferences(output_doc, document()));
|
EXPECT_TRUE(FPDF_ImportPages(output_doc, document(), "1", 0));
|
EXPECT_EQ(1, FPDF_GetPageCount(output_doc));
|
FPDF_CloseDocument(output_doc);
|
|
UnloadPage(page);
|
}
|
|
TEST_F(FPDFPPOEmbeddertest, BadRepeatViewerPref) {
|
ASSERT_TRUE(OpenDocument("repeat_viewer_ref.pdf"));
|
|
FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
|
EXPECT_TRUE(output_doc);
|
EXPECT_TRUE(FPDF_CopyViewerPreferences(output_doc, document()));
|
|
FPDF_FILEWRITE writer;
|
writer.version = 1;
|
writer.WriteBlock = FakeBlockWriter;
|
|
EXPECT_TRUE(FPDF_SaveAsCopy(output_doc, &writer, 0));
|
FPDF_CloseDocument(output_doc);
|
}
|
|
TEST_F(FPDFPPOEmbeddertest, BadCircularViewerPref) {
|
ASSERT_TRUE(OpenDocument("circular_viewer_ref.pdf"));
|
|
FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
|
EXPECT_TRUE(output_doc);
|
EXPECT_TRUE(FPDF_CopyViewerPreferences(output_doc, document()));
|
|
FPDF_FILEWRITE writer;
|
writer.version = 1;
|
writer.WriteBlock = FakeBlockWriter;
|
|
EXPECT_TRUE(FPDF_SaveAsCopy(output_doc, &writer, 0));
|
FPDF_CloseDocument(output_doc);
|
}
|
|
TEST_F(FPDFPPOEmbeddertest, BadRanges) {
|
EXPECT_TRUE(OpenDocument("viewer_ref.pdf"));
|
|
FPDF_PAGE page = LoadPage(0);
|
EXPECT_TRUE(page);
|
|
FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
|
EXPECT_TRUE(output_doc);
|
EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), "clams", 0));
|
EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), "0", 0));
|
EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), "42", 0));
|
EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), "1,2", 0));
|
EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), "1-2", 0));
|
EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), ",1", 0));
|
EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), "1,", 0));
|
EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), "1-", 0));
|
EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), "-1", 0));
|
EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), "-,0,,,1-", 0));
|
FPDF_CloseDocument(output_doc);
|
|
UnloadPage(page);
|
}
|
|
TEST_F(FPDFPPOEmbeddertest, GoodRanges) {
|
EXPECT_TRUE(OpenDocument("viewer_ref.pdf"));
|
|
FPDF_PAGE page = LoadPage(0);
|
EXPECT_TRUE(page);
|
|
FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
|
EXPECT_TRUE(output_doc);
|
EXPECT_TRUE(FPDF_CopyViewerPreferences(output_doc, document()));
|
EXPECT_TRUE(FPDF_ImportPages(output_doc, document(), "1,1,1,1", 0));
|
EXPECT_TRUE(FPDF_ImportPages(output_doc, document(), "1-1", 0));
|
EXPECT_EQ(5, FPDF_GetPageCount(output_doc));
|
FPDF_CloseDocument(output_doc);
|
|
UnloadPage(page);
|
}
|
|
TEST_F(FPDFPPOEmbeddertest, BUG_664284) {
|
EXPECT_TRUE(OpenDocument("bug_664284.pdf"));
|
|
FPDF_PAGE page = LoadPage(0);
|
ASSERT_NE(nullptr, page);
|
|
FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
|
EXPECT_TRUE(output_doc);
|
EXPECT_TRUE(FPDF_ImportPages(output_doc, document(), "1", 0));
|
FPDF_CloseDocument(output_doc);
|
|
UnloadPage(page);
|
}
|
|
TEST_F(FPDFPPOEmbeddertest, BUG_750568) {
|
const char* const kHashes[] = {
|
"64ad08132a1c5a166768298c8a578f57", "83b83e2f6bc80707d0a917c7634140b9",
|
"913cd3723a451e4e46fbc2c05702d1ee", "81fb7cfd4860f855eb468f73dfeb6d60"};
|
|
ASSERT_TRUE(OpenDocument("bug_750568.pdf"));
|
ASSERT_EQ(4, FPDF_GetPageCount(document()));
|
|
for (size_t i = 0; i < 4; ++i) {
|
FPDF_PAGE page = LoadPage(i);
|
ASSERT_NE(nullptr, page);
|
|
FPDF_BITMAP bitmap = RenderPage(page);
|
ASSERT_EQ(200, FPDFBitmap_GetWidth(bitmap));
|
ASSERT_EQ(200, FPDFBitmap_GetHeight(bitmap));
|
ASSERT_EQ(800, FPDFBitmap_GetStride(bitmap));
|
|
std::string digest = HashBitmap(bitmap);
|
FPDFBitmap_Destroy(bitmap);
|
UnloadPage(page);
|
EXPECT_EQ(kHashes[i], digest);
|
}
|
|
FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
|
ASSERT_TRUE(output_doc);
|
EXPECT_TRUE(FPDF_ImportPages(output_doc, document(), "1,2,3,4", 0));
|
ASSERT_EQ(4, FPDF_GetPageCount(output_doc));
|
for (size_t i = 0; i < 4; ++i) {
|
FPDF_PAGE page = FPDF_LoadPage(output_doc, i);
|
ASSERT_NE(nullptr, page);
|
|
FPDF_BITMAP bitmap = RenderPage(page);
|
ASSERT_EQ(200, FPDFBitmap_GetWidth(bitmap));
|
ASSERT_EQ(200, FPDFBitmap_GetHeight(bitmap));
|
ASSERT_EQ(800, FPDFBitmap_GetStride(bitmap));
|
|
std::string digest = HashBitmap(bitmap);
|
FPDFBitmap_Destroy(bitmap);
|
FPDF_ClosePage(page);
|
EXPECT_EQ(kHashes[i], digest);
|
}
|
FPDF_CloseDocument(output_doc);
|
}
|
|
TEST_F(FPDFPPOEmbeddertest, ImportWithZeroLengthStream) {
|
EXPECT_TRUE(OpenDocument("zero_length_stream.pdf"));
|
FPDF_PAGE page = LoadPage(0);
|
ASSERT_NE(nullptr, page);
|
|
FPDF_BITMAP bitmap = RenderPage(page);
|
ASSERT_EQ(200, FPDFBitmap_GetWidth(bitmap));
|
ASSERT_EQ(200, FPDFBitmap_GetHeight(bitmap));
|
ASSERT_EQ(800, FPDFBitmap_GetStride(bitmap));
|
|
std::string digest = HashBitmap(bitmap);
|
FPDFBitmap_Destroy(bitmap);
|
FPDF_ClosePage(page);
|
|
FPDF_DOCUMENT new_doc = FPDF_CreateNewDocument();
|
EXPECT_TRUE(new_doc);
|
EXPECT_TRUE(FPDF_ImportPages(new_doc, document(), "1", 0));
|
|
EXPECT_EQ(1, FPDF_GetPageCount(new_doc));
|
FPDF_PAGE new_page = FPDF_LoadPage(new_doc, 0);
|
ASSERT_NE(nullptr, new_page);
|
FPDF_BITMAP new_bitmap = RenderPage(new_page);
|
ASSERT_EQ(200, FPDFBitmap_GetWidth(new_bitmap));
|
ASSERT_EQ(200, FPDFBitmap_GetHeight(new_bitmap));
|
ASSERT_EQ(800, FPDFBitmap_GetStride(new_bitmap));
|
|
std::string new_digest = HashBitmap(new_bitmap);
|
FPDFBitmap_Destroy(new_bitmap);
|
FPDF_ClosePage(new_page);
|
FPDF_CloseDocument(new_doc);
|
|
EXPECT_EQ(digest, new_digest);
|
}
|
