1 // Copyright 2016 The PDFium Authors
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "build/build_config.h"
6 #include "core/fxge/cfx_defaultrenderdevice.h"
7 #include "public/fpdf_flatten.h"
8 #include "public/fpdfview.h"
9 #include "testing/embedder_test.h"
10 #include "testing/gtest/include/gtest/gtest.h"
11
12 namespace {
13
14 class FPDFFlattenEmbedderTest : public EmbedderTest {};
15
16 } // namespace
17
TEST_F(FPDFFlattenEmbedderTest,FlatNothing)18 TEST_F(FPDFFlattenEmbedderTest, FlatNothing) {
19 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
20 FPDF_PAGE page = LoadPage(0);
21 EXPECT_TRUE(page);
22 EXPECT_EQ(FLATTEN_NOTHINGTODO, FPDFPage_Flatten(page, FLAT_NORMALDISPLAY));
23 UnloadPage(page);
24 }
25
TEST_F(FPDFFlattenEmbedderTest,FlatNormal)26 TEST_F(FPDFFlattenEmbedderTest, FlatNormal) {
27 ASSERT_TRUE(OpenDocument("annotiter.pdf"));
28 FPDF_PAGE page = LoadPage(0);
29 EXPECT_TRUE(page);
30 EXPECT_EQ(FLATTEN_SUCCESS, FPDFPage_Flatten(page, FLAT_NORMALDISPLAY));
31 UnloadPage(page);
32 }
33
TEST_F(FPDFFlattenEmbedderTest,FlatPrint)34 TEST_F(FPDFFlattenEmbedderTest, FlatPrint) {
35 ASSERT_TRUE(OpenDocument("annotiter.pdf"));
36 FPDF_PAGE page = LoadPage(0);
37 EXPECT_TRUE(page);
38 EXPECT_EQ(FLATTEN_SUCCESS, FPDFPage_Flatten(page, FLAT_PRINT));
39 UnloadPage(page);
40 }
41
TEST_F(FPDFFlattenEmbedderTest,BUG_861842)42 TEST_F(FPDFFlattenEmbedderTest, BUG_861842) {
43 const char* checkbox_checksum = []() {
44 if (CFX_DefaultRenderDevice::SkiaIsDefaultRenderer())
45 return "95fdaa000e81c80892b8d370f77be970";
46 #if BUILDFLAG(IS_APPLE)
47 return "6aafcb2d98da222964bcdbf5aa1f4f1f";
48 #else
49 return "594265790b81df2d93120d33b72a6ada";
50 #endif
51 }();
52
53 ASSERT_TRUE(OpenDocument("bug_861842.pdf"));
54 FPDF_PAGE page = LoadPage(0);
55 ASSERT_TRUE(page);
56
57 ScopedFPDFBitmap bitmap = RenderLoadedPageWithFlags(page, FPDF_ANNOT);
58 CompareBitmap(bitmap.get(), 100, 120, checkbox_checksum);
59
60 EXPECT_EQ(FLATTEN_SUCCESS, FPDFPage_Flatten(page, FLAT_PRINT));
61 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
62
63 UnloadPage(page);
64
65 // TODO(crbug.com/861842): This should not render blank.
66 static constexpr char kBlankPageHash[] = "48400809c3862dae64b0cd00d51057a4";
67 VerifySavedDocument(100, 120, kBlankPageHash);
68 }
69
TEST_F(FPDFFlattenEmbedderTest,BUG_889099)70 TEST_F(FPDFFlattenEmbedderTest, BUG_889099) {
71 const char* page_checksum = []() {
72 if (CFX_DefaultRenderDevice::SkiaIsDefaultRenderer())
73 return "de7119d99f42deab2f4215017bdb16af";
74 #if BUILDFLAG(IS_APPLE)
75 return "049ed3f1e21fc72f929af3410c64bc8f";
76 #else
77 return "3db87245e3f4e37f4cb18654bbe22d97";
78 #endif
79 }();
80 const char* flattened_page_checksum = []() {
81 if (CFX_DefaultRenderDevice::SkiaIsDefaultRenderer()) {
82 return "7978c7b3d643a5f0ac0f03ce759c55fe";
83 }
84 #if BUILDFLAG(IS_APPLE)
85 return "41debc60cf2a8f74c710ec6082d77b18";
86 #else
87 return "0832157462ea70fbbf053e14b1d6457f";
88 #endif
89 }();
90
91 ASSERT_TRUE(OpenDocument("bug_889099.pdf"));
92 FPDF_PAGE page = LoadPage(0);
93 ASSERT_TRUE(page);
94
95 // The original document has a malformed media box; the height is -400.
96 ScopedFPDFBitmap bitmap = RenderLoadedPageWithFlags(page, FPDF_ANNOT);
97 CompareBitmap(bitmap.get(), 300, 400, page_checksum);
98
99 EXPECT_EQ(FLATTEN_SUCCESS, FPDFPage_Flatten(page, FLAT_PRINT));
100 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
101
102 UnloadPage(page);
103
104 VerifySavedDocument(300, 400, flattened_page_checksum);
105 }
106
TEST_F(FPDFFlattenEmbedderTest,BUG_890322)107 TEST_F(FPDFFlattenEmbedderTest, BUG_890322) {
108 const char* checksum = []() {
109 if (CFX_DefaultRenderDevice::SkiaIsDefaultRenderer())
110 return "793689536cf64fe792c2f241888c0cf3";
111 return "6c674642154408e877d88c6c082d67e9";
112 }();
113 ASSERT_TRUE(OpenDocument("bug_890322.pdf"));
114 FPDF_PAGE page = LoadPage(0);
115 ASSERT_TRUE(page);
116
117 ScopedFPDFBitmap bitmap = RenderLoadedPageWithFlags(page, FPDF_ANNOT);
118 CompareBitmap(bitmap.get(), 200, 200, checksum);
119
120 EXPECT_EQ(FLATTEN_SUCCESS, FPDFPage_Flatten(page, FLAT_PRINT));
121 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
122
123 UnloadPage(page);
124
125 VerifySavedDocument(200, 200, checksum);
126 }
127
TEST_F(FPDFFlattenEmbedderTest,BUG_896366)128 TEST_F(FPDFFlattenEmbedderTest, BUG_896366) {
129 const char* checksum = []() {
130 if (CFX_DefaultRenderDevice::SkiaIsDefaultRenderer())
131 return "c3cccfadc4c5249e6aa0675e511fa4c3";
132 return "f71ab085c52c8445ae785eca3ec858b1";
133 }();
134 ASSERT_TRUE(OpenDocument("bug_896366.pdf"));
135 FPDF_PAGE page = LoadPage(0);
136 ASSERT_TRUE(page);
137
138 ScopedFPDFBitmap bitmap = RenderLoadedPageWithFlags(page, FPDF_ANNOT);
139 CompareBitmap(bitmap.get(), 612, 792, checksum);
140
141 EXPECT_EQ(FLATTEN_SUCCESS, FPDFPage_Flatten(page, FLAT_PRINT));
142 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
143
144 UnloadPage(page);
145
146 VerifySavedDocument(612, 792, checksum);
147 }
148