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/gmock/include/gmock/gmock.h"
11 #include "testing/gtest/include/gtest/gtest.h"
12
13 using testing::HasSubstr;
14 using testing::Not;
15
16 namespace {
17
18 class FPDFFlattenEmbedderTest : public EmbedderTest {};
19
20 } // namespace
21
TEST_F(FPDFFlattenEmbedderTest,FlatNothing)22 TEST_F(FPDFFlattenEmbedderTest, FlatNothing) {
23 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
24 ScopedEmbedderTestPage page = LoadScopedPage(0);
25 EXPECT_TRUE(page);
26 EXPECT_EQ(FLATTEN_NOTHINGTODO,
27 FPDFPage_Flatten(page.get(), FLAT_NORMALDISPLAY));
28 }
29
TEST_F(FPDFFlattenEmbedderTest,FlatNormal)30 TEST_F(FPDFFlattenEmbedderTest, FlatNormal) {
31 ASSERT_TRUE(OpenDocument("annotiter.pdf"));
32 ScopedEmbedderTestPage page = LoadScopedPage(0);
33 EXPECT_TRUE(page);
34 EXPECT_EQ(FLATTEN_SUCCESS, FPDFPage_Flatten(page.get(), FLAT_NORMALDISPLAY));
35 }
36
TEST_F(FPDFFlattenEmbedderTest,FlatPrint)37 TEST_F(FPDFFlattenEmbedderTest, FlatPrint) {
38 ASSERT_TRUE(OpenDocument("annotiter.pdf"));
39 ScopedEmbedderTestPage page = LoadScopedPage(0);
40 EXPECT_TRUE(page);
41 EXPECT_EQ(FLATTEN_SUCCESS, FPDFPage_Flatten(page.get(), FLAT_PRINT));
42 }
43
TEST_F(FPDFFlattenEmbedderTest,FlatWithBadFont)44 TEST_F(FPDFFlattenEmbedderTest, FlatWithBadFont) {
45 ASSERT_TRUE(OpenDocument("344775293.pdf"));
46 ScopedEmbedderTestPage page = LoadScopedPage(0);
47 EXPECT_TRUE(page);
48
49 FORM_OnLButtonDown(form_handle(), page.get(), 0, 20, 30);
50 FORM_OnLButtonUp(form_handle(), page.get(), 0, 20, 30);
51
52 EXPECT_EQ(FLATTEN_SUCCESS, FPDFPage_Flatten(page.get(), FLAT_PRINT));
53 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
54
55 EXPECT_THAT(GetString(), Not(HasSubstr("/PDFDocEncoding")));
56 }
57
TEST_F(FPDFFlattenEmbedderTest,FlatWithFontNoBaseEncoding)58 TEST_F(FPDFFlattenEmbedderTest, FlatWithFontNoBaseEncoding) {
59 ASSERT_TRUE(OpenDocument("363015187.pdf"));
60 ScopedEmbedderTestPage page = LoadScopedPage(0);
61 EXPECT_TRUE(page);
62
63 EXPECT_EQ(FLATTEN_SUCCESS, FPDFPage_Flatten(page.get(), FLAT_PRINT));
64 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
65
66 EXPECT_THAT(GetString(), HasSubstr("/Differences"));
67 }
68
TEST_F(FPDFFlattenEmbedderTest,Bug861842)69 TEST_F(FPDFFlattenEmbedderTest, Bug861842) {
70 const char* checkbox_checksum = []() {
71 if (CFX_DefaultRenderDevice::UseSkiaRenderer()) {
72 #if BUILDFLAG(IS_APPLE)
73 return "84a527f16649880525a1a8edc6c24c16";
74 #else
75 return "95fdaa000e81c80892b8d370f77be970";
76 #endif
77 }
78 #if BUILDFLAG(IS_APPLE)
79 return "6aafcb2d98da222964bcdbf5aa1f4f1f";
80 #else
81 return "594265790b81df2d93120d33b72a6ada";
82 #endif
83 }();
84
85 ASSERT_TRUE(OpenDocument("bug_861842.pdf"));
86 ScopedEmbedderTestPage page = LoadScopedPage(0);
87 ASSERT_TRUE(page);
88
89 ScopedFPDFBitmap bitmap = RenderLoadedPageWithFlags(page.get(), FPDF_ANNOT);
90 CompareBitmap(bitmap.get(), 100, 120, checkbox_checksum);
91
92 EXPECT_EQ(FLATTEN_SUCCESS, FPDFPage_Flatten(page.get(), FLAT_PRINT));
93 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
94
95
96 // TODO(crbug.com/861842): This should not render blank.
97 static constexpr char kBlankPageHash[] = "48400809c3862dae64b0cd00d51057a4";
98 VerifySavedDocument(100, 120, kBlankPageHash);
99 }
100
TEST_F(FPDFFlattenEmbedderTest,Bug889099)101 TEST_F(FPDFFlattenEmbedderTest, Bug889099) {
102 const char* page_checksum = []() {
103 if (CFX_DefaultRenderDevice::UseSkiaRenderer()) {
104 #if BUILDFLAG(IS_WIN)
105 return "1d83328d2d1ca12b9c9ea5faa62ac515";
106 #elif BUILDFLAG(IS_APPLE)
107 return "3b6f937deec2d27029cbce02111dc065";
108 #else
109 return "de7119d99f42deab2f4215017bdb16af";
110 #endif
111 }
112 #if BUILDFLAG(IS_APPLE)
113 return "049ed3f1e21fc72f929af3410c64bc8f";
114 #else
115 return "3db87245e3f4e37f4cb18654bbe22d97";
116 #endif
117 }();
118 const char* flattened_page_checksum = []() {
119 if (CFX_DefaultRenderDevice::UseSkiaRenderer()) {
120 #if BUILDFLAG(IS_WIN)
121 return "07deccbd4a42aaf6bf45a525f0be388e";
122 #elif BUILDFLAG(IS_APPLE)
123 return "f1bbe115355a2ad6d8ac34c7ff14ba75";
124 #else
125 return "7978c7b3d643a5f0ac0f03ce759c55fe";
126 #endif
127 }
128 #if BUILDFLAG(IS_APPLE)
129 return "41debc60cf2a8f74c710ec6082d77b18";
130 #else
131 return "0832157462ea70fbbf053e14b1d6457f";
132 #endif
133 }();
134
135 ASSERT_TRUE(OpenDocument("bug_889099.pdf"));
136 ScopedEmbedderTestPage page = LoadScopedPage(0);
137 ASSERT_TRUE(page);
138
139 // The original document has a malformed media box; the height is -400.
140 ScopedFPDFBitmap bitmap = RenderLoadedPageWithFlags(page.get(), FPDF_ANNOT);
141 CompareBitmap(bitmap.get(), 300, 400, page_checksum);
142
143 EXPECT_EQ(FLATTEN_SUCCESS, FPDFPage_Flatten(page.get(), FLAT_PRINT));
144 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
145
146
147 VerifySavedDocument(300, 400, flattened_page_checksum);
148 }
149
TEST_F(FPDFFlattenEmbedderTest,Bug890322)150 TEST_F(FPDFFlattenEmbedderTest, Bug890322) {
151 const char* checksum = []() {
152 if (CFX_DefaultRenderDevice::UseSkiaRenderer()) {
153 return "793689536cf64fe792c2f241888c0cf3";
154 }
155 return "6c674642154408e877d88c6c082d67e9";
156 }();
157 ASSERT_TRUE(OpenDocument("bug_890322.pdf"));
158 ScopedEmbedderTestPage page = LoadScopedPage(0);
159 ASSERT_TRUE(page);
160
161 ScopedFPDFBitmap bitmap = RenderLoadedPageWithFlags(page.get(), FPDF_ANNOT);
162 CompareBitmap(bitmap.get(), 200, 200, checksum);
163
164 EXPECT_EQ(FLATTEN_SUCCESS, FPDFPage_Flatten(page.get(), FLAT_PRINT));
165 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
166
167
168 VerifySavedDocument(200, 200, checksum);
169 }
170
TEST_F(FPDFFlattenEmbedderTest,Bug896366)171 TEST_F(FPDFFlattenEmbedderTest, Bug896366) {
172 const char* checksum = []() {
173 if (CFX_DefaultRenderDevice::UseSkiaRenderer()) {
174 return "c3cccfadc4c5249e6aa0675e511fa4c3";
175 }
176 return "f71ab085c52c8445ae785eca3ec858b1";
177 }();
178 ASSERT_TRUE(OpenDocument("bug_896366.pdf"));
179 ScopedEmbedderTestPage page = LoadScopedPage(0);
180 ASSERT_TRUE(page);
181
182 ScopedFPDFBitmap bitmap = RenderLoadedPageWithFlags(page.get(), FPDF_ANNOT);
183 CompareBitmap(bitmap.get(), 612, 792, checksum);
184
185 EXPECT_EQ(FLATTEN_SUCCESS, FPDFPage_Flatten(page.get(), FLAT_PRINT));
186 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
187
188
189 VerifySavedDocument(612, 792, checksum);
190 }
191