Tom Sepez | d483eb4 | 2016-01-06 10:03:59 -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 | |
Dan Sinclair | 85c8e7f | 2016-11-21 13:50:32 -0500 | [diff] [blame] | 5 | #include <memory> |
| 6 | #include <string> |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 7 | #include <utility> |
Jane Liu | 548334e | 2017-08-03 16:33:40 -0400 | [diff] [blame] | 8 | #include <vector> |
Dan Sinclair | 85c8e7f | 2016-11-21 13:50:32 -0500 | [diff] [blame] | 9 | |
Lei Zhang | e4cdac5 | 2019-04-30 16:45:57 +0000 | [diff] [blame] | 10 | #include "build/build_config.h" |
Nicolas Pena | be90aae | 2017-02-27 10:41:41 -0500 | [diff] [blame] | 11 | #include "core/fpdfapi/font/cpdf_font.h" |
Miklos Vajna | 46b4373 | 2018-08-14 19:15:43 +0000 | [diff] [blame] | 12 | #include "core/fpdfapi/page/cpdf_formobject.h" |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 13 | #include "core/fpdfapi/page/cpdf_page.h" |
Henrique Nakashima | 6eb7939 | 2018-06-12 20:27:35 +0000 | [diff] [blame] | 14 | #include "core/fpdfapi/page/cpdf_pageobject.h" |
Nicolas Pena | be90aae | 2017-02-27 10:41:41 -0500 | [diff] [blame] | 15 | #include "core/fpdfapi/parser/cpdf_array.h" |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 16 | #include "core/fpdfapi/parser/cpdf_dictionary.h" |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 17 | #include "core/fpdfapi/parser/cpdf_number.h" |
Nicolas Pena | be90aae | 2017-02-27 10:41:41 -0500 | [diff] [blame] | 18 | #include "core/fpdfapi/parser/cpdf_stream.h" |
Artem Strygin | eababa1 | 2018-06-06 12:31:18 +0000 | [diff] [blame] | 19 | #include "core/fpdfapi/parser/cpdf_stream_acc.h" |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 20 | #include "core/fxcrt/fx_system.h" |
Lei Zhang | 0dcb8fd | 2019-02-07 00:46:04 +0000 | [diff] [blame] | 21 | #include "core/fxge/fx_font.h" |
Dan Sinclair | 00d47a6 | 2018-03-28 18:39:04 +0000 | [diff] [blame] | 22 | #include "fpdfsdk/cpdfsdk_helpers.h" |
Tom Sepez | e08d2b1 | 2018-04-25 18:49:32 +0000 | [diff] [blame] | 23 | #include "public/cpp/fpdf_scopers.h" |
Jane Liu | eda6525 | 2017-06-07 11:31:27 -0400 | [diff] [blame] | 24 | #include "public/fpdf_annot.h" |
Tom Sepez | d483eb4 | 2016-01-06 10:03:59 -0800 | [diff] [blame] | 25 | #include "public/fpdf_edit.h" |
| 26 | #include "public/fpdfview.h" |
| 27 | #include "testing/embedder_test.h" |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 28 | #include "testing/embedder_test_constants.h" |
Lei Zhang | b6992dd | 2019-02-05 23:30:20 +0000 | [diff] [blame] | 29 | #include "testing/fx_string_testhelpers.h" |
Tom Sepez | 0aec19b | 2016-01-07 12:22:44 -0800 | [diff] [blame] | 30 | #include "testing/gmock/include/gmock/gmock-matchers.h" |
Tom Sepez | d483eb4 | 2016-01-06 10:03:59 -0800 | [diff] [blame] | 31 | #include "testing/gtest/include/gtest/gtest.h" |
Lei Zhang | 4c64e96 | 2019-02-05 19:24:12 +0000 | [diff] [blame] | 32 | #include "testing/utils/hash.h" |
Tom Sepez | d483eb4 | 2016-01-06 10:03:59 -0800 | [diff] [blame] | 33 | |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 34 | using pdfium::kHelloWorldChecksum; |
| 35 | |
| 36 | namespace { |
| 37 | |
| 38 | const char kRedRectangleChecksum[] = "66d02eaa6181e2c069ce2ea99beda497"; |
| 39 | |
Hui Yingst | 08c4071 | 2020-04-29 01:37:35 +0000 | [diff] [blame^] | 40 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 41 | #if defined(OS_WIN) |
| 42 | const char kFirstRemovedChecksum[] = "d468228cc55071a9e838f66afc0f58ff"; |
| 43 | #else |
| 44 | const char kFirstRemovedChecksum[] = "0c79dc1065b1d06991e3ac4aaa35d25c"; |
| 45 | #endif // defined(OS_WIN) |
| 46 | #else |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 47 | #if defined(OS_WIN) |
| 48 | const char kFirstRemovedChecksum[] = "aae6c5334721f90ec30d3d59f4ef7deb"; |
| 49 | #elif defined(OS_MACOSX) |
| 50 | const char kFirstRemovedChecksum[] = "17ca3778fd8bb395b46532f1fa17f702"; |
| 51 | #else |
| 52 | const char kFirstRemovedChecksum[] = "b76df015fe88009c3c342395df96abf1"; |
| 53 | #endif |
Hui Yingst | 08c4071 | 2020-04-29 01:37:35 +0000 | [diff] [blame^] | 54 | #endif // defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 55 | |
| 56 | } // namespace |
| 57 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 58 | class FPDFEditEmbedderTest : public EmbedderTest { |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 59 | protected: |
| 60 | FPDF_DOCUMENT CreateNewDocument() { |
| 61 | document_ = FPDF_CreateNewDocument(); |
Tom Sepez | 41066c1 | 2017-05-18 09:28:49 -0700 | [diff] [blame] | 62 | cpdf_doc_ = CPDFDocumentFromFPDFDocument(document_); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 63 | return document_; |
| 64 | } |
| 65 | |
Lei Zhang | 710fa99 | 2018-05-25 16:24:48 +0000 | [diff] [blame] | 66 | void CheckFontDescriptor(const CPDF_Dictionary* font_dict, |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 67 | int font_type, |
| 68 | bool bold, |
| 69 | bool italic, |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 70 | pdfium::span<const uint8_t> span) { |
Lei Zhang | b1ec280 | 2018-05-25 21:55:24 +0000 | [diff] [blame] | 71 | const CPDF_Dictionary* font_desc = font_dict->GetDictFor("FontDescriptor"); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 72 | ASSERT_TRUE(font_desc); |
| 73 | EXPECT_EQ("FontDescriptor", font_desc->GetStringFor("Type")); |
| 74 | EXPECT_EQ(font_dict->GetStringFor("BaseFont"), |
| 75 | font_desc->GetStringFor("FontName")); |
| 76 | |
| 77 | // Check that the font descriptor has the required keys according to spec |
| 78 | // 1.7 Table 5.19 |
| 79 | ASSERT_TRUE(font_desc->KeyExist("Flags")); |
Dan Sinclair | 10e1f05 | 2017-09-28 15:59:42 -0400 | [diff] [blame] | 80 | |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 81 | int font_flags = font_desc->GetIntegerFor("Flags"); |
Lei Zhang | 44d0383 | 2019-10-18 18:45:31 +0000 | [diff] [blame] | 82 | EXPECT_EQ(bold, FontStyleIsForceBold(font_flags)); |
Dan Sinclair | 10e1f05 | 2017-09-28 15:59:42 -0400 | [diff] [blame] | 83 | EXPECT_EQ(italic, FontStyleIsItalic(font_flags)); |
| 84 | EXPECT_TRUE(FontStyleIsNonSymbolic(font_flags)); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 85 | ASSERT_TRUE(font_desc->KeyExist("FontBBox")); |
Nicolás Peña | 5f95f36 | 2017-09-28 13:00:45 +0900 | [diff] [blame] | 86 | |
Lei Zhang | b1ec280 | 2018-05-25 21:55:24 +0000 | [diff] [blame] | 87 | const CPDF_Array* fontBBox = font_desc->GetArrayFor("FontBBox"); |
Nicolás Peña | 5f95f36 | 2017-09-28 13:00:45 +0900 | [diff] [blame] | 88 | ASSERT_TRUE(fontBBox); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 89 | EXPECT_EQ(4u, fontBBox->size()); |
Nicolás Peña | 5f95f36 | 2017-09-28 13:00:45 +0900 | [diff] [blame] | 90 | // Check that the coordinates are in the preferred order according to spec |
| 91 | // 1.7 Section 3.8.4 |
| 92 | EXPECT_TRUE(fontBBox->GetIntegerAt(0) < fontBBox->GetIntegerAt(2)); |
| 93 | EXPECT_TRUE(fontBBox->GetIntegerAt(1) < fontBBox->GetIntegerAt(3)); |
| 94 | |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 95 | EXPECT_TRUE(font_desc->KeyExist("ItalicAngle")); |
| 96 | EXPECT_TRUE(font_desc->KeyExist("Ascent")); |
| 97 | EXPECT_TRUE(font_desc->KeyExist("Descent")); |
| 98 | EXPECT_TRUE(font_desc->KeyExist("CapHeight")); |
| 99 | EXPECT_TRUE(font_desc->KeyExist("StemV")); |
Ryan Harrison | 275e260 | 2017-09-18 14:23:18 -0400 | [diff] [blame] | 100 | ByteString present("FontFile"); |
| 101 | ByteString absent("FontFile2"); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 102 | if (font_type == FPDF_FONT_TRUETYPE) |
| 103 | std::swap(present, absent); |
| 104 | EXPECT_TRUE(font_desc->KeyExist(present)); |
| 105 | EXPECT_FALSE(font_desc->KeyExist(absent)); |
| 106 | |
Artem Strygin | eababa1 | 2018-06-06 12:31:18 +0000 | [diff] [blame] | 107 | auto streamAcc = |
| 108 | pdfium::MakeRetain<CPDF_StreamAcc>(font_desc->GetStreamFor(present)); |
| 109 | streamAcc->LoadAllDataRaw(); |
| 110 | |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 111 | // Check that the font stream is the one that was provided |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 112 | ASSERT_EQ(span.size(), streamAcc->GetSize()); |
Nicolás Peña | 79eab23 | 2017-09-28 13:29:05 +0900 | [diff] [blame] | 113 | if (font_type == FPDF_FONT_TRUETYPE) { |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 114 | ASSERT_EQ(static_cast<int>(span.size()), |
Artem Strygin | eababa1 | 2018-06-06 12:31:18 +0000 | [diff] [blame] | 115 | streamAcc->GetDict()->GetIntegerFor("Length1")); |
Nicolás Peña | 79eab23 | 2017-09-28 13:29:05 +0900 | [diff] [blame] | 116 | } |
Artem Strygin | eababa1 | 2018-06-06 12:31:18 +0000 | [diff] [blame] | 117 | |
| 118 | const uint8_t* stream_data = streamAcc->GetData(); |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 119 | for (size_t j = 0; j < span.size(); j++) |
| 120 | EXPECT_EQ(span[j], stream_data[j]) << " at byte " << j; |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 121 | } |
| 122 | |
Lei Zhang | de579ab | 2018-05-25 21:49:49 +0000 | [diff] [blame] | 123 | void CheckCompositeFontWidths(const CPDF_Array* widths_array, |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 124 | CPDF_Font* typed_font) { |
| 125 | // Check that W array is in a format that conforms to PDF spec 1.7 section |
| 126 | // "Glyph Metrics in CIDFonts" (these checks are not |
| 127 | // implementation-specific). |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 128 | EXPECT_GT(widths_array->size(), 1u); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 129 | int num_cids_checked = 0; |
| 130 | int cur_cid = 0; |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 131 | for (size_t idx = 0; idx < widths_array->size(); idx++) { |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 132 | int cid = widths_array->GetNumberAt(idx); |
| 133 | EXPECT_GE(cid, cur_cid); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 134 | ASSERT_FALSE(++idx == widths_array->size()); |
Lei Zhang | de579ab | 2018-05-25 21:49:49 +0000 | [diff] [blame] | 135 | const CPDF_Object* next = widths_array->GetObjectAt(idx); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 136 | if (next->IsArray()) { |
| 137 | // We are in the c [w1 w2 ...] case |
Lei Zhang | de579ab | 2018-05-25 21:49:49 +0000 | [diff] [blame] | 138 | const CPDF_Array* arr = next->AsArray(); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 139 | int cnt = static_cast<int>(arr->size()); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 140 | size_t inner_idx = 0; |
| 141 | for (cur_cid = cid; cur_cid < cid + cnt; cur_cid++) { |
Nicolas Pena | 2334660 | 2018-01-30 21:42:41 +0000 | [diff] [blame] | 142 | uint32_t width = arr->GetNumberAt(inner_idx++); |
Dan Sinclair | 971a674 | 2018-03-28 19:23:25 +0000 | [diff] [blame] | 143 | EXPECT_EQ(width, typed_font->GetCharWidthF(cur_cid)) |
| 144 | << " at cid " << cur_cid; |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 145 | } |
| 146 | num_cids_checked += cnt; |
| 147 | continue; |
| 148 | } |
| 149 | // Otherwise, are in the c_first c_last w case. |
| 150 | ASSERT_TRUE(next->IsNumber()); |
| 151 | int last_cid = next->AsNumber()->GetInteger(); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 152 | ASSERT_FALSE(++idx == widths_array->size()); |
Nicolas Pena | 2334660 | 2018-01-30 21:42:41 +0000 | [diff] [blame] | 153 | uint32_t width = widths_array->GetNumberAt(idx); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 154 | for (cur_cid = cid; cur_cid <= last_cid; cur_cid++) { |
Dan Sinclair | 971a674 | 2018-03-28 19:23:25 +0000 | [diff] [blame] | 155 | EXPECT_EQ(width, typed_font->GetCharWidthF(cur_cid)) |
| 156 | << " at cid " << cur_cid; |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 157 | } |
| 158 | num_cids_checked += last_cid - cid + 1; |
| 159 | } |
| 160 | // Make sure we have a good amount of cids described |
| 161 | EXPECT_GT(num_cids_checked, 900); |
| 162 | } |
| 163 | CPDF_Document* cpdf_doc() { return cpdf_doc_; } |
| 164 | |
| 165 | private: |
| 166 | CPDF_Document* cpdf_doc_; |
| 167 | }; |
Tom Sepez | d483eb4 | 2016-01-06 10:03:59 -0800 | [diff] [blame] | 168 | |
etienneb | 7712c26 | 2016-04-26 08:13:45 -0700 | [diff] [blame] | 169 | namespace { |
thestig | dc7ec03 | 2016-11-21 15:32:52 -0800 | [diff] [blame] | 170 | |
etienneb | 7712c26 | 2016-04-26 08:13:45 -0700 | [diff] [blame] | 171 | const char kExpectedPDF[] = |
| 172 | "%PDF-1.7\r\n" |
| 173 | "%\xA1\xB3\xC5\xD7\r\n" |
| 174 | "1 0 obj\r\n" |
| 175 | "<</Pages 2 0 R /Type/Catalog>>\r\n" |
| 176 | "endobj\r\n" |
| 177 | "2 0 obj\r\n" |
| 178 | "<</Count 1/Kids\\[ 4 0 R \\]/Type/Pages>>\r\n" |
| 179 | "endobj\r\n" |
| 180 | "3 0 obj\r\n" |
| 181 | "<</CreationDate\\(D:.*\\)/Creator\\(PDFium\\)>>\r\n" |
| 182 | "endobj\r\n" |
| 183 | "4 0 obj\r\n" |
Nicolas Pena | 9ba8fbc | 2017-06-28 15:31:56 -0400 | [diff] [blame] | 184 | "<</MediaBox\\[ 0 0 640 480\\]/Parent 2 0 R " |
| 185 | "/Resources<</ExtGState<</FXE1 5 0 R >>>>" |
Nicolas Pena | d9d6c29 | 2017-06-06 16:12:10 -0400 | [diff] [blame] | 186 | "/Rotate 0/Type/Page" |
etienneb | 7712c26 | 2016-04-26 08:13:45 -0700 | [diff] [blame] | 187 | ">>\r\n" |
| 188 | "endobj\r\n" |
Nicolas Pena | 9ba8fbc | 2017-06-28 15:31:56 -0400 | [diff] [blame] | 189 | "5 0 obj\r\n" |
| 190 | "<</BM/Normal/CA 1/ca 1>>\r\n" |
| 191 | "endobj\r\n" |
etienneb | 7712c26 | 2016-04-26 08:13:45 -0700 | [diff] [blame] | 192 | "xref\r\n" |
Nicolas Pena | 9ba8fbc | 2017-06-28 15:31:56 -0400 | [diff] [blame] | 193 | "0 6\r\n" |
etienneb | 7712c26 | 2016-04-26 08:13:45 -0700 | [diff] [blame] | 194 | "0000000000 65535 f\r\n" |
| 195 | "0000000017 00000 n\r\n" |
| 196 | "0000000066 00000 n\r\n" |
| 197 | "0000000122 00000 n\r\n" |
| 198 | "0000000192 00000 n\r\n" |
Nicolas Pena | 9ba8fbc | 2017-06-28 15:31:56 -0400 | [diff] [blame] | 199 | "0000000311 00000 n\r\n" |
etienneb | 7712c26 | 2016-04-26 08:13:45 -0700 | [diff] [blame] | 200 | "trailer\r\n" |
| 201 | "<<\r\n" |
| 202 | "/Root 1 0 R\r\n" |
| 203 | "/Info 3 0 R\r\n" |
Nicolas Pena | 9ba8fbc | 2017-06-28 15:31:56 -0400 | [diff] [blame] | 204 | "/Size 6/ID\\[<.*><.*>\\]>>\r\n" |
etienneb | 7712c26 | 2016-04-26 08:13:45 -0700 | [diff] [blame] | 205 | "startxref\r\n" |
Nicolas Pena | 9ba8fbc | 2017-06-28 15:31:56 -0400 | [diff] [blame] | 206 | "354\r\n" |
etienneb | 7712c26 | 2016-04-26 08:13:45 -0700 | [diff] [blame] | 207 | "%%EOF\r\n"; |
thestig | dc7ec03 | 2016-11-21 15:32:52 -0800 | [diff] [blame] | 208 | |
etienneb | 7712c26 | 2016-04-26 08:13:45 -0700 | [diff] [blame] | 209 | } // namespace |
| 210 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 211 | TEST_F(FPDFEditEmbedderTest, EmptyCreation) { |
Tom Sepez | d483eb4 | 2016-01-06 10:03:59 -0800 | [diff] [blame] | 212 | EXPECT_TRUE(CreateEmptyDocument()); |
weili | 9b777de | 2016-08-19 16:19:46 -0700 | [diff] [blame] | 213 | FPDF_PAGE page = FPDFPage_New(document(), 0, 640.0, 480.0); |
Tom Sepez | d483eb4 | 2016-01-06 10:03:59 -0800 | [diff] [blame] | 214 | EXPECT_NE(nullptr, page); |
Nicolas Pena | d9d6c29 | 2017-06-06 16:12:10 -0400 | [diff] [blame] | 215 | // The FPDFPage_GenerateContent call should do nothing. |
Tom Sepez | d483eb4 | 2016-01-06 10:03:59 -0800 | [diff] [blame] | 216 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
Tom Sepez | 0aec19b | 2016-01-07 12:22:44 -0800 | [diff] [blame] | 217 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
etienneb | 7712c26 | 2016-04-26 08:13:45 -0700 | [diff] [blame] | 218 | |
Nicolas Pena | d9d6c29 | 2017-06-06 16:12:10 -0400 | [diff] [blame] | 219 | EXPECT_THAT(GetString(), testing::MatchesRegex(std::string( |
| 220 | kExpectedPDF, sizeof(kExpectedPDF)))); |
weili | 9b777de | 2016-08-19 16:19:46 -0700 | [diff] [blame] | 221 | FPDF_ClosePage(page); |
Tom Sepez | d483eb4 | 2016-01-06 10:03:59 -0800 | [diff] [blame] | 222 | } |
thestig | dc7ec03 | 2016-11-21 15:32:52 -0800 | [diff] [blame] | 223 | |
| 224 | // Regression test for https://crbug.com/667012 |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 225 | TEST_F(FPDFEditEmbedderTest, RasterizePDF) { |
thestig | dc7ec03 | 2016-11-21 15:32:52 -0800 | [diff] [blame] | 226 | const char kAllBlackMd5sum[] = "5708fc5c4a8bd0abde99c8e8f0390615"; |
| 227 | |
Lei Zhang | f054289 | 2019-01-17 18:46:27 +0000 | [diff] [blame] | 228 | // Get the bitmap for the original document. |
Tom Sepez | e08d2b1 | 2018-04-25 18:49:32 +0000 | [diff] [blame] | 229 | ScopedFPDFBitmap orig_bitmap; |
thestig | dc7ec03 | 2016-11-21 15:32:52 -0800 | [diff] [blame] | 230 | { |
| 231 | EXPECT_TRUE(OpenDocument("black.pdf")); |
| 232 | FPDF_PAGE orig_page = LoadPage(0); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 233 | ASSERT_TRUE(orig_page); |
| 234 | orig_bitmap = RenderLoadedPage(orig_page); |
| 235 | CompareBitmap(orig_bitmap.get(), 612, 792, kAllBlackMd5sum); |
thestig | dc7ec03 | 2016-11-21 15:32:52 -0800 | [diff] [blame] | 236 | UnloadPage(orig_page); |
| 237 | } |
| 238 | |
| 239 | // Create a new document from |orig_bitmap| and save it. |
| 240 | { |
| 241 | FPDF_DOCUMENT temp_doc = FPDF_CreateNewDocument(); |
| 242 | FPDF_PAGE temp_page = FPDFPage_New(temp_doc, 0, 612, 792); |
| 243 | |
| 244 | // Add the bitmap to an image object and add the image object to the output |
| 245 | // page. |
Lei Zhang | cbd8957 | 2017-03-15 17:35:47 -0700 | [diff] [blame] | 246 | FPDF_PAGEOBJECT temp_img = FPDFPageObj_NewImageObj(temp_doc); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 247 | EXPECT_TRUE( |
| 248 | FPDFImageObj_SetBitmap(&temp_page, 1, temp_img, orig_bitmap.get())); |
thestig | dc7ec03 | 2016-11-21 15:32:52 -0800 | [diff] [blame] | 249 | EXPECT_TRUE(FPDFImageObj_SetMatrix(temp_img, 612, 0, 0, 792, 0, 0)); |
| 250 | FPDFPage_InsertObject(temp_page, temp_img); |
| 251 | EXPECT_TRUE(FPDFPage_GenerateContent(temp_page)); |
| 252 | EXPECT_TRUE(FPDF_SaveAsCopy(temp_doc, this, 0)); |
| 253 | FPDF_ClosePage(temp_page); |
| 254 | FPDF_CloseDocument(temp_doc); |
| 255 | } |
thestig | dc7ec03 | 2016-11-21 15:32:52 -0800 | [diff] [blame] | 256 | |
| 257 | // Get the generated content. Make sure it is at least as big as the original |
| 258 | // PDF. |
Lei Zhang | d72fd58 | 2018-07-27 19:37:27 +0000 | [diff] [blame] | 259 | EXPECT_GT(GetString().size(), 923u); |
Dan Sinclair | 04e4dc8 | 2017-10-18 12:17:14 -0400 | [diff] [blame] | 260 | VerifySavedDocument(612, 792, kAllBlackMd5sum); |
thestig | dc7ec03 | 2016-11-21 15:32:52 -0800 | [diff] [blame] | 261 | } |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 262 | |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 263 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 264 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 265 | #define MAYBE_AddPaths DISABLED_AddPaths |
| 266 | #else |
| 267 | #define MAYBE_AddPaths AddPaths |
| 268 | #endif |
| 269 | TEST_F(FPDFEditEmbedderTest, MAYBE_AddPaths) { |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 270 | // Start with a blank page |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 271 | FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 272 | ASSERT_TRUE(page); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 273 | |
| 274 | // We will first add a red rectangle |
| 275 | FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(10, 10, 20, 20); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 276 | ASSERT_TRUE(red_rect); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 277 | // Expect false when trying to set colors out of range |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 278 | EXPECT_FALSE(FPDFPageObj_SetStrokeColor(red_rect, 100, 100, 100, 300)); |
| 279 | EXPECT_FALSE(FPDFPageObj_SetFillColor(red_rect, 200, 256, 200, 0)); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 280 | |
| 281 | // Fill rectangle with red and insert to the page |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 282 | EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 0, 0, 255)); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 283 | EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0)); |
Miklos Vajna | 491112b | 2018-05-30 13:30:10 +0000 | [diff] [blame] | 284 | |
| 285 | int fillmode = FPDF_FILLMODE_NONE; |
| 286 | FPDF_BOOL stroke = true; |
| 287 | EXPECT_TRUE(FPDFPath_GetDrawMode(red_rect, &fillmode, &stroke)); |
| 288 | EXPECT_EQ(FPDF_FILLMODE_ALTERNATE, fillmode); |
| 289 | EXPECT_FALSE(stroke); |
| 290 | |
Lei Zhang | 8da9823 | 2019-12-11 23:29:33 +0000 | [diff] [blame] | 291 | static const FS_MATRIX kMatrix = {1, 2, 3, 4, 5, 6}; |
| 292 | EXPECT_FALSE(FPDFPath_SetMatrix(nullptr, &kMatrix)); |
| 293 | EXPECT_TRUE(FPDFPath_SetMatrix(red_rect, &kMatrix)); |
| 294 | |
| 295 | FS_MATRIX matrix; |
| 296 | EXPECT_FALSE(FPDFPath_GetMatrix(nullptr, &matrix)); |
| 297 | EXPECT_TRUE(FPDFPath_GetMatrix(red_rect, &matrix)); |
| 298 | EXPECT_FLOAT_EQ(1.0f, matrix.a); |
| 299 | EXPECT_FLOAT_EQ(2.0f, matrix.b); |
| 300 | EXPECT_FLOAT_EQ(3.0f, matrix.c); |
| 301 | EXPECT_FLOAT_EQ(4.0f, matrix.d); |
| 302 | EXPECT_FLOAT_EQ(5.0f, matrix.e); |
| 303 | EXPECT_FLOAT_EQ(6.0f, matrix.f); |
| 304 | |
| 305 | // Set back the identity matrix. |
| 306 | matrix = {1.0f, 0.0f, 0.0f, 1.0f, 0.0f, 0.0f}; |
| 307 | EXPECT_TRUE(FPDFPath_SetMatrix(red_rect, &matrix)); |
Miklos Vajna | 97f4d67 | 2018-06-04 14:47:17 +0000 | [diff] [blame] | 308 | |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 309 | FPDFPage_InsertObject(page, red_rect); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 310 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 311 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 312 | CompareBitmap(page_bitmap.get(), 612, 792, kRedRectangleChecksum); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 313 | } |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 314 | |
| 315 | // Now add to that a green rectangle with some medium alpha |
| 316 | FPDF_PAGEOBJECT green_rect = FPDFPageObj_CreateNewRect(100, 100, 40, 40); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 317 | EXPECT_TRUE(FPDFPageObj_SetFillColor(green_rect, 0, 255, 0, 128)); |
Miklos Vajna | ed4705b | 2017-04-05 09:24:50 +0200 | [diff] [blame] | 318 | |
Miklos Vajna | 1ef04c9 | 2017-05-08 18:14:19 +0200 | [diff] [blame] | 319 | // Make sure the type of the rectangle is a path. |
| 320 | EXPECT_EQ(FPDF_PAGEOBJ_PATH, FPDFPageObj_GetType(green_rect)); |
| 321 | |
Miklos Vajna | ed4705b | 2017-04-05 09:24:50 +0200 | [diff] [blame] | 322 | // Make sure we get back the same color we set previously. |
| 323 | unsigned int R; |
| 324 | unsigned int G; |
| 325 | unsigned int B; |
| 326 | unsigned int A; |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 327 | EXPECT_TRUE(FPDFPageObj_GetFillColor(green_rect, &R, &G, &B, &A)); |
Lei Zhang | d72fd58 | 2018-07-27 19:37:27 +0000 | [diff] [blame] | 328 | EXPECT_EQ(0u, R); |
| 329 | EXPECT_EQ(255u, G); |
| 330 | EXPECT_EQ(0u, B); |
| 331 | EXPECT_EQ(128u, A); |
Miklos Vajna | ed4705b | 2017-04-05 09:24:50 +0200 | [diff] [blame] | 332 | |
Miklos Vajna | 12abfd0 | 2017-09-15 07:49:03 +0200 | [diff] [blame] | 333 | // Make sure the path has 5 points (1 FXPT_TYPE::MoveTo and 4 |
| 334 | // FXPT_TYPE::LineTo). |
Miklos Vajna | 0150a54 | 2017-09-21 21:46:56 +0200 | [diff] [blame] | 335 | ASSERT_EQ(5, FPDFPath_CountSegments(green_rect)); |
Miklos Vajna | 36eed87 | 2017-09-20 22:52:43 +0200 | [diff] [blame] | 336 | // Verify actual coordinates. |
| 337 | FPDF_PATHSEGMENT segment = FPDFPath_GetPathSegment(green_rect, 0); |
| 338 | float x; |
| 339 | float y; |
| 340 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 341 | EXPECT_EQ(100, x); |
| 342 | EXPECT_EQ(100, y); |
| 343 | EXPECT_EQ(FPDF_SEGMENT_MOVETO, FPDFPathSegment_GetType(segment)); |
| 344 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 345 | segment = FPDFPath_GetPathSegment(green_rect, 1); |
| 346 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 347 | EXPECT_EQ(100, x); |
| 348 | EXPECT_EQ(140, y); |
| 349 | EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment)); |
| 350 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 351 | segment = FPDFPath_GetPathSegment(green_rect, 2); |
| 352 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 353 | EXPECT_EQ(140, x); |
| 354 | EXPECT_EQ(140, y); |
| 355 | EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment)); |
| 356 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 357 | segment = FPDFPath_GetPathSegment(green_rect, 3); |
| 358 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 359 | EXPECT_EQ(140, x); |
| 360 | EXPECT_EQ(100, y); |
| 361 | EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment)); |
| 362 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 363 | segment = FPDFPath_GetPathSegment(green_rect, 4); |
| 364 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 365 | EXPECT_EQ(100, x); |
| 366 | EXPECT_EQ(100, y); |
| 367 | EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment)); |
| 368 | EXPECT_TRUE(FPDFPathSegment_GetClose(segment)); |
Miklos Vajna | 12abfd0 | 2017-09-15 07:49:03 +0200 | [diff] [blame] | 369 | |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 370 | EXPECT_TRUE(FPDFPath_SetDrawMode(green_rect, FPDF_FILLMODE_WINDING, 0)); |
| 371 | FPDFPage_InsertObject(page, green_rect); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 372 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 373 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 374 | CompareBitmap(page_bitmap.get(), 612, 792, |
| 375 | "7b0b87604594e773add528fae567a558"); |
| 376 | } |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 377 | |
| 378 | // Add a black triangle. |
| 379 | FPDF_PAGEOBJECT black_path = FPDFPageObj_CreateNewPath(400, 100); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 380 | EXPECT_TRUE(FPDFPageObj_SetFillColor(black_path, 0, 0, 0, 200)); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 381 | EXPECT_TRUE(FPDFPath_SetDrawMode(black_path, FPDF_FILLMODE_ALTERNATE, 0)); |
| 382 | EXPECT_TRUE(FPDFPath_LineTo(black_path, 400, 200)); |
| 383 | EXPECT_TRUE(FPDFPath_LineTo(black_path, 300, 100)); |
| 384 | EXPECT_TRUE(FPDFPath_Close(black_path)); |
Miklos Vajna | 12abfd0 | 2017-09-15 07:49:03 +0200 | [diff] [blame] | 385 | |
| 386 | // Make sure the path has 3 points (1 FXPT_TYPE::MoveTo and 2 |
| 387 | // FXPT_TYPE::LineTo). |
Miklos Vajna | 0150a54 | 2017-09-21 21:46:56 +0200 | [diff] [blame] | 388 | ASSERT_EQ(3, FPDFPath_CountSegments(black_path)); |
Miklos Vajna | 36eed87 | 2017-09-20 22:52:43 +0200 | [diff] [blame] | 389 | // Verify actual coordinates. |
| 390 | segment = FPDFPath_GetPathSegment(black_path, 0); |
| 391 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 392 | EXPECT_EQ(400, x); |
| 393 | EXPECT_EQ(100, y); |
| 394 | EXPECT_EQ(FPDF_SEGMENT_MOVETO, FPDFPathSegment_GetType(segment)); |
| 395 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 396 | segment = FPDFPath_GetPathSegment(black_path, 1); |
| 397 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 398 | EXPECT_EQ(400, x); |
| 399 | EXPECT_EQ(200, y); |
| 400 | EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment)); |
| 401 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 402 | segment = FPDFPath_GetPathSegment(black_path, 2); |
| 403 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 404 | EXPECT_EQ(300, x); |
| 405 | EXPECT_EQ(100, y); |
| 406 | EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment)); |
| 407 | EXPECT_TRUE(FPDFPathSegment_GetClose(segment)); |
| 408 | // Make sure out of bounds index access fails properly. |
| 409 | EXPECT_EQ(nullptr, FPDFPath_GetPathSegment(black_path, 3)); |
Miklos Vajna | 12abfd0 | 2017-09-15 07:49:03 +0200 | [diff] [blame] | 410 | |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 411 | FPDFPage_InsertObject(page, black_path); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 412 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 413 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 414 | CompareBitmap(page_bitmap.get(), 612, 792, |
| 415 | "eadc8020a14dfcf091da2688733d8806"); |
| 416 | } |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 417 | |
| 418 | // Now add a more complex blue path. |
| 419 | FPDF_PAGEOBJECT blue_path = FPDFPageObj_CreateNewPath(200, 200); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 420 | EXPECT_TRUE(FPDFPageObj_SetFillColor(blue_path, 0, 0, 255, 255)); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 421 | EXPECT_TRUE(FPDFPath_SetDrawMode(blue_path, FPDF_FILLMODE_WINDING, 0)); |
| 422 | EXPECT_TRUE(FPDFPath_LineTo(blue_path, 230, 230)); |
| 423 | EXPECT_TRUE(FPDFPath_BezierTo(blue_path, 250, 250, 280, 280, 300, 300)); |
| 424 | EXPECT_TRUE(FPDFPath_LineTo(blue_path, 325, 325)); |
| 425 | EXPECT_TRUE(FPDFPath_LineTo(blue_path, 350, 325)); |
| 426 | EXPECT_TRUE(FPDFPath_BezierTo(blue_path, 375, 330, 390, 360, 400, 400)); |
| 427 | EXPECT_TRUE(FPDFPath_Close(blue_path)); |
| 428 | FPDFPage_InsertObject(page, blue_path); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 429 | const char kLastMD5[] = "9823e1a21bd9b72b6a442ba4f12af946"; |
| 430 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 431 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 432 | CompareBitmap(page_bitmap.get(), 612, 792, kLastMD5); |
| 433 | } |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 434 | |
| 435 | // Now save the result, closing the page and document |
Nicolas Pena | 207b727 | 2017-05-26 17:37:06 -0400 | [diff] [blame] | 436 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 437 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 438 | FPDF_ClosePage(page); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 439 | |
| 440 | // Render the saved result |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 441 | VerifySavedDocument(612, 792, kLastMD5); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 442 | } |
| 443 | |
Miklos Vajna | 45501f3 | 2019-07-30 07:41:02 +0000 | [diff] [blame] | 444 | TEST_F(FPDFEditEmbedderTest, ClipPath) { |
| 445 | // Load document with a clipped rectangle. |
| 446 | EXPECT_TRUE(OpenDocument("clip_path.pdf")); |
| 447 | FPDF_PAGE page = LoadPage(0); |
| 448 | ASSERT_TRUE(page); |
| 449 | |
| 450 | ASSERT_EQ(1, FPDFPage_CountObjects(page)); |
| 451 | |
| 452 | FPDF_PAGEOBJECT triangle = FPDFPage_GetObject(page, 0); |
| 453 | ASSERT_TRUE(triangle); |
| 454 | |
| 455 | // Test that we got the expected triangle. |
| 456 | ASSERT_EQ(4, FPDFPath_CountSegments(triangle)); |
| 457 | |
| 458 | FPDF_PATHSEGMENT segment = FPDFPath_GetPathSegment(triangle, 0); |
| 459 | float x; |
| 460 | float y; |
| 461 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 462 | EXPECT_EQ(10, x); |
| 463 | EXPECT_EQ(10, y); |
| 464 | EXPECT_EQ(FPDF_SEGMENT_MOVETO, FPDFPathSegment_GetType(segment)); |
| 465 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 466 | |
| 467 | segment = FPDFPath_GetPathSegment(triangle, 1); |
| 468 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 469 | EXPECT_EQ(25, x); |
| 470 | EXPECT_EQ(40, y); |
| 471 | EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment)); |
| 472 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 473 | |
| 474 | segment = FPDFPath_GetPathSegment(triangle, 2); |
| 475 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 476 | EXPECT_EQ(40, x); |
| 477 | EXPECT_EQ(10, y); |
| 478 | EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment)); |
| 479 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 480 | |
| 481 | segment = FPDFPath_GetPathSegment(triangle, 3); |
| 482 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 483 | EXPECT_TRUE(FPDFPathSegment_GetClose(segment)); |
| 484 | |
| 485 | // Test FPDFPageObj_GetClipPath(). |
| 486 | ASSERT_EQ(nullptr, FPDFPageObj_GetClipPath(nullptr)); |
| 487 | |
| 488 | FPDF_CLIPPATH clip_path = FPDFPageObj_GetClipPath(triangle); |
| 489 | ASSERT_TRUE(clip_path); |
| 490 | |
| 491 | // Test FPDFClipPath_CountPaths(). |
| 492 | ASSERT_EQ(-1, FPDFClipPath_CountPaths(nullptr)); |
| 493 | ASSERT_EQ(1, FPDFClipPath_CountPaths(clip_path)); |
| 494 | |
| 495 | // Test FPDFClipPath_CountPathSegments(). |
| 496 | ASSERT_EQ(-1, FPDFClipPath_CountPathSegments(nullptr, 0)); |
| 497 | ASSERT_EQ(-1, FPDFClipPath_CountPathSegments(clip_path, -1)); |
| 498 | ASSERT_EQ(-1, FPDFClipPath_CountPathSegments(clip_path, 1)); |
| 499 | ASSERT_EQ(4, FPDFClipPath_CountPathSegments(clip_path, 0)); |
| 500 | |
| 501 | // FPDFClipPath_GetPathSegment() negative testing. |
| 502 | ASSERT_EQ(nullptr, FPDFClipPath_GetPathSegment(nullptr, 0, 0)); |
| 503 | ASSERT_EQ(nullptr, FPDFClipPath_GetPathSegment(clip_path, -1, 0)); |
| 504 | ASSERT_EQ(nullptr, FPDFClipPath_GetPathSegment(clip_path, 1, 0)); |
| 505 | ASSERT_EQ(nullptr, FPDFClipPath_GetPathSegment(clip_path, 0, -1)); |
| 506 | ASSERT_EQ(nullptr, FPDFClipPath_GetPathSegment(clip_path, 0, 4)); |
| 507 | |
| 508 | // FPDFClipPath_GetPathSegment() positive testing. |
| 509 | segment = FPDFClipPath_GetPathSegment(clip_path, 0, 0); |
| 510 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 511 | EXPECT_EQ(10, x); |
| 512 | EXPECT_EQ(15, y); |
| 513 | EXPECT_EQ(FPDF_SEGMENT_MOVETO, FPDFPathSegment_GetType(segment)); |
| 514 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 515 | |
| 516 | segment = FPDFClipPath_GetPathSegment(clip_path, 0, 1); |
| 517 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 518 | EXPECT_EQ(40, x); |
| 519 | EXPECT_EQ(15, y); |
| 520 | EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment)); |
| 521 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 522 | |
| 523 | segment = FPDFClipPath_GetPathSegment(clip_path, 0, 2); |
| 524 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 525 | EXPECT_EQ(40, x); |
| 526 | EXPECT_EQ(35, y); |
| 527 | EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment)); |
| 528 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 529 | |
| 530 | segment = FPDFClipPath_GetPathSegment(clip_path, 0, 3); |
| 531 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 532 | EXPECT_EQ(10, x); |
| 533 | EXPECT_EQ(35, y); |
| 534 | EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment)); |
| 535 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 536 | |
| 537 | UnloadPage(page); |
| 538 | } |
| 539 | |
Lei Zhang | 571f932 | 2019-09-25 12:40:25 +0000 | [diff] [blame] | 540 | TEST_F(FPDFEditEmbedderTest, BUG_1399) { |
| 541 | // Load document with a clipped rectangle. |
| 542 | EXPECT_TRUE(OpenDocument("bug_1399.pdf")); |
| 543 | FPDF_PAGE page = LoadPage(0); |
| 544 | ASSERT_TRUE(page); |
| 545 | |
| 546 | ASSERT_EQ(7, FPDFPage_CountObjects(page)); |
| 547 | |
| 548 | FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 0); |
| 549 | ASSERT_TRUE(obj); |
| 550 | |
| 551 | ASSERT_EQ(2, FPDFPath_CountSegments(obj)); |
| 552 | |
| 553 | FPDF_PATHSEGMENT segment = FPDFPath_GetPathSegment(obj, 0); |
| 554 | float x; |
| 555 | float y; |
| 556 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 557 | EXPECT_FLOAT_EQ(107.718f, x); |
| 558 | EXPECT_FLOAT_EQ(719.922f, y); |
| 559 | EXPECT_EQ(FPDF_SEGMENT_MOVETO, FPDFPathSegment_GetType(segment)); |
| 560 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 561 | |
| 562 | segment = FPDFPath_GetPathSegment(obj, 1); |
| 563 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 564 | EXPECT_FLOAT_EQ(394.718f, x); |
| 565 | EXPECT_FLOAT_EQ(719.922f, y); |
| 566 | EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment)); |
| 567 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 568 | |
| 569 | FPDF_CLIPPATH clip_path = FPDFPageObj_GetClipPath(obj); |
| 570 | ASSERT_TRUE(clip_path); |
| 571 | |
| 572 | EXPECT_EQ(-1, FPDFClipPath_CountPaths(clip_path)); |
| 573 | EXPECT_EQ(-1, FPDFClipPath_CountPathSegments(clip_path, 0)); |
| 574 | EXPECT_FALSE(FPDFClipPath_GetPathSegment(clip_path, 0, 0)); |
| 575 | |
| 576 | UnloadPage(page); |
| 577 | } |
| 578 | |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 579 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 580 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 581 | #define MAYBE_SetText DISABLED_SetText |
| 582 | #else |
| 583 | #define MAYBE_SetText SetText |
| 584 | #endif |
| 585 | TEST_F(FPDFEditEmbedderTest, MAYBE_SetText) { |
Henrique Nakashima | 5ebfd64 | 2018-06-07 15:18:55 +0000 | [diff] [blame] | 586 | // Load document with some text. |
| 587 | EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
| 588 | FPDF_PAGE page = LoadPage(0); |
| 589 | ASSERT_TRUE(page); |
| 590 | |
| 591 | // Get the "Hello, world!" text object and change it. |
| 592 | ASSERT_EQ(2, FPDFPage_CountObjects(page)); |
| 593 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0); |
| 594 | ASSERT_TRUE(page_object); |
Lei Zhang | f0f6768 | 2019-04-08 17:03:21 +0000 | [diff] [blame] | 595 | ScopedFPDFWideString text1 = GetFPDFWideString(L"Changed for SetText test"); |
Henrique Nakashima | 5ebfd64 | 2018-06-07 15:18:55 +0000 | [diff] [blame] | 596 | EXPECT_TRUE(FPDFText_SetText(page_object, text1.get())); |
| 597 | |
| 598 | // Verify the "Hello, world!" text is gone and "Changed for SetText test" is |
| 599 | // now displayed. |
| 600 | ASSERT_EQ(2, FPDFPage_CountObjects(page)); |
Lei Zhang | e4cdac5 | 2019-04-30 16:45:57 +0000 | [diff] [blame] | 601 | #if defined(OS_MACOSX) |
Lei Zhang | 0f6b434 | 2020-02-25 20:00:39 +0000 | [diff] [blame] | 602 | const char kChangedMD5[] = "904132275a1144ea06b0694537c80b4c"; |
Lei Zhang | e67bcc7 | 2019-04-30 18:55:58 +0000 | [diff] [blame] | 603 | #elif defined(OS_WIN) |
Lei Zhang | 5e2c51c | 2018-07-27 22:33:34 +0000 | [diff] [blame] | 604 | const char kChangedMD5[] = "3137fdb27962671f5c3963a5e965eff5"; |
Henrique Nakashima | 5ebfd64 | 2018-06-07 15:18:55 +0000 | [diff] [blame] | 605 | #else |
| 606 | const char kChangedMD5[] = "a0c4ea6620772991f66bf7130379b08a"; |
| 607 | #endif |
| 608 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 609 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Henrique Nakashima | 5ebfd64 | 2018-06-07 15:18:55 +0000 | [diff] [blame] | 610 | CompareBitmap(page_bitmap.get(), 200, 200, kChangedMD5); |
| 611 | } |
| 612 | |
| 613 | // Now save the result. |
| 614 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 615 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 616 | |
| 617 | UnloadPage(page); |
| 618 | |
| 619 | // Re-open the file and check the changes were kept in the saved .pdf. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 620 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | 5ebfd64 | 2018-06-07 15:18:55 +0000 | [diff] [blame] | 621 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 622 | EXPECT_EQ(2, FPDFPage_CountObjects(saved_page)); |
| 623 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 624 | ScopedFPDFBitmap page_bitmap = RenderPage(saved_page); |
Henrique Nakashima | 5ebfd64 | 2018-06-07 15:18:55 +0000 | [diff] [blame] | 625 | CompareBitmap(page_bitmap.get(), 200, 200, kChangedMD5); |
| 626 | } |
| 627 | |
| 628 | CloseSavedPage(saved_page); |
| 629 | CloseSavedDocument(); |
| 630 | } |
| 631 | |
Hui Yingst | 08c4071 | 2020-04-29 01:37:35 +0000 | [diff] [blame^] | 632 | TEST_F(FPDFEditEmbedderTest, RemovePageObject) { |
Henrique Nakashima | 35841fa | 2018-03-15 15:25:16 +0000 | [diff] [blame] | 633 | // Load document with some text. |
| 634 | EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
| 635 | FPDF_PAGE page = LoadPage(0); |
| 636 | ASSERT_TRUE(page); |
| 637 | |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 638 | // Show what the original file looks like. |
| 639 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 640 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 641 | CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldChecksum); |
Henrique Nakashima | 35841fa | 2018-03-15 15:25:16 +0000 | [diff] [blame] | 642 | } |
| 643 | |
| 644 | // Get the "Hello, world!" text object and remove it. |
| 645 | ASSERT_EQ(2, FPDFPage_CountObjects(page)); |
| 646 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0); |
| 647 | ASSERT_TRUE(page_object); |
| 648 | EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object)); |
| 649 | |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 650 | // Verify the "Hello, world!" text is gone. |
| 651 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 652 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 653 | CompareBitmap(page_bitmap.get(), 200, 200, kFirstRemovedChecksum); |
Henrique Nakashima | 35841fa | 2018-03-15 15:25:16 +0000 | [diff] [blame] | 654 | } |
| 655 | ASSERT_EQ(1, FPDFPage_CountObjects(page)); |
| 656 | |
| 657 | UnloadPage(page); |
| 658 | FPDFPageObj_Destroy(page_object); |
| 659 | } |
| 660 | |
Henrique Nakashima | cbed949 | 2018-07-10 21:54:26 +0000 | [diff] [blame] | 661 | void CheckMarkCounts(FPDF_PAGE page, |
| 662 | int start_from, |
| 663 | int expected_object_count, |
| 664 | size_t expected_prime_count, |
| 665 | size_t expected_square_count, |
| 666 | size_t expected_greater_than_ten_count, |
| 667 | size_t expected_bounds_count) { |
| 668 | int object_count = FPDFPage_CountObjects(page); |
| 669 | ASSERT_EQ(expected_object_count, object_count); |
| 670 | |
| 671 | size_t prime_count = 0; |
| 672 | size_t square_count = 0; |
| 673 | size_t greater_than_ten_count = 0; |
| 674 | size_t bounds_count = 0; |
| 675 | for (int i = 0; i < object_count; ++i) { |
| 676 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i); |
| 677 | |
| 678 | int mark_count = FPDFPageObj_CountMarks(page_object); |
| 679 | for (int j = 0; j < mark_count; ++j) { |
| 680 | FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, j); |
| 681 | |
| 682 | char buffer[256]; |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 683 | unsigned long name_len = 999u; |
| 684 | ASSERT_TRUE( |
| 685 | FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len)); |
| 686 | EXPECT_GT(name_len, 0u); |
| 687 | EXPECT_NE(999u, name_len); |
Henrique Nakashima | cbed949 | 2018-07-10 21:54:26 +0000 | [diff] [blame] | 688 | std::wstring name = |
| 689 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)); |
| 690 | if (name == L"Prime") { |
| 691 | prime_count++; |
| 692 | } else if (name == L"Square") { |
| 693 | square_count++; |
| 694 | int expected_square = start_from + i; |
| 695 | EXPECT_EQ(1, FPDFPageObjMark_CountParams(mark)); |
| 696 | |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 697 | unsigned long get_param_key_return = 999u; |
| 698 | ASSERT_TRUE(FPDFPageObjMark_GetParamKey(mark, 0, buffer, sizeof(buffer), |
| 699 | &get_param_key_return)); |
Henrique Nakashima | cbed949 | 2018-07-10 21:54:26 +0000 | [diff] [blame] | 700 | EXPECT_EQ((6u + 1u) * 2u, get_param_key_return); |
| 701 | std::wstring key = |
| 702 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)); |
| 703 | EXPECT_EQ(L"Factor", key); |
| 704 | |
| 705 | EXPECT_EQ(FPDF_OBJECT_NUMBER, |
Henrique Nakashima | 94230e5 | 2018-07-11 22:02:02 +0000 | [diff] [blame] | 706 | FPDFPageObjMark_GetParamValueType(mark, "Factor")); |
Henrique Nakashima | 140dead | 2018-07-11 21:40:03 +0000 | [diff] [blame] | 707 | int square_root; |
Henrique Nakashima | 94230e5 | 2018-07-11 22:02:02 +0000 | [diff] [blame] | 708 | EXPECT_TRUE( |
| 709 | FPDFPageObjMark_GetParamIntValue(mark, "Factor", &square_root)); |
Henrique Nakashima | cbed949 | 2018-07-10 21:54:26 +0000 | [diff] [blame] | 710 | EXPECT_EQ(expected_square, square_root * square_root); |
| 711 | } else if (name == L"GreaterThanTen") { |
| 712 | greater_than_ten_count++; |
| 713 | } else if (name == L"Bounds") { |
| 714 | bounds_count++; |
| 715 | EXPECT_EQ(1, FPDFPageObjMark_CountParams(mark)); |
| 716 | |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 717 | unsigned long get_param_key_return = 999u; |
| 718 | ASSERT_TRUE(FPDFPageObjMark_GetParamKey(mark, 0, buffer, sizeof(buffer), |
| 719 | &get_param_key_return)); |
Henrique Nakashima | cbed949 | 2018-07-10 21:54:26 +0000 | [diff] [blame] | 720 | EXPECT_EQ((8u + 1u) * 2u, get_param_key_return); |
| 721 | std::wstring key = |
| 722 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)); |
| 723 | EXPECT_EQ(L"Position", key); |
| 724 | |
Henrique Nakashima | cbed949 | 2018-07-10 21:54:26 +0000 | [diff] [blame] | 725 | EXPECT_EQ(FPDF_OBJECT_STRING, |
Henrique Nakashima | 94230e5 | 2018-07-11 22:02:02 +0000 | [diff] [blame] | 726 | FPDFPageObjMark_GetParamValueType(mark, "Position")); |
Henrique Nakashima | cbed949 | 2018-07-10 21:54:26 +0000 | [diff] [blame] | 727 | unsigned long length; |
Henrique Nakashima | a340677 | 2018-07-13 19:10:53 +0000 | [diff] [blame] | 728 | EXPECT_TRUE(FPDFPageObjMark_GetParamStringValue( |
| 729 | mark, "Position", buffer, sizeof(buffer), &length)); |
Henrique Nakashima | cbed949 | 2018-07-10 21:54:26 +0000 | [diff] [blame] | 730 | ASSERT_GT(length, 0u); |
Henrique Nakashima | 140dead | 2018-07-11 21:40:03 +0000 | [diff] [blame] | 731 | std::wstring value = |
| 732 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)); |
Henrique Nakashima | d8df8c3 | 2018-07-12 22:15:09 +0000 | [diff] [blame] | 733 | |
Henrique Nakashima | a340677 | 2018-07-13 19:10:53 +0000 | [diff] [blame] | 734 | // "Position" can be "First", "Last", or "End". |
Henrique Nakashima | d8df8c3 | 2018-07-12 22:15:09 +0000 | [diff] [blame] | 735 | if (i == 0) { |
| 736 | EXPECT_EQ((5u + 1u) * 2u, length); |
| 737 | EXPECT_EQ(L"First", value); |
| 738 | } else if (i == object_count - 1) { |
Henrique Nakashima | a340677 | 2018-07-13 19:10:53 +0000 | [diff] [blame] | 739 | if (length == (4u + 1u) * 2u) { |
| 740 | EXPECT_EQ(L"Last", value); |
| 741 | } else if (length == (3u + 1u) * 2u) { |
| 742 | EXPECT_EQ(L"End", value); |
| 743 | } else { |
| 744 | FAIL(); |
| 745 | } |
Henrique Nakashima | d8df8c3 | 2018-07-12 22:15:09 +0000 | [diff] [blame] | 746 | } else { |
| 747 | FAIL(); |
| 748 | } |
Henrique Nakashima | cbed949 | 2018-07-10 21:54:26 +0000 | [diff] [blame] | 749 | } else { |
| 750 | FAIL(); |
| 751 | } |
| 752 | } |
| 753 | } |
| 754 | |
| 755 | // Expect certain number of tagged objects. The test file contains strings |
| 756 | // from 1 to 19. |
| 757 | EXPECT_EQ(expected_prime_count, prime_count); |
| 758 | EXPECT_EQ(expected_square_count, square_count); |
| 759 | EXPECT_EQ(expected_greater_than_ten_count, greater_than_ten_count); |
| 760 | EXPECT_EQ(expected_bounds_count, bounds_count); |
| 761 | } |
| 762 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 763 | TEST_F(FPDFEditEmbedderTest, ReadMarkedObjectsIndirectDict) { |
Henrique Nakashima | cbed949 | 2018-07-10 21:54:26 +0000 | [diff] [blame] | 764 | // Load document with some text marked with an indirect property. |
| 765 | EXPECT_TRUE(OpenDocument("text_in_page_marked_indirect.pdf")); |
| 766 | FPDF_PAGE page = LoadPage(0); |
| 767 | ASSERT_TRUE(page); |
| 768 | |
| 769 | CheckMarkCounts(page, 1, 19, 8, 4, 9, 1); |
| 770 | |
| 771 | UnloadPage(page); |
| 772 | } |
| 773 | |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 774 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 775 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 776 | #define MAYBE_RemoveMarkedObjectsPrime DISABLED_RemoveMarkedObjectsPrime |
| 777 | #else |
| 778 | #define MAYBE_RemoveMarkedObjectsPrime RemoveMarkedObjectsPrime |
| 779 | #endif |
| 780 | TEST_F(FPDFEditEmbedderTest, MAYBE_RemoveMarkedObjectsPrime) { |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 781 | // Load document with some text. |
| 782 | EXPECT_TRUE(OpenDocument("text_in_page_marked.pdf")); |
| 783 | FPDF_PAGE page = LoadPage(0); |
| 784 | ASSERT_TRUE(page); |
| 785 | |
| 786 | // Show what the original file looks like. |
| 787 | { |
Lei Zhang | e4cdac5 | 2019-04-30 16:45:57 +0000 | [diff] [blame] | 788 | #if defined(OS_MACOSX) |
Lei Zhang | 0f6b434 | 2020-02-25 20:00:39 +0000 | [diff] [blame] | 789 | const char kOriginalMD5[] = "adf815e53c788a5272b4df07c610a1da"; |
Lei Zhang | e67bcc7 | 2019-04-30 18:55:58 +0000 | [diff] [blame] | 790 | #elif defined(OS_WIN) |
Lei Zhang | a2b7073 | 2019-06-25 08:34:22 +0000 | [diff] [blame] | 791 | const char kOriginalMD5[] = "00542ee435b37749c4453be63bf7bdb6"; |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 792 | #else |
Lei Zhang | a2b7073 | 2019-06-25 08:34:22 +0000 | [diff] [blame] | 793 | const char kOriginalMD5[] = "41647268d5911d049801803b15c2dfb0"; |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 794 | #endif |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 795 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 796 | CompareBitmap(page_bitmap.get(), 200, 200, kOriginalMD5); |
| 797 | } |
| 798 | |
Henrique Nakashima | cbed949 | 2018-07-10 21:54:26 +0000 | [diff] [blame] | 799 | constexpr int expected_object_count = 19; |
| 800 | CheckMarkCounts(page, 1, expected_object_count, 8, 4, 9, 1); |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 801 | |
Henrique Nakashima | cbed949 | 2018-07-10 21:54:26 +0000 | [diff] [blame] | 802 | // Get all objects marked with "Prime" |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 803 | std::vector<FPDF_PAGEOBJECT> primes; |
Henrique Nakashima | cbed949 | 2018-07-10 21:54:26 +0000 | [diff] [blame] | 804 | for (int i = 0; i < expected_object_count; ++i) { |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 805 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i); |
| 806 | |
| 807 | int mark_count = FPDFPageObj_CountMarks(page_object); |
| 808 | for (int j = 0; j < mark_count; ++j) { |
| 809 | FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, j); |
| 810 | |
| 811 | char buffer[256]; |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 812 | unsigned long name_len = 999u; |
| 813 | ASSERT_TRUE( |
| 814 | FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len)); |
| 815 | EXPECT_GT(name_len, 0u); |
| 816 | EXPECT_NE(999u, name_len); |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 817 | std::wstring name = |
| 818 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)); |
| 819 | if (name == L"Prime") { |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 820 | primes.push_back(page_object); |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 821 | } |
| 822 | } |
| 823 | } |
| 824 | |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 825 | // Remove all objects marked with "Prime". |
| 826 | for (FPDF_PAGEOBJECT page_object : primes) { |
| 827 | EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object)); |
| 828 | FPDFPageObj_Destroy(page_object); |
| 829 | } |
| 830 | |
| 831 | EXPECT_EQ(11, FPDFPage_CountObjects(page)); |
| 832 | |
Lei Zhang | e4cdac5 | 2019-04-30 16:45:57 +0000 | [diff] [blame] | 833 | #if defined(OS_MACOSX) |
Lei Zhang | 0f6b434 | 2020-02-25 20:00:39 +0000 | [diff] [blame] | 834 | const char kNonPrimesMD5[] = "d29e2ddff56e0d12f340794d26796400"; |
| 835 | const char kNonPrimesAfterSaveMD5[] = "10eff2cd0037b661496981779601fa6f"; |
Lei Zhang | e67bcc7 | 2019-04-30 18:55:58 +0000 | [diff] [blame] | 836 | #elif defined(OS_WIN) |
Lei Zhang | a2b7073 | 2019-06-25 08:34:22 +0000 | [diff] [blame] | 837 | const char kNonPrimesMD5[] = "86e371fdae30c2471f476631f3f93413"; |
| 838 | const char kNonPrimesAfterSaveMD5[] = "86e371fdae30c2471f476631f3f93413"; |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 839 | #else |
Lei Zhang | a2b7073 | 2019-06-25 08:34:22 +0000 | [diff] [blame] | 840 | const char kNonPrimesMD5[] = "67ab13115d0cc34e99a1003c28047b40"; |
| 841 | const char kNonPrimesAfterSaveMD5[] = "67ab13115d0cc34e99a1003c28047b40"; |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 842 | #endif |
Henrique Nakashima | a5078b7 | 2018-09-10 17:09:42 +0000 | [diff] [blame] | 843 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 844 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 845 | CompareBitmap(page_bitmap.get(), 200, 200, kNonPrimesMD5); |
| 846 | } |
| 847 | |
Henrique Nakashima | a5078b7 | 2018-09-10 17:09:42 +0000 | [diff] [blame] | 848 | // Save the file. |
| 849 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 850 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 851 | UnloadPage(page); |
Henrique Nakashima | a5078b7 | 2018-09-10 17:09:42 +0000 | [diff] [blame] | 852 | |
| 853 | // Re-open the file and check the prime marks are not there anymore. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 854 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | a5078b7 | 2018-09-10 17:09:42 +0000 | [diff] [blame] | 855 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 856 | EXPECT_EQ(11, FPDFPage_CountObjects(saved_page)); |
| 857 | |
| 858 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 859 | ScopedFPDFBitmap page_bitmap = RenderPage(saved_page); |
Manoj Biswas | 7daf7a2 | 2019-03-22 23:02:34 +0000 | [diff] [blame] | 860 | CompareBitmap(page_bitmap.get(), 200, 200, kNonPrimesAfterSaveMD5); |
Henrique Nakashima | a5078b7 | 2018-09-10 17:09:42 +0000 | [diff] [blame] | 861 | } |
| 862 | |
| 863 | CloseSavedPage(saved_page); |
| 864 | CloseSavedDocument(); |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 865 | } |
| 866 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 867 | TEST_F(FPDFEditEmbedderTest, RemoveMarks) { |
Henrique Nakashima | fed4adb | 2018-07-13 19:47:22 +0000 | [diff] [blame] | 868 | // Load document with some text. |
| 869 | EXPECT_TRUE(OpenDocument("text_in_page_marked.pdf")); |
| 870 | FPDF_PAGE page = LoadPage(0); |
| 871 | ASSERT_TRUE(page); |
| 872 | |
| 873 | constexpr int kExpectedObjectCount = 19; |
| 874 | CheckMarkCounts(page, 1, kExpectedObjectCount, 8, 4, 9, 1); |
| 875 | |
| 876 | // Remove all "Prime" content marks. |
| 877 | for (int i = 0; i < kExpectedObjectCount; ++i) { |
| 878 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i); |
| 879 | |
| 880 | int mark_count = FPDFPageObj_CountMarks(page_object); |
| 881 | for (int j = mark_count - 1; j >= 0; --j) { |
| 882 | FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, j); |
| 883 | |
| 884 | char buffer[256]; |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 885 | unsigned long name_len = 999u; |
| 886 | ASSERT_TRUE( |
| 887 | FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len)); |
| 888 | EXPECT_GT(name_len, 0u); |
| 889 | EXPECT_NE(999u, name_len); |
Henrique Nakashima | fed4adb | 2018-07-13 19:47:22 +0000 | [diff] [blame] | 890 | std::wstring name = |
| 891 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)); |
| 892 | if (name == L"Prime") { |
| 893 | // Remove mark. |
| 894 | EXPECT_TRUE(FPDFPageObj_RemoveMark(page_object, mark)); |
| 895 | |
| 896 | // Verify there is now one fewer mark in the page object. |
| 897 | EXPECT_EQ(mark_count - 1, FPDFPageObj_CountMarks(page_object)); |
| 898 | } |
| 899 | } |
| 900 | } |
| 901 | |
| 902 | // Verify there are 0 "Prime" content marks now. |
| 903 | CheckMarkCounts(page, 1, kExpectedObjectCount, 0, 4, 9, 1); |
| 904 | |
| 905 | // Save the file. |
| 906 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 907 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 908 | UnloadPage(page); |
| 909 | |
| 910 | // Re-open the file and check the prime marks are not there anymore. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 911 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | fed4adb | 2018-07-13 19:47:22 +0000 | [diff] [blame] | 912 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 913 | |
| 914 | CheckMarkCounts(saved_page, 1, kExpectedObjectCount, 0, 4, 9, 1); |
| 915 | |
| 916 | CloseSavedPage(saved_page); |
| 917 | CloseSavedDocument(); |
| 918 | } |
| 919 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 920 | TEST_F(FPDFEditEmbedderTest, RemoveMarkParam) { |
Henrique Nakashima | cf403ba | 2018-07-13 20:12:41 +0000 | [diff] [blame] | 921 | // Load document with some text. |
| 922 | EXPECT_TRUE(OpenDocument("text_in_page_marked.pdf")); |
| 923 | FPDF_PAGE page = LoadPage(0); |
| 924 | ASSERT_TRUE(page); |
| 925 | |
| 926 | constexpr int kExpectedObjectCount = 19; |
| 927 | CheckMarkCounts(page, 1, kExpectedObjectCount, 8, 4, 9, 1); |
| 928 | |
| 929 | // Remove all "Square" content marks parameters. |
| 930 | for (int i = 0; i < kExpectedObjectCount; ++i) { |
| 931 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i); |
| 932 | |
| 933 | int mark_count = FPDFPageObj_CountMarks(page_object); |
| 934 | for (int j = 0; j < mark_count; ++j) { |
| 935 | FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, j); |
| 936 | |
| 937 | char buffer[256]; |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 938 | unsigned long name_len = 999u; |
| 939 | ASSERT_TRUE( |
| 940 | FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len)); |
| 941 | EXPECT_GT(name_len, 0u); |
| 942 | EXPECT_NE(999u, name_len); |
Henrique Nakashima | cf403ba | 2018-07-13 20:12:41 +0000 | [diff] [blame] | 943 | std::wstring name = |
| 944 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)); |
| 945 | if (name == L"Square") { |
| 946 | // Show the mark has a "Factor" parameter. |
| 947 | int out_value; |
| 948 | EXPECT_TRUE( |
| 949 | FPDFPageObjMark_GetParamIntValue(mark, "Factor", &out_value)); |
| 950 | |
| 951 | // Remove parameter. |
| 952 | EXPECT_TRUE(FPDFPageObjMark_RemoveParam(page_object, mark, "Factor")); |
| 953 | |
| 954 | // Verify the "Factor" parameter is gone. |
| 955 | EXPECT_FALSE( |
| 956 | FPDFPageObjMark_GetParamIntValue(mark, "Factor", &out_value)); |
| 957 | } |
| 958 | } |
| 959 | } |
| 960 | |
| 961 | // Save the file. |
| 962 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 963 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 964 | UnloadPage(page); |
| 965 | |
| 966 | // Re-open the file and check the "Factor" parameters are still gone. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 967 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | cf403ba | 2018-07-13 20:12:41 +0000 | [diff] [blame] | 968 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 969 | |
| 970 | size_t square_count = 0; |
| 971 | for (int i = 0; i < kExpectedObjectCount; ++i) { |
| 972 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(saved_page, i); |
| 973 | |
| 974 | int mark_count = FPDFPageObj_CountMarks(page_object); |
| 975 | for (int j = 0; j < mark_count; ++j) { |
| 976 | FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, j); |
| 977 | |
| 978 | char buffer[256]; |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 979 | unsigned long name_len = 999u; |
| 980 | ASSERT_TRUE( |
| 981 | FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len)); |
| 982 | EXPECT_GT(name_len, 0u); |
| 983 | EXPECT_NE(999u, name_len); |
Henrique Nakashima | cf403ba | 2018-07-13 20:12:41 +0000 | [diff] [blame] | 984 | std::wstring name = |
| 985 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)); |
| 986 | if (name == L"Square") { |
| 987 | // Verify the "Factor" parameter is still gone. |
| 988 | int out_value; |
| 989 | EXPECT_FALSE( |
| 990 | FPDFPageObjMark_GetParamIntValue(mark, "Factor", &out_value)); |
| 991 | |
| 992 | ++square_count; |
| 993 | } |
| 994 | } |
| 995 | } |
| 996 | |
| 997 | // Verify the parameters are gone, but the marks are not. |
| 998 | EXPECT_EQ(4u, square_count); |
| 999 | |
| 1000 | CloseSavedPage(saved_page); |
| 1001 | CloseSavedDocument(); |
| 1002 | } |
| 1003 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 1004 | TEST_F(FPDFEditEmbedderTest, MaintainMarkedObjects) { |
Henrique Nakashima | b4bcf69 | 2018-07-11 21:19:22 +0000 | [diff] [blame] | 1005 | // Load document with some text. |
| 1006 | EXPECT_TRUE(OpenDocument("text_in_page_marked.pdf")); |
| 1007 | FPDF_PAGE page = LoadPage(0); |
| 1008 | ASSERT_TRUE(page); |
| 1009 | |
| 1010 | // Iterate over all objects, counting the number of times each content mark |
| 1011 | // name appears. |
| 1012 | CheckMarkCounts(page, 1, 19, 8, 4, 9, 1); |
| 1013 | |
| 1014 | // Remove first page object. |
| 1015 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0); |
| 1016 | EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object)); |
| 1017 | FPDFPageObj_Destroy(page_object); |
| 1018 | |
| 1019 | CheckMarkCounts(page, 2, 18, 8, 3, 9, 1); |
| 1020 | |
| 1021 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1022 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1023 | |
| 1024 | UnloadPage(page); |
| 1025 | |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1026 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | b4bcf69 | 2018-07-11 21:19:22 +0000 | [diff] [blame] | 1027 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1028 | |
| 1029 | CheckMarkCounts(saved_page, 2, 18, 8, 3, 9, 1); |
| 1030 | |
| 1031 | CloseSavedPage(saved_page); |
| 1032 | CloseSavedDocument(); |
| 1033 | } |
| 1034 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 1035 | TEST_F(FPDFEditEmbedderTest, MaintainIndirectMarkedObjects) { |
Henrique Nakashima | b4bcf69 | 2018-07-11 21:19:22 +0000 | [diff] [blame] | 1036 | // Load document with some text. |
| 1037 | EXPECT_TRUE(OpenDocument("text_in_page_marked_indirect.pdf")); |
| 1038 | FPDF_PAGE page = LoadPage(0); |
| 1039 | ASSERT_TRUE(page); |
| 1040 | |
| 1041 | // Iterate over all objects, counting the number of times each content mark |
| 1042 | // name appears. |
| 1043 | CheckMarkCounts(page, 1, 19, 8, 4, 9, 1); |
| 1044 | |
| 1045 | // Remove first page object. |
| 1046 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0); |
| 1047 | EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object)); |
| 1048 | FPDFPageObj_Destroy(page_object); |
| 1049 | |
| 1050 | CheckMarkCounts(page, 2, 18, 8, 3, 9, 1); |
| 1051 | |
| 1052 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1053 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1054 | |
| 1055 | UnloadPage(page); |
| 1056 | |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1057 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | b4bcf69 | 2018-07-11 21:19:22 +0000 | [diff] [blame] | 1058 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1059 | |
| 1060 | CheckMarkCounts(saved_page, 2, 18, 8, 3, 9, 1); |
| 1061 | |
| 1062 | CloseSavedPage(saved_page); |
| 1063 | CloseSavedDocument(); |
| 1064 | } |
| 1065 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 1066 | TEST_F(FPDFEditEmbedderTest, RemoveExistingPageObject) { |
Henrique Nakashima | c49e62e | 2018-04-16 20:58:47 +0000 | [diff] [blame] | 1067 | // Load document with some text. |
| 1068 | EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
| 1069 | FPDF_PAGE page = LoadPage(0); |
| 1070 | ASSERT_TRUE(page); |
| 1071 | |
| 1072 | // Get the "Hello, world!" text object and remove it. |
| 1073 | ASSERT_EQ(2, FPDFPage_CountObjects(page)); |
| 1074 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0); |
| 1075 | ASSERT_TRUE(page_object); |
| 1076 | EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object)); |
| 1077 | |
| 1078 | // Verify the "Hello, world!" text is gone. |
| 1079 | ASSERT_EQ(1, FPDFPage_CountObjects(page)); |
| 1080 | |
| 1081 | // Save the file |
| 1082 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1083 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1084 | UnloadPage(page); |
| 1085 | FPDFPageObj_Destroy(page_object); |
| 1086 | |
| 1087 | // Re-open the file and check the page object count is still 1. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1088 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | c49e62e | 2018-04-16 20:58:47 +0000 | [diff] [blame] | 1089 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1090 | EXPECT_EQ(1, FPDFPage_CountObjects(saved_page)); |
| 1091 | CloseSavedPage(saved_page); |
| 1092 | CloseSavedDocument(); |
| 1093 | } |
| 1094 | |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 1095 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 1096 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 1097 | #define MAYBE_RemoveExistingPageObjectSplitStreamsNotLonely \ |
| 1098 | DISABLED_RemoveExistingPageObjectSplitStreamsNotLonely |
| 1099 | #else |
| 1100 | #define MAYBE_RemoveExistingPageObjectSplitStreamsNotLonely \ |
| 1101 | RemoveExistingPageObjectSplitStreamsNotLonely |
| 1102 | #endif |
| 1103 | TEST_F(FPDFEditEmbedderTest, |
| 1104 | MAYBE_RemoveExistingPageObjectSplitStreamsNotLonely) { |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1105 | // Load document with some text. |
| 1106 | EXPECT_TRUE(OpenDocument("hello_world_split_streams.pdf")); |
| 1107 | FPDF_PAGE page = LoadPage(0); |
| 1108 | ASSERT_TRUE(page); |
| 1109 | |
| 1110 | // Get the "Hello, world!" text object and remove it. There is another object |
| 1111 | // in the same stream that says "Goodbye, world!" |
| 1112 | ASSERT_EQ(3, FPDFPage_CountObjects(page)); |
| 1113 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0); |
| 1114 | ASSERT_TRUE(page_object); |
| 1115 | EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object)); |
| 1116 | |
| 1117 | // Verify the "Hello, world!" text is gone. |
| 1118 | ASSERT_EQ(2, FPDFPage_CountObjects(page)); |
Lei Zhang | e4cdac5 | 2019-04-30 16:45:57 +0000 | [diff] [blame] | 1119 | #if defined(OS_MACOSX) |
Lei Zhang | 0f6b434 | 2020-02-25 20:00:39 +0000 | [diff] [blame] | 1120 | const char kHelloRemovedMD5[] = "3b3b27602a86dfe5996a33c42c59885b"; |
Lei Zhang | e67bcc7 | 2019-04-30 18:55:58 +0000 | [diff] [blame] | 1121 | #elif defined(OS_WIN) |
Lei Zhang | 5e2c51c | 2018-07-27 22:33:34 +0000 | [diff] [blame] | 1122 | const char kHelloRemovedMD5[] = "a97d4c72c969ba373c2dce675d277e65"; |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1123 | #else |
| 1124 | const char kHelloRemovedMD5[] = "95b92950647a2190e1230911e7a1a0e9"; |
| 1125 | #endif |
| 1126 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1127 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1128 | CompareBitmap(page_bitmap.get(), 200, 200, kHelloRemovedMD5); |
| 1129 | } |
| 1130 | |
| 1131 | // Save the file |
| 1132 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1133 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1134 | UnloadPage(page); |
| 1135 | FPDFPageObj_Destroy(page_object); |
| 1136 | |
| 1137 | // Re-open the file and check the page object count is still 2. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1138 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1139 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1140 | |
| 1141 | EXPECT_EQ(2, FPDFPage_CountObjects(saved_page)); |
| 1142 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1143 | ScopedFPDFBitmap page_bitmap = RenderPage(saved_page); |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1144 | CompareBitmap(page_bitmap.get(), 200, 200, kHelloRemovedMD5); |
| 1145 | } |
| 1146 | |
| 1147 | CloseSavedPage(saved_page); |
| 1148 | CloseSavedDocument(); |
| 1149 | } |
| 1150 | |
Hui Yingst | 08c4071 | 2020-04-29 01:37:35 +0000 | [diff] [blame^] | 1151 | TEST_F(FPDFEditEmbedderTest, RemoveExistingPageObjectSplitStreamsLonely) { |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1152 | // Load document with some text. |
| 1153 | EXPECT_TRUE(OpenDocument("hello_world_split_streams.pdf")); |
| 1154 | FPDF_PAGE page = LoadPage(0); |
| 1155 | ASSERT_TRUE(page); |
| 1156 | |
| 1157 | // Get the "Greetings, world!" text object and remove it. This is the only |
| 1158 | // object in the stream. |
| 1159 | ASSERT_EQ(3, FPDFPage_CountObjects(page)); |
| 1160 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 2); |
| 1161 | ASSERT_TRUE(page_object); |
| 1162 | EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object)); |
| 1163 | |
| 1164 | // Verify the "Greetings, world!" text is gone. |
| 1165 | ASSERT_EQ(2, FPDFPage_CountObjects(page)); |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1166 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1167 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1168 | CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldChecksum); |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1169 | } |
| 1170 | |
| 1171 | // Save the file |
| 1172 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1173 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1174 | UnloadPage(page); |
| 1175 | FPDFPageObj_Destroy(page_object); |
| 1176 | |
| 1177 | // Re-open the file and check the page object count is still 2. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1178 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1179 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1180 | |
| 1181 | EXPECT_EQ(2, FPDFPage_CountObjects(saved_page)); |
| 1182 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1183 | ScopedFPDFBitmap page_bitmap = RenderPage(saved_page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1184 | CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldChecksum); |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1185 | } |
| 1186 | |
| 1187 | CloseSavedPage(saved_page); |
| 1188 | CloseSavedDocument(); |
| 1189 | } |
| 1190 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 1191 | TEST_F(FPDFEditEmbedderTest, GetContentStream) { |
Henrique Nakashima | 6eb7939 | 2018-06-12 20:27:35 +0000 | [diff] [blame] | 1192 | // Load document with some text split across streams. |
| 1193 | EXPECT_TRUE(OpenDocument("split_streams.pdf")); |
| 1194 | FPDF_PAGE page = LoadPage(0); |
| 1195 | ASSERT_TRUE(page); |
| 1196 | |
| 1197 | // Content stream 0: page objects 0-14. |
| 1198 | // Content stream 1: page objects 15-17. |
| 1199 | // Content stream 2: page object 18. |
| 1200 | ASSERT_EQ(19, FPDFPage_CountObjects(page)); |
| 1201 | for (int i = 0; i < 19; i++) { |
| 1202 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i); |
| 1203 | ASSERT_TRUE(page_object); |
| 1204 | CPDF_PageObject* cpdf_page_object = |
| 1205 | CPDFPageObjectFromFPDFPageObject(page_object); |
| 1206 | if (i < 15) |
| 1207 | EXPECT_EQ(0, cpdf_page_object->GetContentStream()) << i; |
| 1208 | else if (i < 18) |
| 1209 | EXPECT_EQ(1, cpdf_page_object->GetContentStream()) << i; |
| 1210 | else |
| 1211 | EXPECT_EQ(2, cpdf_page_object->GetContentStream()) << i; |
| 1212 | } |
| 1213 | |
| 1214 | UnloadPage(page); |
| 1215 | } |
| 1216 | |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 1217 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 1218 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 1219 | #define MAYBE_RemoveAllFromStream DISABLED_RemoveAllFromStream |
| 1220 | #else |
| 1221 | #define MAYBE_RemoveAllFromStream RemoveAllFromStream |
| 1222 | #endif |
| 1223 | TEST_F(FPDFEditEmbedderTest, MAYBE_RemoveAllFromStream) { |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1224 | // Load document with some text split across streams. |
| 1225 | EXPECT_TRUE(OpenDocument("split_streams.pdf")); |
| 1226 | FPDF_PAGE page = LoadPage(0); |
| 1227 | ASSERT_TRUE(page); |
| 1228 | |
| 1229 | // Content stream 0: page objects 0-14. |
| 1230 | // Content stream 1: page objects 15-17. |
| 1231 | // Content stream 2: page object 18. |
| 1232 | ASSERT_EQ(19, FPDFPage_CountObjects(page)); |
| 1233 | |
| 1234 | // Loop backwards because objects will being removed, which shifts the indexes |
| 1235 | // after the removed position. |
| 1236 | for (int i = 18; i >= 0; i--) { |
| 1237 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i); |
| 1238 | ASSERT_TRUE(page_object); |
| 1239 | CPDF_PageObject* cpdf_page_object = |
| 1240 | CPDFPageObjectFromFPDFPageObject(page_object); |
| 1241 | |
| 1242 | // Empty content stream 1. |
| 1243 | if (cpdf_page_object->GetContentStream() == 1) { |
| 1244 | EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object)); |
| 1245 | FPDFPageObj_Destroy(page_object); |
| 1246 | } |
| 1247 | } |
| 1248 | |
| 1249 | // Content stream 0: page objects 0-14. |
| 1250 | // Content stream 2: page object 15. |
| 1251 | ASSERT_EQ(16, FPDFPage_CountObjects(page)); |
| 1252 | for (int i = 0; i < 16; i++) { |
| 1253 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i); |
| 1254 | ASSERT_TRUE(page_object); |
| 1255 | CPDF_PageObject* cpdf_page_object = |
| 1256 | CPDFPageObjectFromFPDFPageObject(page_object); |
| 1257 | if (i < 15) |
| 1258 | EXPECT_EQ(0, cpdf_page_object->GetContentStream()) << i; |
| 1259 | else |
| 1260 | EXPECT_EQ(2, cpdf_page_object->GetContentStream()) << i; |
| 1261 | } |
| 1262 | |
| 1263 | // Generate contents should remove the empty stream and update the page |
| 1264 | // objects' contents stream indexes. |
| 1265 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1266 | |
| 1267 | // Content stream 0: page objects 0-14. |
| 1268 | // Content stream 1: page object 15. |
| 1269 | ASSERT_EQ(16, FPDFPage_CountObjects(page)); |
| 1270 | for (int i = 0; i < 16; i++) { |
| 1271 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i); |
| 1272 | ASSERT_TRUE(page_object); |
| 1273 | CPDF_PageObject* cpdf_page_object = |
| 1274 | CPDFPageObjectFromFPDFPageObject(page_object); |
| 1275 | if (i < 15) |
| 1276 | EXPECT_EQ(0, cpdf_page_object->GetContentStream()) << i; |
| 1277 | else |
| 1278 | EXPECT_EQ(1, cpdf_page_object->GetContentStream()) << i; |
| 1279 | } |
| 1280 | |
Lei Zhang | e4cdac5 | 2019-04-30 16:45:57 +0000 | [diff] [blame] | 1281 | #if defined(OS_MACOSX) |
Lei Zhang | 0f6b434 | 2020-02-25 20:00:39 +0000 | [diff] [blame] | 1282 | const char kStream1RemovedMD5[] = "0e8856ca9abc7049412e64f9230c7c43"; |
Lei Zhang | e67bcc7 | 2019-04-30 18:55:58 +0000 | [diff] [blame] | 1283 | #elif defined(OS_WIN) |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1284 | const char kStream1RemovedMD5[] = "b4140f203523e38793283a5943d8075b"; |
| 1285 | #else |
| 1286 | const char kStream1RemovedMD5[] = "e86a3efc160ede6cfcb1f59bcacf1105"; |
| 1287 | #endif |
| 1288 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1289 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1290 | CompareBitmap(page_bitmap.get(), 200, 200, kStream1RemovedMD5); |
| 1291 | } |
| 1292 | |
| 1293 | // Save the file |
| 1294 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1295 | UnloadPage(page); |
| 1296 | |
| 1297 | // Re-open the file and check the page object count is still 16, and that |
| 1298 | // content stream 1 was removed. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1299 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1300 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1301 | |
| 1302 | // Content stream 0: page objects 0-14. |
| 1303 | // Content stream 1: page object 15. |
| 1304 | EXPECT_EQ(16, FPDFPage_CountObjects(saved_page)); |
| 1305 | for (int i = 0; i < 16; i++) { |
| 1306 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(saved_page, i); |
| 1307 | ASSERT_TRUE(page_object); |
| 1308 | CPDF_PageObject* cpdf_page_object = |
| 1309 | CPDFPageObjectFromFPDFPageObject(page_object); |
| 1310 | if (i < 15) |
| 1311 | EXPECT_EQ(0, cpdf_page_object->GetContentStream()) << i; |
| 1312 | else |
| 1313 | EXPECT_EQ(1, cpdf_page_object->GetContentStream()) << i; |
| 1314 | } |
| 1315 | |
| 1316 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1317 | ScopedFPDFBitmap page_bitmap = RenderPage(saved_page); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1318 | CompareBitmap(page_bitmap.get(), 200, 200, kStream1RemovedMD5); |
| 1319 | } |
| 1320 | |
| 1321 | CloseSavedPage(saved_page); |
| 1322 | CloseSavedDocument(); |
| 1323 | } |
| 1324 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 1325 | TEST_F(FPDFEditEmbedderTest, RemoveAllFromSingleStream) { |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1326 | // Load document with a single stream. |
| 1327 | EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
| 1328 | FPDF_PAGE page = LoadPage(0); |
| 1329 | ASSERT_TRUE(page); |
| 1330 | |
| 1331 | // Content stream 0: page objects 0-1. |
| 1332 | ASSERT_EQ(2, FPDFPage_CountObjects(page)); |
| 1333 | |
| 1334 | // Loop backwards because objects will being removed, which shifts the indexes |
| 1335 | // after the removed position. |
| 1336 | for (int i = 1; i >= 0; i--) { |
| 1337 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i); |
| 1338 | ASSERT_TRUE(page_object); |
| 1339 | CPDF_PageObject* cpdf_page_object = |
| 1340 | CPDFPageObjectFromFPDFPageObject(page_object); |
| 1341 | ASSERT_EQ(0, cpdf_page_object->GetContentStream()); |
| 1342 | ASSERT_TRUE(FPDFPage_RemoveObject(page, page_object)); |
| 1343 | FPDFPageObj_Destroy(page_object); |
| 1344 | } |
| 1345 | |
| 1346 | // No more objects in the stream |
| 1347 | ASSERT_EQ(0, FPDFPage_CountObjects(page)); |
| 1348 | |
| 1349 | // Generate contents should remove the empty stream and update the page |
| 1350 | // objects' contents stream indexes. |
| 1351 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1352 | |
| 1353 | ASSERT_EQ(0, FPDFPage_CountObjects(page)); |
| 1354 | |
| 1355 | const char kAllRemovedMD5[] = "eee4600ac08b458ac7ac2320e225674c"; |
| 1356 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1357 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1358 | CompareBitmap(page_bitmap.get(), 200, 200, kAllRemovedMD5); |
| 1359 | } |
| 1360 | |
| 1361 | // Save the file |
| 1362 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1363 | UnloadPage(page); |
| 1364 | |
| 1365 | // Re-open the file and check the page object count is still 0. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1366 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1367 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1368 | |
| 1369 | EXPECT_EQ(0, FPDFPage_CountObjects(saved_page)); |
| 1370 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1371 | ScopedFPDFBitmap page_bitmap = RenderPage(saved_page); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1372 | CompareBitmap(page_bitmap.get(), 200, 200, kAllRemovedMD5); |
| 1373 | } |
| 1374 | |
| 1375 | CloseSavedPage(saved_page); |
| 1376 | CloseSavedDocument(); |
| 1377 | } |
| 1378 | |
Hui Yingst | 08c4071 | 2020-04-29 01:37:35 +0000 | [diff] [blame^] | 1379 | TEST_F(FPDFEditEmbedderTest, RemoveFirstFromSingleStream) { |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1380 | // Load document with a single stream. |
| 1381 | EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
| 1382 | FPDF_PAGE page = LoadPage(0); |
| 1383 | ASSERT_TRUE(page); |
| 1384 | |
| 1385 | // Content stream 0: page objects 0-1. |
| 1386 | ASSERT_EQ(2, FPDFPage_CountObjects(page)); |
| 1387 | |
| 1388 | // Remove first object. |
| 1389 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0); |
| 1390 | ASSERT_TRUE(page_object); |
| 1391 | CPDF_PageObject* cpdf_page_object = |
| 1392 | CPDFPageObjectFromFPDFPageObject(page_object); |
| 1393 | ASSERT_EQ(0, cpdf_page_object->GetContentStream()); |
| 1394 | ASSERT_TRUE(FPDFPage_RemoveObject(page, page_object)); |
| 1395 | FPDFPageObj_Destroy(page_object); |
| 1396 | |
| 1397 | // One object left in the stream. |
| 1398 | ASSERT_EQ(1, FPDFPage_CountObjects(page)); |
| 1399 | page_object = FPDFPage_GetObject(page, 0); |
| 1400 | ASSERT_TRUE(page_object); |
| 1401 | cpdf_page_object = CPDFPageObjectFromFPDFPageObject(page_object); |
| 1402 | ASSERT_EQ(0, cpdf_page_object->GetContentStream()); |
| 1403 | |
| 1404 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1405 | |
| 1406 | // Still one object left in the stream. |
| 1407 | ASSERT_EQ(1, FPDFPage_CountObjects(page)); |
| 1408 | page_object = FPDFPage_GetObject(page, 0); |
| 1409 | ASSERT_TRUE(page_object); |
| 1410 | cpdf_page_object = CPDFPageObjectFromFPDFPageObject(page_object); |
| 1411 | ASSERT_EQ(0, cpdf_page_object->GetContentStream()); |
| 1412 | |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1413 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1414 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1415 | CompareBitmap(page_bitmap.get(), 200, 200, kFirstRemovedChecksum); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1416 | } |
| 1417 | |
| 1418 | // Save the file |
| 1419 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1420 | UnloadPage(page); |
| 1421 | |
| 1422 | // Re-open the file and check the page object count is still 0. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1423 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1424 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1425 | |
| 1426 | ASSERT_EQ(1, FPDFPage_CountObjects(saved_page)); |
| 1427 | page_object = FPDFPage_GetObject(saved_page, 0); |
| 1428 | ASSERT_TRUE(page_object); |
| 1429 | cpdf_page_object = CPDFPageObjectFromFPDFPageObject(page_object); |
| 1430 | ASSERT_EQ(0, cpdf_page_object->GetContentStream()); |
| 1431 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1432 | ScopedFPDFBitmap page_bitmap = RenderPage(saved_page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1433 | CompareBitmap(page_bitmap.get(), 200, 200, kFirstRemovedChecksum); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1434 | } |
| 1435 | |
| 1436 | CloseSavedPage(saved_page); |
| 1437 | CloseSavedDocument(); |
| 1438 | } |
| 1439 | |
Hui Yingst | 08c4071 | 2020-04-29 01:37:35 +0000 | [diff] [blame^] | 1440 | TEST_F(FPDFEditEmbedderTest, RemoveLastFromSingleStream) { |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1441 | // Load document with a single stream. |
| 1442 | EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
| 1443 | FPDF_PAGE page = LoadPage(0); |
| 1444 | ASSERT_TRUE(page); |
| 1445 | |
| 1446 | // Content stream 0: page objects 0-1. |
| 1447 | ASSERT_EQ(2, FPDFPage_CountObjects(page)); |
| 1448 | |
| 1449 | // Remove last object |
| 1450 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 1); |
| 1451 | ASSERT_TRUE(page_object); |
| 1452 | CPDF_PageObject* cpdf_page_object = |
| 1453 | CPDFPageObjectFromFPDFPageObject(page_object); |
| 1454 | ASSERT_EQ(0, cpdf_page_object->GetContentStream()); |
| 1455 | ASSERT_TRUE(FPDFPage_RemoveObject(page, page_object)); |
| 1456 | FPDFPageObj_Destroy(page_object); |
| 1457 | |
| 1458 | // One object left in the stream. |
| 1459 | ASSERT_EQ(1, FPDFPage_CountObjects(page)); |
| 1460 | page_object = FPDFPage_GetObject(page, 0); |
| 1461 | ASSERT_TRUE(page_object); |
| 1462 | cpdf_page_object = CPDFPageObjectFromFPDFPageObject(page_object); |
| 1463 | ASSERT_EQ(0, cpdf_page_object->GetContentStream()); |
| 1464 | |
| 1465 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1466 | |
| 1467 | // Still one object left in the stream. |
| 1468 | ASSERT_EQ(1, FPDFPage_CountObjects(page)); |
| 1469 | page_object = FPDFPage_GetObject(page, 0); |
| 1470 | ASSERT_TRUE(page_object); |
| 1471 | cpdf_page_object = CPDFPageObjectFromFPDFPageObject(page_object); |
| 1472 | ASSERT_EQ(0, cpdf_page_object->GetContentStream()); |
| 1473 | |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1474 | using pdfium::kHelloWorldRemovedChecksum; |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1475 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1476 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1477 | CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldRemovedChecksum); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1478 | } |
| 1479 | |
| 1480 | // Save the file |
| 1481 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1482 | UnloadPage(page); |
| 1483 | |
| 1484 | // Re-open the file and check the page object count is still 0. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1485 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1486 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1487 | |
| 1488 | ASSERT_EQ(1, FPDFPage_CountObjects(saved_page)); |
| 1489 | page_object = FPDFPage_GetObject(saved_page, 0); |
| 1490 | ASSERT_TRUE(page_object); |
| 1491 | cpdf_page_object = CPDFPageObjectFromFPDFPageObject(page_object); |
| 1492 | ASSERT_EQ(0, cpdf_page_object->GetContentStream()); |
| 1493 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1494 | ScopedFPDFBitmap page_bitmap = RenderPage(saved_page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1495 | CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldRemovedChecksum); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1496 | } |
| 1497 | |
| 1498 | CloseSavedPage(saved_page); |
| 1499 | CloseSavedDocument(); |
| 1500 | } |
| 1501 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 1502 | TEST_F(FPDFEditEmbedderTest, RemoveAllFromMultipleStreams) { |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1503 | // Load document with some text. |
| 1504 | EXPECT_TRUE(OpenDocument("hello_world_split_streams.pdf")); |
| 1505 | FPDF_PAGE page = LoadPage(0); |
| 1506 | ASSERT_TRUE(page); |
| 1507 | |
| 1508 | // Content stream 0: page objects 0-1. |
| 1509 | // Content stream 1: page object 2. |
| 1510 | ASSERT_EQ(3, FPDFPage_CountObjects(page)); |
| 1511 | |
| 1512 | // Loop backwards because objects will being removed, which shifts the indexes |
| 1513 | // after the removed position. |
| 1514 | for (int i = 2; i >= 0; i--) { |
| 1515 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i); |
| 1516 | ASSERT_TRUE(page_object); |
| 1517 | ASSERT_TRUE(FPDFPage_RemoveObject(page, page_object)); |
| 1518 | FPDFPageObj_Destroy(page_object); |
| 1519 | } |
| 1520 | |
| 1521 | // No more objects in the page. |
| 1522 | ASSERT_EQ(0, FPDFPage_CountObjects(page)); |
| 1523 | |
| 1524 | // Generate contents should remove the empty streams and update the page |
| 1525 | // objects' contents stream indexes. |
| 1526 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1527 | |
| 1528 | ASSERT_EQ(0, FPDFPage_CountObjects(page)); |
| 1529 | |
| 1530 | const char kAllRemovedMD5[] = "eee4600ac08b458ac7ac2320e225674c"; |
| 1531 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1532 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1533 | CompareBitmap(page_bitmap.get(), 200, 200, kAllRemovedMD5); |
| 1534 | } |
| 1535 | |
| 1536 | // Save the file |
| 1537 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1538 | UnloadPage(page); |
| 1539 | |
| 1540 | // Re-open the file and check the page object count is still 0. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1541 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1542 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1543 | |
| 1544 | EXPECT_EQ(0, FPDFPage_CountObjects(saved_page)); |
| 1545 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1546 | ScopedFPDFBitmap page_bitmap = RenderPage(saved_page); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1547 | CompareBitmap(page_bitmap.get(), 200, 200, kAllRemovedMD5); |
| 1548 | } |
| 1549 | |
| 1550 | CloseSavedPage(saved_page); |
| 1551 | CloseSavedDocument(); |
| 1552 | } |
| 1553 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 1554 | TEST_F(FPDFEditEmbedderTest, InsertPageObjectAndSave) { |
Henrique Nakashima | c49e62e | 2018-04-16 20:58:47 +0000 | [diff] [blame] | 1555 | // Load document with some text. |
| 1556 | EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
| 1557 | FPDF_PAGE page = LoadPage(0); |
| 1558 | ASSERT_TRUE(page); |
| 1559 | |
| 1560 | // Add a red rectangle. |
| 1561 | ASSERT_EQ(2, FPDFPage_CountObjects(page)); |
| 1562 | FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(20, 100, 50, 50); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1563 | EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 0, 0, 255)); |
Henrique Nakashima | c49e62e | 2018-04-16 20:58:47 +0000 | [diff] [blame] | 1564 | EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0)); |
| 1565 | FPDFPage_InsertObject(page, red_rect); |
| 1566 | |
| 1567 | // Verify the red rectangle was added. |
| 1568 | ASSERT_EQ(3, FPDFPage_CountObjects(page)); |
| 1569 | |
| 1570 | // Save the file |
| 1571 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1572 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1573 | UnloadPage(page); |
| 1574 | |
| 1575 | // Re-open the file and check the page object count is still 3. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1576 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | c49e62e | 2018-04-16 20:58:47 +0000 | [diff] [blame] | 1577 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1578 | EXPECT_EQ(3, FPDFPage_CountObjects(saved_page)); |
| 1579 | CloseSavedPage(saved_page); |
| 1580 | CloseSavedDocument(); |
| 1581 | } |
| 1582 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 1583 | TEST_F(FPDFEditEmbedderTest, InsertPageObjectEditAndSave) { |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1584 | // Load document with some text. |
| 1585 | EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
| 1586 | FPDF_PAGE page = LoadPage(0); |
| 1587 | ASSERT_TRUE(page); |
| 1588 | |
| 1589 | // Add a red rectangle. |
| 1590 | ASSERT_EQ(2, FPDFPage_CountObjects(page)); |
| 1591 | FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(20, 100, 50, 50); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1592 | EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 100, 100, 255)); |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1593 | EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0)); |
| 1594 | FPDFPage_InsertObject(page, red_rect); |
| 1595 | |
| 1596 | // Verify the red rectangle was added. |
| 1597 | ASSERT_EQ(3, FPDFPage_CountObjects(page)); |
| 1598 | |
| 1599 | // Generate content but change it again |
| 1600 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1601 | EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 0, 0, 255)); |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1602 | |
| 1603 | // Save the file |
| 1604 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1605 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1606 | UnloadPage(page); |
| 1607 | |
| 1608 | // Re-open the file and check the page object count is still 3. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1609 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1610 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1611 | EXPECT_EQ(3, FPDFPage_CountObjects(saved_page)); |
| 1612 | CloseSavedPage(saved_page); |
| 1613 | CloseSavedDocument(); |
| 1614 | } |
| 1615 | |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 1616 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 1617 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 1618 | #define MAYBE_InsertAndRemoveLargeFile DISABLED_InsertAndRemoveLargeFile |
| 1619 | #else |
| 1620 | #define MAYBE_InsertAndRemoveLargeFile InsertAndRemoveLargeFile |
| 1621 | #endif |
| 1622 | TEST_F(FPDFEditEmbedderTest, MAYBE_InsertAndRemoveLargeFile) { |
Henrique Nakashima | 657a1aa | 2018-09-12 16:19:22 +0000 | [diff] [blame] | 1623 | const int kOriginalObjectCount = 600; |
| 1624 | |
| 1625 | // Load document with many objects. |
| 1626 | EXPECT_TRUE(OpenDocument("many_rectangles.pdf")); |
| 1627 | FPDF_PAGE page = LoadPage(0); |
| 1628 | ASSERT_TRUE(page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1629 | |
| 1630 | using pdfium::kManyRectanglesChecksum; |
Henrique Nakashima | 657a1aa | 2018-09-12 16:19:22 +0000 | [diff] [blame] | 1631 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1632 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1633 | CompareBitmap(page_bitmap.get(), 200, 300, kManyRectanglesChecksum); |
Henrique Nakashima | 657a1aa | 2018-09-12 16:19:22 +0000 | [diff] [blame] | 1634 | } |
| 1635 | |
| 1636 | // Add a black rectangle. |
| 1637 | ASSERT_EQ(kOriginalObjectCount, FPDFPage_CountObjects(page)); |
| 1638 | FPDF_PAGEOBJECT black_rect = FPDFPageObj_CreateNewRect(20, 100, 50, 50); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1639 | EXPECT_TRUE(FPDFPageObj_SetFillColor(black_rect, 0, 0, 0, 255)); |
Henrique Nakashima | 657a1aa | 2018-09-12 16:19:22 +0000 | [diff] [blame] | 1640 | EXPECT_TRUE(FPDFPath_SetDrawMode(black_rect, FPDF_FILLMODE_ALTERNATE, 0)); |
| 1641 | FPDFPage_InsertObject(page, black_rect); |
| 1642 | |
| 1643 | // Verify the black rectangle was added. |
| 1644 | ASSERT_EQ(kOriginalObjectCount + 1, FPDFPage_CountObjects(page)); |
| 1645 | const char kPlusRectangleMD5[] = "6b9396ab570754b32b04ca629e902f77"; |
| 1646 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1647 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Henrique Nakashima | 657a1aa | 2018-09-12 16:19:22 +0000 | [diff] [blame] | 1648 | CompareBitmap(page_bitmap.get(), 200, 300, kPlusRectangleMD5); |
| 1649 | } |
| 1650 | |
| 1651 | // Save the file. |
| 1652 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1653 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1654 | UnloadPage(page); |
| 1655 | |
| 1656 | // Re-open the file and check the rectangle added is still there. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1657 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | 657a1aa | 2018-09-12 16:19:22 +0000 | [diff] [blame] | 1658 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1659 | EXPECT_EQ(kOriginalObjectCount + 1, FPDFPage_CountObjects(saved_page)); |
| 1660 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1661 | ScopedFPDFBitmap page_bitmap = RenderPage(saved_page); |
Henrique Nakashima | 657a1aa | 2018-09-12 16:19:22 +0000 | [diff] [blame] | 1662 | CompareBitmap(page_bitmap.get(), 200, 300, kPlusRectangleMD5); |
| 1663 | } |
| 1664 | |
| 1665 | // Remove the added rectangle. |
| 1666 | FPDF_PAGEOBJECT added_object = |
| 1667 | FPDFPage_GetObject(saved_page, kOriginalObjectCount); |
| 1668 | EXPECT_TRUE(FPDFPage_RemoveObject(saved_page, added_object)); |
| 1669 | FPDFPageObj_Destroy(added_object); |
| 1670 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1671 | ScopedFPDFBitmap page_bitmap = RenderPage(saved_page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1672 | CompareBitmap(page_bitmap.get(), 200, 300, kManyRectanglesChecksum); |
Henrique Nakashima | 657a1aa | 2018-09-12 16:19:22 +0000 | [diff] [blame] | 1673 | } |
| 1674 | EXPECT_EQ(kOriginalObjectCount, FPDFPage_CountObjects(saved_page)); |
| 1675 | |
| 1676 | // Save the file again. |
| 1677 | EXPECT_TRUE(FPDFPage_GenerateContent(saved_page)); |
| 1678 | EXPECT_TRUE(FPDF_SaveAsCopy(saved_document_, this, 0)); |
| 1679 | |
| 1680 | CloseSavedPage(saved_page); |
| 1681 | CloseSavedDocument(); |
| 1682 | |
| 1683 | // Re-open the file (again) and check the black rectangle was removed and the |
| 1684 | // rest is intact. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1685 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | 657a1aa | 2018-09-12 16:19:22 +0000 | [diff] [blame] | 1686 | saved_page = LoadSavedPage(0); |
| 1687 | EXPECT_EQ(kOriginalObjectCount, FPDFPage_CountObjects(saved_page)); |
| 1688 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1689 | ScopedFPDFBitmap page_bitmap = RenderPage(saved_page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1690 | CompareBitmap(page_bitmap.get(), 200, 300, kManyRectanglesChecksum); |
Henrique Nakashima | 657a1aa | 2018-09-12 16:19:22 +0000 | [diff] [blame] | 1691 | } |
| 1692 | |
| 1693 | CloseSavedPage(saved_page); |
| 1694 | CloseSavedDocument(); |
| 1695 | } |
| 1696 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 1697 | TEST_F(FPDFEditEmbedderTest, AddAndRemovePaths) { |
Henrique Nakashima | 35841fa | 2018-03-15 15:25:16 +0000 | [diff] [blame] | 1698 | // Start with a blank page. |
| 1699 | FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792); |
| 1700 | ASSERT_TRUE(page); |
| 1701 | |
| 1702 | // Render the blank page and verify it's a blank bitmap. |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1703 | using pdfium::kBlankPage612By792Checksum; |
Henrique Nakashima | 35841fa | 2018-03-15 15:25:16 +0000 | [diff] [blame] | 1704 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1705 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1706 | CompareBitmap(page_bitmap.get(), 612, 792, kBlankPage612By792Checksum); |
Henrique Nakashima | 35841fa | 2018-03-15 15:25:16 +0000 | [diff] [blame] | 1707 | } |
| 1708 | ASSERT_EQ(0, FPDFPage_CountObjects(page)); |
| 1709 | |
| 1710 | // Add a red rectangle. |
| 1711 | FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(10, 10, 20, 20); |
| 1712 | ASSERT_TRUE(red_rect); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1713 | EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 0, 0, 255)); |
Henrique Nakashima | 35841fa | 2018-03-15 15:25:16 +0000 | [diff] [blame] | 1714 | EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0)); |
| 1715 | FPDFPage_InsertObject(page, red_rect); |
Henrique Nakashima | 35841fa | 2018-03-15 15:25:16 +0000 | [diff] [blame] | 1716 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1717 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1718 | CompareBitmap(page_bitmap.get(), 612, 792, kRedRectangleChecksum); |
Henrique Nakashima | 35841fa | 2018-03-15 15:25:16 +0000 | [diff] [blame] | 1719 | } |
| 1720 | EXPECT_EQ(1, FPDFPage_CountObjects(page)); |
| 1721 | |
| 1722 | // Remove rectangle and verify it does not render anymore and the bitmap is |
| 1723 | // back to a blank one. |
| 1724 | EXPECT_TRUE(FPDFPage_RemoveObject(page, red_rect)); |
| 1725 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1726 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1727 | CompareBitmap(page_bitmap.get(), 612, 792, kBlankPage612By792Checksum); |
Henrique Nakashima | 35841fa | 2018-03-15 15:25:16 +0000 | [diff] [blame] | 1728 | } |
| 1729 | EXPECT_EQ(0, FPDFPage_CountObjects(page)); |
| 1730 | |
| 1731 | // Trying to remove an object not in the page should return false. |
| 1732 | EXPECT_FALSE(FPDFPage_RemoveObject(page, red_rect)); |
| 1733 | |
| 1734 | FPDF_ClosePage(page); |
| 1735 | FPDFPageObj_Destroy(red_rect); |
| 1736 | } |
| 1737 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 1738 | TEST_F(FPDFEditEmbedderTest, PathsPoints) { |
Miklos Vajna | 12abfd0 | 2017-09-15 07:49:03 +0200 | [diff] [blame] | 1739 | CreateNewDocument(); |
| 1740 | FPDF_PAGEOBJECT img = FPDFPageObj_NewImageObj(document_); |
| 1741 | // This should fail gracefully, even if img is not a path. |
Miklos Vajna | 0150a54 | 2017-09-21 21:46:56 +0200 | [diff] [blame] | 1742 | ASSERT_EQ(-1, FPDFPath_CountSegments(img)); |
Miklos Vajna | 12abfd0 | 2017-09-15 07:49:03 +0200 | [diff] [blame] | 1743 | |
| 1744 | // This should fail gracefully, even if path is NULL. |
Miklos Vajna | 0150a54 | 2017-09-21 21:46:56 +0200 | [diff] [blame] | 1745 | ASSERT_EQ(-1, FPDFPath_CountSegments(nullptr)); |
Miklos Vajna | 12abfd0 | 2017-09-15 07:49:03 +0200 | [diff] [blame] | 1746 | |
Miklos Vajna | 36eed87 | 2017-09-20 22:52:43 +0200 | [diff] [blame] | 1747 | // FPDFPath_GetPathSegment() with a non-path. |
| 1748 | ASSERT_EQ(nullptr, FPDFPath_GetPathSegment(img, 0)); |
| 1749 | // FPDFPath_GetPathSegment() with a NULL path. |
| 1750 | ASSERT_EQ(nullptr, FPDFPath_GetPathSegment(nullptr, 0)); |
| 1751 | float x; |
| 1752 | float y; |
| 1753 | // FPDFPathSegment_GetPoint() with a NULL segment. |
| 1754 | EXPECT_FALSE(FPDFPathSegment_GetPoint(nullptr, &x, &y)); |
| 1755 | |
| 1756 | // FPDFPathSegment_GetType() with a NULL segment. |
| 1757 | ASSERT_EQ(FPDF_SEGMENT_UNKNOWN, FPDFPathSegment_GetType(nullptr)); |
| 1758 | |
| 1759 | // FPDFPathSegment_GetClose() with a NULL segment. |
| 1760 | EXPECT_FALSE(FPDFPathSegment_GetClose(nullptr)); |
| 1761 | |
Miklos Vajna | 12abfd0 | 2017-09-15 07:49:03 +0200 | [diff] [blame] | 1762 | FPDFPageObj_Destroy(img); |
| 1763 | } |
| 1764 | |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 1765 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 1766 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 1767 | #define MAYBE_PathOnTopOfText DISABLED_PathOnTopOfText |
| 1768 | #else |
| 1769 | #define MAYBE_PathOnTopOfText PathOnTopOfText |
| 1770 | #endif |
| 1771 | TEST_F(FPDFEditEmbedderTest, MAYBE_PathOnTopOfText) { |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 1772 | // Load document with some text |
| 1773 | EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
| 1774 | FPDF_PAGE page = LoadPage(0); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1775 | ASSERT_TRUE(page); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 1776 | |
| 1777 | // Add an opaque rectangle on top of some of the text. |
| 1778 | FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(20, 100, 50, 50); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1779 | EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 0, 0, 255)); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 1780 | EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0)); |
| 1781 | FPDFPage_InsertObject(page, red_rect); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 1782 | |
| 1783 | // Add a transparent triangle on top of other part of the text. |
| 1784 | FPDF_PAGEOBJECT black_path = FPDFPageObj_CreateNewPath(20, 50); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1785 | EXPECT_TRUE(FPDFPageObj_SetFillColor(black_path, 0, 0, 0, 100)); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 1786 | EXPECT_TRUE(FPDFPath_SetDrawMode(black_path, FPDF_FILLMODE_ALTERNATE, 0)); |
| 1787 | EXPECT_TRUE(FPDFPath_LineTo(black_path, 30, 80)); |
| 1788 | EXPECT_TRUE(FPDFPath_LineTo(black_path, 40, 10)); |
| 1789 | EXPECT_TRUE(FPDFPath_Close(black_path)); |
| 1790 | FPDFPage_InsertObject(page, black_path); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 1791 | |
| 1792 | // Render and check the result. Text is slightly different on Mac. |
Tom Sepez | e08d2b1 | 2018-04-25 18:49:32 +0000 | [diff] [blame] | 1793 | ScopedFPDFBitmap bitmap = RenderLoadedPage(page); |
Lei Zhang | e4cdac5 | 2019-04-30 16:45:57 +0000 | [diff] [blame] | 1794 | #if defined(OS_MACOSX) |
Lei Zhang | 0f6b434 | 2020-02-25 20:00:39 +0000 | [diff] [blame] | 1795 | const char md5[] = "e55bcd1facb7243dc6e16dd5f912265b"; |
Lei Zhang | e67bcc7 | 2019-04-30 18:55:58 +0000 | [diff] [blame] | 1796 | #elif defined(OS_WIN) |
Lei Zhang | 5e2c51c | 2018-07-27 22:33:34 +0000 | [diff] [blame] | 1797 | const char md5[] = "74dd9c393b8b2578d2b7feb032b7daad"; |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 1798 | #else |
Henrique Nakashima | 09b4192 | 2017-10-27 20:39:29 +0000 | [diff] [blame] | 1799 | const char md5[] = "aa71b09b93b55f467f1290e5111babee"; |
Nicolas Pena | 5bcd9a3 | 2017-03-22 11:04:35 -0400 | [diff] [blame] | 1800 | #endif |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1801 | CompareBitmap(bitmap.get(), 200, 200, md5); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 1802 | UnloadPage(page); |
| 1803 | } |
Nicolas Pena | 2eb1a70 | 2017-02-09 18:17:33 -0500 | [diff] [blame] | 1804 | |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 1805 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 1806 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 1807 | #define MAYBE_EditOverExistingContent DISABLED_EditOverExistingContent |
| 1808 | #else |
| 1809 | #define MAYBE_EditOverExistingContent EditOverExistingContent |
| 1810 | #endif |
| 1811 | TEST_F(FPDFEditEmbedderTest, MAYBE_EditOverExistingContent) { |
wileyrya | e858aa4 | 2017-05-31 14:49:05 -0500 | [diff] [blame] | 1812 | // Load document with existing content |
| 1813 | EXPECT_TRUE(OpenDocument("bug_717.pdf")); |
| 1814 | FPDF_PAGE page = LoadPage(0); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1815 | ASSERT_TRUE(page); |
wileyrya | e858aa4 | 2017-05-31 14:49:05 -0500 | [diff] [blame] | 1816 | |
| 1817 | // Add a transparent rectangle on top of the existing content |
| 1818 | FPDF_PAGEOBJECT red_rect2 = FPDFPageObj_CreateNewRect(90, 700, 25, 50); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1819 | EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect2, 255, 0, 0, 100)); |
wileyrya | e858aa4 | 2017-05-31 14:49:05 -0500 | [diff] [blame] | 1820 | EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect2, FPDF_FILLMODE_ALTERNATE, 0)); |
| 1821 | FPDFPage_InsertObject(page, red_rect2); |
| 1822 | |
| 1823 | // Add an opaque rectangle on top of the existing content |
| 1824 | FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(115, 700, 25, 50); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1825 | EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 0, 0, 255)); |
wileyrya | e858aa4 | 2017-05-31 14:49:05 -0500 | [diff] [blame] | 1826 | EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0)); |
| 1827 | FPDFPage_InsertObject(page, red_rect); |
| 1828 | |
Tom Sepez | e08d2b1 | 2018-04-25 18:49:32 +0000 | [diff] [blame] | 1829 | ScopedFPDFBitmap bitmap = RenderLoadedPage(page); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1830 | CompareBitmap(bitmap.get(), 612, 792, "ad04e5bd0f471a9a564fb034bd0fb073"); |
wileyrya | e858aa4 | 2017-05-31 14:49:05 -0500 | [diff] [blame] | 1831 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1832 | |
| 1833 | // Now save the result, closing the page and document |
| 1834 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
Nicolas Pena | 3ff5400 | 2017-07-05 11:55:35 -0400 | [diff] [blame] | 1835 | UnloadPage(page); |
wileyrya | e858aa4 | 2017-05-31 14:49:05 -0500 | [diff] [blame] | 1836 | |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1837 | ASSERT_TRUE(OpenSavedDocument()); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1838 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1839 | VerifySavedRendering(saved_page, 612, 792, |
| 1840 | "ad04e5bd0f471a9a564fb034bd0fb073"); |
wileyrya | e858aa4 | 2017-05-31 14:49:05 -0500 | [diff] [blame] | 1841 | |
| 1842 | ClearString(); |
| 1843 | // Add another opaque rectangle on top of the existing content |
| 1844 | FPDF_PAGEOBJECT green_rect = FPDFPageObj_CreateNewRect(150, 700, 25, 50); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1845 | EXPECT_TRUE(FPDFPageObj_SetFillColor(green_rect, 0, 255, 0, 255)); |
wileyrya | e858aa4 | 2017-05-31 14:49:05 -0500 | [diff] [blame] | 1846 | EXPECT_TRUE(FPDFPath_SetDrawMode(green_rect, FPDF_FILLMODE_ALTERNATE, 0)); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1847 | FPDFPage_InsertObject(saved_page, green_rect); |
wileyrya | e858aa4 | 2017-05-31 14:49:05 -0500 | [diff] [blame] | 1848 | |
| 1849 | // Add another transparent rectangle on top of existing content |
| 1850 | FPDF_PAGEOBJECT green_rect2 = FPDFPageObj_CreateNewRect(175, 700, 25, 50); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1851 | EXPECT_TRUE(FPDFPageObj_SetFillColor(green_rect2, 0, 255, 0, 100)); |
wileyrya | e858aa4 | 2017-05-31 14:49:05 -0500 | [diff] [blame] | 1852 | EXPECT_TRUE(FPDFPath_SetDrawMode(green_rect2, FPDF_FILLMODE_ALTERNATE, 0)); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1853 | FPDFPage_InsertObject(saved_page, green_rect2); |
Lei Zhang | c113c7a | 2018-02-12 14:58:44 +0000 | [diff] [blame] | 1854 | const char kLastMD5[] = "4b5b00f824620f8c9b8801ebb98e1cdd"; |
| 1855 | { |
Tom Sepez | e08d2b1 | 2018-04-25 18:49:32 +0000 | [diff] [blame] | 1856 | ScopedFPDFBitmap new_bitmap = RenderSavedPage(saved_page); |
Lei Zhang | c113c7a | 2018-02-12 14:58:44 +0000 | [diff] [blame] | 1857 | CompareBitmap(new_bitmap.get(), 612, 792, kLastMD5); |
| 1858 | } |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1859 | EXPECT_TRUE(FPDFPage_GenerateContent(saved_page)); |
wileyrya | e858aa4 | 2017-05-31 14:49:05 -0500 | [diff] [blame] | 1860 | |
| 1861 | // Now save the result, closing the page and document |
Lei Zhang | 0729be2 | 2018-02-05 21:13:51 +0000 | [diff] [blame] | 1862 | EXPECT_TRUE(FPDF_SaveAsCopy(saved_document_, this, 0)); |
Dan Sinclair | 04e4dc8 | 2017-10-18 12:17:14 -0400 | [diff] [blame] | 1863 | |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1864 | CloseSavedPage(saved_page); |
Dan Sinclair | 04e4dc8 | 2017-10-18 12:17:14 -0400 | [diff] [blame] | 1865 | CloseSavedDocument(); |
wileyrya | e858aa4 | 2017-05-31 14:49:05 -0500 | [diff] [blame] | 1866 | |
| 1867 | // Render the saved result |
Lei Zhang | c113c7a | 2018-02-12 14:58:44 +0000 | [diff] [blame] | 1868 | VerifySavedDocument(612, 792, kLastMD5); |
wileyrya | e858aa4 | 2017-05-31 14:49:05 -0500 | [diff] [blame] | 1869 | } |
| 1870 | |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 1871 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 1872 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 1873 | #define MAYBE_AddStrokedPaths DISABLED_AddStrokedPaths |
| 1874 | #else |
| 1875 | #define MAYBE_AddStrokedPaths AddStrokedPaths |
| 1876 | #endif |
| 1877 | TEST_F(FPDFEditEmbedderTest, MAYBE_AddStrokedPaths) { |
Nicolas Pena | 2eb1a70 | 2017-02-09 18:17:33 -0500 | [diff] [blame] | 1878 | // Start with a blank page |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 1879 | FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792); |
Nicolas Pena | 2eb1a70 | 2017-02-09 18:17:33 -0500 | [diff] [blame] | 1880 | |
| 1881 | // Add a large stroked rectangle (fill color should not affect it). |
| 1882 | FPDF_PAGEOBJECT rect = FPDFPageObj_CreateNewRect(20, 20, 200, 400); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1883 | EXPECT_TRUE(FPDFPageObj_SetFillColor(rect, 255, 0, 0, 255)); |
| 1884 | EXPECT_TRUE(FPDFPageObj_SetStrokeColor(rect, 0, 255, 0, 255)); |
| 1885 | EXPECT_TRUE(FPDFPageObj_SetStrokeWidth(rect, 15.0f)); |
Miklos Vajna | 366df7f | 2018-05-22 14:27:29 +0000 | [diff] [blame] | 1886 | |
| 1887 | float width = 0; |
| 1888 | EXPECT_TRUE(FPDFPageObj_GetStrokeWidth(rect, &width)); |
| 1889 | EXPECT_EQ(15.0f, width); |
| 1890 | |
Nicolas Pena | 2eb1a70 | 2017-02-09 18:17:33 -0500 | [diff] [blame] | 1891 | EXPECT_TRUE(FPDFPath_SetDrawMode(rect, 0, 1)); |
| 1892 | FPDFPage_InsertObject(page, rect); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1893 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1894 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1895 | CompareBitmap(page_bitmap.get(), 612, 792, |
| 1896 | "64bd31f862a89e0a9e505a5af6efd506"); |
| 1897 | } |
Nicolas Pena | 2eb1a70 | 2017-02-09 18:17:33 -0500 | [diff] [blame] | 1898 | |
| 1899 | // Add crossed-checkmark |
| 1900 | FPDF_PAGEOBJECT check = FPDFPageObj_CreateNewPath(300, 500); |
| 1901 | EXPECT_TRUE(FPDFPath_LineTo(check, 400, 400)); |
| 1902 | EXPECT_TRUE(FPDFPath_LineTo(check, 600, 600)); |
| 1903 | EXPECT_TRUE(FPDFPath_MoveTo(check, 400, 600)); |
| 1904 | EXPECT_TRUE(FPDFPath_LineTo(check, 600, 400)); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1905 | EXPECT_TRUE(FPDFPageObj_SetStrokeColor(check, 128, 128, 128, 180)); |
| 1906 | EXPECT_TRUE(FPDFPageObj_SetStrokeWidth(check, 8.35f)); |
Nicolas Pena | 2eb1a70 | 2017-02-09 18:17:33 -0500 | [diff] [blame] | 1907 | EXPECT_TRUE(FPDFPath_SetDrawMode(check, 0, 1)); |
| 1908 | FPDFPage_InsertObject(page, check); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1909 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1910 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1911 | CompareBitmap(page_bitmap.get(), 612, 792, |
| 1912 | "4b6f3b9d25c4e194821217d5016c3724"); |
| 1913 | } |
Nicolas Pena | 2eb1a70 | 2017-02-09 18:17:33 -0500 | [diff] [blame] | 1914 | |
| 1915 | // Add stroked and filled oval-ish path. |
| 1916 | FPDF_PAGEOBJECT path = FPDFPageObj_CreateNewPath(250, 100); |
| 1917 | EXPECT_TRUE(FPDFPath_BezierTo(path, 180, 166, 180, 233, 250, 300)); |
| 1918 | EXPECT_TRUE(FPDFPath_LineTo(path, 255, 305)); |
| 1919 | EXPECT_TRUE(FPDFPath_BezierTo(path, 325, 233, 325, 166, 255, 105)); |
| 1920 | EXPECT_TRUE(FPDFPath_Close(path)); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1921 | EXPECT_TRUE(FPDFPageObj_SetFillColor(path, 200, 128, 128, 100)); |
| 1922 | EXPECT_TRUE(FPDFPageObj_SetStrokeColor(path, 128, 200, 128, 150)); |
| 1923 | EXPECT_TRUE(FPDFPageObj_SetStrokeWidth(path, 10.5f)); |
Nicolas Pena | 2eb1a70 | 2017-02-09 18:17:33 -0500 | [diff] [blame] | 1924 | EXPECT_TRUE(FPDFPath_SetDrawMode(path, FPDF_FILLMODE_ALTERNATE, 1)); |
| 1925 | FPDFPage_InsertObject(page, path); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1926 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1927 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1928 | CompareBitmap(page_bitmap.get(), 612, 792, |
| 1929 | "ff3e6a22326754944cc6e56609acd73b"); |
| 1930 | } |
Nicolas Pena | 2eb1a70 | 2017-02-09 18:17:33 -0500 | [diff] [blame] | 1931 | FPDF_ClosePage(page); |
Nicolas Pena | 2eb1a70 | 2017-02-09 18:17:33 -0500 | [diff] [blame] | 1932 | } |
Nicolas Pena | 4905840 | 2017-02-14 18:26:20 -0500 | [diff] [blame] | 1933 | |
Nicolas Pena | 4c48b10 | 2018-06-13 18:23:46 +0000 | [diff] [blame] | 1934 | // Tests adding text from standard font using FPDFPageObj_NewTextObj. |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 1935 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 1936 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 1937 | #define MAYBE_AddStandardFontText DISABLED_AddStandardFontText |
| 1938 | #else |
| 1939 | #define MAYBE_AddStandardFontText AddStandardFontText |
| 1940 | #endif |
| 1941 | TEST_F(FPDFEditEmbedderTest, MAYBE_AddStandardFontText) { |
Nicolas Pena | 4905840 | 2017-02-14 18:26:20 -0500 | [diff] [blame] | 1942 | // Start with a blank page |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 1943 | FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792); |
Nicolas Pena | 4905840 | 2017-02-14 18:26:20 -0500 | [diff] [blame] | 1944 | |
| 1945 | // Add some text to the page |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 1946 | FPDF_PAGEOBJECT text_object1 = |
| 1947 | FPDFPageObj_NewTextObj(document(), "Arial", 12.0f); |
| 1948 | EXPECT_TRUE(text_object1); |
Lei Zhang | f0f6768 | 2019-04-08 17:03:21 +0000 | [diff] [blame] | 1949 | ScopedFPDFWideString text1 = |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 1950 | GetFPDFWideString(L"I'm at the bottom of the page"); |
| 1951 | EXPECT_TRUE(FPDFText_SetText(text_object1, text1.get())); |
| 1952 | FPDFPageObj_Transform(text_object1, 1, 0, 0, 1, 20, 20); |
| 1953 | FPDFPage_InsertObject(page, text_object1); |
Lei Zhang | e039bab | 2019-03-18 19:57:56 +0000 | [diff] [blame] | 1954 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1955 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1956 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Lei Zhang | e4cdac5 | 2019-04-30 16:45:57 +0000 | [diff] [blame] | 1957 | #if defined(OS_MACOSX) |
Lei Zhang | 0f6b434 | 2020-02-25 20:00:39 +0000 | [diff] [blame] | 1958 | const char md5[] = "324e1db8164a040cf6104538baa95ba6"; |
Lei Zhang | e67bcc7 | 2019-04-30 18:55:58 +0000 | [diff] [blame] | 1959 | #elif defined(OS_WIN) |
Lei Zhang | 5e2c51c | 2018-07-27 22:33:34 +0000 | [diff] [blame] | 1960 | const char md5[] = "08d1ff3e5a42801bee6077fd366bef00"; |
Nicolas Pena | 4905840 | 2017-02-14 18:26:20 -0500 | [diff] [blame] | 1961 | #else |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1962 | const char md5[] = "eacaa24573b8ce997b3882595f096f00"; |
Nicolas Pena | 5bcd9a3 | 2017-03-22 11:04:35 -0400 | [diff] [blame] | 1963 | #endif |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1964 | CompareBitmap(page_bitmap.get(), 612, 792, md5); |
Lei Zhang | e039bab | 2019-03-18 19:57:56 +0000 | [diff] [blame] | 1965 | |
| 1966 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1967 | VerifySavedDocument(612, 792, md5); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1968 | } |
Nicolas Pena | 4905840 | 2017-02-14 18:26:20 -0500 | [diff] [blame] | 1969 | |
| 1970 | // Try another font |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 1971 | FPDF_PAGEOBJECT text_object2 = |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 1972 | FPDFPageObj_NewTextObj(document(), "TimesNewRomanBold", 15.0f); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 1973 | EXPECT_TRUE(text_object2); |
Lei Zhang | f0f6768 | 2019-04-08 17:03:21 +0000 | [diff] [blame] | 1974 | ScopedFPDFWideString text2 = |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 1975 | GetFPDFWideString(L"Hi, I'm Bold. Times New Roman Bold."); |
| 1976 | EXPECT_TRUE(FPDFText_SetText(text_object2, text2.get())); |
| 1977 | FPDFPageObj_Transform(text_object2, 1, 0, 0, 1, 100, 600); |
| 1978 | FPDFPage_InsertObject(page, text_object2); |
Lei Zhang | e039bab | 2019-03-18 19:57:56 +0000 | [diff] [blame] | 1979 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1980 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1981 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Lei Zhang | e4cdac5 | 2019-04-30 16:45:57 +0000 | [diff] [blame] | 1982 | #if defined(OS_MACOSX) |
Lei Zhang | 0f6b434 | 2020-02-25 20:00:39 +0000 | [diff] [blame] | 1983 | const char md5[] = "26a516d923b0a18fbea0a24e3aca5562"; |
Lei Zhang | e67bcc7 | 2019-04-30 18:55:58 +0000 | [diff] [blame] | 1984 | #elif defined(OS_WIN) |
Lei Zhang | e039bab | 2019-03-18 19:57:56 +0000 | [diff] [blame] | 1985 | const char md5[] = "3755dd35abd4c605755369401ee85b2d"; |
Nicolas Pena | 4905840 | 2017-02-14 18:26:20 -0500 | [diff] [blame] | 1986 | #else |
Lei Zhang | e039bab | 2019-03-18 19:57:56 +0000 | [diff] [blame] | 1987 | const char md5[] = "76fcc7d08aa15445efd2e2ceb7c6cc3b"; |
Nicolas Pena | 5bcd9a3 | 2017-03-22 11:04:35 -0400 | [diff] [blame] | 1988 | #endif |
Lei Zhang | e039bab | 2019-03-18 19:57:56 +0000 | [diff] [blame] | 1989 | CompareBitmap(page_bitmap.get(), 612, 792, md5); |
| 1990 | |
| 1991 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1992 | VerifySavedDocument(612, 792, md5); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1993 | } |
Nicolas Pena | 4905840 | 2017-02-14 18:26:20 -0500 | [diff] [blame] | 1994 | |
| 1995 | // And some randomly transformed text |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 1996 | FPDF_PAGEOBJECT text_object3 = |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 1997 | FPDFPageObj_NewTextObj(document(), "Courier-Bold", 20.0f); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 1998 | EXPECT_TRUE(text_object3); |
Lei Zhang | f0f6768 | 2019-04-08 17:03:21 +0000 | [diff] [blame] | 1999 | ScopedFPDFWideString text3 = GetFPDFWideString(L"Can you read me? <:)>"); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2000 | EXPECT_TRUE(FPDFText_SetText(text_object3, text3.get())); |
| 2001 | FPDFPageObj_Transform(text_object3, 1, 1.5, 2, 0.5, 200, 200); |
| 2002 | FPDFPage_InsertObject(page, text_object3); |
Lei Zhang | e039bab | 2019-03-18 19:57:56 +0000 | [diff] [blame] | 2003 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2004 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 2005 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Lei Zhang | e4cdac5 | 2019-04-30 16:45:57 +0000 | [diff] [blame] | 2006 | #if defined(OS_MACOSX) |
Lei Zhang | 0f6b434 | 2020-02-25 20:00:39 +0000 | [diff] [blame] | 2007 | const char md5[] = "532024c9ded47843313bb64a060118f3"; |
Lei Zhang | e67bcc7 | 2019-04-30 18:55:58 +0000 | [diff] [blame] | 2008 | #elif defined(OS_WIN) |
Lei Zhang | a2b7073 | 2019-06-25 08:34:22 +0000 | [diff] [blame] | 2009 | const char md5[] = "5ded49fe157f89627903553771431e3d"; |
Nicolas Pena | 4905840 | 2017-02-14 18:26:20 -0500 | [diff] [blame] | 2010 | #else |
Lei Zhang | a2b7073 | 2019-06-25 08:34:22 +0000 | [diff] [blame] | 2011 | const char md5[] = "344534539aa7c5cc78404cfff4bde7fb"; |
Nicolas Pena | 5bcd9a3 | 2017-03-22 11:04:35 -0400 | [diff] [blame] | 2012 | #endif |
Lei Zhang | e039bab | 2019-03-18 19:57:56 +0000 | [diff] [blame] | 2013 | CompareBitmap(page_bitmap.get(), 612, 792, md5); |
| 2014 | |
| 2015 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 2016 | VerifySavedDocument(612, 792, md5); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2017 | } |
Nicolas Pena | 4905840 | 2017-02-14 18:26:20 -0500 | [diff] [blame] | 2018 | |
Lei Zhang | 8da9823 | 2019-12-11 23:29:33 +0000 | [diff] [blame] | 2019 | FS_MATRIX matrix; |
| 2020 | EXPECT_FALSE(FPDFTextObj_GetMatrix(nullptr, &matrix)); |
| 2021 | EXPECT_TRUE(FPDFTextObj_GetMatrix(text_object3, &matrix)); |
| 2022 | EXPECT_FLOAT_EQ(1.0f, matrix.a); |
| 2023 | EXPECT_FLOAT_EQ(1.5f, matrix.b); |
| 2024 | EXPECT_FLOAT_EQ(2.0f, matrix.c); |
| 2025 | EXPECT_FLOAT_EQ(0.5f, matrix.d); |
| 2026 | EXPECT_FLOAT_EQ(200.0f, matrix.e); |
| 2027 | EXPECT_FLOAT_EQ(200.0f, matrix.f); |
Miklos Vajna | c765d2a | 2018-06-19 15:45:42 +0000 | [diff] [blame] | 2028 | |
Miklos Vajna | 8625d3b | 2018-06-26 15:12:48 +0000 | [diff] [blame] | 2029 | EXPECT_EQ(0, FPDFTextObj_GetFontSize(nullptr)); |
| 2030 | EXPECT_EQ(20, FPDFTextObj_GetFontSize(text_object3)); |
| 2031 | |
Nicolas Pena | 4905840 | 2017-02-14 18:26:20 -0500 | [diff] [blame] | 2032 | // TODO(npm): Why are there issues with text rotated by 90 degrees? |
| 2033 | // TODO(npm): FPDF_SaveAsCopy not giving the desired result after this. |
| 2034 | FPDF_ClosePage(page); |
Nicolas Pena | 4905840 | 2017-02-14 18:26:20 -0500 | [diff] [blame] | 2035 | } |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2036 | |
Daniel Hosseinian | 8cb6a65 | 2019-12-18 00:50:41 +0000 | [diff] [blame] | 2037 | TEST_F(FPDFEditEmbedderTest, GetTextRenderMode) { |
Miklos Vajna | 1448cc1 | 2018-07-03 13:52:33 +0000 | [diff] [blame] | 2038 | EXPECT_TRUE(OpenDocument("text_render_mode.pdf")); |
| 2039 | FPDF_PAGE page = LoadPage(0); |
| 2040 | ASSERT_TRUE(page); |
| 2041 | ASSERT_EQ(2, FPDFPage_CountObjects(page)); |
| 2042 | |
Daniel Hosseinian | e83d8b4 | 2020-01-08 00:26:15 +0000 | [diff] [blame] | 2043 | EXPECT_EQ(FPDF_TEXTRENDERMODE_UNKNOWN, |
Daniel Hosseinian | 8cb6a65 | 2019-12-18 00:50:41 +0000 | [diff] [blame] | 2044 | FPDFTextObj_GetTextRenderMode(nullptr)); |
Miklos Vajna | 1448cc1 | 2018-07-03 13:52:33 +0000 | [diff] [blame] | 2045 | |
| 2046 | FPDF_PAGEOBJECT fill = FPDFPage_GetObject(page, 0); |
Daniel Hosseinian | e83d8b4 | 2020-01-08 00:26:15 +0000 | [diff] [blame] | 2047 | EXPECT_EQ(FPDF_TEXTRENDERMODE_FILL, FPDFTextObj_GetTextRenderMode(fill)); |
Miklos Vajna | 1448cc1 | 2018-07-03 13:52:33 +0000 | [diff] [blame] | 2048 | |
| 2049 | FPDF_PAGEOBJECT stroke = FPDFPage_GetObject(page, 1); |
Daniel Hosseinian | e83d8b4 | 2020-01-08 00:26:15 +0000 | [diff] [blame] | 2050 | EXPECT_EQ(FPDF_TEXTRENDERMODE_STROKE, FPDFTextObj_GetTextRenderMode(stroke)); |
| 2051 | |
| 2052 | UnloadPage(page); |
| 2053 | } |
| 2054 | |
Hui Yingst | 609f7d6 | 2020-04-23 23:14:13 +0000 | [diff] [blame] | 2055 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 2056 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 2057 | #define MAYBE_SetTextRenderMode DISABLED_SetTextRenderMode |
| 2058 | #else |
| 2059 | #define MAYBE_SetTextRenderMode SetTextRenderMode |
| 2060 | #endif |
| 2061 | TEST_F(FPDFEditEmbedderTest, MAYBE_SetTextRenderMode) { |
Daniel Hosseinian | a933ebf | 2020-01-31 21:28:25 +0000 | [diff] [blame] | 2062 | // Platform specific original bitmaps |
Daniel Hosseinian | e83d8b4 | 2020-01-08 00:26:15 +0000 | [diff] [blame] | 2063 | #if defined(OS_MACOSX) |
Lei Zhang | 0f6b434 | 2020-02-25 20:00:39 +0000 | [diff] [blame] | 2064 | const char md5[] = "2453a9a524ee3f5f525dd21345ec4d81"; |
Daniel Hosseinian | e83d8b4 | 2020-01-08 00:26:15 +0000 | [diff] [blame] | 2065 | #elif defined(OS_WIN) |
Daniel Hosseinian | a933ebf | 2020-01-31 21:28:25 +0000 | [diff] [blame] | 2066 | const char md5[] = "de6e86bad3e9fda753a8471a45cfbb58"; |
Daniel Hosseinian | e83d8b4 | 2020-01-08 00:26:15 +0000 | [diff] [blame] | 2067 | #else |
Daniel Hosseinian | a933ebf | 2020-01-31 21:28:25 +0000 | [diff] [blame] | 2068 | const char md5[] = "5a012d2920ac075c39ffa9437ea42faa"; |
Daniel Hosseinian | e83d8b4 | 2020-01-08 00:26:15 +0000 | [diff] [blame] | 2069 | #endif |
Daniel Hosseinian | a933ebf | 2020-01-31 21:28:25 +0000 | [diff] [blame] | 2070 | const char md5_stroke[] = "412e52e621b46bd77baf2162e1fb1a1d"; |
Daniel Hosseinian | e83d8b4 | 2020-01-08 00:26:15 +0000 | [diff] [blame] | 2071 | |
Daniel Hosseinian | e83d8b4 | 2020-01-08 00:26:15 +0000 | [diff] [blame] | 2072 | { |
Daniel Hosseinian | a933ebf | 2020-01-31 21:28:25 +0000 | [diff] [blame] | 2073 | EXPECT_TRUE(OpenDocument("text_render_mode.pdf")); |
| 2074 | FPDF_PAGE page = LoadPage(0); |
| 2075 | ASSERT_TRUE(page); |
| 2076 | ASSERT_EQ(2, FPDFPage_CountObjects(page)); |
| 2077 | |
| 2078 | // Check the bitmap |
| 2079 | { |
| 2080 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
| 2081 | CompareBitmap(page_bitmap.get(), 612, 446, md5); |
| 2082 | } |
| 2083 | |
| 2084 | // Cannot set on a null object. |
| 2085 | EXPECT_FALSE( |
| 2086 | FPDFTextObj_SetTextRenderMode(nullptr, FPDF_TEXTRENDERMODE_UNKNOWN)); |
| 2087 | EXPECT_FALSE( |
| 2088 | FPDFTextObj_SetTextRenderMode(nullptr, FPDF_TEXTRENDERMODE_INVISIBLE)); |
| 2089 | |
| 2090 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0); |
| 2091 | ASSERT_TRUE(page_object); |
| 2092 | EXPECT_EQ(FPDF_TEXTRENDERMODE_FILL, |
| 2093 | FPDFTextObj_GetTextRenderMode(page_object)); |
| 2094 | |
| 2095 | // Cannot set UNKNOWN as a render mode. |
| 2096 | EXPECT_FALSE(FPDFTextObj_SetTextRenderMode(page_object, |
| 2097 | FPDF_TEXTRENDERMODE_UNKNOWN)); |
| 2098 | |
| 2099 | EXPECT_TRUE( |
| 2100 | FPDFTextObj_SetTextRenderMode(page_object, FPDF_TEXTRENDERMODE_STROKE)); |
| 2101 | EXPECT_EQ(FPDF_TEXTRENDERMODE_STROKE, |
| 2102 | FPDFTextObj_GetTextRenderMode(page_object)); |
| 2103 | |
| 2104 | // Check that bitmap displays changed content |
| 2105 | { |
| 2106 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
| 2107 | CompareBitmap(page_bitmap.get(), 612, 446, md5_stroke); |
| 2108 | } |
| 2109 | |
| 2110 | UnloadPage(page); |
Daniel Hosseinian | e83d8b4 | 2020-01-08 00:26:15 +0000 | [diff] [blame] | 2111 | } |
Miklos Vajna | 1448cc1 | 2018-07-03 13:52:33 +0000 | [diff] [blame] | 2112 | |
Daniel Hosseinian | a933ebf | 2020-01-31 21:28:25 +0000 | [diff] [blame] | 2113 | { |
| 2114 | // Save a copy, open the copy, and render it. |
| 2115 | // Note that the text render mode should be set to |
| 2116 | // |FPDF_TEXTRENDERMODE_STROKE|, but isn't. |
| 2117 | // TODO(crbug.com/pdfium/1470): Fix FPDFTextObj_SetTextRenderMode() to work |
| 2118 | // on saved documents. |
| 2119 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 2120 | ASSERT_TRUE(OpenSavedDocument()); |
| 2121 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 2122 | ASSERT_TRUE(saved_page); |
| 2123 | |
| 2124 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(saved_page, 0); |
| 2125 | EXPECT_TRUE(page_object); |
| 2126 | EXPECT_EQ(FPDF_TEXTRENDERMODE_FILL, |
| 2127 | FPDFTextObj_GetTextRenderMode(page_object)); |
| 2128 | |
| 2129 | ScopedFPDFBitmap bitmap = RenderSavedPage(saved_page); |
| 2130 | CompareBitmap(bitmap.get(), 612, 446, md5); |
| 2131 | |
| 2132 | CloseSavedPage(saved_page); |
| 2133 | CloseSavedDocument(); |
| 2134 | } |
Miklos Vajna | 1448cc1 | 2018-07-03 13:52:33 +0000 | [diff] [blame] | 2135 | } |
| 2136 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 2137 | TEST_F(FPDFEditEmbedderTest, TestGetTextFontName) { |
Miklos Vajna | 53d4f0a | 2018-08-01 01:28:49 +0000 | [diff] [blame] | 2138 | EXPECT_TRUE(OpenDocument("text_font.pdf")); |
| 2139 | FPDF_PAGE page = LoadPage(0); |
| 2140 | ASSERT_TRUE(page); |
| 2141 | ASSERT_EQ(1, FPDFPage_CountObjects(page)); |
| 2142 | |
| 2143 | // FPDFTextObj_GetFontName() positive testing. |
| 2144 | FPDF_PAGEOBJECT text = FPDFPage_GetObject(page, 0); |
| 2145 | unsigned long size = FPDFTextObj_GetFontName(text, nullptr, 0); |
| 2146 | const char kExpectedFontName[] = "Liberation Serif"; |
| 2147 | ASSERT_EQ(sizeof(kExpectedFontName), size); |
| 2148 | std::vector<char> font_name(size); |
| 2149 | ASSERT_EQ(size, FPDFTextObj_GetFontName(text, font_name.data(), size)); |
| 2150 | ASSERT_STREQ(kExpectedFontName, font_name.data()); |
| 2151 | |
| 2152 | // FPDFTextObj_GetFontName() negative testing. |
| 2153 | ASSERT_EQ(0U, FPDFTextObj_GetFontName(nullptr, nullptr, 0)); |
| 2154 | |
| 2155 | font_name.resize(2); |
| 2156 | font_name[0] = 'x'; |
| 2157 | font_name[1] = '\0'; |
| 2158 | size = FPDFTextObj_GetFontName(text, font_name.data(), font_name.size()); |
| 2159 | ASSERT_EQ(sizeof(kExpectedFontName), size); |
| 2160 | ASSERT_EQ(std::string("x"), std::string(font_name.data())); |
| 2161 | |
| 2162 | UnloadPage(page); |
| 2163 | } |
| 2164 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 2165 | TEST_F(FPDFEditEmbedderTest, TestFormGetObjects) { |
Miklos Vajna | b66077d | 2018-07-11 13:25:02 +0000 | [diff] [blame] | 2166 | EXPECT_TRUE(OpenDocument("form_object.pdf")); |
| 2167 | FPDF_PAGE page = LoadPage(0); |
| 2168 | ASSERT_TRUE(page); |
| 2169 | ASSERT_EQ(1, FPDFPage_CountObjects(page)); |
| 2170 | |
| 2171 | FPDF_PAGEOBJECT form = FPDFPage_GetObject(page, 0); |
| 2172 | EXPECT_EQ(FPDF_PAGEOBJ_FORM, FPDFPageObj_GetType(form)); |
| 2173 | ASSERT_EQ(-1, FPDFFormObj_CountObjects(nullptr)); |
| 2174 | ASSERT_EQ(2, FPDFFormObj_CountObjects(form)); |
| 2175 | |
Miklos Vajna | 1d273f1 | 2018-07-16 19:20:36 +0000 | [diff] [blame] | 2176 | // FPDFFormObj_GetObject() positive testing. |
| 2177 | FPDF_PAGEOBJECT text1 = FPDFFormObj_GetObject(form, 0); |
| 2178 | ASSERT_TRUE(text1); |
| 2179 | float left = 0; |
| 2180 | float bottom = 0; |
| 2181 | float right = 0; |
| 2182 | float top = 0; |
| 2183 | ASSERT_TRUE(FPDFPageObj_GetBounds(text1, &left, &bottom, &right, &top)); |
| 2184 | ASSERT_EQ(271, static_cast<int>(top)); |
| 2185 | |
| 2186 | FPDF_PAGEOBJECT text2 = FPDFFormObj_GetObject(form, 1); |
| 2187 | ASSERT_TRUE(text2); |
| 2188 | ASSERT_TRUE(FPDFPageObj_GetBounds(text2, &left, &bottom, &right, &top)); |
| 2189 | ASSERT_EQ(221, static_cast<int>(top)); |
| 2190 | |
| 2191 | // FPDFFormObj_GetObject() negative testing. |
| 2192 | ASSERT_EQ(nullptr, FPDFFormObj_GetObject(nullptr, 0)); |
| 2193 | ASSERT_EQ(nullptr, FPDFFormObj_GetObject(form, -1)); |
| 2194 | ASSERT_EQ(nullptr, FPDFFormObj_GetObject(form, 2)); |
| 2195 | |
Miklos Vajna | 46b4373 | 2018-08-14 19:15:43 +0000 | [diff] [blame] | 2196 | // Reset the form object matrix to identity. |
| 2197 | auto* pPageObj = CPDFPageObjectFromFPDFPageObject(form); |
| 2198 | CPDF_FormObject* pFormObj = pPageObj->AsForm(); |
| 2199 | pFormObj->Transform(pFormObj->form_matrix().GetInverse()); |
| 2200 | |
| 2201 | // FPDFFormObj_GetMatrix() positive testing. |
Lei Zhang | 8da9823 | 2019-12-11 23:29:33 +0000 | [diff] [blame] | 2202 | static constexpr FS_MATRIX kMatrix = {1.0f, 1.5f, 2.0f, 2.5f, 100.0f, 200.0f}; |
| 2203 | pFormObj->Transform(CFXMatrixFromFSMatrix(kMatrix)); |
Miklos Vajna | 46b4373 | 2018-08-14 19:15:43 +0000 | [diff] [blame] | 2204 | |
Lei Zhang | 8da9823 | 2019-12-11 23:29:33 +0000 | [diff] [blame] | 2205 | FS_MATRIX matrix; |
| 2206 | EXPECT_TRUE(FPDFFormObj_GetMatrix(form, &matrix)); |
| 2207 | EXPECT_FLOAT_EQ(kMatrix.a, matrix.a); |
| 2208 | EXPECT_FLOAT_EQ(kMatrix.b, matrix.b); |
| 2209 | EXPECT_FLOAT_EQ(kMatrix.c, matrix.c); |
| 2210 | EXPECT_FLOAT_EQ(kMatrix.d, matrix.d); |
| 2211 | EXPECT_FLOAT_EQ(kMatrix.e, matrix.e); |
| 2212 | EXPECT_FLOAT_EQ(kMatrix.f, matrix.f); |
Miklos Vajna | 46b4373 | 2018-08-14 19:15:43 +0000 | [diff] [blame] | 2213 | |
| 2214 | // FPDFFormObj_GetMatrix() negative testing. |
Lei Zhang | 8da9823 | 2019-12-11 23:29:33 +0000 | [diff] [blame] | 2215 | EXPECT_FALSE(FPDFFormObj_GetMatrix(nullptr, &matrix)); |
| 2216 | EXPECT_FALSE(FPDFFormObj_GetMatrix(form, nullptr)); |
| 2217 | EXPECT_FALSE(FPDFFormObj_GetMatrix(nullptr, nullptr)); |
Miklos Vajna | 46b4373 | 2018-08-14 19:15:43 +0000 | [diff] [blame] | 2218 | |
Miklos Vajna | b66077d | 2018-07-11 13:25:02 +0000 | [diff] [blame] | 2219 | UnloadPage(page); |
| 2220 | } |
| 2221 | |
Nicolas Pena | 4c48b10 | 2018-06-13 18:23:46 +0000 | [diff] [blame] | 2222 | // Tests adding text from standard font using FPDFText_LoadStandardFont. |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 2223 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 2224 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 2225 | #define MAYBE_AddStandardFontText2 DISABLED_AddStandardFontText2 |
| 2226 | #else |
| 2227 | #define MAYBE_AddStandardFontText2 AddStandardFontText2 |
| 2228 | #endif |
| 2229 | TEST_F(FPDFEditEmbedderTest, MAYBE_AddStandardFontText2) { |
Nicolas Pena | 4c48b10 | 2018-06-13 18:23:46 +0000 | [diff] [blame] | 2230 | // Start with a blank page |
| 2231 | ScopedFPDFPage page(FPDFPage_New(CreateNewDocument(), 0, 612, 792)); |
| 2232 | |
| 2233 | // Load a standard font. |
Tom Sepez | ffff6c5 | 2019-07-30 21:56:19 +0000 | [diff] [blame] | 2234 | ScopedFPDFFont font(FPDFText_LoadStandardFont(document(), "Helvetica")); |
Nicolas Pena | 4c48b10 | 2018-06-13 18:23:46 +0000 | [diff] [blame] | 2235 | ASSERT_TRUE(font); |
| 2236 | |
| 2237 | // Add some text to the page. |
| 2238 | FPDF_PAGEOBJECT text_object = |
Tom Sepez | ffff6c5 | 2019-07-30 21:56:19 +0000 | [diff] [blame] | 2239 | FPDFPageObj_CreateTextObj(document(), font.get(), 12.0f); |
Nicolas Pena | 4c48b10 | 2018-06-13 18:23:46 +0000 | [diff] [blame] | 2240 | EXPECT_TRUE(text_object); |
Lei Zhang | f0f6768 | 2019-04-08 17:03:21 +0000 | [diff] [blame] | 2241 | ScopedFPDFWideString text = |
Nicolas Pena | 4c48b10 | 2018-06-13 18:23:46 +0000 | [diff] [blame] | 2242 | GetFPDFWideString(L"I'm at the bottom of the page"); |
| 2243 | EXPECT_TRUE(FPDFText_SetText(text_object, text.get())); |
| 2244 | FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 20, 20); |
| 2245 | FPDFPage_InsertObject(page.get(), text_object); |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 2246 | ScopedFPDFBitmap page_bitmap = RenderPage(page.get()); |
Lei Zhang | e4cdac5 | 2019-04-30 16:45:57 +0000 | [diff] [blame] | 2247 | #if defined(OS_MACOSX) |
Lei Zhang | 0f6b434 | 2020-02-25 20:00:39 +0000 | [diff] [blame] | 2248 | const char md5[] = "324e1db8164a040cf6104538baa95ba6"; |
Lei Zhang | e67bcc7 | 2019-04-30 18:55:58 +0000 | [diff] [blame] | 2249 | #elif defined(OS_WIN) |
Lei Zhang | 5e2c51c | 2018-07-27 22:33:34 +0000 | [diff] [blame] | 2250 | const char md5[] = "08d1ff3e5a42801bee6077fd366bef00"; |
Nicolas Pena | 4c48b10 | 2018-06-13 18:23:46 +0000 | [diff] [blame] | 2251 | #else |
| 2252 | const char md5[] = "eacaa24573b8ce997b3882595f096f00"; |
| 2253 | #endif |
| 2254 | CompareBitmap(page_bitmap.get(), 612, 792, md5); |
| 2255 | } |
| 2256 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 2257 | TEST_F(FPDFEditEmbedderTest, LoadStandardFonts) { |
Nicolas Pena | 4c48b10 | 2018-06-13 18:23:46 +0000 | [diff] [blame] | 2258 | CreateNewDocument(); |
Lei Zhang | 590f424 | 2019-05-15 20:57:26 +0000 | [diff] [blame] | 2259 | static constexpr const char* kStandardFontNames[] = { |
Lei Zhang | d72fd58 | 2018-07-27 19:37:27 +0000 | [diff] [blame] | 2260 | "Arial", |
| 2261 | "Arial-Bold", |
| 2262 | "Arial-BoldItalic", |
| 2263 | "Arial-Italic", |
| 2264 | "Courier", |
| 2265 | "Courier-BoldOblique", |
| 2266 | "Courier-Oblique", |
| 2267 | "Courier-Bold", |
| 2268 | "CourierNew", |
| 2269 | "CourierNew-Bold", |
| 2270 | "CourierNew-BoldItalic", |
| 2271 | "CourierNew-Italic", |
| 2272 | "Helvetica", |
| 2273 | "Helvetica-Bold", |
| 2274 | "Helvetica-BoldOblique", |
| 2275 | "Helvetica-Oblique", |
| 2276 | "Symbol", |
| 2277 | "TimesNewRoman", |
| 2278 | "TimesNewRoman-Bold", |
| 2279 | "TimesNewRoman-BoldItalic", |
| 2280 | "TimesNewRoman-Italic", |
| 2281 | "ZapfDingbats"}; |
Lei Zhang | 590f424 | 2019-05-15 20:57:26 +0000 | [diff] [blame] | 2282 | for (const char* font_name : kStandardFontNames) { |
Tom Sepez | ffff6c5 | 2019-07-30 21:56:19 +0000 | [diff] [blame] | 2283 | ScopedFPDFFont font(FPDFText_LoadStandardFont(document(), font_name)); |
Nicolas Pena | 4c48b10 | 2018-06-13 18:23:46 +0000 | [diff] [blame] | 2284 | EXPECT_TRUE(font) << font_name << " should be considered a standard font."; |
| 2285 | } |
Lei Zhang | 590f424 | 2019-05-15 20:57:26 +0000 | [diff] [blame] | 2286 | static constexpr const char* kNotStandardFontNames[] = { |
Nicolas Pena | 4c48b10 | 2018-06-13 18:23:46 +0000 | [diff] [blame] | 2287 | "Abcdefg", "ArialB", "Arial-Style", |
| 2288 | "Font Name", "FontArial", "NotAStandardFontName", |
| 2289 | "TestFontName", "Quack", "Symbol-Italic", |
| 2290 | "Zapf"}; |
Lei Zhang | 590f424 | 2019-05-15 20:57:26 +0000 | [diff] [blame] | 2291 | for (const char* font_name : kNotStandardFontNames) { |
Tom Sepez | ffff6c5 | 2019-07-30 21:56:19 +0000 | [diff] [blame] | 2292 | ScopedFPDFFont font(FPDFText_LoadStandardFont(document(), font_name)); |
Nicolas Pena | 4c48b10 | 2018-06-13 18:23:46 +0000 | [diff] [blame] | 2293 | EXPECT_FALSE(font) << font_name |
| 2294 | << " should not be considered a standard font."; |
| 2295 | } |
| 2296 | } |
| 2297 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 2298 | TEST_F(FPDFEditEmbedderTest, GraphicsData) { |
Nicolas Pena | 603a31d | 2017-06-14 11:41:18 -0400 | [diff] [blame] | 2299 | // New page |
Tom Sepez | e08d2b1 | 2018-04-25 18:49:32 +0000 | [diff] [blame] | 2300 | ScopedFPDFPage page(FPDFPage_New(CreateNewDocument(), 0, 612, 792)); |
Nicolas Pena | 603a31d | 2017-06-14 11:41:18 -0400 | [diff] [blame] | 2301 | |
| 2302 | // Create a rect with nontrivial graphics |
| 2303 | FPDF_PAGEOBJECT rect1 = FPDFPageObj_CreateNewRect(10, 10, 100, 100); |
| 2304 | FPDFPageObj_SetBlendMode(rect1, "Color"); |
| 2305 | FPDFPage_InsertObject(page.get(), rect1); |
| 2306 | EXPECT_TRUE(FPDFPage_GenerateContent(page.get())); |
| 2307 | |
| 2308 | // Check that the ExtGState was created |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2309 | CPDF_Page* cpage = CPDFPageFromFPDFPage(page.get()); |
| 2310 | CPDF_Dictionary* graphics_dict = cpage->m_pResources->GetDictFor("ExtGState"); |
Nicolas Pena | 603a31d | 2017-06-14 11:41:18 -0400 | [diff] [blame] | 2311 | ASSERT_TRUE(graphics_dict); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2312 | EXPECT_EQ(2u, graphics_dict->size()); |
Nicolas Pena | 603a31d | 2017-06-14 11:41:18 -0400 | [diff] [blame] | 2313 | |
| 2314 | // Add a text object causing no change to the graphics dictionary |
| 2315 | FPDF_PAGEOBJECT text1 = FPDFPageObj_NewTextObj(document(), "Arial", 12.0f); |
| 2316 | // Only alpha, the last component, matters for the graphics dictionary. And |
| 2317 | // the default value is 255. |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 2318 | EXPECT_TRUE(FPDFPageObj_SetFillColor(text1, 100, 100, 100, 255)); |
Nicolas Pena | 603a31d | 2017-06-14 11:41:18 -0400 | [diff] [blame] | 2319 | FPDFPage_InsertObject(page.get(), text1); |
| 2320 | EXPECT_TRUE(FPDFPage_GenerateContent(page.get())); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2321 | EXPECT_EQ(2u, graphics_dict->size()); |
Nicolas Pena | 603a31d | 2017-06-14 11:41:18 -0400 | [diff] [blame] | 2322 | |
| 2323 | // Add a text object increasing the size of the graphics dictionary |
| 2324 | FPDF_PAGEOBJECT text2 = |
| 2325 | FPDFPageObj_NewTextObj(document(), "Times-Roman", 12.0f); |
| 2326 | FPDFPage_InsertObject(page.get(), text2); |
| 2327 | FPDFPageObj_SetBlendMode(text2, "Darken"); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 2328 | EXPECT_TRUE(FPDFPageObj_SetFillColor(text2, 0, 0, 255, 150)); |
Nicolas Pena | 603a31d | 2017-06-14 11:41:18 -0400 | [diff] [blame] | 2329 | EXPECT_TRUE(FPDFPage_GenerateContent(page.get())); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2330 | EXPECT_EQ(3u, graphics_dict->size()); |
Nicolas Pena | 603a31d | 2017-06-14 11:41:18 -0400 | [diff] [blame] | 2331 | |
| 2332 | // Add a path that should reuse graphics |
Nicolas Pena | ce67be4 | 2017-06-14 14:52:49 -0400 | [diff] [blame] | 2333 | FPDF_PAGEOBJECT path = FPDFPageObj_CreateNewPath(400, 100); |
Nicolas Pena | 603a31d | 2017-06-14 11:41:18 -0400 | [diff] [blame] | 2334 | FPDFPageObj_SetBlendMode(path, "Darken"); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 2335 | EXPECT_TRUE(FPDFPageObj_SetFillColor(path, 200, 200, 100, 150)); |
Nicolas Pena | 603a31d | 2017-06-14 11:41:18 -0400 | [diff] [blame] | 2336 | FPDFPage_InsertObject(page.get(), path); |
| 2337 | EXPECT_TRUE(FPDFPage_GenerateContent(page.get())); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2338 | EXPECT_EQ(3u, graphics_dict->size()); |
Nicolas Pena | 603a31d | 2017-06-14 11:41:18 -0400 | [diff] [blame] | 2339 | |
| 2340 | // Add a rect increasing the size of the graphics dictionary |
| 2341 | FPDF_PAGEOBJECT rect2 = FPDFPageObj_CreateNewRect(10, 10, 100, 100); |
| 2342 | FPDFPageObj_SetBlendMode(rect2, "Darken"); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 2343 | EXPECT_TRUE(FPDFPageObj_SetFillColor(rect2, 0, 0, 255, 150)); |
| 2344 | EXPECT_TRUE(FPDFPageObj_SetStrokeColor(rect2, 0, 0, 0, 200)); |
Nicolas Pena | 603a31d | 2017-06-14 11:41:18 -0400 | [diff] [blame] | 2345 | FPDFPage_InsertObject(page.get(), rect2); |
| 2346 | EXPECT_TRUE(FPDFPage_GenerateContent(page.get())); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2347 | EXPECT_EQ(4u, graphics_dict->size()); |
Nicolas Pena | 603a31d | 2017-06-14 11:41:18 -0400 | [diff] [blame] | 2348 | } |
| 2349 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 2350 | TEST_F(FPDFEditEmbedderTest, DoubleGenerating) { |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2351 | // Start with a blank page |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2352 | FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792); |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2353 | |
| 2354 | // Add a red rectangle with some non-default alpha |
| 2355 | FPDF_PAGEOBJECT rect = FPDFPageObj_CreateNewRect(10, 10, 100, 100); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 2356 | EXPECT_TRUE(FPDFPageObj_SetFillColor(rect, 255, 0, 0, 128)); |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2357 | EXPECT_TRUE(FPDFPath_SetDrawMode(rect, FPDF_FILLMODE_WINDING, 0)); |
| 2358 | FPDFPage_InsertObject(page, rect); |
| 2359 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 2360 | |
| 2361 | // Check the ExtGState |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2362 | CPDF_Page* cpage = CPDFPageFromFPDFPage(page); |
| 2363 | CPDF_Dictionary* graphics_dict = cpage->m_pResources->GetDictFor("ExtGState"); |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2364 | ASSERT_TRUE(graphics_dict); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2365 | EXPECT_EQ(2u, graphics_dict->size()); |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2366 | |
| 2367 | // Check the bitmap |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2368 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 2369 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2370 | CompareBitmap(page_bitmap.get(), 612, 792, |
| 2371 | "5384da3406d62360ffb5cac4476fff1c"); |
| 2372 | } |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2373 | |
| 2374 | // Never mind, my new favorite color is blue, increase alpha |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 2375 | EXPECT_TRUE(FPDFPageObj_SetFillColor(rect, 0, 0, 255, 180)); |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2376 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2377 | EXPECT_EQ(3u, graphics_dict->size()); |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2378 | |
| 2379 | // Check that bitmap displays changed content |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2380 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 2381 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2382 | CompareBitmap(page_bitmap.get(), 612, 792, |
| 2383 | "2e51656f5073b0bee611d9cd086aa09c"); |
| 2384 | } |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2385 | |
| 2386 | // And now generate, without changes |
| 2387 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2388 | EXPECT_EQ(3u, graphics_dict->size()); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2389 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 2390 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2391 | CompareBitmap(page_bitmap.get(), 612, 792, |
| 2392 | "2e51656f5073b0bee611d9cd086aa09c"); |
| 2393 | } |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2394 | |
| 2395 | // Add some text to the page |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2396 | FPDF_PAGEOBJECT text_object = |
| 2397 | FPDFPageObj_NewTextObj(document(), "Arial", 12.0f); |
Lei Zhang | f0f6768 | 2019-04-08 17:03:21 +0000 | [diff] [blame] | 2398 | ScopedFPDFWideString text = |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2399 | GetFPDFWideString(L"Something something #text# something"); |
| 2400 | EXPECT_TRUE(FPDFText_SetText(text_object, text.get())); |
| 2401 | FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 300, 300); |
| 2402 | FPDFPage_InsertObject(page, text_object); |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2403 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2404 | CPDF_Dictionary* font_dict = cpage->m_pResources->GetDictFor("Font"); |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2405 | ASSERT_TRUE(font_dict); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2406 | EXPECT_EQ(1u, font_dict->size()); |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2407 | |
| 2408 | // Generate yet again, check dicts are reasonably sized |
| 2409 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2410 | EXPECT_EQ(3u, graphics_dict->size()); |
| 2411 | EXPECT_EQ(1u, font_dict->size()); |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2412 | FPDF_ClosePage(page); |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2413 | } |
Nicolas Pena | be90aae | 2017-02-27 10:41:41 -0500 | [diff] [blame] | 2414 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 2415 | TEST_F(FPDFEditEmbedderTest, LoadSimpleType1Font) { |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2416 | CreateNewDocument(); |
| 2417 | // TODO(npm): use other fonts after disallowing loading any font as any type |
Tom Sepez | ffff6c5 | 2019-07-30 21:56:19 +0000 | [diff] [blame] | 2418 | RetainPtr<CPDF_Font> stock_font = |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2419 | CPDF_Font::GetStockFont(cpdf_doc(), "Times-Bold"); |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 2420 | pdfium::span<const uint8_t> span = stock_font->GetFont()->GetFontSpan(); |
| 2421 | ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(), |
| 2422 | FPDF_FONT_TYPE1, false)); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2423 | ASSERT_TRUE(font.get()); |
Tom Sepez | 525147a | 2018-05-03 17:19:53 +0000 | [diff] [blame] | 2424 | CPDF_Font* typed_font = CPDFFontFromFPDFFont(font.get()); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2425 | EXPECT_TRUE(typed_font->IsType1Font()); |
Nicolas Pena | be90aae | 2017-02-27 10:41:41 -0500 | [diff] [blame] | 2426 | |
Lei Zhang | 710fa99 | 2018-05-25 16:24:48 +0000 | [diff] [blame] | 2427 | const CPDF_Dictionary* font_dict = typed_font->GetFontDict(); |
Nicolas Pena | be90aae | 2017-02-27 10:41:41 -0500 | [diff] [blame] | 2428 | EXPECT_EQ("Font", font_dict->GetStringFor("Type")); |
| 2429 | EXPECT_EQ("Type1", font_dict->GetStringFor("Subtype")); |
KDr2 | 7a62988 | 2019-02-01 01:16:38 +0000 | [diff] [blame] | 2430 | EXPECT_EQ("TimesNewRomanPS-BoldMT", font_dict->GetStringFor("BaseFont")); |
Nicolas Pena | be90aae | 2017-02-27 10:41:41 -0500 | [diff] [blame] | 2431 | ASSERT_TRUE(font_dict->KeyExist("FirstChar")); |
| 2432 | ASSERT_TRUE(font_dict->KeyExist("LastChar")); |
| 2433 | EXPECT_EQ(32, font_dict->GetIntegerFor("FirstChar")); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2434 | EXPECT_EQ(255, font_dict->GetIntegerFor("LastChar")); |
| 2435 | |
Lei Zhang | de579ab | 2018-05-25 21:49:49 +0000 | [diff] [blame] | 2436 | const CPDF_Array* widths_array = font_dict->GetArrayFor("Widths"); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2437 | ASSERT_TRUE(widths_array); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2438 | ASSERT_EQ(224u, widths_array->size()); |
Nicolas Pena | be90aae | 2017-02-27 10:41:41 -0500 | [diff] [blame] | 2439 | EXPECT_EQ(250, widths_array->GetNumberAt(0)); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2440 | EXPECT_EQ(569, widths_array->GetNumberAt(11)); |
| 2441 | EXPECT_EQ(500, widths_array->GetNumberAt(223)); |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 2442 | CheckFontDescriptor(font_dict, FPDF_FONT_TYPE1, true, false, span); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2443 | } |
Nicolas Pena | be90aae | 2017-02-27 10:41:41 -0500 | [diff] [blame] | 2444 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 2445 | TEST_F(FPDFEditEmbedderTest, LoadSimpleTrueTypeFont) { |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2446 | CreateNewDocument(); |
Tom Sepez | ffff6c5 | 2019-07-30 21:56:19 +0000 | [diff] [blame] | 2447 | RetainPtr<CPDF_Font> stock_font = |
| 2448 | CPDF_Font::GetStockFont(cpdf_doc(), "Courier"); |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 2449 | pdfium::span<const uint8_t> span = stock_font->GetFont()->GetFontSpan(); |
| 2450 | ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(), |
| 2451 | FPDF_FONT_TRUETYPE, false)); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2452 | ASSERT_TRUE(font.get()); |
Tom Sepez | 525147a | 2018-05-03 17:19:53 +0000 | [diff] [blame] | 2453 | CPDF_Font* typed_font = CPDFFontFromFPDFFont(font.get()); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2454 | EXPECT_TRUE(typed_font->IsTrueTypeFont()); |
Nicolas Pena | be90aae | 2017-02-27 10:41:41 -0500 | [diff] [blame] | 2455 | |
Lei Zhang | 710fa99 | 2018-05-25 16:24:48 +0000 | [diff] [blame] | 2456 | const CPDF_Dictionary* font_dict = typed_font->GetFontDict(); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2457 | EXPECT_EQ("Font", font_dict->GetStringFor("Type")); |
| 2458 | EXPECT_EQ("TrueType", font_dict->GetStringFor("Subtype")); |
KDr2 | 7a62988 | 2019-02-01 01:16:38 +0000 | [diff] [blame] | 2459 | EXPECT_EQ("CourierNewPSMT", font_dict->GetStringFor("BaseFont")); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2460 | ASSERT_TRUE(font_dict->KeyExist("FirstChar")); |
| 2461 | ASSERT_TRUE(font_dict->KeyExist("LastChar")); |
| 2462 | EXPECT_EQ(32, font_dict->GetIntegerFor("FirstChar")); |
| 2463 | EXPECT_EQ(255, font_dict->GetIntegerFor("LastChar")); |
Nicolas Pena | be90aae | 2017-02-27 10:41:41 -0500 | [diff] [blame] | 2464 | |
Lei Zhang | de579ab | 2018-05-25 21:49:49 +0000 | [diff] [blame] | 2465 | const CPDF_Array* widths_array = font_dict->GetArrayFor("Widths"); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2466 | ASSERT_TRUE(widths_array); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2467 | ASSERT_EQ(224u, widths_array->size()); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2468 | EXPECT_EQ(600, widths_array->GetNumberAt(33)); |
| 2469 | EXPECT_EQ(600, widths_array->GetNumberAt(74)); |
| 2470 | EXPECT_EQ(600, widths_array->GetNumberAt(223)); |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 2471 | CheckFontDescriptor(font_dict, FPDF_FONT_TRUETYPE, false, false, span); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2472 | } |
| 2473 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 2474 | TEST_F(FPDFEditEmbedderTest, LoadCIDType0Font) { |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2475 | CreateNewDocument(); |
Tom Sepez | ffff6c5 | 2019-07-30 21:56:19 +0000 | [diff] [blame] | 2476 | RetainPtr<CPDF_Font> stock_font = |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2477 | CPDF_Font::GetStockFont(cpdf_doc(), "Times-Roman"); |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 2478 | pdfium::span<const uint8_t> span = stock_font->GetFont()->GetFontSpan(); |
| 2479 | ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(), |
| 2480 | FPDF_FONT_TYPE1, 1)); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2481 | ASSERT_TRUE(font.get()); |
Tom Sepez | 525147a | 2018-05-03 17:19:53 +0000 | [diff] [blame] | 2482 | CPDF_Font* typed_font = CPDFFontFromFPDFFont(font.get()); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2483 | EXPECT_TRUE(typed_font->IsCIDFont()); |
| 2484 | |
| 2485 | // Check font dictionary entries |
Lei Zhang | 710fa99 | 2018-05-25 16:24:48 +0000 | [diff] [blame] | 2486 | const CPDF_Dictionary* font_dict = typed_font->GetFontDict(); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2487 | EXPECT_EQ("Font", font_dict->GetStringFor("Type")); |
| 2488 | EXPECT_EQ("Type0", font_dict->GetStringFor("Subtype")); |
Lei Zhang | b6992dd | 2019-02-05 23:30:20 +0000 | [diff] [blame] | 2489 | EXPECT_EQ("TimesNewRomanPSMT-Identity-H", |
| 2490 | font_dict->GetStringFor("BaseFont")); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2491 | EXPECT_EQ("Identity-H", font_dict->GetStringFor("Encoding")); |
Lei Zhang | de579ab | 2018-05-25 21:49:49 +0000 | [diff] [blame] | 2492 | const CPDF_Array* descendant_array = |
| 2493 | font_dict->GetArrayFor("DescendantFonts"); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2494 | ASSERT_TRUE(descendant_array); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2495 | EXPECT_EQ(1u, descendant_array->size()); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2496 | |
| 2497 | // Check the CIDFontDict |
Lei Zhang | de579ab | 2018-05-25 21:49:49 +0000 | [diff] [blame] | 2498 | const CPDF_Dictionary* cidfont_dict = descendant_array->GetDictAt(0); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2499 | EXPECT_EQ("Font", cidfont_dict->GetStringFor("Type")); |
| 2500 | EXPECT_EQ("CIDFontType0", cidfont_dict->GetStringFor("Subtype")); |
KDr2 | 7a62988 | 2019-02-01 01:16:38 +0000 | [diff] [blame] | 2501 | EXPECT_EQ("TimesNewRomanPSMT", cidfont_dict->GetStringFor("BaseFont")); |
Lei Zhang | b1ec280 | 2018-05-25 21:55:24 +0000 | [diff] [blame] | 2502 | const CPDF_Dictionary* cidinfo_dict = |
| 2503 | cidfont_dict->GetDictFor("CIDSystemInfo"); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2504 | ASSERT_TRUE(cidinfo_dict); |
Lei Zhang | 9c950b1 | 2019-01-16 19:06:37 +0000 | [diff] [blame] | 2505 | const CPDF_Object* registry = cidinfo_dict->GetObjectFor("Registry"); |
| 2506 | ASSERT_TRUE(registry); |
KDr2 | 8da0e1b | 2019-01-17 03:44:29 +0000 | [diff] [blame] | 2507 | EXPECT_EQ(CPDF_Object::kString, registry->GetType()); |
Lei Zhang | 9c950b1 | 2019-01-16 19:06:37 +0000 | [diff] [blame] | 2508 | EXPECT_EQ("Adobe", registry->GetString()); |
| 2509 | const CPDF_Object* ordering = cidinfo_dict->GetObjectFor("Ordering"); |
| 2510 | ASSERT_TRUE(ordering); |
KDr2 | 8da0e1b | 2019-01-17 03:44:29 +0000 | [diff] [blame] | 2511 | EXPECT_EQ(CPDF_Object::kString, ordering->GetType()); |
Lei Zhang | 9c950b1 | 2019-01-16 19:06:37 +0000 | [diff] [blame] | 2512 | EXPECT_EQ("Identity", ordering->GetString()); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2513 | EXPECT_EQ(0, cidinfo_dict->GetNumberFor("Supplement")); |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 2514 | CheckFontDescriptor(cidfont_dict, FPDF_FONT_TYPE1, false, false, span); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2515 | |
| 2516 | // Check widths |
Lei Zhang | de579ab | 2018-05-25 21:49:49 +0000 | [diff] [blame] | 2517 | const CPDF_Array* widths_array = cidfont_dict->GetArrayFor("W"); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2518 | ASSERT_TRUE(widths_array); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2519 | EXPECT_GT(widths_array->size(), 1u); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2520 | CheckCompositeFontWidths(widths_array, typed_font); |
| 2521 | } |
| 2522 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 2523 | TEST_F(FPDFEditEmbedderTest, LoadCIDType2Font) { |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2524 | CreateNewDocument(); |
Tom Sepez | ffff6c5 | 2019-07-30 21:56:19 +0000 | [diff] [blame] | 2525 | RetainPtr<CPDF_Font> stock_font = |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2526 | CPDF_Font::GetStockFont(cpdf_doc(), "Helvetica-Oblique"); |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 2527 | pdfium::span<const uint8_t> span = stock_font->GetFont()->GetFontSpan(); |
| 2528 | ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(), |
| 2529 | FPDF_FONT_TRUETYPE, 1)); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2530 | ASSERT_TRUE(font.get()); |
Tom Sepez | 525147a | 2018-05-03 17:19:53 +0000 | [diff] [blame] | 2531 | CPDF_Font* typed_font = CPDFFontFromFPDFFont(font.get()); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2532 | EXPECT_TRUE(typed_font->IsCIDFont()); |
| 2533 | |
| 2534 | // Check font dictionary entries |
Lei Zhang | 710fa99 | 2018-05-25 16:24:48 +0000 | [diff] [blame] | 2535 | const CPDF_Dictionary* font_dict = typed_font->GetFontDict(); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2536 | EXPECT_EQ("Font", font_dict->GetStringFor("Type")); |
| 2537 | EXPECT_EQ("Type0", font_dict->GetStringFor("Subtype")); |
KDr2 | 7a62988 | 2019-02-01 01:16:38 +0000 | [diff] [blame] | 2538 | EXPECT_EQ("Arial-ItalicMT", font_dict->GetStringFor("BaseFont")); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2539 | EXPECT_EQ("Identity-H", font_dict->GetStringFor("Encoding")); |
Lei Zhang | de579ab | 2018-05-25 21:49:49 +0000 | [diff] [blame] | 2540 | const CPDF_Array* descendant_array = |
| 2541 | font_dict->GetArrayFor("DescendantFonts"); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2542 | ASSERT_TRUE(descendant_array); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2543 | EXPECT_EQ(1u, descendant_array->size()); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2544 | |
| 2545 | // Check the CIDFontDict |
Lei Zhang | de579ab | 2018-05-25 21:49:49 +0000 | [diff] [blame] | 2546 | const CPDF_Dictionary* cidfont_dict = descendant_array->GetDictAt(0); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2547 | EXPECT_EQ("Font", cidfont_dict->GetStringFor("Type")); |
| 2548 | EXPECT_EQ("CIDFontType2", cidfont_dict->GetStringFor("Subtype")); |
KDr2 | 7a62988 | 2019-02-01 01:16:38 +0000 | [diff] [blame] | 2549 | EXPECT_EQ("Arial-ItalicMT", cidfont_dict->GetStringFor("BaseFont")); |
Lei Zhang | b1ec280 | 2018-05-25 21:55:24 +0000 | [diff] [blame] | 2550 | const CPDF_Dictionary* cidinfo_dict = |
| 2551 | cidfont_dict->GetDictFor("CIDSystemInfo"); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2552 | ASSERT_TRUE(cidinfo_dict); |
| 2553 | EXPECT_EQ("Adobe", cidinfo_dict->GetStringFor("Registry")); |
| 2554 | EXPECT_EQ("Identity", cidinfo_dict->GetStringFor("Ordering")); |
| 2555 | EXPECT_EQ(0, cidinfo_dict->GetNumberFor("Supplement")); |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 2556 | CheckFontDescriptor(cidfont_dict, FPDF_FONT_TRUETYPE, false, true, span); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2557 | |
| 2558 | // Check widths |
Lei Zhang | de579ab | 2018-05-25 21:49:49 +0000 | [diff] [blame] | 2559 | const CPDF_Array* widths_array = cidfont_dict->GetArrayFor("W"); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2560 | ASSERT_TRUE(widths_array); |
| 2561 | CheckCompositeFontWidths(widths_array, typed_font); |
Nicolas Pena | be90aae | 2017-02-27 10:41:41 -0500 | [diff] [blame] | 2562 | } |
rbpotter | ce8e51e | 2017-04-28 12:42:47 -0700 | [diff] [blame] | 2563 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 2564 | TEST_F(FPDFEditEmbedderTest, NormalizeNegativeRotation) { |
rbpotter | ce8e51e | 2017-04-28 12:42:47 -0700 | [diff] [blame] | 2565 | // Load document with a -90 degree rotation |
| 2566 | EXPECT_TRUE(OpenDocument("bug_713197.pdf")); |
| 2567 | FPDF_PAGE page = LoadPage(0); |
| 2568 | EXPECT_NE(nullptr, page); |
| 2569 | |
| 2570 | EXPECT_EQ(3, FPDFPage_GetRotation(page)); |
| 2571 | UnloadPage(page); |
| 2572 | } |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2573 | |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 2574 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 2575 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 2576 | #define MAYBE_AddTrueTypeFontText DISABLED_AddTrueTypeFontText |
| 2577 | #else |
| 2578 | #define MAYBE_AddTrueTypeFontText AddTrueTypeFontText |
| 2579 | #endif |
| 2580 | TEST_F(FPDFEditEmbedderTest, MAYBE_AddTrueTypeFontText) { |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2581 | // Start with a blank page |
| 2582 | FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792); |
| 2583 | { |
Tom Sepez | ffff6c5 | 2019-07-30 21:56:19 +0000 | [diff] [blame] | 2584 | RetainPtr<CPDF_Font> stock_font = |
| 2585 | CPDF_Font::GetStockFont(cpdf_doc(), "Arial"); |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 2586 | pdfium::span<const uint8_t> span = stock_font->GetFont()->GetFontSpan(); |
| 2587 | ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(), |
| 2588 | FPDF_FONT_TRUETYPE, 0)); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2589 | ASSERT_TRUE(font.get()); |
| 2590 | |
| 2591 | // Add some text to the page |
| 2592 | FPDF_PAGEOBJECT text_object = |
| 2593 | FPDFPageObj_CreateTextObj(document(), font.get(), 12.0f); |
| 2594 | EXPECT_TRUE(text_object); |
Lei Zhang | f0f6768 | 2019-04-08 17:03:21 +0000 | [diff] [blame] | 2595 | ScopedFPDFWideString text = |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2596 | GetFPDFWideString(L"I am testing my loaded font, WEE."); |
| 2597 | EXPECT_TRUE(FPDFText_SetText(text_object, text.get())); |
| 2598 | FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 400, 400); |
| 2599 | FPDFPage_InsertObject(page, text_object); |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 2600 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Lei Zhang | e4cdac5 | 2019-04-30 16:45:57 +0000 | [diff] [blame] | 2601 | #if defined(OS_MACOSX) |
Lei Zhang | 0f6b434 | 2020-02-25 20:00:39 +0000 | [diff] [blame] | 2602 | const char md5[] = "fc921c0bbdde73986ac13c15a85db4c3"; |
Lei Zhang | e67bcc7 | 2019-04-30 18:55:58 +0000 | [diff] [blame] | 2603 | #elif defined(OS_WIN) |
Lei Zhang | 5e2c51c | 2018-07-27 22:33:34 +0000 | [diff] [blame] | 2604 | const char md5[] = "d60ba39f9698e32360d99e727dd93165"; |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2605 | #else |
Henrique Nakashima | 09b4192 | 2017-10-27 20:39:29 +0000 | [diff] [blame] | 2606 | const char md5[] = "70592859010ffbf532a2237b8118bcc4"; |
Lei Zhang | e4cdac5 | 2019-04-30 16:45:57 +0000 | [diff] [blame] | 2607 | #endif |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2608 | CompareBitmap(page_bitmap.get(), 612, 792, md5); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2609 | |
| 2610 | // Add some more text, same font |
| 2611 | FPDF_PAGEOBJECT text_object2 = |
| 2612 | FPDFPageObj_CreateTextObj(document(), font.get(), 15.0f); |
Lei Zhang | f0f6768 | 2019-04-08 17:03:21 +0000 | [diff] [blame] | 2613 | ScopedFPDFWideString text2 = GetFPDFWideString(L"Bigger font size"); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2614 | EXPECT_TRUE(FPDFText_SetText(text_object2, text2.get())); |
| 2615 | FPDFPageObj_Transform(text_object2, 1, 0, 0, 1, 200, 200); |
| 2616 | FPDFPage_InsertObject(page, text_object2); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2617 | } |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 2618 | ScopedFPDFBitmap page_bitmap2 = RenderPage(page); |
Lei Zhang | e4cdac5 | 2019-04-30 16:45:57 +0000 | [diff] [blame] | 2619 | #if defined(OS_MACOSX) |
Lei Zhang | 0f6b434 | 2020-02-25 20:00:39 +0000 | [diff] [blame] | 2620 | const char md5_2[] = "9a1a7dfebe659513691aadd0d95b8d50"; |
Lei Zhang | e67bcc7 | 2019-04-30 18:55:58 +0000 | [diff] [blame] | 2621 | #elif defined(OS_WIN) |
Lei Zhang | 5e2c51c | 2018-07-27 22:33:34 +0000 | [diff] [blame] | 2622 | const char md5_2[] = "2199b579c49ab5f80c246a586a80ee90"; |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2623 | #else |
Henrique Nakashima | 09b4192 | 2017-10-27 20:39:29 +0000 | [diff] [blame] | 2624 | const char md5_2[] = "c1d10cce1761c4a998a16b2562030568"; |
Lei Zhang | e4cdac5 | 2019-04-30 16:45:57 +0000 | [diff] [blame] | 2625 | #endif |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2626 | CompareBitmap(page_bitmap2.get(), 612, 792, md5_2); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2627 | |
Nicolas Pena | 207b727 | 2017-05-26 17:37:06 -0400 | [diff] [blame] | 2628 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2629 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 2630 | FPDF_ClosePage(page); |
Dan Sinclair | 04e4dc8 | 2017-10-18 12:17:14 -0400 | [diff] [blame] | 2631 | |
| 2632 | VerifySavedDocument(612, 792, md5_2); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2633 | } |
Nicolas Pena | f45ade3 | 2017-05-03 10:23:49 -0400 | [diff] [blame] | 2634 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 2635 | TEST_F(FPDFEditEmbedderTest, TransformAnnot) { |
Jane Liu | eda6525 | 2017-06-07 11:31:27 -0400 | [diff] [blame] | 2636 | // Open a file with one annotation and load its first page. |
| 2637 | ASSERT_TRUE(OpenDocument("annotation_highlight_long_content.pdf")); |
Lei Zhang | 75c8171 | 2018-02-08 17:22:39 +0000 | [diff] [blame] | 2638 | FPDF_PAGE page = LoadPage(0); |
Jane Liu | eda6525 | 2017-06-07 11:31:27 -0400 | [diff] [blame] | 2639 | ASSERT_TRUE(page); |
| 2640 | |
Lei Zhang | a21d593 | 2018-02-05 18:28:38 +0000 | [diff] [blame] | 2641 | { |
| 2642 | // Add an underline annotation to the page without specifying its rectangle. |
Tom Sepez | e08d2b1 | 2018-04-25 18:49:32 +0000 | [diff] [blame] | 2643 | ScopedFPDFAnnotation annot( |
Lei Zhang | a21d593 | 2018-02-05 18:28:38 +0000 | [diff] [blame] | 2644 | FPDFPage_CreateAnnot(page, FPDF_ANNOT_UNDERLINE)); |
| 2645 | ASSERT_TRUE(annot); |
Jane Liu | eda6525 | 2017-06-07 11:31:27 -0400 | [diff] [blame] | 2646 | |
Lei Zhang | a21d593 | 2018-02-05 18:28:38 +0000 | [diff] [blame] | 2647 | // FPDFPage_TransformAnnots() should run without errors when modifying |
| 2648 | // annotation rectangles. |
| 2649 | FPDFPage_TransformAnnots(page, 1, 2, 3, 4, 5, 6); |
| 2650 | } |
Jane Liu | eda6525 | 2017-06-07 11:31:27 -0400 | [diff] [blame] | 2651 | UnloadPage(page); |
| 2652 | } |
| 2653 | |
Nicolas Pena | f45ade3 | 2017-05-03 10:23:49 -0400 | [diff] [blame] | 2654 | // TODO(npm): Add tests using Japanese fonts in other OS. |
Dan Sinclair | 698aed7 | 2017-09-26 16:24:49 -0400 | [diff] [blame] | 2655 | #if _FX_PLATFORM_ == _FX_PLATFORM_LINUX_ |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 2656 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 2657 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 2658 | #define MAYBE_AddCIDFontText DISABLED_AddCIDFontText |
| 2659 | #else |
| 2660 | #define MAYBE_AddCIDFontText AddCIDFontText |
| 2661 | #endif |
| 2662 | TEST_F(FPDFEditEmbedderTest, MAYBE_AddCIDFontText) { |
Nicolas Pena | f45ade3 | 2017-05-03 10:23:49 -0400 | [diff] [blame] | 2663 | // Start with a blank page |
| 2664 | FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792); |
| 2665 | CFX_Font CIDfont; |
| 2666 | { |
| 2667 | // First, get the data from the font |
| 2668 | CIDfont.LoadSubst("IPAGothic", 1, 0, 400, 0, 932, 0); |
| 2669 | EXPECT_EQ("IPAGothic", CIDfont.GetFaceName()); |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 2670 | pdfium::span<const uint8_t> span = CIDfont.GetFontSpan(); |
Nicolas Pena | f45ade3 | 2017-05-03 10:23:49 -0400 | [diff] [blame] | 2671 | |
| 2672 | // Load the data into a FPDF_Font. |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 2673 | ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(), |
| 2674 | FPDF_FONT_TRUETYPE, 1)); |
Nicolas Pena | f45ade3 | 2017-05-03 10:23:49 -0400 | [diff] [blame] | 2675 | ASSERT_TRUE(font.get()); |
| 2676 | |
| 2677 | // Add some text to the page |
| 2678 | FPDF_PAGEOBJECT text_object = |
| 2679 | FPDFPageObj_CreateTextObj(document(), font.get(), 12.0f); |
| 2680 | ASSERT_TRUE(text_object); |
| 2681 | std::wstring wstr = L"ABCDEFGhijklmnop."; |
Lei Zhang | f0f6768 | 2019-04-08 17:03:21 +0000 | [diff] [blame] | 2682 | ScopedFPDFWideString text = GetFPDFWideString(wstr); |
Nicolas Pena | f45ade3 | 2017-05-03 10:23:49 -0400 | [diff] [blame] | 2683 | EXPECT_TRUE(FPDFText_SetText(text_object, text.get())); |
| 2684 | FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 200, 200); |
| 2685 | FPDFPage_InsertObject(page, text_object); |
| 2686 | |
| 2687 | // And add some Japanese characters |
| 2688 | FPDF_PAGEOBJECT text_object2 = |
| 2689 | FPDFPageObj_CreateTextObj(document(), font.get(), 18.0f); |
| 2690 | ASSERT_TRUE(text_object2); |
| 2691 | std::wstring wstr2 = |
| 2692 | L"\u3053\u3093\u306B\u3061\u306f\u4e16\u754C\u3002\u3053\u3053\u306B1" |
| 2693 | L"\u756A"; |
Lei Zhang | f0f6768 | 2019-04-08 17:03:21 +0000 | [diff] [blame] | 2694 | ScopedFPDFWideString text2 = GetFPDFWideString(wstr2); |
Nicolas Pena | f45ade3 | 2017-05-03 10:23:49 -0400 | [diff] [blame] | 2695 | EXPECT_TRUE(FPDFText_SetText(text_object2, text2.get())); |
| 2696 | FPDFPageObj_Transform(text_object2, 1, 0, 0, 1, 100, 500); |
| 2697 | FPDFPage_InsertObject(page, text_object2); |
| 2698 | } |
| 2699 | |
Nicolas Pena | 207b727 | 2017-05-26 17:37:06 -0400 | [diff] [blame] | 2700 | // Check that the text renders properly. |
Lei Zhang | a2b7073 | 2019-06-25 08:34:22 +0000 | [diff] [blame] | 2701 | const char md5[] = "5159a72903fe57bf0cf645c894de8a74"; |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2702 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 2703 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2704 | CompareBitmap(page_bitmap.get(), 612, 792, md5); |
| 2705 | } |
Nicolas Pena | f45ade3 | 2017-05-03 10:23:49 -0400 | [diff] [blame] | 2706 | |
| 2707 | // Save the document, close the page. |
Nicolas Pena | 207b727 | 2017-05-26 17:37:06 -0400 | [diff] [blame] | 2708 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
Nicolas Pena | f45ade3 | 2017-05-03 10:23:49 -0400 | [diff] [blame] | 2709 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 2710 | FPDF_ClosePage(page); |
Dan Sinclair | 04e4dc8 | 2017-10-18 12:17:14 -0400 | [diff] [blame] | 2711 | |
| 2712 | VerifySavedDocument(612, 792, md5); |
Nicolas Pena | f45ade3 | 2017-05-03 10:23:49 -0400 | [diff] [blame] | 2713 | } |
Dan Sinclair | 698aed7 | 2017-09-26 16:24:49 -0400 | [diff] [blame] | 2714 | #endif // _FX_PLATFORM_ == _FX_PLATFORM_LINUX_ |
Nicolas Pena | 9ba8fbc | 2017-06-28 15:31:56 -0400 | [diff] [blame] | 2715 | |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 2716 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 2717 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 2718 | #define MAYBE_SaveAndRender DISABLED_SaveAndRender |
| 2719 | #else |
| 2720 | #define MAYBE_SaveAndRender SaveAndRender |
| 2721 | #endif |
| 2722 | TEST_F(FPDFEditEmbedderTest, MAYBE_SaveAndRender) { |
Nicolas Pena | a0b48aa | 2017-06-29 11:01:46 -0400 | [diff] [blame] | 2723 | const char md5[] = "3c20472b0552c0c22b88ab1ed8c6202b"; |
Nicolas Pena | 9ba8fbc | 2017-06-28 15:31:56 -0400 | [diff] [blame] | 2724 | { |
| 2725 | EXPECT_TRUE(OpenDocument("bug_779.pdf")); |
| 2726 | FPDF_PAGE page = LoadPage(0); |
| 2727 | ASSERT_NE(nullptr, page); |
| 2728 | |
| 2729 | // Now add a more complex blue path. |
| 2730 | FPDF_PAGEOBJECT green_path = FPDFPageObj_CreateNewPath(20, 20); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 2731 | EXPECT_TRUE(FPDFPageObj_SetFillColor(green_path, 0, 255, 0, 200)); |
Nicolas Pena | 9ba8fbc | 2017-06-28 15:31:56 -0400 | [diff] [blame] | 2732 | // TODO(npm): stroking will cause the MD5s to differ. |
| 2733 | EXPECT_TRUE(FPDFPath_SetDrawMode(green_path, FPDF_FILLMODE_WINDING, 0)); |
| 2734 | EXPECT_TRUE(FPDFPath_LineTo(green_path, 20, 63)); |
| 2735 | EXPECT_TRUE(FPDFPath_BezierTo(green_path, 55, 55, 78, 78, 90, 90)); |
| 2736 | EXPECT_TRUE(FPDFPath_LineTo(green_path, 133, 133)); |
| 2737 | EXPECT_TRUE(FPDFPath_LineTo(green_path, 133, 33)); |
| 2738 | EXPECT_TRUE(FPDFPath_BezierTo(green_path, 38, 33, 39, 36, 40, 40)); |
| 2739 | EXPECT_TRUE(FPDFPath_Close(green_path)); |
| 2740 | FPDFPage_InsertObject(page, green_path); |
Tom Sepez | e08d2b1 | 2018-04-25 18:49:32 +0000 | [diff] [blame] | 2741 | ScopedFPDFBitmap page_bitmap = RenderLoadedPage(page); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2742 | CompareBitmap(page_bitmap.get(), 612, 792, md5); |
Nicolas Pena | 9ba8fbc | 2017-06-28 15:31:56 -0400 | [diff] [blame] | 2743 | |
| 2744 | // Now save the result, closing the page and document |
| 2745 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 2746 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 2747 | UnloadPage(page); |
| 2748 | } |
Dan Sinclair | 04e4dc8 | 2017-10-18 12:17:14 -0400 | [diff] [blame] | 2749 | |
| 2750 | VerifySavedDocument(612, 792, md5); |
Nicolas Pena | 9ba8fbc | 2017-06-28 15:31:56 -0400 | [diff] [blame] | 2751 | } |
Jane Liu | 28fb7ba | 2017-08-02 21:45:57 -0400 | [diff] [blame] | 2752 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 2753 | TEST_F(FPDFEditEmbedderTest, AddMark) { |
Henrique Nakashima | d8df8c3 | 2018-07-12 22:15:09 +0000 | [diff] [blame] | 2754 | // Load document with some text. |
| 2755 | EXPECT_TRUE(OpenDocument("text_in_page_marked.pdf")); |
| 2756 | FPDF_PAGE page = LoadPage(0); |
| 2757 | ASSERT_TRUE(page); |
| 2758 | |
Lei Zhang | 2697cb1 | 2019-07-03 18:14:29 +0000 | [diff] [blame] | 2759 | CheckMarkCounts(page, 1, 19, 8, 4, 9, 1); |
Henrique Nakashima | d8df8c3 | 2018-07-12 22:15:09 +0000 | [diff] [blame] | 2760 | |
| 2761 | // Add to the first page object a "Bounds" mark with "Position": "First". |
| 2762 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0); |
| 2763 | FPDF_PAGEOBJECTMARK mark = FPDFPageObj_AddMark(page_object, "Bounds"); |
| 2764 | EXPECT_TRUE(mark); |
Henrique Nakashima | a340677 | 2018-07-13 19:10:53 +0000 | [diff] [blame] | 2765 | EXPECT_TRUE(FPDFPageObjMark_SetStringParam(document(), page_object, mark, |
| 2766 | "Position", "First")); |
Henrique Nakashima | d8df8c3 | 2018-07-12 22:15:09 +0000 | [diff] [blame] | 2767 | |
Lei Zhang | 2697cb1 | 2019-07-03 18:14:29 +0000 | [diff] [blame] | 2768 | CheckMarkCounts(page, 1, 19, 8, 4, 9, 2); |
Henrique Nakashima | d8df8c3 | 2018-07-12 22:15:09 +0000 | [diff] [blame] | 2769 | |
| 2770 | // Save the file |
| 2771 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 2772 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 2773 | UnloadPage(page); |
| 2774 | |
| 2775 | // Re-open the file and check the new mark is present. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 2776 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | d8df8c3 | 2018-07-12 22:15:09 +0000 | [diff] [blame] | 2777 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 2778 | |
Lei Zhang | 2697cb1 | 2019-07-03 18:14:29 +0000 | [diff] [blame] | 2779 | CheckMarkCounts(saved_page, 1, 19, 8, 4, 9, 2); |
| 2780 | |
| 2781 | CloseSavedPage(saved_page); |
| 2782 | CloseSavedDocument(); |
| 2783 | } |
| 2784 | |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 2785 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 2786 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 2787 | #define MAYBE_AddMarkCompressedStream DISABLED_AddMarkCompressedStream |
| 2788 | #else |
| 2789 | #define MAYBE_AddMarkCompressedStream AddMarkCompressedStream |
| 2790 | #endif |
| 2791 | TEST_F(FPDFEditEmbedderTest, MAYBE_AddMarkCompressedStream) { |
Lei Zhang | 2697cb1 | 2019-07-03 18:14:29 +0000 | [diff] [blame] | 2792 | |
| 2793 | // Load document with some text in a compressed stream. |
| 2794 | EXPECT_TRUE(OpenDocument("hello_world_compressed_stream.pdf")); |
| 2795 | FPDF_PAGE page = LoadPage(0); |
| 2796 | ASSERT_TRUE(page); |
| 2797 | |
| 2798 | // Render and check there are no marks. |
| 2799 | { |
| 2800 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 2801 | CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldChecksum); |
Lei Zhang | 2697cb1 | 2019-07-03 18:14:29 +0000 | [diff] [blame] | 2802 | } |
| 2803 | CheckMarkCounts(page, 0, 2, 0, 0, 0, 0); |
| 2804 | |
| 2805 | // Add to the first page object a "Bounds" mark with "Position": "First". |
| 2806 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0); |
| 2807 | FPDF_PAGEOBJECTMARK mark = FPDFPageObj_AddMark(page_object, "Bounds"); |
| 2808 | EXPECT_TRUE(mark); |
| 2809 | EXPECT_TRUE(FPDFPageObjMark_SetStringParam(document(), page_object, mark, |
| 2810 | "Position", "First")); |
| 2811 | |
| 2812 | // Render and check there is 1 mark. |
| 2813 | { |
| 2814 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 2815 | CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldChecksum); |
Lei Zhang | 2697cb1 | 2019-07-03 18:14:29 +0000 | [diff] [blame] | 2816 | } |
| 2817 | CheckMarkCounts(page, 0, 2, 0, 0, 0, 1); |
| 2818 | |
| 2819 | // Save the file. |
| 2820 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 2821 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 2822 | UnloadPage(page); |
| 2823 | |
| 2824 | // Re-open the file and check the new mark is present. |
| 2825 | ASSERT_TRUE(OpenSavedDocument()); |
| 2826 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 2827 | |
| 2828 | { |
| 2829 | ScopedFPDFBitmap page_bitmap = RenderPage(saved_page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 2830 | CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldChecksum); |
Lei Zhang | 2697cb1 | 2019-07-03 18:14:29 +0000 | [diff] [blame] | 2831 | } |
| 2832 | CheckMarkCounts(saved_page, 0, 2, 0, 0, 0, 1); |
Henrique Nakashima | d8df8c3 | 2018-07-12 22:15:09 +0000 | [diff] [blame] | 2833 | |
| 2834 | CloseSavedPage(saved_page); |
| 2835 | CloseSavedDocument(); |
| 2836 | } |
| 2837 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 2838 | TEST_F(FPDFEditEmbedderTest, SetMarkParam) { |
Henrique Nakashima | a340677 | 2018-07-13 19:10:53 +0000 | [diff] [blame] | 2839 | // Load document with some text. |
| 2840 | EXPECT_TRUE(OpenDocument("text_in_page_marked.pdf")); |
| 2841 | FPDF_PAGE page = LoadPage(0); |
| 2842 | ASSERT_TRUE(page); |
| 2843 | |
| 2844 | constexpr int kExpectedObjectCount = 19; |
| 2845 | CheckMarkCounts(page, 1, kExpectedObjectCount, 8, 4, 9, 1); |
| 2846 | |
| 2847 | // Check the "Bounds" mark's "Position" param is "Last". |
| 2848 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 18); |
| 2849 | FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, 1); |
| 2850 | ASSERT_TRUE(mark); |
| 2851 | char buffer[256]; |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 2852 | unsigned long name_len = 999u; |
| 2853 | ASSERT_TRUE(FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len)); |
| 2854 | EXPECT_EQ((6u + 1u) * 2u, name_len); |
Henrique Nakashima | a340677 | 2018-07-13 19:10:53 +0000 | [diff] [blame] | 2855 | ASSERT_EQ(L"Bounds", |
| 2856 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer))); |
| 2857 | unsigned long out_buffer_len; |
| 2858 | ASSERT_TRUE(FPDFPageObjMark_GetParamStringValue( |
| 2859 | mark, "Position", buffer, sizeof(buffer), &out_buffer_len)); |
| 2860 | ASSERT_EQ(L"Last", |
| 2861 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer))); |
| 2862 | |
| 2863 | // Set is to "End". |
| 2864 | EXPECT_TRUE(FPDFPageObjMark_SetStringParam(document(), page_object, mark, |
| 2865 | "Position", "End")); |
| 2866 | |
| 2867 | // Verify the object passed must correspond to the mark passed. |
| 2868 | FPDF_PAGEOBJECT another_page_object = FPDFPage_GetObject(page, 17); |
| 2869 | EXPECT_FALSE(FPDFPageObjMark_SetStringParam(document(), another_page_object, |
| 2870 | mark, "Position", "End")); |
| 2871 | |
| 2872 | // Verify nothing else changed. |
| 2873 | CheckMarkCounts(page, 1, kExpectedObjectCount, 8, 4, 9, 1); |
| 2874 | |
| 2875 | // Verify "Position" now maps to "End". |
| 2876 | EXPECT_TRUE(FPDFPageObjMark_GetParamStringValue( |
| 2877 | mark, "Position", buffer, sizeof(buffer), &out_buffer_len)); |
| 2878 | EXPECT_EQ(L"End", |
| 2879 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer))); |
| 2880 | |
| 2881 | // Save the file |
| 2882 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 2883 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 2884 | UnloadPage(page); |
| 2885 | |
| 2886 | // Re-open the file and cerify "Position" still maps to "End". |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 2887 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | a340677 | 2018-07-13 19:10:53 +0000 | [diff] [blame] | 2888 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 2889 | |
| 2890 | CheckMarkCounts(saved_page, 1, kExpectedObjectCount, 8, 4, 9, 1); |
| 2891 | page_object = FPDFPage_GetObject(saved_page, 18); |
| 2892 | mark = FPDFPageObj_GetMark(page_object, 1); |
| 2893 | EXPECT_TRUE(FPDFPageObjMark_GetParamStringValue( |
| 2894 | mark, "Position", buffer, sizeof(buffer), &out_buffer_len)); |
| 2895 | EXPECT_EQ(L"End", |
| 2896 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer))); |
| 2897 | |
| 2898 | CloseSavedPage(saved_page); |
| 2899 | CloseSavedDocument(); |
| 2900 | } |
| 2901 | |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 2902 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 2903 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 2904 | #define MAYBE_AddMarkedText DISABLED_AddMarkedText |
| 2905 | #else |
| 2906 | #define MAYBE_AddMarkedText AddMarkedText |
| 2907 | #endif |
| 2908 | TEST_F(FPDFEditEmbedderTest, MAYBE_AddMarkedText) { |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2909 | // Start with a blank page. |
| 2910 | FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792); |
| 2911 | |
Tom Sepez | ffff6c5 | 2019-07-30 21:56:19 +0000 | [diff] [blame] | 2912 | RetainPtr<CPDF_Font> stock_font = |
| 2913 | CPDF_Font::GetStockFont(cpdf_doc(), "Arial"); |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 2914 | pdfium::span<const uint8_t> span = stock_font->GetFont()->GetFontSpan(); |
| 2915 | ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(), |
| 2916 | FPDF_FONT_TRUETYPE, 0)); |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2917 | ASSERT_TRUE(font.get()); |
| 2918 | |
| 2919 | // Add some text to the page. |
| 2920 | FPDF_PAGEOBJECT text_object = |
| 2921 | FPDFPageObj_CreateTextObj(document(), font.get(), 12.0f); |
| 2922 | |
| 2923 | EXPECT_TRUE(text_object); |
Lei Zhang | f0f6768 | 2019-04-08 17:03:21 +0000 | [diff] [blame] | 2924 | ScopedFPDFWideString text1 = |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2925 | GetFPDFWideString(L"I am testing my loaded font, WEE."); |
| 2926 | EXPECT_TRUE(FPDFText_SetText(text_object, text1.get())); |
| 2927 | FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 400, 400); |
| 2928 | FPDFPage_InsertObject(page, text_object); |
| 2929 | |
| 2930 | // Add a mark with the tag "TestMarkName" to that text. |
| 2931 | EXPECT_EQ(0, FPDFPageObj_CountMarks(text_object)); |
Henrique Nakashima | 6fc8d87 | 2018-09-18 16:48:31 +0000 | [diff] [blame] | 2932 | FPDF_PAGEOBJECTMARK mark = FPDFPageObj_AddMark(text_object, "Test Mark Name"); |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2933 | EXPECT_TRUE(mark); |
| 2934 | EXPECT_EQ(1, FPDFPageObj_CountMarks(text_object)); |
| 2935 | EXPECT_EQ(mark, FPDFPageObj_GetMark(text_object, 0)); |
| 2936 | char buffer[256]; |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 2937 | unsigned long name_len = 999u; |
| 2938 | ASSERT_TRUE(FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len)); |
| 2939 | EXPECT_EQ((14u + 1u) * 2, name_len); |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2940 | std::wstring name = |
| 2941 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)); |
Henrique Nakashima | 6fc8d87 | 2018-09-18 16:48:31 +0000 | [diff] [blame] | 2942 | EXPECT_EQ(L"Test Mark Name", name); |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2943 | |
| 2944 | // Add parameters: |
| 2945 | // - int "IntKey" : 42 |
| 2946 | // - string "StringKey": "StringValue" |
Henrique Nakashima | 07520f6 | 2018-07-12 19:45:29 +0000 | [diff] [blame] | 2947 | // - blob "BlobKey": "\x01\x02\x03\0BlobValue1\0\0\0BlobValue2\0" |
Lei Zhang | 590f424 | 2019-05-15 20:57:26 +0000 | [diff] [blame] | 2948 | constexpr size_t kBlobLen = 28; |
Lei Zhang | d3b028b | 2018-11-30 22:22:00 +0000 | [diff] [blame] | 2949 | char block_value[kBlobLen]; |
| 2950 | memcpy(block_value, "\x01\x02\x03\0BlobValue1\0\0\0BlobValue2\0", kBlobLen); |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2951 | EXPECT_EQ(0, FPDFPageObjMark_CountParams(mark)); |
Henrique Nakashima | a340677 | 2018-07-13 19:10:53 +0000 | [diff] [blame] | 2952 | EXPECT_TRUE( |
| 2953 | FPDFPageObjMark_SetIntParam(document(), text_object, mark, "IntKey", 42)); |
| 2954 | EXPECT_TRUE(FPDFPageObjMark_SetStringParam(document(), text_object, mark, |
| 2955 | "StringKey", "StringValue")); |
| 2956 | EXPECT_TRUE(FPDFPageObjMark_SetBlobParam(document(), text_object, mark, |
Lei Zhang | d3b028b | 2018-11-30 22:22:00 +0000 | [diff] [blame] | 2957 | "BlobKey", block_value, kBlobLen)); |
Henrique Nakashima | 07520f6 | 2018-07-12 19:45:29 +0000 | [diff] [blame] | 2958 | EXPECT_EQ(3, FPDFPageObjMark_CountParams(mark)); |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2959 | |
| 2960 | // Check the two parameters can be retrieved. |
| 2961 | EXPECT_EQ(FPDF_OBJECT_NUMBER, |
Henrique Nakashima | 94230e5 | 2018-07-11 22:02:02 +0000 | [diff] [blame] | 2962 | FPDFPageObjMark_GetParamValueType(mark, "IntKey")); |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2963 | int int_value; |
Henrique Nakashima | 94230e5 | 2018-07-11 22:02:02 +0000 | [diff] [blame] | 2964 | EXPECT_TRUE(FPDFPageObjMark_GetParamIntValue(mark, "IntKey", &int_value)); |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2965 | EXPECT_EQ(42, int_value); |
| 2966 | |
| 2967 | EXPECT_EQ(FPDF_OBJECT_STRING, |
Henrique Nakashima | 94230e5 | 2018-07-11 22:02:02 +0000 | [diff] [blame] | 2968 | FPDFPageObjMark_GetParamValueType(mark, "StringKey")); |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 2969 | unsigned long out_buffer_len = 999u; |
Henrique Nakashima | a340677 | 2018-07-13 19:10:53 +0000 | [diff] [blame] | 2970 | EXPECT_TRUE(FPDFPageObjMark_GetParamStringValue( |
| 2971 | mark, "StringKey", buffer, sizeof(buffer), &out_buffer_len)); |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2972 | EXPECT_GT(out_buffer_len, 0u); |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 2973 | EXPECT_NE(999u, out_buffer_len); |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2974 | name = GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)); |
| 2975 | EXPECT_EQ(L"StringValue", name); |
| 2976 | |
Henrique Nakashima | 07520f6 | 2018-07-12 19:45:29 +0000 | [diff] [blame] | 2977 | EXPECT_EQ(FPDF_OBJECT_STRING, |
| 2978 | FPDFPageObjMark_GetParamValueType(mark, "BlobKey")); |
| 2979 | out_buffer_len = 0; |
Henrique Nakashima | a340677 | 2018-07-13 19:10:53 +0000 | [diff] [blame] | 2980 | EXPECT_TRUE(FPDFPageObjMark_GetParamBlobValue( |
| 2981 | mark, "BlobKey", buffer, sizeof(buffer), &out_buffer_len)); |
Henrique Nakashima | 07520f6 | 2018-07-12 19:45:29 +0000 | [diff] [blame] | 2982 | EXPECT_EQ(kBlobLen, out_buffer_len); |
Lei Zhang | d3b028b | 2018-11-30 22:22:00 +0000 | [diff] [blame] | 2983 | EXPECT_EQ(0, memcmp(block_value, buffer, kBlobLen)); |
Henrique Nakashima | 07520f6 | 2018-07-12 19:45:29 +0000 | [diff] [blame] | 2984 | |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2985 | // Render and check the bitmap is the expected one. |
Lei Zhang | e4cdac5 | 2019-04-30 16:45:57 +0000 | [diff] [blame] | 2986 | #if defined(OS_MACOSX) |
Lei Zhang | 0f6b434 | 2020-02-25 20:00:39 +0000 | [diff] [blame] | 2987 | const char md5[] = "fc921c0bbdde73986ac13c15a85db4c3"; |
Lei Zhang | e67bcc7 | 2019-04-30 18:55:58 +0000 | [diff] [blame] | 2988 | #elif defined(OS_WIN) |
Manoj Biswas | 7daf7a2 | 2019-03-22 23:02:34 +0000 | [diff] [blame] | 2989 | const char md5[] = "d60ba39f9698e32360d99e727dd93165"; |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2990 | #else |
| 2991 | const char md5[] = "70592859010ffbf532a2237b8118bcc4"; |
| 2992 | #endif |
| 2993 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 2994 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2995 | CompareBitmap(page_bitmap.get(), 612, 792, md5); |
| 2996 | } |
| 2997 | |
Henrique Nakashima | b4bcf69 | 2018-07-11 21:19:22 +0000 | [diff] [blame] | 2998 | // Now save the result. |
| 2999 | EXPECT_EQ(1, FPDFPage_CountObjects(page)); |
| 3000 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 3001 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 3002 | |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 3003 | FPDF_ClosePage(page); |
| 3004 | |
Henrique Nakashima | b4bcf69 | 2018-07-11 21:19:22 +0000 | [diff] [blame] | 3005 | // Re-open the file and check the changes were kept in the saved .pdf. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 3006 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | b4bcf69 | 2018-07-11 21:19:22 +0000 | [diff] [blame] | 3007 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 3008 | EXPECT_EQ(1, FPDFPage_CountObjects(saved_page)); |
| 3009 | |
| 3010 | text_object = FPDFPage_GetObject(saved_page, 0); |
| 3011 | EXPECT_TRUE(text_object); |
| 3012 | EXPECT_EQ(1, FPDFPageObj_CountMarks(text_object)); |
| 3013 | mark = FPDFPageObj_GetMark(text_object, 0); |
| 3014 | EXPECT_TRUE(mark); |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 3015 | |
| 3016 | name_len = 999u; |
| 3017 | ASSERT_TRUE(FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len)); |
| 3018 | EXPECT_EQ((14u + 1u) * 2, name_len); |
Henrique Nakashima | b4bcf69 | 2018-07-11 21:19:22 +0000 | [diff] [blame] | 3019 | name = GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)); |
Henrique Nakashima | 6fc8d87 | 2018-09-18 16:48:31 +0000 | [diff] [blame] | 3020 | EXPECT_EQ(L"Test Mark Name", name); |
Henrique Nakashima | b4bcf69 | 2018-07-11 21:19:22 +0000 | [diff] [blame] | 3021 | |
| 3022 | CloseSavedPage(saved_page); |
| 3023 | CloseSavedDocument(); |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 3024 | } |
| 3025 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 3026 | TEST_F(FPDFEditEmbedderTest, MarkGetName) { |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 3027 | EXPECT_TRUE(OpenDocument("text_in_page_marked.pdf")); |
| 3028 | FPDF_PAGE page = LoadPage(0); |
| 3029 | ASSERT_TRUE(page); |
| 3030 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 18); |
| 3031 | FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, 1); |
| 3032 | ASSERT_TRUE(mark); |
| 3033 | |
| 3034 | char buffer[256]; |
| 3035 | unsigned long out_len; |
| 3036 | |
| 3037 | // Show the positive cases of FPDFPageObjMark_GetName. |
| 3038 | out_len = 999u; |
| 3039 | EXPECT_TRUE(FPDFPageObjMark_GetName(mark, nullptr, 0, &out_len)); |
| 3040 | EXPECT_EQ((6u + 1u) * 2u, out_len); |
| 3041 | |
| 3042 | out_len = 999u; |
| 3043 | EXPECT_TRUE(FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &out_len)); |
| 3044 | EXPECT_EQ(L"Bounds", |
| 3045 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer))); |
| 3046 | EXPECT_EQ((6u + 1u) * 2u, out_len); |
| 3047 | |
| 3048 | // Show the negative cases of FPDFPageObjMark_GetName. |
| 3049 | out_len = 999u; |
| 3050 | EXPECT_FALSE( |
| 3051 | FPDFPageObjMark_GetName(nullptr, buffer, sizeof(buffer), &out_len)); |
| 3052 | EXPECT_EQ(999u, out_len); |
| 3053 | |
| 3054 | EXPECT_FALSE(FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), nullptr)); |
| 3055 | |
| 3056 | UnloadPage(page); |
| 3057 | } |
| 3058 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 3059 | TEST_F(FPDFEditEmbedderTest, MarkGetParamKey) { |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 3060 | EXPECT_TRUE(OpenDocument("text_in_page_marked.pdf")); |
| 3061 | FPDF_PAGE page = LoadPage(0); |
| 3062 | ASSERT_TRUE(page); |
| 3063 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 18); |
| 3064 | FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, 1); |
| 3065 | ASSERT_TRUE(mark); |
| 3066 | |
| 3067 | char buffer[256]; |
| 3068 | unsigned long out_len; |
| 3069 | |
| 3070 | // Show the positive cases of FPDFPageObjMark_GetParamKey. |
| 3071 | out_len = 999u; |
| 3072 | EXPECT_TRUE(FPDFPageObjMark_GetParamKey(mark, 0, nullptr, 0, &out_len)); |
| 3073 | EXPECT_EQ((8u + 1u) * 2u, out_len); |
| 3074 | |
| 3075 | out_len = 999u; |
| 3076 | EXPECT_TRUE( |
| 3077 | FPDFPageObjMark_GetParamKey(mark, 0, buffer, sizeof(buffer), &out_len)); |
| 3078 | EXPECT_EQ(L"Position", |
| 3079 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer))); |
| 3080 | EXPECT_EQ((8u + 1u) * 2u, out_len); |
| 3081 | |
| 3082 | // Show the negative cases of FPDFPageObjMark_GetParamKey. |
| 3083 | out_len = 999u; |
| 3084 | EXPECT_FALSE(FPDFPageObjMark_GetParamKey(nullptr, 0, buffer, sizeof(buffer), |
| 3085 | &out_len)); |
| 3086 | EXPECT_EQ(999u, out_len); |
| 3087 | |
| 3088 | out_len = 999u; |
| 3089 | EXPECT_FALSE( |
| 3090 | FPDFPageObjMark_GetParamKey(mark, 1, buffer, sizeof(buffer), &out_len)); |
| 3091 | EXPECT_EQ(999u, out_len); |
| 3092 | |
| 3093 | EXPECT_FALSE( |
| 3094 | FPDFPageObjMark_GetParamKey(mark, 0, buffer, sizeof(buffer), nullptr)); |
| 3095 | |
| 3096 | UnloadPage(page); |
| 3097 | } |
| 3098 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 3099 | TEST_F(FPDFEditEmbedderTest, MarkGetIntParam) { |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 3100 | EXPECT_TRUE(OpenDocument("text_in_page_marked.pdf")); |
| 3101 | FPDF_PAGE page = LoadPage(0); |
| 3102 | ASSERT_TRUE(page); |
| 3103 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 8); |
| 3104 | FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, 0); |
| 3105 | ASSERT_TRUE(mark); |
| 3106 | |
| 3107 | int out_value; |
| 3108 | |
| 3109 | // Show the positive cases of FPDFPageObjMark_GetParamIntValue. |
| 3110 | out_value = 999; |
| 3111 | EXPECT_TRUE(FPDFPageObjMark_GetParamIntValue(mark, "Factor", &out_value)); |
| 3112 | EXPECT_EQ(3, out_value); |
| 3113 | |
| 3114 | // Show the negative cases of FPDFPageObjMark_GetParamIntValue. |
| 3115 | out_value = 999; |
| 3116 | EXPECT_FALSE(FPDFPageObjMark_GetParamIntValue(nullptr, "Factor", &out_value)); |
| 3117 | EXPECT_EQ(999, out_value); |
| 3118 | |
| 3119 | out_value = 999; |
| 3120 | EXPECT_FALSE(FPDFPageObjMark_GetParamIntValue(mark, "ParamThatDoesNotExist", |
| 3121 | &out_value)); |
| 3122 | EXPECT_EQ(999, out_value); |
| 3123 | |
| 3124 | EXPECT_FALSE(FPDFPageObjMark_GetParamIntValue(mark, "Factor", nullptr)); |
| 3125 | |
| 3126 | page_object = FPDFPage_GetObject(page, 18); |
| 3127 | mark = FPDFPageObj_GetMark(page_object, 1); |
| 3128 | out_value = 999; |
| 3129 | EXPECT_FALSE(FPDFPageObjMark_GetParamIntValue(mark, "Position", &out_value)); |
| 3130 | EXPECT_EQ(999, out_value); |
| 3131 | |
| 3132 | UnloadPage(page); |
| 3133 | } |
| 3134 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 3135 | TEST_F(FPDFEditEmbedderTest, MarkGetStringParam) { |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 3136 | EXPECT_TRUE(OpenDocument("text_in_page_marked.pdf")); |
| 3137 | FPDF_PAGE page = LoadPage(0); |
| 3138 | ASSERT_TRUE(page); |
| 3139 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 18); |
| 3140 | FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, 1); |
| 3141 | ASSERT_TRUE(mark); |
| 3142 | |
| 3143 | char buffer[256]; |
| 3144 | unsigned long out_len; |
| 3145 | |
| 3146 | // Show the positive cases of FPDFPageObjMark_GetParamStringValue. |
| 3147 | out_len = 999u; |
| 3148 | EXPECT_TRUE(FPDFPageObjMark_GetParamStringValue(mark, "Position", nullptr, 0, |
| 3149 | &out_len)); |
| 3150 | EXPECT_EQ((4u + 1u) * 2u, out_len); |
| 3151 | |
| 3152 | out_len = 999u; |
| 3153 | EXPECT_TRUE(FPDFPageObjMark_GetParamStringValue(mark, "Position", buffer, |
| 3154 | sizeof(buffer), &out_len)); |
| 3155 | EXPECT_EQ(L"Last", |
| 3156 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer))); |
| 3157 | EXPECT_EQ((4u + 1u) * 2u, out_len); |
| 3158 | |
| 3159 | // Show the negative cases of FPDFPageObjMark_GetParamStringValue. |
| 3160 | out_len = 999u; |
| 3161 | EXPECT_FALSE(FPDFPageObjMark_GetParamStringValue(nullptr, "Position", buffer, |
| 3162 | sizeof(buffer), &out_len)); |
| 3163 | EXPECT_EQ(999u, out_len); |
| 3164 | |
| 3165 | out_len = 999u; |
| 3166 | EXPECT_FALSE(FPDFPageObjMark_GetParamStringValue( |
| 3167 | mark, "ParamThatDoesNotExist", buffer, sizeof(buffer), &out_len)); |
| 3168 | EXPECT_EQ(999u, out_len); |
| 3169 | |
| 3170 | EXPECT_FALSE(FPDFPageObjMark_GetParamStringValue(mark, "Position", buffer, |
| 3171 | sizeof(buffer), nullptr)); |
| 3172 | |
| 3173 | page_object = FPDFPage_GetObject(page, 8); |
| 3174 | mark = FPDFPageObj_GetMark(page_object, 0); |
| 3175 | out_len = 999u; |
| 3176 | EXPECT_FALSE(FPDFPageObjMark_GetParamStringValue(mark, "Factor", buffer, |
| 3177 | sizeof(buffer), &out_len)); |
| 3178 | EXPECT_EQ(999u, out_len); |
| 3179 | |
| 3180 | UnloadPage(page); |
| 3181 | } |
| 3182 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 3183 | TEST_F(FPDFEditEmbedderTest, ExtractImageBitmap) { |
Jane Liu | 28fb7ba | 2017-08-02 21:45:57 -0400 | [diff] [blame] | 3184 | ASSERT_TRUE(OpenDocument("embedded_images.pdf")); |
| 3185 | FPDF_PAGE page = LoadPage(0); |
| 3186 | ASSERT_TRUE(page); |
Miklos Vajna | 9262761 | 2017-09-25 12:59:29 +0200 | [diff] [blame] | 3187 | ASSERT_EQ(39, FPDFPage_CountObjects(page)); |
Jane Liu | 28fb7ba | 2017-08-02 21:45:57 -0400 | [diff] [blame] | 3188 | |
| 3189 | FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 32); |
| 3190 | EXPECT_NE(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3191 | EXPECT_FALSE(FPDFImageObj_GetBitmap(obj)); |
| 3192 | |
| 3193 | obj = FPDFPage_GetObject(page, 33); |
| 3194 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3195 | FPDF_BITMAP bitmap = FPDFImageObj_GetBitmap(obj); |
| 3196 | EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap)); |
Lei Zhang | bf98014 | 2019-12-20 01:05:42 +0000 | [diff] [blame] | 3197 | CompareBitmap(bitmap, 109, 88, "cb3637934bb3b95a6e4ae1ea9eb9e56e"); |
Jane Liu | 28fb7ba | 2017-08-02 21:45:57 -0400 | [diff] [blame] | 3198 | FPDFBitmap_Destroy(bitmap); |
| 3199 | |
| 3200 | obj = FPDFPage_GetObject(page, 34); |
| 3201 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3202 | bitmap = FPDFImageObj_GetBitmap(obj); |
| 3203 | EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap)); |
Lei Zhang | bf98014 | 2019-12-20 01:05:42 +0000 | [diff] [blame] | 3204 | CompareBitmap(bitmap, 103, 75, "c8d51fa6821ceb2a67f08446ff236c40"); |
Jane Liu | 28fb7ba | 2017-08-02 21:45:57 -0400 | [diff] [blame] | 3205 | FPDFBitmap_Destroy(bitmap); |
| 3206 | |
| 3207 | obj = FPDFPage_GetObject(page, 35); |
| 3208 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3209 | bitmap = FPDFImageObj_GetBitmap(obj); |
| 3210 | EXPECT_EQ(FPDFBitmap_Gray, FPDFBitmap_GetFormat(bitmap)); |
| 3211 | CompareBitmap(bitmap, 92, 68, "9c6d76cb1e37ef8514f9455d759391f3"); |
| 3212 | FPDFBitmap_Destroy(bitmap); |
| 3213 | |
| 3214 | obj = FPDFPage_GetObject(page, 36); |
| 3215 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3216 | bitmap = FPDFImageObj_GetBitmap(obj); |
| 3217 | EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap)); |
Lei Zhang | bf98014 | 2019-12-20 01:05:42 +0000 | [diff] [blame] | 3218 | CompareBitmap(bitmap, 79, 60, "f4e72fb783a01c7b4614cdc25eaa98ac"); |
Jane Liu | 28fb7ba | 2017-08-02 21:45:57 -0400 | [diff] [blame] | 3219 | FPDFBitmap_Destroy(bitmap); |
| 3220 | |
| 3221 | obj = FPDFPage_GetObject(page, 37); |
| 3222 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3223 | bitmap = FPDFImageObj_GetBitmap(obj); |
| 3224 | EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap)); |
Lei Zhang | bf98014 | 2019-12-20 01:05:42 +0000 | [diff] [blame] | 3225 | CompareBitmap(bitmap, 126, 106, "2cf9e66414c72461f4ccbf9cdebdfa1b"); |
Jane Liu | 28fb7ba | 2017-08-02 21:45:57 -0400 | [diff] [blame] | 3226 | FPDFBitmap_Destroy(bitmap); |
| 3227 | |
| 3228 | obj = FPDFPage_GetObject(page, 38); |
| 3229 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3230 | bitmap = FPDFImageObj_GetBitmap(obj); |
| 3231 | EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap)); |
Lei Zhang | bf98014 | 2019-12-20 01:05:42 +0000 | [diff] [blame] | 3232 | CompareBitmap(bitmap, 194, 119, "a8f3a126cec274dab8242fd2ccdc1b8b"); |
Jane Liu | 28fb7ba | 2017-08-02 21:45:57 -0400 | [diff] [blame] | 3233 | FPDFBitmap_Destroy(bitmap); |
| 3234 | UnloadPage(page); |
| 3235 | } |
Jane Liu | 548334e | 2017-08-03 16:33:40 -0400 | [diff] [blame] | 3236 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 3237 | TEST_F(FPDFEditEmbedderTest, ExtractJBigImageBitmap) { |
Lei Zhang | 53341dd | 2018-03-01 15:42:47 +0000 | [diff] [blame] | 3238 | ASSERT_TRUE(OpenDocument("bug_631912.pdf")); |
| 3239 | FPDF_PAGE page = LoadPage(0); |
| 3240 | ASSERT_TRUE(page); |
| 3241 | ASSERT_EQ(1, FPDFPage_CountObjects(page)); |
| 3242 | |
| 3243 | FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 0); |
| 3244 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3245 | { |
Tom Sepez | e08d2b1 | 2018-04-25 18:49:32 +0000 | [diff] [blame] | 3246 | ScopedFPDFBitmap bitmap(FPDFImageObj_GetBitmap(obj)); |
Lei Zhang | 1330ebb | 2018-03-05 15:16:37 +0000 | [diff] [blame] | 3247 | ASSERT_TRUE(bitmap); |
| 3248 | EXPECT_EQ(FPDFBitmap_Gray, FPDFBitmap_GetFormat(bitmap.get())); |
| 3249 | CompareBitmap(bitmap.get(), 1152, 720, "3f6a48e2b3e91b799bf34567f55cb4de"); |
Lei Zhang | 53341dd | 2018-03-01 15:42:47 +0000 | [diff] [blame] | 3250 | } |
| 3251 | |
| 3252 | UnloadPage(page); |
| 3253 | } |
| 3254 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 3255 | TEST_F(FPDFEditEmbedderTest, GetImageData) { |
Jane Liu | 548334e | 2017-08-03 16:33:40 -0400 | [diff] [blame] | 3256 | EXPECT_TRUE(OpenDocument("embedded_images.pdf")); |
| 3257 | FPDF_PAGE page = LoadPage(0); |
| 3258 | ASSERT_TRUE(page); |
Miklos Vajna | 9262761 | 2017-09-25 12:59:29 +0200 | [diff] [blame] | 3259 | ASSERT_EQ(39, FPDFPage_CountObjects(page)); |
Jane Liu | 548334e | 2017-08-03 16:33:40 -0400 | [diff] [blame] | 3260 | |
| 3261 | // Retrieve an image object with flate-encoded data stream. |
| 3262 | FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 33); |
| 3263 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3264 | |
| 3265 | // Check that the raw image data has the correct length and hash value. |
| 3266 | unsigned long len = FPDFImageObj_GetImageDataRaw(obj, nullptr, 0); |
| 3267 | std::vector<char> buf(len); |
| 3268 | EXPECT_EQ(4091u, FPDFImageObj_GetImageDataRaw(obj, buf.data(), len)); |
| 3269 | EXPECT_EQ("f73802327d2e88e890f653961bcda81a", |
| 3270 | GenerateMD5Base16(reinterpret_cast<uint8_t*>(buf.data()), len)); |
| 3271 | |
| 3272 | // Check that the decoded image data has the correct length and hash value. |
| 3273 | len = FPDFImageObj_GetImageDataDecoded(obj, nullptr, 0); |
| 3274 | buf.clear(); |
| 3275 | buf.resize(len); |
| 3276 | EXPECT_EQ(28776u, FPDFImageObj_GetImageDataDecoded(obj, buf.data(), len)); |
| 3277 | EXPECT_EQ("cb3637934bb3b95a6e4ae1ea9eb9e56e", |
| 3278 | GenerateMD5Base16(reinterpret_cast<uint8_t*>(buf.data()), len)); |
| 3279 | |
Lei Zhang | 351e8b0 | 2018-12-20 01:10:06 +0000 | [diff] [blame] | 3280 | // Retrieve an image object with DCTDecode-encoded data stream. |
Jane Liu | 548334e | 2017-08-03 16:33:40 -0400 | [diff] [blame] | 3281 | obj = FPDFPage_GetObject(page, 37); |
| 3282 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3283 | |
| 3284 | // Check that the raw image data has the correct length and hash value. |
| 3285 | len = FPDFImageObj_GetImageDataRaw(obj, nullptr, 0); |
| 3286 | buf.clear(); |
| 3287 | buf.resize(len); |
| 3288 | EXPECT_EQ(4370u, FPDFImageObj_GetImageDataRaw(obj, buf.data(), len)); |
| 3289 | EXPECT_EQ("6aae1f3710335023a9e12191be66b64b", |
| 3290 | GenerateMD5Base16(reinterpret_cast<uint8_t*>(buf.data()), len)); |
| 3291 | |
| 3292 | // Check that the decoded image data has the correct length and hash value, |
| 3293 | // which should be the same as those of the raw data, since this image is |
| 3294 | // encoded by a single DCTDecode filter and decoding is a noop. |
| 3295 | len = FPDFImageObj_GetImageDataDecoded(obj, nullptr, 0); |
| 3296 | buf.clear(); |
| 3297 | buf.resize(len); |
| 3298 | EXPECT_EQ(4370u, FPDFImageObj_GetImageDataDecoded(obj, buf.data(), len)); |
| 3299 | EXPECT_EQ("6aae1f3710335023a9e12191be66b64b", |
| 3300 | GenerateMD5Base16(reinterpret_cast<uint8_t*>(buf.data()), len)); |
| 3301 | |
| 3302 | UnloadPage(page); |
| 3303 | } |
Jane Liu | 2e5f0ae | 2017-08-08 15:23:27 -0400 | [diff] [blame] | 3304 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 3305 | TEST_F(FPDFEditEmbedderTest, GetImageMatrix) { |
Lei Zhang | 866d688 | 2018-10-24 17:31:01 +0000 | [diff] [blame] | 3306 | ASSERT_TRUE(OpenDocument("embedded_images.pdf")); |
| 3307 | FPDF_PAGE page = LoadPage(0); |
| 3308 | ASSERT_TRUE(page); |
| 3309 | ASSERT_EQ(39, FPDFPage_CountObjects(page)); |
| 3310 | |
| 3311 | FPDF_PAGEOBJECT obj; |
| 3312 | double a; |
| 3313 | double b; |
| 3314 | double c; |
| 3315 | double d; |
| 3316 | double e; |
| 3317 | double f; |
| 3318 | |
| 3319 | obj = FPDFPage_GetObject(page, 33); |
| 3320 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3321 | EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f)); |
| 3322 | EXPECT_DOUBLE_EQ(53.0, a); |
| 3323 | EXPECT_DOUBLE_EQ(0.0, b); |
| 3324 | EXPECT_DOUBLE_EQ(0.0, c); |
| 3325 | EXPECT_DOUBLE_EQ(43.0, d); |
| 3326 | EXPECT_DOUBLE_EQ(72.0, e); |
| 3327 | EXPECT_DOUBLE_EQ(646.510009765625, f); |
| 3328 | |
| 3329 | obj = FPDFPage_GetObject(page, 34); |
| 3330 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3331 | EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f)); |
| 3332 | EXPECT_DOUBLE_EQ(70.0, a); |
| 3333 | EXPECT_DOUBLE_EQ(0.0, b); |
| 3334 | EXPECT_DOUBLE_EQ(0.0, c); |
| 3335 | EXPECT_DOUBLE_EQ(51.0, d); |
| 3336 | EXPECT_DOUBLE_EQ(216.0, e); |
| 3337 | EXPECT_DOUBLE_EQ(646.510009765625, f); |
| 3338 | |
| 3339 | obj = FPDFPage_GetObject(page, 35); |
| 3340 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3341 | EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f)); |
| 3342 | EXPECT_DOUBLE_EQ(69.0, a); |
| 3343 | EXPECT_DOUBLE_EQ(0.0, b); |
| 3344 | EXPECT_DOUBLE_EQ(0.0, c); |
| 3345 | EXPECT_DOUBLE_EQ(51.0, d); |
| 3346 | EXPECT_DOUBLE_EQ(360.0, e); |
| 3347 | EXPECT_DOUBLE_EQ(646.510009765625, f); |
| 3348 | |
| 3349 | obj = FPDFPage_GetObject(page, 36); |
| 3350 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3351 | EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f)); |
| 3352 | EXPECT_DOUBLE_EQ(59.0, a); |
| 3353 | EXPECT_DOUBLE_EQ(0.0, b); |
| 3354 | EXPECT_DOUBLE_EQ(0.0, c); |
| 3355 | EXPECT_DOUBLE_EQ(45.0, d); |
| 3356 | EXPECT_DOUBLE_EQ(72.0, e); |
| 3357 | EXPECT_DOUBLE_EQ(553.510009765625, f); |
| 3358 | |
| 3359 | obj = FPDFPage_GetObject(page, 37); |
| 3360 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3361 | EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f)); |
| 3362 | EXPECT_DOUBLE_EQ(55.94000244140625, a); |
| 3363 | EXPECT_DOUBLE_EQ(0.0, b); |
| 3364 | EXPECT_DOUBLE_EQ(0.0, c); |
| 3365 | EXPECT_DOUBLE_EQ(46.950000762939453, d); |
| 3366 | EXPECT_DOUBLE_EQ(216.0, e); |
| 3367 | EXPECT_DOUBLE_EQ(552.510009765625, f); |
| 3368 | |
| 3369 | obj = FPDFPage_GetObject(page, 38); |
| 3370 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3371 | EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f)); |
| 3372 | EXPECT_DOUBLE_EQ(70.528999328613281, a); |
| 3373 | EXPECT_DOUBLE_EQ(0.0, b); |
| 3374 | EXPECT_DOUBLE_EQ(0.0, c); |
| 3375 | EXPECT_DOUBLE_EQ(43.149997711181641, d); |
| 3376 | EXPECT_DOUBLE_EQ(360.0, e); |
| 3377 | EXPECT_DOUBLE_EQ(553.3599853515625, f); |
| 3378 | |
| 3379 | UnloadPage(page); |
| 3380 | } |
| 3381 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 3382 | TEST_F(FPDFEditEmbedderTest, DestroyPageObject) { |
Jane Liu | 2e5f0ae | 2017-08-08 15:23:27 -0400 | [diff] [blame] | 3383 | FPDF_PAGEOBJECT rect = FPDFPageObj_CreateNewRect(10, 10, 20, 20); |
| 3384 | ASSERT_TRUE(rect); |
| 3385 | |
| 3386 | // There should be no memory leaks with a call to FPDFPageObj_Destroy(). |
| 3387 | FPDFPageObj_Destroy(rect); |
| 3388 | } |
Jane Liu | be63ab9 | 2017-08-09 14:09:34 -0400 | [diff] [blame] | 3389 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 3390 | TEST_F(FPDFEditEmbedderTest, GetImageFilters) { |
Jane Liu | be63ab9 | 2017-08-09 14:09:34 -0400 | [diff] [blame] | 3391 | EXPECT_TRUE(OpenDocument("embedded_images.pdf")); |
| 3392 | FPDF_PAGE page = LoadPage(0); |
| 3393 | ASSERT_TRUE(page); |
| 3394 | |
| 3395 | // Verify that retrieving the filter of a non-image object would fail. |
| 3396 | FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 32); |
| 3397 | ASSERT_NE(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3398 | ASSERT_EQ(0, FPDFImageObj_GetImageFilterCount(obj)); |
| 3399 | EXPECT_EQ(0u, FPDFImageObj_GetImageFilter(obj, 0, nullptr, 0)); |
| 3400 | |
| 3401 | // Verify the returned filter string for an image object with a single filter. |
| 3402 | obj = FPDFPage_GetObject(page, 33); |
| 3403 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3404 | ASSERT_EQ(1, FPDFImageObj_GetImageFilterCount(obj)); |
| 3405 | unsigned long len = FPDFImageObj_GetImageFilter(obj, 0, nullptr, 0); |
| 3406 | std::vector<char> buf(len); |
Lei Zhang | 0733a1b | 2017-08-31 12:36:31 -0700 | [diff] [blame] | 3407 | static constexpr char kFlateDecode[] = "FlateDecode"; |
| 3408 | EXPECT_EQ(sizeof(kFlateDecode), |
| 3409 | FPDFImageObj_GetImageFilter(obj, 0, buf.data(), len)); |
| 3410 | EXPECT_STREQ(kFlateDecode, buf.data()); |
Jane Liu | be63ab9 | 2017-08-09 14:09:34 -0400 | [diff] [blame] | 3411 | EXPECT_EQ(0u, FPDFImageObj_GetImageFilter(obj, 1, nullptr, 0)); |
| 3412 | |
| 3413 | // Verify all the filters for an image object with a list of filters. |
| 3414 | obj = FPDFPage_GetObject(page, 38); |
| 3415 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3416 | ASSERT_EQ(2, FPDFImageObj_GetImageFilterCount(obj)); |
| 3417 | len = FPDFImageObj_GetImageFilter(obj, 0, nullptr, 0); |
| 3418 | buf.clear(); |
| 3419 | buf.resize(len); |
Lei Zhang | 0733a1b | 2017-08-31 12:36:31 -0700 | [diff] [blame] | 3420 | static constexpr char kASCIIHexDecode[] = "ASCIIHexDecode"; |
| 3421 | EXPECT_EQ(sizeof(kASCIIHexDecode), |
| 3422 | FPDFImageObj_GetImageFilter(obj, 0, buf.data(), len)); |
| 3423 | EXPECT_STREQ(kASCIIHexDecode, buf.data()); |
Jane Liu | be63ab9 | 2017-08-09 14:09:34 -0400 | [diff] [blame] | 3424 | |
| 3425 | len = FPDFImageObj_GetImageFilter(obj, 1, nullptr, 0); |
| 3426 | buf.clear(); |
| 3427 | buf.resize(len); |
Lei Zhang | 0733a1b | 2017-08-31 12:36:31 -0700 | [diff] [blame] | 3428 | static constexpr char kDCTDecode[] = "DCTDecode"; |
| 3429 | EXPECT_EQ(sizeof(kDCTDecode), |
| 3430 | FPDFImageObj_GetImageFilter(obj, 1, buf.data(), len)); |
| 3431 | EXPECT_STREQ(kDCTDecode, buf.data()); |
Jane Liu | be63ab9 | 2017-08-09 14:09:34 -0400 | [diff] [blame] | 3432 | |
| 3433 | UnloadPage(page); |
| 3434 | } |
Jane Liu | ca89829 | 2017-08-16 11:25:35 -0400 | [diff] [blame] | 3435 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 3436 | TEST_F(FPDFEditEmbedderTest, GetImageMetadata) { |
Jane Liu | ca89829 | 2017-08-16 11:25:35 -0400 | [diff] [blame] | 3437 | ASSERT_TRUE(OpenDocument("embedded_images.pdf")); |
| 3438 | FPDF_PAGE page = LoadPage(0); |
| 3439 | ASSERT_TRUE(page); |
| 3440 | |
| 3441 | // Check that getting the metadata of a null object would fail. |
| 3442 | FPDF_IMAGEOBJ_METADATA metadata; |
| 3443 | EXPECT_FALSE(FPDFImageObj_GetImageMetadata(nullptr, page, &metadata)); |
| 3444 | |
| 3445 | // Check that receiving the metadata with a null metadata object would fail. |
| 3446 | FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 35); |
| 3447 | EXPECT_FALSE(FPDFImageObj_GetImageMetadata(obj, page, nullptr)); |
| 3448 | |
| 3449 | // Check that when retrieving an image object's metadata without passing in |
| 3450 | // |page|, all values are correct, with the last two being default values. |
| 3451 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3452 | ASSERT_TRUE(FPDFImageObj_GetImageMetadata(obj, nullptr, &metadata)); |
Julian Lunger | ecd063e | 2017-12-27 10:18:50 -0500 | [diff] [blame] | 3453 | EXPECT_EQ(7, metadata.marked_content_id); |
Jane Liu | ca89829 | 2017-08-16 11:25:35 -0400 | [diff] [blame] | 3454 | EXPECT_EQ(92u, metadata.width); |
| 3455 | EXPECT_EQ(68u, metadata.height); |
Lei Zhang | 351e8b0 | 2018-12-20 01:10:06 +0000 | [diff] [blame] | 3456 | EXPECT_FLOAT_EQ(96.0f, metadata.horizontal_dpi); |
| 3457 | EXPECT_FLOAT_EQ(96.0f, metadata.vertical_dpi); |
Jane Liu | ca89829 | 2017-08-16 11:25:35 -0400 | [diff] [blame] | 3458 | EXPECT_EQ(0u, metadata.bits_per_pixel); |
| 3459 | EXPECT_EQ(FPDF_COLORSPACE_UNKNOWN, metadata.colorspace); |
| 3460 | |
| 3461 | // Verify the metadata of a bitmap image with indexed colorspace. |
| 3462 | ASSERT_TRUE(FPDFImageObj_GetImageMetadata(obj, page, &metadata)); |
Julian Lunger | ecd063e | 2017-12-27 10:18:50 -0500 | [diff] [blame] | 3463 | EXPECT_EQ(7, metadata.marked_content_id); |
Jane Liu | ca89829 | 2017-08-16 11:25:35 -0400 | [diff] [blame] | 3464 | EXPECT_EQ(92u, metadata.width); |
| 3465 | EXPECT_EQ(68u, metadata.height); |
Lei Zhang | 351e8b0 | 2018-12-20 01:10:06 +0000 | [diff] [blame] | 3466 | EXPECT_FLOAT_EQ(96.0f, metadata.horizontal_dpi); |
| 3467 | EXPECT_FLOAT_EQ(96.0f, metadata.vertical_dpi); |
Jane Liu | ca89829 | 2017-08-16 11:25:35 -0400 | [diff] [blame] | 3468 | EXPECT_EQ(1u, metadata.bits_per_pixel); |
| 3469 | EXPECT_EQ(FPDF_COLORSPACE_INDEXED, metadata.colorspace); |
| 3470 | |
| 3471 | // Verify the metadata of an image with RGB colorspace. |
| 3472 | obj = FPDFPage_GetObject(page, 37); |
| 3473 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3474 | ASSERT_TRUE(FPDFImageObj_GetImageMetadata(obj, page, &metadata)); |
Julian Lunger | ecd063e | 2017-12-27 10:18:50 -0500 | [diff] [blame] | 3475 | EXPECT_EQ(9, metadata.marked_content_id); |
Jane Liu | ca89829 | 2017-08-16 11:25:35 -0400 | [diff] [blame] | 3476 | EXPECT_EQ(126u, metadata.width); |
| 3477 | EXPECT_EQ(106u, metadata.height); |
Lei Zhang | 351e8b0 | 2018-12-20 01:10:06 +0000 | [diff] [blame] | 3478 | EXPECT_FLOAT_EQ(162.173752f, metadata.horizontal_dpi); |
| 3479 | EXPECT_FLOAT_EQ(162.555878f, metadata.vertical_dpi); |
Jane Liu | ca89829 | 2017-08-16 11:25:35 -0400 | [diff] [blame] | 3480 | EXPECT_EQ(24u, metadata.bits_per_pixel); |
| 3481 | EXPECT_EQ(FPDF_COLORSPACE_DEVICERGB, metadata.colorspace); |
| 3482 | |
| 3483 | UnloadPage(page); |
| 3484 | } |