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 #include <string>
5
6 #include "public/fpdf_edit.h"
7 #include "public/fpdf_ppo.h"
8 #include "public/fpdf_save.h"
9 #include "public/fpdfview.h"
10 #include "testing/embedder_test.h"
11 #include "testing/gtest/include/gtest/gtest.h"
12 #include "testing/test_support.h"
13
14 namespace {
15
16 class FPDFPPOEmbeddertest : public EmbedderTest {};
17
FakeBlockWriter(FPDF_FILEWRITE * pThis,const void * pData,unsigned long size)18 int FakeBlockWriter(FPDF_FILEWRITE* pThis,
19 const void* pData,
20 unsigned long size) {
21 return size;
22 }
23
24 } // namespace
25
TEST_F(FPDFPPOEmbeddertest,NoViewerPreferences)26 TEST_F(FPDFPPOEmbeddertest, NoViewerPreferences) {
27 EXPECT_TRUE(OpenDocument("hello_world.pdf"));
28
29 FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
30 EXPECT_TRUE(output_doc);
31 EXPECT_FALSE(FPDF_CopyViewerPreferences(output_doc, document()));
32 FPDF_CloseDocument(output_doc);
33 }
34
TEST_F(FPDFPPOEmbeddertest,ViewerPreferences)35 TEST_F(FPDFPPOEmbeddertest, ViewerPreferences) {
36 EXPECT_TRUE(OpenDocument("viewer_ref.pdf"));
37
38 FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
39 EXPECT_TRUE(output_doc);
40 EXPECT_TRUE(FPDF_CopyViewerPreferences(output_doc, document()));
41 FPDF_CloseDocument(output_doc);
42 }
43
TEST_F(FPDFPPOEmbeddertest,ImportPages)44 TEST_F(FPDFPPOEmbeddertest, ImportPages) {
45 ASSERT_TRUE(OpenDocument("viewer_ref.pdf"));
46
47 FPDF_PAGE page = LoadPage(0);
48 EXPECT_TRUE(page);
49
50 FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
51 ASSERT_TRUE(output_doc);
52 EXPECT_TRUE(FPDF_CopyViewerPreferences(output_doc, document()));
53 EXPECT_TRUE(FPDF_ImportPages(output_doc, document(), "1", 0));
54 EXPECT_EQ(1, FPDF_GetPageCount(output_doc));
55 FPDF_CloseDocument(output_doc);
56
57 UnloadPage(page);
58 }
59
TEST_F(FPDFPPOEmbeddertest,BadRepeatViewerPref)60 TEST_F(FPDFPPOEmbeddertest, BadRepeatViewerPref) {
61 ASSERT_TRUE(OpenDocument("repeat_viewer_ref.pdf"));
62
63 FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
64 EXPECT_TRUE(output_doc);
65 EXPECT_TRUE(FPDF_CopyViewerPreferences(output_doc, document()));
66
67 FPDF_FILEWRITE writer;
68 writer.version = 1;
69 writer.WriteBlock = FakeBlockWriter;
70
71 EXPECT_TRUE(FPDF_SaveAsCopy(output_doc, &writer, 0));
72 FPDF_CloseDocument(output_doc);
73 }
74
TEST_F(FPDFPPOEmbeddertest,BadCircularViewerPref)75 TEST_F(FPDFPPOEmbeddertest, BadCircularViewerPref) {
76 ASSERT_TRUE(OpenDocument("circular_viewer_ref.pdf"));
77
78 FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
79 EXPECT_TRUE(output_doc);
80 EXPECT_TRUE(FPDF_CopyViewerPreferences(output_doc, document()));
81
82 FPDF_FILEWRITE writer;
83 writer.version = 1;
84 writer.WriteBlock = FakeBlockWriter;
85
86 EXPECT_TRUE(FPDF_SaveAsCopy(output_doc, &writer, 0));
87 FPDF_CloseDocument(output_doc);
88 }
89
TEST_F(FPDFPPOEmbeddertest,BadRanges)90 TEST_F(FPDFPPOEmbeddertest, BadRanges) {
91 EXPECT_TRUE(OpenDocument("viewer_ref.pdf"));
92
93 FPDF_PAGE page = LoadPage(0);
94 EXPECT_TRUE(page);
95
96 FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
97 EXPECT_TRUE(output_doc);
98 EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), "clams", 0));
99 EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), "0", 0));
100 EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), "42", 0));
101 EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), "1,2", 0));
102 EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), "1-2", 0));
103 EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), ",1", 0));
104 EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), "1,", 0));
105 EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), "1-", 0));
106 EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), "-1", 0));
107 EXPECT_FALSE(FPDF_ImportPages(output_doc, document(), "-,0,,,1-", 0));
108 FPDF_CloseDocument(output_doc);
109
110 UnloadPage(page);
111 }
112
TEST_F(FPDFPPOEmbeddertest,GoodRanges)113 TEST_F(FPDFPPOEmbeddertest, GoodRanges) {
114 EXPECT_TRUE(OpenDocument("viewer_ref.pdf"));
115
116 FPDF_PAGE page = LoadPage(0);
117 EXPECT_TRUE(page);
118
119 FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
120 EXPECT_TRUE(output_doc);
121 EXPECT_TRUE(FPDF_CopyViewerPreferences(output_doc, document()));
122 EXPECT_TRUE(FPDF_ImportPages(output_doc, document(), "1,1,1,1", 0));
123 EXPECT_TRUE(FPDF_ImportPages(output_doc, document(), "1-1", 0));
124 EXPECT_EQ(5, FPDF_GetPageCount(output_doc));
125 FPDF_CloseDocument(output_doc);
126
127 UnloadPage(page);
128 }
129
TEST_F(FPDFPPOEmbeddertest,BUG_664284)130 TEST_F(FPDFPPOEmbeddertest, BUG_664284) {
131 EXPECT_TRUE(OpenDocument("bug_664284.pdf"));
132
133 FPDF_PAGE page = LoadPage(0);
134 ASSERT_NE(nullptr, page);
135
136 FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
137 EXPECT_TRUE(output_doc);
138 EXPECT_TRUE(FPDF_ImportPages(output_doc, document(), "1", 0));
139 FPDF_CloseDocument(output_doc);
140
141 UnloadPage(page);
142 }
143
TEST_F(FPDFPPOEmbeddertest,BUG_750568)144 TEST_F(FPDFPPOEmbeddertest, BUG_750568) {
145 const char* const kHashes[] = {
146 "64ad08132a1c5a166768298c8a578f57", "83b83e2f6bc80707d0a917c7634140b9",
147 "913cd3723a451e4e46fbc2c05702d1ee", "81fb7cfd4860f855eb468f73dfeb6d60"};
148
149 ASSERT_TRUE(OpenDocument("bug_750568.pdf"));
150 ASSERT_EQ(4, FPDF_GetPageCount(document()));
151
152 for (size_t i = 0; i < 4; ++i) {
153 FPDF_PAGE page = LoadPage(i);
154 ASSERT_NE(nullptr, page);
155
156 FPDF_BITMAP bitmap = RenderPage(page);
157 ASSERT_EQ(200, FPDFBitmap_GetWidth(bitmap));
158 ASSERT_EQ(200, FPDFBitmap_GetHeight(bitmap));
159 ASSERT_EQ(800, FPDFBitmap_GetStride(bitmap));
160
161 std::string digest = HashBitmap(bitmap);
162 FPDFBitmap_Destroy(bitmap);
163 UnloadPage(page);
164 EXPECT_EQ(kHashes[i], digest);
165 }
166
167 FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
168 ASSERT_TRUE(output_doc);
169 EXPECT_TRUE(FPDF_ImportPages(output_doc, document(), "1,2,3,4", 0));
170 ASSERT_EQ(4, FPDF_GetPageCount(output_doc));
171 for (size_t i = 0; i < 4; ++i) {
172 FPDF_PAGE page = FPDF_LoadPage(output_doc, i);
173 ASSERT_NE(nullptr, page);
174
175 FPDF_BITMAP bitmap = RenderPage(page);
176 ASSERT_EQ(200, FPDFBitmap_GetWidth(bitmap));
177 ASSERT_EQ(200, FPDFBitmap_GetHeight(bitmap));
178 ASSERT_EQ(800, FPDFBitmap_GetStride(bitmap));
179
180 std::string digest = HashBitmap(bitmap);
181 FPDFBitmap_Destroy(bitmap);
182 FPDF_ClosePage(page);
183 EXPECT_EQ(kHashes[i], digest);
184 }
185 FPDF_CloseDocument(output_doc);
186 }
187
TEST_F(FPDFPPOEmbeddertest,ImportWithZeroLengthStream)188 TEST_F(FPDFPPOEmbeddertest, ImportWithZeroLengthStream) {
189 EXPECT_TRUE(OpenDocument("zero_length_stream.pdf"));
190 FPDF_PAGE page = LoadPage(0);
191 ASSERT_NE(nullptr, page);
192
193 FPDF_BITMAP bitmap = RenderPage(page);
194 ASSERT_EQ(200, FPDFBitmap_GetWidth(bitmap));
195 ASSERT_EQ(200, FPDFBitmap_GetHeight(bitmap));
196 ASSERT_EQ(800, FPDFBitmap_GetStride(bitmap));
197
198 std::string digest = HashBitmap(bitmap);
199 FPDFBitmap_Destroy(bitmap);
200 FPDF_ClosePage(page);
201
202 FPDF_DOCUMENT new_doc = FPDF_CreateNewDocument();
203 EXPECT_TRUE(new_doc);
204 EXPECT_TRUE(FPDF_ImportPages(new_doc, document(), "1", 0));
205
206 EXPECT_EQ(1, FPDF_GetPageCount(new_doc));
207 FPDF_PAGE new_page = FPDF_LoadPage(new_doc, 0);
208 ASSERT_NE(nullptr, new_page);
209 FPDF_BITMAP new_bitmap = RenderPage(new_page);
210 ASSERT_EQ(200, FPDFBitmap_GetWidth(new_bitmap));
211 ASSERT_EQ(200, FPDFBitmap_GetHeight(new_bitmap));
212 ASSERT_EQ(800, FPDFBitmap_GetStride(new_bitmap));
213
214 std::string new_digest = HashBitmap(new_bitmap);
215 FPDFBitmap_Destroy(new_bitmap);
216 FPDF_ClosePage(new_page);
217 FPDF_CloseDocument(new_doc);
218
219 EXPECT_EQ(digest, new_digest);
220 }
221