Tom Sepez | 9eb811f | 2016-01-05 14:48:31 -0800 | [diff] [blame] | 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 | |
Lei Zhang | a98e366 | 2018-02-07 20:28:35 +0000 | [diff] [blame] | 5 | #include <memory> |
Henrique Nakashima | 9fa5036 | 2017-11-10 22:40:44 +0000 | [diff] [blame] | 6 | #include <string> |
Tom Sepez | 9eb811f | 2016-01-05 14:48:31 -0800 | [diff] [blame] | 7 | |
dsinclair | a52ab74 | 2016-09-29 13:59:29 -0700 | [diff] [blame] | 8 | #include "core/fxcrt/fx_string.h" |
Tom Sepez | e08d2b1 | 2018-04-25 18:49:32 +0000 | [diff] [blame] | 9 | #include "public/cpp/fpdf_scopers.h" |
tsepez | 6e1d603 | 2016-11-11 17:55:40 -0800 | [diff] [blame] | 10 | #include "public/fpdf_edit.h" |
| 11 | #include "public/fpdf_ppo.h" |
Lei Zhang | a98e366 | 2018-02-07 20:28:35 +0000 | [diff] [blame] | 12 | #include "public/fpdf_save.h" |
Tom Sepez | 9eb811f | 2016-01-05 14:48:31 -0800 | [diff] [blame] | 13 | #include "public/fpdfview.h" |
| 14 | #include "testing/embedder_test.h" |
Tom Sepez | 0aec19b | 2016-01-07 12:22:44 -0800 | [diff] [blame] | 15 | #include "testing/gmock/include/gmock/gmock-matchers.h" |
Tom Sepez | 9eb811f | 2016-01-05 14:48:31 -0800 | [diff] [blame] | 16 | #include "testing/gtest/include/gtest/gtest.h" |
| 17 | |
Nicolas Pena | 3ff5400 | 2017-07-05 11:55:35 -0400 | [diff] [blame] | 18 | class FPDFSaveEmbedderTest : public EmbedderTest {}; |
Tom Sepez | 9eb811f | 2016-01-05 14:48:31 -0800 | [diff] [blame] | 19 | |
| 20 | TEST_F(FPDFSaveEmbedderTest, SaveSimpleDoc) { |
| 21 | EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
Tom Sepez | 0aec19b | 2016-01-07 12:22:44 -0800 | [diff] [blame] | 22 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 23 | EXPECT_THAT(GetString(), testing::StartsWith("%PDF-1.7\r\n")); |
Artem Strygin | d24b97e | 2017-08-09 18:50:59 +0300 | [diff] [blame] | 24 | EXPECT_EQ(805u, GetString().length()); |
Tom Sepez | 9eb811f | 2016-01-05 14:48:31 -0800 | [diff] [blame] | 25 | } |
| 26 | |
| 27 | TEST_F(FPDFSaveEmbedderTest, SaveSimpleDocWithVersion) { |
| 28 | EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
Tom Sepez | 0aec19b | 2016-01-07 12:22:44 -0800 | [diff] [blame] | 29 | EXPECT_TRUE(FPDF_SaveWithVersion(document(), this, 0, 14)); |
| 30 | EXPECT_THAT(GetString(), testing::StartsWith("%PDF-1.4\r\n")); |
Artem Strygin | d24b97e | 2017-08-09 18:50:59 +0300 | [diff] [blame] | 31 | EXPECT_EQ(805u, GetString().length()); |
Tom Sepez | 9eb811f | 2016-01-05 14:48:31 -0800 | [diff] [blame] | 32 | } |
Tom Sepez | 9eb811f | 2016-01-05 14:48:31 -0800 | [diff] [blame] | 33 | TEST_F(FPDFSaveEmbedderTest, SaveSimpleDocWithBadVersion) { |
| 34 | EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
Tom Sepez | 0aec19b | 2016-01-07 12:22:44 -0800 | [diff] [blame] | 35 | EXPECT_TRUE(FPDF_SaveWithVersion(document(), this, 0, -1)); |
| 36 | EXPECT_THAT(GetString(), testing::StartsWith("%PDF-1.7\r\n")); |
| 37 | |
| 38 | ClearString(); |
| 39 | EXPECT_TRUE(FPDF_SaveWithVersion(document(), this, 0, 0)); |
| 40 | EXPECT_THAT(GetString(), testing::StartsWith("%PDF-1.7\r\n")); |
| 41 | |
| 42 | ClearString(); |
| 43 | EXPECT_TRUE(FPDF_SaveWithVersion(document(), this, 0, 18)); |
| 44 | EXPECT_THAT(GetString(), testing::StartsWith("%PDF-1.7\r\n")); |
Tom Sepez | 9eb811f | 2016-01-05 14:48:31 -0800 | [diff] [blame] | 45 | } |
| 46 | |
tsepez | 6e1d603 | 2016-11-11 17:55:40 -0800 | [diff] [blame] | 47 | TEST_F(FPDFSaveEmbedderTest, SaveCopiedDoc) { |
| 48 | EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
| 49 | |
| 50 | FPDF_PAGE page = LoadPage(0); |
| 51 | EXPECT_TRUE(page); |
| 52 | |
| 53 | FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument(); |
| 54 | EXPECT_TRUE(output_doc); |
| 55 | EXPECT_TRUE(FPDF_ImportPages(output_doc, document(), "1", 0)); |
| 56 | EXPECT_TRUE(FPDF_SaveAsCopy(output_doc, this, 0)); |
| 57 | FPDF_CloseDocument(output_doc); |
| 58 | |
| 59 | UnloadPage(page); |
| 60 | } |
| 61 | |
Henrique Nakashima | 9fa5036 | 2017-11-10 22:40:44 +0000 | [diff] [blame] | 62 | TEST_F(FPDFSaveEmbedderTest, SaveLinearizedDoc) { |
| 63 | const int kPageCount = 3; |
| 64 | std::string original_md5[kPageCount]; |
| 65 | |
| 66 | EXPECT_TRUE(OpenDocument("linearized.pdf")); |
| 67 | for (int i = 0; i < kPageCount; ++i) { |
| 68 | FPDF_PAGE page = LoadPage(i); |
Lei Zhang | 6183a6e | 2018-02-07 23:00:07 +0000 | [diff] [blame] | 69 | ASSERT_TRUE(page); |
Tom Sepez | e08d2b1 | 2018-04-25 18:49:32 +0000 | [diff] [blame] | 70 | ScopedFPDFBitmap bitmap = RenderLoadedPage(page); |
Lei Zhang | 6183a6e | 2018-02-07 23:00:07 +0000 | [diff] [blame] | 71 | EXPECT_EQ(612, FPDFBitmap_GetWidth(bitmap.get())); |
| 72 | EXPECT_EQ(792, FPDFBitmap_GetHeight(bitmap.get())); |
| 73 | original_md5[i] = HashBitmap(bitmap.get()); |
Henrique Nakashima | 9fa5036 | 2017-11-10 22:40:44 +0000 | [diff] [blame] | 74 | UnloadPage(page); |
| 75 | } |
| 76 | |
| 77 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 78 | EXPECT_THAT(GetString(), testing::StartsWith("%PDF-1.6\r\n")); |
| 79 | EXPECT_THAT(GetString(), testing::HasSubstr("/Root ")); |
| 80 | EXPECT_THAT(GetString(), testing::HasSubstr("/Info ")); |
| 81 | EXPECT_EQ(8219u, GetString().length()); |
| 82 | |
| 83 | // Make sure new document renders the same as the old one. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 84 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | 9fa5036 | 2017-11-10 22:40:44 +0000 | [diff] [blame] | 85 | for (int i = 0; i < kPageCount; ++i) { |
| 86 | FPDF_PAGE page = LoadSavedPage(i); |
Lei Zhang | a98e366 | 2018-02-07 20:28:35 +0000 | [diff] [blame] | 87 | ASSERT_TRUE(page); |
Tom Sepez | e08d2b1 | 2018-04-25 18:49:32 +0000 | [diff] [blame] | 88 | ScopedFPDFBitmap bitmap = RenderSavedPage(page); |
Lei Zhang | a98e366 | 2018-02-07 20:28:35 +0000 | [diff] [blame] | 89 | EXPECT_EQ(original_md5[i], HashBitmap(bitmap.get())); |
Henrique Nakashima | 9fa5036 | 2017-11-10 22:40:44 +0000 | [diff] [blame] | 90 | CloseSavedPage(page); |
| 91 | } |
| 92 | CloseSavedDocument(); |
| 93 | } |
| 94 | |
Tom Sepez | 9eb811f | 2016-01-05 14:48:31 -0800 | [diff] [blame] | 95 | TEST_F(FPDFSaveEmbedderTest, BUG_342) { |
| 96 | EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
Tom Sepez | 0aec19b | 2016-01-07 12:22:44 -0800 | [diff] [blame] | 97 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 98 | EXPECT_THAT(GetString(), testing::HasSubstr("0000000000 65535 f\r\n")); |
| 99 | EXPECT_THAT(GetString(), |
| 100 | testing::Not(testing::HasSubstr("0000000000 65536 f\r\n"))); |
Tom Sepez | 9eb811f | 2016-01-05 14:48:31 -0800 | [diff] [blame] | 101 | } |
Lei Zhang | c160069 | 2018-11-14 23:12:58 +0000 | [diff] [blame] | 102 | |
| 103 | TEST_F(FPDFSaveEmbedderTest, BUG_905142) { |
| 104 | EXPECT_TRUE(OpenDocument("bug_905142.pdf")); |
| 105 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 106 | EXPECT_THAT(GetString(), testing::HasSubstr("/Length 0")); |
| 107 | } |