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 <string>
6
7 #include "core/fxcrt/fx_string.h"
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/embedder_test_constants.h"
15 #include "testing/gmock/include/gmock/gmock-matchers.h"
16 #include "testing/gtest/include/gtest/gtest.h"
17
18 using testing::HasSubstr;
19 using testing::Not;
20 using testing::StartsWith;
21
22 class FPDFSaveEmbedderTest : public EmbedderTest {};
23
TEST_F(FPDFSaveEmbedderTest,SaveSimpleDoc)24 TEST_F(FPDFSaveEmbedderTest, SaveSimpleDoc) {
25 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
26 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
27 EXPECT_THAT(GetString(), StartsWith("%PDF-1.7\r\n"));
28 EXPECT_EQ(805u, GetString().size());
29 }
30
TEST_F(FPDFSaveEmbedderTest,SaveSimpleDocWithVersion)31 TEST_F(FPDFSaveEmbedderTest, SaveSimpleDocWithVersion) {
32 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
33 EXPECT_TRUE(FPDF_SaveWithVersion(document(), this, 0, 14));
34 EXPECT_THAT(GetString(), StartsWith("%PDF-1.4\r\n"));
35 EXPECT_EQ(805u, GetString().size());
36 }
37
TEST_F(FPDFSaveEmbedderTest,SaveSimpleDocWithBadVersion)38 TEST_F(FPDFSaveEmbedderTest, SaveSimpleDocWithBadVersion) {
39 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
40 EXPECT_TRUE(FPDF_SaveWithVersion(document(), this, 0, -1));
41 EXPECT_THAT(GetString(), StartsWith("%PDF-1.7\r\n"));
42
43 ClearString();
44 EXPECT_TRUE(FPDF_SaveWithVersion(document(), this, 0, 0));
45 EXPECT_THAT(GetString(), StartsWith("%PDF-1.7\r\n"));
46
47 ClearString();
48 EXPECT_TRUE(FPDF_SaveWithVersion(document(), this, 0, 18));
49 EXPECT_THAT(GetString(), StartsWith("%PDF-1.7\r\n"));
50 }
51
TEST_F(FPDFSaveEmbedderTest,SaveSimpleDocIncremental)52 TEST_F(FPDFSaveEmbedderTest, SaveSimpleDocIncremental) {
53 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
54 EXPECT_TRUE(FPDF_SaveWithVersion(document(), this, FPDF_INCREMENTAL, 14));
55 // Version gets taken as-is from input document.
56 EXPECT_THAT(GetString(), StartsWith("%PDF-1.7\n%\xa0\xf2\xa4\xf4"));
57 // Additional output produced vs. non incremental.
58 EXPECT_EQ(985u, GetString().size());
59 }
60
TEST_F(FPDFSaveEmbedderTest,SaveSimpleDocNoIncremental)61 TEST_F(FPDFSaveEmbedderTest, SaveSimpleDocNoIncremental) {
62 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
63 EXPECT_TRUE(FPDF_SaveWithVersion(document(), this, FPDF_NO_INCREMENTAL, 14));
64 EXPECT_THAT(GetString(), StartsWith("%PDF-1.4\r\n"));
65 EXPECT_EQ(805u, GetString().size());
66 }
67
TEST_F(FPDFSaveEmbedderTest,SaveSimpleDocRemoveSecurity)68 TEST_F(FPDFSaveEmbedderTest, SaveSimpleDocRemoveSecurity) {
69 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
70 EXPECT_TRUE(FPDF_SaveWithVersion(document(), this, FPDF_REMOVE_SECURITY, 14));
71 EXPECT_THAT(GetString(), StartsWith("%PDF-1.4\r\n"));
72 EXPECT_EQ(805u, GetString().size());
73 }
74
TEST_F(FPDFSaveEmbedderTest,SaveSimpleDocBadFlags)75 TEST_F(FPDFSaveEmbedderTest, SaveSimpleDocBadFlags) {
76 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
77 EXPECT_TRUE(FPDF_SaveWithVersion(document(), this, 999999, 14));
78 EXPECT_THAT(GetString(), StartsWith("%PDF-1.4\r\n"));
79 EXPECT_EQ(805u, GetString().size());
80 }
81
TEST_F(FPDFSaveEmbedderTest,SaveCopiedDoc)82 TEST_F(FPDFSaveEmbedderTest, SaveCopiedDoc) {
83 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
84
85 FPDF_PAGE page = LoadPage(0);
86 EXPECT_TRUE(page);
87
88 FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument();
89 EXPECT_TRUE(output_doc);
90 EXPECT_TRUE(FPDF_ImportPages(output_doc, document(), "1", 0));
91 EXPECT_TRUE(FPDF_SaveAsCopy(output_doc, this, 0));
92 FPDF_CloseDocument(output_doc);
93
94 UnloadPage(page);
95 }
96
TEST_F(FPDFSaveEmbedderTest,SaveLinearizedDoc)97 TEST_F(FPDFSaveEmbedderTest, SaveLinearizedDoc) {
98 const int kPageCount = 3;
99 std::string original_md5[kPageCount];
100
101 ASSERT_TRUE(OpenDocument("linearized.pdf"));
102 for (int i = 0; i < kPageCount; ++i) {
103 FPDF_PAGE page = LoadPage(i);
104 ASSERT_TRUE(page);
105 ScopedFPDFBitmap bitmap = RenderLoadedPage(page);
106 EXPECT_EQ(612, FPDFBitmap_GetWidth(bitmap.get()));
107 EXPECT_EQ(792, FPDFBitmap_GetHeight(bitmap.get()));
108 original_md5[i] = HashBitmap(bitmap.get());
109 UnloadPage(page);
110 }
111
112 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
113 EXPECT_THAT(GetString(), StartsWith("%PDF-1.6\r\n"));
114 EXPECT_THAT(GetString(), HasSubstr("/Root "));
115 EXPECT_THAT(GetString(), HasSubstr("/Info "));
116 EXPECT_THAT(GetString(), HasSubstr("/Size 37"));
117 EXPECT_THAT(GetString(), HasSubstr("35 0 obj"));
118 EXPECT_THAT(GetString(), HasSubstr("36 0 obj"));
119 EXPECT_THAT(GetString(), Not(HasSubstr("37 0 obj")));
120 EXPECT_THAT(GetString(), Not(HasSubstr("38 0 obj")));
121 EXPECT_EQ(7908u, GetString().size());
122
123 // Make sure new document renders the same as the old one.
124 ASSERT_TRUE(OpenSavedDocument());
125 for (int i = 0; i < kPageCount; ++i) {
126 FPDF_PAGE page = LoadSavedPage(i);
127 ASSERT_TRUE(page);
128 ScopedFPDFBitmap bitmap = RenderSavedPage(page);
129 EXPECT_EQ(original_md5[i], HashBitmap(bitmap.get()));
130 CloseSavedPage(page);
131 }
132 CloseSavedDocument();
133 }
134
TEST_F(FPDFSaveEmbedderTest,Bug1409)135 TEST_F(FPDFSaveEmbedderTest, Bug1409) {
136 ASSERT_TRUE(OpenDocument("jpx_lzw.pdf"));
137 FPDF_PAGE page = LoadPage(0);
138 ASSERT_TRUE(page);
139 while (FPDFPage_CountObjects(page) > 0) {
140 ScopedFPDFPageObject object(FPDFPage_GetObject(page, 0));
141 ASSERT_TRUE(object);
142 ASSERT_TRUE(FPDFPage_RemoveObject(page, object.get()));
143 }
144 ASSERT_TRUE(FPDFPage_GenerateContent(page));
145 UnloadPage(page);
146
147 ASSERT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
148
149 // The new document should render as empty.
150 ASSERT_TRUE(OpenSavedDocument());
151 FPDF_PAGE saved_page = LoadSavedPage(0);
152 ASSERT_TRUE(saved_page);
153 ScopedFPDFBitmap bitmap = RenderSavedPage(saved_page);
154 EXPECT_EQ(pdfium::kBlankPage612By792Checksum, HashBitmap(bitmap.get()));
155 CloseSavedPage(saved_page);
156 CloseSavedDocument();
157
158 EXPECT_THAT(GetString(), StartsWith("%PDF-1.7\r\n"));
159 EXPECT_THAT(GetString(), HasSubstr("/Root "));
160 EXPECT_THAT(GetString(), Not(HasSubstr("/Image")));
161 EXPECT_LT(GetString().size(), 600u);
162 }
163
164 #ifdef PDF_ENABLE_XFA
TEST_F(FPDFSaveEmbedderTest,SaveXFADoc)165 TEST_F(FPDFSaveEmbedderTest, SaveXFADoc) {
166 ASSERT_TRUE(OpenDocument("simple_xfa.pdf"));
167 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
168 EXPECT_THAT(GetString(), StartsWith("%PDF-1.7\r\n"));
169 ASSERT_TRUE(OpenSavedDocument());
170 // TODO(tsepez): check for XFA forms in document
171 CloseSavedDocument();
172 }
173 #endif // PDF_ENABLE_XFA
174
TEST_F(FPDFSaveEmbedderTest,BUG_342)175 TEST_F(FPDFSaveEmbedderTest, BUG_342) {
176 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
177 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
178 EXPECT_THAT(GetString(), HasSubstr("0000000000 65535 f\r\n"));
179 EXPECT_THAT(GetString(), Not(HasSubstr("0000000000 65536 f\r\n")));
180 }
181
TEST_F(FPDFSaveEmbedderTest,BUG_905142)182 TEST_F(FPDFSaveEmbedderTest, BUG_905142) {
183 ASSERT_TRUE(OpenDocument("bug_905142.pdf"));
184 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
185 EXPECT_THAT(GetString(), HasSubstr("/Length 0"));
186 }
187
188 // Should not trigger a DCHECK() failure in CFX_FileBufferArchive.
189 // Fails because the PDF is malformed.
TEST_F(FPDFSaveEmbedderTest,Bug1328389)190 TEST_F(FPDFSaveEmbedderTest, Bug1328389) {
191 ASSERT_TRUE(OpenDocument("bug_1328389.pdf"));
192 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
193 EXPECT_THAT(GetString(), HasSubstr("/Foo/"));
194 }
195