1 // Copyright 2016 PDFium Authors. All rights reserved.
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 <memory>
6 #include <string>
7
8 #include "public/cpp/fpdf_scopers.h"
9 #include "public/fpdf_edit.h"
10 #include "public/fpdf_ppo.h"
11 #include "public/fpdf_save.h"
12 #include "public/fpdfview.h"
13 #include "testing/embedder_test.h"
14 #include "testing/gtest/include/gtest/gtest.h"
15
16 namespace {
17
18 class FPDFPPOEmbedderTest : public EmbedderTest {};
19
FakeBlockWriter(FPDF_FILEWRITE * pThis,const void * pData,unsigned long size)20 int FakeBlockWriter(FPDF_FILEWRITE* pThis,
21 const void* pData,
22 unsigned long size) {
23 return size;
24 }
25
26 } // namespace
27
TEST_F(FPDFPPOEmbedderTest,NoViewerPreferences)28 TEST_F(FPDFPPOEmbedderTest, NoViewerPreferences) {
29 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
30
31 FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
32 EXPECT_TRUE(output_doc);
33 EXPECT_FALSE(FPDF_CopyViewerPreferences(output_doc, document()));
34 FPDF_CloseDocument(output_doc);
35 }
36
TEST_F(FPDFPPOEmbedderTest,ViewerPreferences)37 TEST_F(FPDFPPOEmbedderTest, ViewerPreferences) {
38 ASSERT_TRUE(OpenDocument("viewer_ref.pdf"));
39
40 FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
41 EXPECT_TRUE(output_doc);
42 EXPECT_TRUE(FPDF_CopyViewerPreferences(output_doc, document()));
43 FPDF_CloseDocument(output_doc);
44 }
45
TEST_F(FPDFPPOEmbedderTest,ImportPages)46 TEST_F(FPDFPPOEmbedderTest, ImportPages) {
47 ASSERT_TRUE(OpenDocument("viewer_ref.pdf"));
48
49 FPDF_PAGE page = LoadPage(0);
50 EXPECT_TRUE(page);
51
52 FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
53 ASSERT_TRUE(output_doc);
54 EXPECT_TRUE(FPDF_CopyViewerPreferences(output_doc, document()));
55 EXPECT_TRUE(FPDF_ImportPages(output_doc, document(), "1", 0));
56 EXPECT_EQ(1, FPDF_GetPageCount(output_doc));
57 FPDF_CloseDocument(output_doc);
58
59 UnloadPage(page);
60 }
61
TEST_F(FPDFPPOEmbedderTest,ImportNPages)62 TEST_F(FPDFPPOEmbedderTest, ImportNPages) {
63 ASSERT_TRUE(OpenDocument("rectangles_multi_pages.pdf"));
64
65 ScopedFPDFDocument output_doc_2up(
66 FPDF_ImportNPagesToOne(document(), 612, 792, 2, 1));
67 ASSERT_TRUE(output_doc_2up);
68 EXPECT_EQ(3, FPDF_GetPageCount(output_doc_2up.get()));
69 ScopedFPDFDocument output_doc_5up(
70 FPDF_ImportNPagesToOne(document(), 612, 792, 5, 1));
71 ASSERT_TRUE(output_doc_5up);
72 EXPECT_EQ(1, FPDF_GetPageCount(output_doc_5up.get()));
73 ScopedFPDFDocument output_doc_8up(
74 FPDF_ImportNPagesToOne(document(), 792, 612, 8, 1));
75 ASSERT_TRUE(output_doc_8up);
76 EXPECT_EQ(1, FPDF_GetPageCount(output_doc_8up.get()));
77 ScopedFPDFDocument output_doc_128up(
78 FPDF_ImportNPagesToOne(document(), 792, 612, 128, 1));
79 ASSERT_TRUE(output_doc_128up);
80 EXPECT_EQ(1, FPDF_GetPageCount(output_doc_128up.get()));
81 }
82
TEST_F(FPDFPPOEmbedderTest,BadNupParams)83 TEST_F(FPDFPPOEmbedderTest, BadNupParams) {
84 ASSERT_TRUE(OpenDocument("rectangles_multi_pages.pdf"));
85
86 FPDF_DOCUMENT output_doc_zero_row =
87 FPDF_ImportNPagesToOne(document(), 612, 792, 0, 3);
88 ASSERT_FALSE(output_doc_zero_row);
89 FPDF_DOCUMENT output_doc_zero_col =
90 FPDF_ImportNPagesToOne(document(), 612, 792, 2, 0);
91 ASSERT_FALSE(output_doc_zero_col);
92 FPDF_DOCUMENT output_doc_zero_width =
93 FPDF_ImportNPagesToOne(document(), 0, 792, 2, 1);
94 ASSERT_FALSE(output_doc_zero_width);
95 FPDF_DOCUMENT output_doc_zero_height =
96 FPDF_ImportNPagesToOne(document(), 612, 0, 7, 1);
97 ASSERT_FALSE(output_doc_zero_height);
98 }
99
100 // TODO(Xlou): Add more tests to check output doc content of
101 // FPDF_ImportNPagesToOne()
TEST_F(FPDFPPOEmbedderTest,NupRenderImage)102 TEST_F(FPDFPPOEmbedderTest, NupRenderImage) {
103 ASSERT_TRUE(OpenDocument("rectangles_multi_pages.pdf"));
104 const int kPageCount = 2;
105 #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
106 static constexpr const char* kExpectedMD5s[kPageCount] = {
107 "bf8fa88dc85a9897931273168e8e1a30", "4fa6a7507e9f3ef4f28719a7d656c3a5"};
108 #else
109 static constexpr const char* kExpectedMD5s[kPageCount] = {
110 "4d225b961da0f1bced7c83273e64c9b6", "fb18142190d770cfbc329d2b071aee4d"};
111 #endif
112 ScopedFPDFDocument output_doc_3up(
113 FPDF_ImportNPagesToOne(document(), 792, 612, 3, 1));
114 ASSERT_TRUE(output_doc_3up);
115 ASSERT_EQ(kPageCount, FPDF_GetPageCount(output_doc_3up.get()));
116 for (int i = 0; i < kPageCount; ++i) {
117 ScopedFPDFPage page(FPDF_LoadPage(output_doc_3up.get(), i));
118 ASSERT_TRUE(page);
119 ScopedFPDFBitmap bitmap = RenderPage(page.get());
120 EXPECT_EQ(792, FPDFBitmap_GetWidth(bitmap.get()));
121 EXPECT_EQ(612, FPDFBitmap_GetHeight(bitmap.get()));
122 EXPECT_EQ(kExpectedMD5s[i], HashBitmap(bitmap.get()));
123 }
124 }
125
TEST_F(FPDFPPOEmbedderTest,BUG_925981)126 TEST_F(FPDFPPOEmbedderTest, BUG_925981) {
127 ASSERT_TRUE(OpenDocument("bug_925981.pdf"));
128 ScopedFPDFDocument output_doc_2up(
129 FPDF_ImportNPagesToOne(document(), 612, 792, 2, 1));
130 EXPECT_EQ(1, FPDF_GetPageCount(output_doc_2up.get()));
131 }
132
TEST_F(FPDFPPOEmbedderTest,BadRepeatViewerPref)133 TEST_F(FPDFPPOEmbedderTest, BadRepeatViewerPref) {
134 ASSERT_TRUE(OpenDocument("repeat_viewer_ref.pdf"));
135
136 FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
137 EXPECT_TRUE(output_doc);
138 EXPECT_TRUE(FPDF_CopyViewerPreferences(output_doc, document()));
139
140 FPDF_FILEWRITE writer;
141 writer.version = 1;
142 writer.WriteBlock = FakeBlockWriter;
143
144 EXPECT_TRUE(FPDF_SaveAsCopy(output_doc, &writer, 0));
145 FPDF_CloseDocument(output_doc);
146 }
147
TEST_F(FPDFPPOEmbedderTest,BadCircularViewerPref)148 TEST_F(FPDFPPOEmbedderTest, BadCircularViewerPref) {
149 ASSERT_TRUE(OpenDocument("circular_viewer_ref.pdf"));
150
151 FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
152 EXPECT_TRUE(output_doc);
153 EXPECT_TRUE(FPDF_CopyViewerPreferences(output_doc, document()));
154
155 FPDF_FILEWRITE writer;
156 writer.version = 1;
157 writer.WriteBlock = FakeBlockWriter;
158
159 EXPECT_TRUE(FPDF_SaveAsCopy(output_doc, &writer, 0));
160 FPDF_CloseDocument(output_doc);
161 }
162
TEST_F(FPDFPPOEmbedderTest,BadRanges)163 TEST_F(FPDFPPOEmbedderTest, BadRanges) {
164 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
165
166 FPDF_PAGE page = LoadPage(0);
167 EXPECT_TRUE(page);
168
169 FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
170 EXPECT_TRUE(output_doc);
171 EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), "clams", 0));
172 EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), "0", 0));
173 EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), "42", 0));
174 EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), "1,2", 0));
175 EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), "1-2", 0));
176 EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), ",1", 0));
177 EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), "1,", 0));
178 EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), "1-", 0));
179 EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), "-1", 0));
180 EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), "-,0,,,1-", 0));
181 FPDF_CloseDocument(output_doc);
182
183 UnloadPage(page);
184 }
185
TEST_F(FPDFPPOEmbedderTest,GoodRanges)186 TEST_F(FPDFPPOEmbedderTest, GoodRanges) {
187 ASSERT_TRUE(OpenDocument("viewer_ref.pdf"));
188
189 FPDF_PAGE page = LoadPage(0);
190 EXPECT_TRUE(page);
191
192 FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
193 EXPECT_TRUE(output_doc);
194 EXPECT_TRUE(FPDF_CopyViewerPreferences(output_doc, document()));
195 EXPECT_TRUE(FPDF_ImportPages(output_doc, document(), "1,1,1,1", 0));
196 EXPECT_EQ(4, FPDF_GetPageCount(output_doc));
197 EXPECT_TRUE(FPDF_ImportPages(output_doc, document(), "1-1", 0));
198 EXPECT_EQ(5, FPDF_GetPageCount(output_doc));
199 EXPECT_TRUE(FPDF_ImportPages(output_doc, document(), "5-5", 0));
200 EXPECT_EQ(6, FPDF_GetPageCount(output_doc));
201 EXPECT_TRUE(FPDF_ImportPages(output_doc, document(), "2-4", 0));
202 EXPECT_EQ(9, FPDF_GetPageCount(output_doc));
203 FPDF_CloseDocument(output_doc);
204
205 UnloadPage(page);
206 }
207
TEST_F(FPDFPPOEmbedderTest,BUG_664284)208 TEST_F(FPDFPPOEmbedderTest, BUG_664284) {
209 ASSERT_TRUE(OpenDocument("bug_664284.pdf"));
210
211 FPDF_PAGE page = LoadPage(0);
212 ASSERT_NE(nullptr, page);
213
214 FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
215 EXPECT_TRUE(output_doc);
216 EXPECT_TRUE(FPDF_ImportPages(output_doc, document(), "1", 0));
217 FPDF_CloseDocument(output_doc);
218
219 UnloadPage(page);
220 }
221
TEST_F(FPDFPPOEmbedderTest,BUG_750568)222 TEST_F(FPDFPPOEmbedderTest, BUG_750568) {
223 #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
224 const char* const kHashes[] = {
225 "eaa139e944eafb43d31e8742a0e158de", "226485e9d4fa6a67dfe0a88723f12060",
226 "c5601a3492ae5dcc5dd25140fc463bfe", "1f60055b54de4fac8a59c65e90da156e"};
227 #else
228 const char* const kHashes[] = {
229 "64ad08132a1c5a166768298c8a578f57", "83b83e2f6bc80707d0a917c7634140b9",
230 "913cd3723a451e4e46fbc2c05702d1ee", "81fb7cfd4860f855eb468f73dfeb6d60"};
231 #endif
232
233 ASSERT_TRUE(OpenDocument("bug_750568.pdf"));
234 ASSERT_EQ(4, FPDF_GetPageCount(document()));
235
236 for (size_t i = 0; i < 4; ++i) {
237 FPDF_PAGE page = LoadPage(i);
238 ASSERT_TRUE(page);
239
240 ScopedFPDFBitmap bitmap = RenderLoadedPage(page);
241 ASSERT_EQ(200, FPDFBitmap_GetWidth(bitmap.get()));
242 ASSERT_EQ(200, FPDFBitmap_GetHeight(bitmap.get()));
243 ASSERT_EQ(800, FPDFBitmap_GetStride(bitmap.get()));
244
245 EXPECT_EQ(kHashes[i], HashBitmap(bitmap.get()));
246 UnloadPage(page);
247 }
248
249 FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
250 ASSERT_TRUE(output_doc);
251 EXPECT_TRUE(FPDF_ImportPages(output_doc, document(), "1,2,3,4", 0));
252 ASSERT_EQ(4, FPDF_GetPageCount(output_doc));
253 for (size_t i = 0; i < 4; ++i) {
254 FPDF_PAGE page = FPDF_LoadPage(output_doc, i);
255 ASSERT_TRUE(page);
256
257 ScopedFPDFBitmap bitmap = RenderPage(page);
258 ASSERT_EQ(200, FPDFBitmap_GetWidth(bitmap.get()));
259 ASSERT_EQ(200, FPDFBitmap_GetHeight(bitmap.get()));
260 ASSERT_EQ(800, FPDFBitmap_GetStride(bitmap.get()));
261
262 EXPECT_EQ(kHashes[i], HashBitmap(bitmap.get()));
263 FPDF_ClosePage(page);
264 }
265 FPDF_CloseDocument(output_doc);
266 }
267
TEST_F(FPDFPPOEmbedderTest,ImportWithZeroLengthStream)268 TEST_F(FPDFPPOEmbedderTest, ImportWithZeroLengthStream) {
269 ASSERT_TRUE(OpenDocument("zero_length_stream.pdf"));
270 FPDF_PAGE page = LoadPage(0);
271 ASSERT_TRUE(page);
272
273 ScopedFPDFBitmap bitmap = RenderLoadedPage(page);
274 ASSERT_EQ(200, FPDFBitmap_GetWidth(bitmap.get()));
275 ASSERT_EQ(200, FPDFBitmap_GetHeight(bitmap.get()));
276 ASSERT_EQ(800, FPDFBitmap_GetStride(bitmap.get()));
277
278 std::string digest = HashBitmap(bitmap.get());
279 UnloadPage(page);
280
281 FPDF_DOCUMENT new_doc = FPDF_CreateNewDocument();
282 EXPECT_TRUE(new_doc);
283 EXPECT_TRUE(FPDF_ImportPages(new_doc, document(), "1", 0));
284
285 EXPECT_EQ(1, FPDF_GetPageCount(new_doc));
286 FPDF_PAGE new_page = FPDF_LoadPage(new_doc, 0);
287 ASSERT_NE(nullptr, new_page);
288 ScopedFPDFBitmap new_bitmap = RenderPage(new_page);
289 ASSERT_EQ(200, FPDFBitmap_GetWidth(new_bitmap.get()));
290 ASSERT_EQ(200, FPDFBitmap_GetHeight(new_bitmap.get()));
291 ASSERT_EQ(800, FPDFBitmap_GetStride(new_bitmap.get()));
292
293 EXPECT_EQ(digest, HashBitmap(new_bitmap.get()));
294 FPDF_ClosePage(new_page);
295 FPDF_CloseDocument(new_doc);
296 }
297