blob: b7ff7ed5c5a7f977aad3264085aa8f7aa44abcde [file] [log] [blame]
Tom Sepezd483eb42016-01-06 10:03:59 -08001// 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 Sinclair85c8e7f2016-11-21 13:50:32 -05005#include <memory>
6#include <string>
Nicolas Penad03ca422017-03-06 13:54:33 -05007#include <utility>
Jane Liu548334e2017-08-03 16:33:40 -04008#include <vector>
Dan Sinclair85c8e7f2016-11-21 13:50:32 -05009
Lei Zhange4cdac52019-04-30 16:45:57 +000010#include "build/build_config.h"
Nicolas Penabe90aae2017-02-27 10:41:41 -050011#include "core/fpdfapi/font/cpdf_font.h"
Miklos Vajna46b43732018-08-14 19:15:43 +000012#include "core/fpdfapi/page/cpdf_formobject.h"
Nicolas Penaa4ad01f2017-02-15 16:26:48 -050013#include "core/fpdfapi/page/cpdf_page.h"
Henrique Nakashima6eb79392018-06-12 20:27:35 +000014#include "core/fpdfapi/page/cpdf_pageobject.h"
Nicolas Penabe90aae2017-02-27 10:41:41 -050015#include "core/fpdfapi/parser/cpdf_array.h"
Nicolas Penaa4ad01f2017-02-15 16:26:48 -050016#include "core/fpdfapi/parser/cpdf_dictionary.h"
Nicolas Penad03ca422017-03-06 13:54:33 -050017#include "core/fpdfapi/parser/cpdf_number.h"
Nicolas Penabe90aae2017-02-27 10:41:41 -050018#include "core/fpdfapi/parser/cpdf_stream.h"
Artem Strygineababa12018-06-06 12:31:18 +000019#include "core/fpdfapi/parser/cpdf_stream_acc.h"
Nicolas Pena0fc185e2017-02-08 12:13:20 -050020#include "core/fxcrt/fx_system.h"
Lei Zhang0dcb8fd2019-02-07 00:46:04 +000021#include "core/fxge/fx_font.h"
Dan Sinclair00d47a62018-03-28 18:39:04 +000022#include "fpdfsdk/cpdfsdk_helpers.h"
Tom Sepeze08d2b12018-04-25 18:49:32 +000023#include "public/cpp/fpdf_scopers.h"
Jane Liueda65252017-06-07 11:31:27 -040024#include "public/fpdf_annot.h"
Tom Sepezd483eb42016-01-06 10:03:59 -080025#include "public/fpdf_edit.h"
26#include "public/fpdfview.h"
27#include "testing/embedder_test.h"
Hui Yingstb4baceb2020-04-28 23:46:10 +000028#include "testing/embedder_test_constants.h"
Lei Zhangb6992dd2019-02-05 23:30:20 +000029#include "testing/fx_string_testhelpers.h"
Tom Sepez0aec19b2016-01-07 12:22:44 -080030#include "testing/gmock/include/gmock/gmock-matchers.h"
Tom Sepezd483eb42016-01-06 10:03:59 -080031#include "testing/gtest/include/gtest/gtest.h"
Hui Yingst8e88e382020-11-02 18:19:56 +000032#include "testing/utils/file_util.h"
Lei Zhang4c64e962019-02-05 19:24:12 +000033#include "testing/utils/hash.h"
Hui Yingst8e88e382020-11-02 18:19:56 +000034#include "testing/utils/path_service.h"
Tom Sepezd483eb42016-01-06 10:03:59 -080035
Hui Yingstb4baceb2020-04-28 23:46:10 +000036using pdfium::kHelloWorldChecksum;
37
38namespace {
39
Hui Yingst3b6136a2020-06-09 00:39:33 +000040const char kAllRemovedChecksum[] = "eee4600ac08b458ac7ac2320e225674c";
41
42const wchar_t kBottomText[] = L"I'm at the bottom of the page";
43
Hui Yingst5ce6a3f2020-11-11 02:29:09 +000044#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
45static constexpr char kBottomTextChecksum[] =
46 "5df7be86df1e18819723cdd9c81c2c7d";
Hui Yingst3b6136a2020-06-09 00:39:33 +000047#else
Hui Yingst5ce6a3f2020-11-11 02:29:09 +000048#if defined(OS_WIN)
49static constexpr char kBottomTextChecksum[] =
50 "08d1ff3e5a42801bee6077fd366bef00";
51#elif defined(OS_APPLE)
52static constexpr char kBottomTextChecksum[] =
53 "324e1db8164a040cf6104538baa95ba6";
54#else
55static constexpr char kBottomTextChecksum[] =
56 "eacaa24573b8ce997b3882595f096f00";
Hui Yingst3b6136a2020-06-09 00:39:33 +000057#endif
Hui Yingst5ce6a3f2020-11-11 02:29:09 +000058#endif // defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
Hui Yingstb4baceb2020-04-28 23:46:10 +000059
Hui Yingst08c40712020-04-29 01:37:35 +000060#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
61#if defined(OS_WIN)
62const char kFirstRemovedChecksum[] = "d468228cc55071a9e838f66afc0f58ff";
63#else
64const char kFirstRemovedChecksum[] = "0c79dc1065b1d06991e3ac4aaa35d25c";
65#endif // defined(OS_WIN)
66#else
Hui Yingstb4baceb2020-04-28 23:46:10 +000067#if defined(OS_WIN)
68const char kFirstRemovedChecksum[] = "aae6c5334721f90ec30d3d59f4ef7deb";
Lei Zhang0c03d632020-07-30 17:05:36 +000069#elif defined(OS_APPLE)
Hui Yingstb4baceb2020-04-28 23:46:10 +000070const char kFirstRemovedChecksum[] = "17ca3778fd8bb395b46532f1fa17f702";
71#else
72const char kFirstRemovedChecksum[] = "b76df015fe88009c3c342395df96abf1";
73#endif
Hui Yingst08c40712020-04-29 01:37:35 +000074#endif // defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
Hui Yingstb4baceb2020-04-28 23:46:10 +000075
Hui Yingst3b6136a2020-06-09 00:39:33 +000076const wchar_t kLoadedFontText[] = L"I am testing my loaded font, WEE.";
77
Hui Yingst9d6ec8f2020-11-05 23:35:38 +000078#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
79const char kLoadedFontTextChecksum[] = "a1dffe52c1b9ded5fe8d77eb10d8cc19";
80#else
Hui Yingst3b6136a2020-06-09 00:39:33 +000081#if defined(OS_WIN)
82const char kLoadedFontTextChecksum[] = "d60ba39f9698e32360d99e727dd93165";
Lei Zhang0c03d632020-07-30 17:05:36 +000083#elif defined(OS_APPLE)
Hui Yingst3b6136a2020-06-09 00:39:33 +000084const char kLoadedFontTextChecksum[] = "fc921c0bbdde73986ac13c15a85db4c3";
85#else
86const char kLoadedFontTextChecksum[] = "70592859010ffbf532a2237b8118bcc4";
87#endif
Hui Yingst9d6ec8f2020-11-05 23:35:38 +000088#endif // defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
Hui Yingst3b6136a2020-06-09 00:39:33 +000089
90const char kRedRectangleChecksum[] = "66d02eaa6181e2c069ce2ea99beda497";
91
Lei Zhang85166512020-07-14 23:28:56 +000092// In embedded_images.pdf.
93const char kEmbeddedImage33Checksum[] = "cb3637934bb3b95a6e4ae1ea9eb9e56e";
94
Hui Yingstb4baceb2020-04-28 23:46:10 +000095} // namespace
96
Lei Zhangab41f252018-12-23 03:10:50 +000097class FPDFEditEmbedderTest : public EmbedderTest {
Nicolas Penad03ca422017-03-06 13:54:33 -050098 protected:
99 FPDF_DOCUMENT CreateNewDocument() {
100 document_ = FPDF_CreateNewDocument();
Tom Sepez41066c12017-05-18 09:28:49 -0700101 cpdf_doc_ = CPDFDocumentFromFPDFDocument(document_);
Nicolas Penad03ca422017-03-06 13:54:33 -0500102 return document_;
103 }
104
Lei Zhang710fa992018-05-25 16:24:48 +0000105 void CheckFontDescriptor(const CPDF_Dictionary* font_dict,
Nicolas Penad03ca422017-03-06 13:54:33 -0500106 int font_type,
107 bool bold,
108 bool italic,
Tom Sepez20c41a52018-08-29 23:53:53 +0000109 pdfium::span<const uint8_t> span) {
Lei Zhangb1ec2802018-05-25 21:55:24 +0000110 const CPDF_Dictionary* font_desc = font_dict->GetDictFor("FontDescriptor");
Nicolas Penad03ca422017-03-06 13:54:33 -0500111 ASSERT_TRUE(font_desc);
Lei Zhanga228ff32020-06-24 17:39:33 +0000112 EXPECT_EQ("FontDescriptor", font_desc->GetNameFor("Type"));
113 ByteString font_name = font_desc->GetNameFor("FontName");
114 EXPECT_FALSE(font_name.IsEmpty());
115 EXPECT_EQ(font_dict->GetNameFor("BaseFont"), font_name);
Nicolas Penad03ca422017-03-06 13:54:33 -0500116
117 // Check that the font descriptor has the required keys according to spec
118 // 1.7 Table 5.19
119 ASSERT_TRUE(font_desc->KeyExist("Flags"));
Dan Sinclair10e1f052017-09-28 15:59:42 -0400120
Nicolas Penad03ca422017-03-06 13:54:33 -0500121 int font_flags = font_desc->GetIntegerFor("Flags");
Lei Zhang44d03832019-10-18 18:45:31 +0000122 EXPECT_EQ(bold, FontStyleIsForceBold(font_flags));
Dan Sinclair10e1f052017-09-28 15:59:42 -0400123 EXPECT_EQ(italic, FontStyleIsItalic(font_flags));
124 EXPECT_TRUE(FontStyleIsNonSymbolic(font_flags));
Nicolas Penad03ca422017-03-06 13:54:33 -0500125 ASSERT_TRUE(font_desc->KeyExist("FontBBox"));
Nicolás Peña5f95f362017-09-28 13:00:45 +0900126
Lei Zhangb1ec2802018-05-25 21:55:24 +0000127 const CPDF_Array* fontBBox = font_desc->GetArrayFor("FontBBox");
Nicolás Peña5f95f362017-09-28 13:00:45 +0900128 ASSERT_TRUE(fontBBox);
Lei Zhangf40380f2018-10-12 18:31:51 +0000129 EXPECT_EQ(4u, fontBBox->size());
Nicolás Peña5f95f362017-09-28 13:00:45 +0900130 // Check that the coordinates are in the preferred order according to spec
131 // 1.7 Section 3.8.4
132 EXPECT_TRUE(fontBBox->GetIntegerAt(0) < fontBBox->GetIntegerAt(2));
133 EXPECT_TRUE(fontBBox->GetIntegerAt(1) < fontBBox->GetIntegerAt(3));
134
Nicolas Penad03ca422017-03-06 13:54:33 -0500135 EXPECT_TRUE(font_desc->KeyExist("ItalicAngle"));
136 EXPECT_TRUE(font_desc->KeyExist("Ascent"));
137 EXPECT_TRUE(font_desc->KeyExist("Descent"));
138 EXPECT_TRUE(font_desc->KeyExist("CapHeight"));
139 EXPECT_TRUE(font_desc->KeyExist("StemV"));
Ryan Harrison275e2602017-09-18 14:23:18 -0400140 ByteString present("FontFile");
141 ByteString absent("FontFile2");
Nicolas Penad03ca422017-03-06 13:54:33 -0500142 if (font_type == FPDF_FONT_TRUETYPE)
143 std::swap(present, absent);
144 EXPECT_TRUE(font_desc->KeyExist(present));
145 EXPECT_FALSE(font_desc->KeyExist(absent));
146
Artem Strygineababa12018-06-06 12:31:18 +0000147 auto streamAcc =
148 pdfium::MakeRetain<CPDF_StreamAcc>(font_desc->GetStreamFor(present));
149 streamAcc->LoadAllDataRaw();
150
Nicolas Penad03ca422017-03-06 13:54:33 -0500151 // Check that the font stream is the one that was provided
Tom Sepez20c41a52018-08-29 23:53:53 +0000152 ASSERT_EQ(span.size(), streamAcc->GetSize());
Nicolás Peña79eab232017-09-28 13:29:05 +0900153 if (font_type == FPDF_FONT_TRUETYPE) {
Tom Sepez20c41a52018-08-29 23:53:53 +0000154 ASSERT_EQ(static_cast<int>(span.size()),
Artem Strygineababa12018-06-06 12:31:18 +0000155 streamAcc->GetDict()->GetIntegerFor("Length1"));
Nicolás Peña79eab232017-09-28 13:29:05 +0900156 }
Artem Strygineababa12018-06-06 12:31:18 +0000157
158 const uint8_t* stream_data = streamAcc->GetData();
Tom Sepez20c41a52018-08-29 23:53:53 +0000159 for (size_t j = 0; j < span.size(); j++)
160 EXPECT_EQ(span[j], stream_data[j]) << " at byte " << j;
Nicolas Penad03ca422017-03-06 13:54:33 -0500161 }
162
Lei Zhangde579ab2018-05-25 21:49:49 +0000163 void CheckCompositeFontWidths(const CPDF_Array* widths_array,
Nicolas Penad03ca422017-03-06 13:54:33 -0500164 CPDF_Font* typed_font) {
165 // Check that W array is in a format that conforms to PDF spec 1.7 section
166 // "Glyph Metrics in CIDFonts" (these checks are not
167 // implementation-specific).
Lei Zhangf40380f2018-10-12 18:31:51 +0000168 EXPECT_GT(widths_array->size(), 1u);
Nicolas Penad03ca422017-03-06 13:54:33 -0500169 int num_cids_checked = 0;
170 int cur_cid = 0;
Lei Zhangf40380f2018-10-12 18:31:51 +0000171 for (size_t idx = 0; idx < widths_array->size(); idx++) {
Nicolas Penad03ca422017-03-06 13:54:33 -0500172 int cid = widths_array->GetNumberAt(idx);
173 EXPECT_GE(cid, cur_cid);
Lei Zhangf40380f2018-10-12 18:31:51 +0000174 ASSERT_FALSE(++idx == widths_array->size());
Lei Zhangde579ab2018-05-25 21:49:49 +0000175 const CPDF_Object* next = widths_array->GetObjectAt(idx);
Nicolas Penad03ca422017-03-06 13:54:33 -0500176 if (next->IsArray()) {
177 // We are in the c [w1 w2 ...] case
Lei Zhangde579ab2018-05-25 21:49:49 +0000178 const CPDF_Array* arr = next->AsArray();
Lei Zhangf40380f2018-10-12 18:31:51 +0000179 int cnt = static_cast<int>(arr->size());
Nicolas Penad03ca422017-03-06 13:54:33 -0500180 size_t inner_idx = 0;
181 for (cur_cid = cid; cur_cid < cid + cnt; cur_cid++) {
Lei Zhang70e66192020-09-18 20:50:51 +0000182 int width = arr->GetNumberAt(inner_idx++);
Dan Sinclair971a6742018-03-28 19:23:25 +0000183 EXPECT_EQ(width, typed_font->GetCharWidthF(cur_cid))
184 << " at cid " << cur_cid;
Nicolas Penad03ca422017-03-06 13:54:33 -0500185 }
186 num_cids_checked += cnt;
187 continue;
188 }
189 // Otherwise, are in the c_first c_last w case.
190 ASSERT_TRUE(next->IsNumber());
191 int last_cid = next->AsNumber()->GetInteger();
Lei Zhangf40380f2018-10-12 18:31:51 +0000192 ASSERT_FALSE(++idx == widths_array->size());
Lei Zhang70e66192020-09-18 20:50:51 +0000193 int width = widths_array->GetNumberAt(idx);
Nicolas Penad03ca422017-03-06 13:54:33 -0500194 for (cur_cid = cid; cur_cid <= last_cid; cur_cid++) {
Dan Sinclair971a6742018-03-28 19:23:25 +0000195 EXPECT_EQ(width, typed_font->GetCharWidthF(cur_cid))
196 << " at cid " << cur_cid;
Nicolas Penad03ca422017-03-06 13:54:33 -0500197 }
198 num_cids_checked += last_cid - cid + 1;
199 }
200 // Make sure we have a good amount of cids described
201 EXPECT_GT(num_cids_checked, 900);
202 }
203 CPDF_Document* cpdf_doc() { return cpdf_doc_; }
204
205 private:
206 CPDF_Document* cpdf_doc_;
207};
Tom Sepezd483eb42016-01-06 10:03:59 -0800208
etienneb7712c262016-04-26 08:13:45 -0700209namespace {
thestigdc7ec032016-11-21 15:32:52 -0800210
etienneb7712c262016-04-26 08:13:45 -0700211const char kExpectedPDF[] =
212 "%PDF-1.7\r\n"
213 "%\xA1\xB3\xC5\xD7\r\n"
214 "1 0 obj\r\n"
215 "<</Pages 2 0 R /Type/Catalog>>\r\n"
216 "endobj\r\n"
217 "2 0 obj\r\n"
218 "<</Count 1/Kids\\[ 4 0 R \\]/Type/Pages>>\r\n"
219 "endobj\r\n"
220 "3 0 obj\r\n"
221 "<</CreationDate\\(D:.*\\)/Creator\\(PDFium\\)>>\r\n"
222 "endobj\r\n"
223 "4 0 obj\r\n"
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400224 "<</MediaBox\\[ 0 0 640 480\\]/Parent 2 0 R "
225 "/Resources<</ExtGState<</FXE1 5 0 R >>>>"
Nicolas Penad9d6c292017-06-06 16:12:10 -0400226 "/Rotate 0/Type/Page"
etienneb7712c262016-04-26 08:13:45 -0700227 ">>\r\n"
228 "endobj\r\n"
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400229 "5 0 obj\r\n"
230 "<</BM/Normal/CA 1/ca 1>>\r\n"
231 "endobj\r\n"
etienneb7712c262016-04-26 08:13:45 -0700232 "xref\r\n"
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400233 "0 6\r\n"
etienneb7712c262016-04-26 08:13:45 -0700234 "0000000000 65535 f\r\n"
235 "0000000017 00000 n\r\n"
236 "0000000066 00000 n\r\n"
237 "0000000122 00000 n\r\n"
238 "0000000192 00000 n\r\n"
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400239 "0000000311 00000 n\r\n"
etienneb7712c262016-04-26 08:13:45 -0700240 "trailer\r\n"
241 "<<\r\n"
242 "/Root 1 0 R\r\n"
243 "/Info 3 0 R\r\n"
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400244 "/Size 6/ID\\[<.*><.*>\\]>>\r\n"
etienneb7712c262016-04-26 08:13:45 -0700245 "startxref\r\n"
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400246 "354\r\n"
etienneb7712c262016-04-26 08:13:45 -0700247 "%%EOF\r\n";
thestigdc7ec032016-11-21 15:32:52 -0800248
etienneb7712c262016-04-26 08:13:45 -0700249} // namespace
250
Hui Yingst8e88e382020-11-02 18:19:56 +0000251TEST_F(FPDFEditEmbedderTest, EmbedNotoSansSCFont) {
252 EXPECT_TRUE(CreateEmptyDocument());
253 ScopedFPDFPage page(FPDFPage_New(document(), 0, 400, 400));
254 std::string font_path;
Hui Yingst8532ef02020-11-04 19:17:17 +0000255 ASSERT_TRUE(PathService::GetThirdPartyFilePath(
256 "NotoSansCJK/NotoSansSC-Regular.subset.otf", &font_path));
Hui Yingst8e88e382020-11-02 18:19:56 +0000257
258 size_t file_length = 0;
259 std::unique_ptr<char, pdfium::FreeDeleter> font_data =
260 GetFileContents(font_path.c_str(), &file_length);
261 ASSERT(font_data);
262
263 ScopedFPDFFont font(FPDFText_LoadFont(
264 document(), reinterpret_cast<const uint8_t*>(font_data.get()),
265 file_length, FPDF_FONT_TRUETYPE, /*cid=*/true));
266 FPDF_PAGEOBJECT text_object =
267 FPDFPageObj_CreateTextObj(document(), font.get(), 20.0f);
268 EXPECT_TRUE(text_object);
269
270 // Test the characters which are either mapped to one single unicode or
271 // multiple unicodes in the embedded font.
272 ScopedFPDFWideString text = GetFPDFWideString(L"这是第一句。 这是第二行。");
273 EXPECT_TRUE(FPDFText_SetText(text_object, text.get()));
274
275 FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 50, 200);
276 FPDFPage_InsertObject(page.get(), text_object);
277 EXPECT_TRUE(FPDFPage_GenerateContent(page.get()));
278
279#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
280#if defined(OS_APPLE)
281 const char kChecksum[] = "9a31fb87d1c6d2346bba22d1196041cd";
282#else
283 const char kChecksum[] = "5bb65e15fc0a685934cd5006dec08a76";
284#endif // defined(OS_APPLE)
285#else
286#if defined(OS_WIN)
287 const char kChecksum[] = "89e8eef5d6ad18c542a92a0519954d0f";
288#else
289 const char kChecksum[] = "9a31fb87d1c6d2346bba22d1196041cd";
290#endif // defined(OS_WIN)
291#endif // defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
292 ScopedFPDFBitmap page_bitmap = RenderPage(page.get());
293 CompareBitmap(page_bitmap.get(), 400, 400, kChecksum);
294
295 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
296 VerifySavedDocument(400, 400, kChecksum);
297}
298
Lei Zhangab41f252018-12-23 03:10:50 +0000299TEST_F(FPDFEditEmbedderTest, EmptyCreation) {
Tom Sepezd483eb42016-01-06 10:03:59 -0800300 EXPECT_TRUE(CreateEmptyDocument());
weili9b777de2016-08-19 16:19:46 -0700301 FPDF_PAGE page = FPDFPage_New(document(), 0, 640.0, 480.0);
Tom Sepezd483eb42016-01-06 10:03:59 -0800302 EXPECT_NE(nullptr, page);
Nicolas Penad9d6c292017-06-06 16:12:10 -0400303 // The FPDFPage_GenerateContent call should do nothing.
Tom Sepezd483eb42016-01-06 10:03:59 -0800304 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Tom Sepez0aec19b2016-01-07 12:22:44 -0800305 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
etienneb7712c262016-04-26 08:13:45 -0700306
Nicolas Penad9d6c292017-06-06 16:12:10 -0400307 EXPECT_THAT(GetString(), testing::MatchesRegex(std::string(
308 kExpectedPDF, sizeof(kExpectedPDF))));
weili9b777de2016-08-19 16:19:46 -0700309 FPDF_ClosePage(page);
Tom Sepezd483eb42016-01-06 10:03:59 -0800310}
thestigdc7ec032016-11-21 15:32:52 -0800311
312// Regression test for https://crbug.com/667012
Lei Zhangab41f252018-12-23 03:10:50 +0000313TEST_F(FPDFEditEmbedderTest, RasterizePDF) {
thestigdc7ec032016-11-21 15:32:52 -0800314 const char kAllBlackMd5sum[] = "5708fc5c4a8bd0abde99c8e8f0390615";
315
Lei Zhangf0542892019-01-17 18:46:27 +0000316 // Get the bitmap for the original document.
Tom Sepeze08d2b12018-04-25 18:49:32 +0000317 ScopedFPDFBitmap orig_bitmap;
thestigdc7ec032016-11-21 15:32:52 -0800318 {
Daniel Hosseinian5af51b62020-07-18 00:53:43 +0000319 ASSERT_TRUE(OpenDocument("black.pdf"));
thestigdc7ec032016-11-21 15:32:52 -0800320 FPDF_PAGE orig_page = LoadPage(0);
Lei Zhang107fa7b2018-02-09 21:48:15 +0000321 ASSERT_TRUE(orig_page);
322 orig_bitmap = RenderLoadedPage(orig_page);
323 CompareBitmap(orig_bitmap.get(), 612, 792, kAllBlackMd5sum);
thestigdc7ec032016-11-21 15:32:52 -0800324 UnloadPage(orig_page);
325 }
326
327 // Create a new document from |orig_bitmap| and save it.
328 {
329 FPDF_DOCUMENT temp_doc = FPDF_CreateNewDocument();
330 FPDF_PAGE temp_page = FPDFPage_New(temp_doc, 0, 612, 792);
331
332 // Add the bitmap to an image object and add the image object to the output
333 // page.
Lei Zhangcbd89572017-03-15 17:35:47 -0700334 FPDF_PAGEOBJECT temp_img = FPDFPageObj_NewImageObj(temp_doc);
Lei Zhang107fa7b2018-02-09 21:48:15 +0000335 EXPECT_TRUE(
336 FPDFImageObj_SetBitmap(&temp_page, 1, temp_img, orig_bitmap.get()));
thestigdc7ec032016-11-21 15:32:52 -0800337 EXPECT_TRUE(FPDFImageObj_SetMatrix(temp_img, 612, 0, 0, 792, 0, 0));
338 FPDFPage_InsertObject(temp_page, temp_img);
339 EXPECT_TRUE(FPDFPage_GenerateContent(temp_page));
340 EXPECT_TRUE(FPDF_SaveAsCopy(temp_doc, this, 0));
341 FPDF_ClosePage(temp_page);
342 FPDF_CloseDocument(temp_doc);
343 }
thestigdc7ec032016-11-21 15:32:52 -0800344
345 // Get the generated content. Make sure it is at least as big as the original
346 // PDF.
Lei Zhangd72fd582018-07-27 19:37:27 +0000347 EXPECT_GT(GetString().size(), 923u);
Dan Sinclair04e4dc82017-10-18 12:17:14 -0400348 VerifySavedDocument(612, 792, kAllBlackMd5sum);
thestigdc7ec032016-11-21 15:32:52 -0800349}
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500350
Lei Zhang03e5e682019-09-16 19:45:55 +0000351// TODO(crbug.com/pdfium/11): Fix this test and enable.
352#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
353#define MAYBE_AddPaths DISABLED_AddPaths
354#else
355#define MAYBE_AddPaths AddPaths
356#endif
357TEST_F(FPDFEditEmbedderTest, MAYBE_AddPaths) {
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500358 // Start with a blank page
Nicolas Penad03ca422017-03-06 13:54:33 -0500359 FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792);
Lei Zhang107fa7b2018-02-09 21:48:15 +0000360 ASSERT_TRUE(page);
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500361
362 // We will first add a red rectangle
363 FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(10, 10, 20, 20);
Lei Zhang107fa7b2018-02-09 21:48:15 +0000364 ASSERT_TRUE(red_rect);
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500365 // Expect false when trying to set colors out of range
Lei Zhang3475b482019-05-13 18:30:57 +0000366 EXPECT_FALSE(FPDFPageObj_SetStrokeColor(red_rect, 100, 100, 100, 300));
367 EXPECT_FALSE(FPDFPageObj_SetFillColor(red_rect, 200, 256, 200, 0));
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500368
369 // Fill rectangle with red and insert to the page
Lei Zhang3475b482019-05-13 18:30:57 +0000370 EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 0, 0, 255));
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500371 EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0));
Miklos Vajna491112b2018-05-30 13:30:10 +0000372
373 int fillmode = FPDF_FILLMODE_NONE;
374 FPDF_BOOL stroke = true;
375 EXPECT_TRUE(FPDFPath_GetDrawMode(red_rect, &fillmode, &stroke));
376 EXPECT_EQ(FPDF_FILLMODE_ALTERNATE, fillmode);
377 EXPECT_FALSE(stroke);
378
Lei Zhang8da98232019-12-11 23:29:33 +0000379 static const FS_MATRIX kMatrix = {1, 2, 3, 4, 5, 6};
380 EXPECT_FALSE(FPDFPath_SetMatrix(nullptr, &kMatrix));
381 EXPECT_TRUE(FPDFPath_SetMatrix(red_rect, &kMatrix));
382
383 FS_MATRIX matrix;
384 EXPECT_FALSE(FPDFPath_GetMatrix(nullptr, &matrix));
385 EXPECT_TRUE(FPDFPath_GetMatrix(red_rect, &matrix));
386 EXPECT_FLOAT_EQ(1.0f, matrix.a);
387 EXPECT_FLOAT_EQ(2.0f, matrix.b);
388 EXPECT_FLOAT_EQ(3.0f, matrix.c);
389 EXPECT_FLOAT_EQ(4.0f, matrix.d);
390 EXPECT_FLOAT_EQ(5.0f, matrix.e);
391 EXPECT_FLOAT_EQ(6.0f, matrix.f);
392
393 // Set back the identity matrix.
394 matrix = {1.0f, 0.0f, 0.0f, 1.0f, 0.0f, 0.0f};
395 EXPECT_TRUE(FPDFPath_SetMatrix(red_rect, &matrix));
Miklos Vajna97f4d672018-06-04 14:47:17 +0000396
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500397 FPDFPage_InsertObject(page, red_rect);
Lei Zhang107fa7b2018-02-09 21:48:15 +0000398 {
Lei Zhang30ff2532019-01-31 21:37:55 +0000399 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstb4baceb2020-04-28 23:46:10 +0000400 CompareBitmap(page_bitmap.get(), 612, 792, kRedRectangleChecksum);
Lei Zhang107fa7b2018-02-09 21:48:15 +0000401 }
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500402
403 // Now add to that a green rectangle with some medium alpha
404 FPDF_PAGEOBJECT green_rect = FPDFPageObj_CreateNewRect(100, 100, 40, 40);
Lei Zhang3475b482019-05-13 18:30:57 +0000405 EXPECT_TRUE(FPDFPageObj_SetFillColor(green_rect, 0, 255, 0, 128));
Miklos Vajnaed4705b2017-04-05 09:24:50 +0200406
Miklos Vajna1ef04c92017-05-08 18:14:19 +0200407 // Make sure the type of the rectangle is a path.
408 EXPECT_EQ(FPDF_PAGEOBJ_PATH, FPDFPageObj_GetType(green_rect));
409
Miklos Vajnaed4705b2017-04-05 09:24:50 +0200410 // Make sure we get back the same color we set previously.
411 unsigned int R;
412 unsigned int G;
413 unsigned int B;
414 unsigned int A;
Lei Zhang3475b482019-05-13 18:30:57 +0000415 EXPECT_TRUE(FPDFPageObj_GetFillColor(green_rect, &R, &G, &B, &A));
Lei Zhangd72fd582018-07-27 19:37:27 +0000416 EXPECT_EQ(0u, R);
417 EXPECT_EQ(255u, G);
418 EXPECT_EQ(0u, B);
419 EXPECT_EQ(128u, A);
Miklos Vajnaed4705b2017-04-05 09:24:50 +0200420
Miklos Vajna12abfd02017-09-15 07:49:03 +0200421 // Make sure the path has 5 points (1 FXPT_TYPE::MoveTo and 4
422 // FXPT_TYPE::LineTo).
Miklos Vajna0150a542017-09-21 21:46:56 +0200423 ASSERT_EQ(5, FPDFPath_CountSegments(green_rect));
Miklos Vajna36eed872017-09-20 22:52:43 +0200424 // Verify actual coordinates.
425 FPDF_PATHSEGMENT segment = FPDFPath_GetPathSegment(green_rect, 0);
426 float x;
427 float y;
428 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
429 EXPECT_EQ(100, x);
430 EXPECT_EQ(100, y);
431 EXPECT_EQ(FPDF_SEGMENT_MOVETO, FPDFPathSegment_GetType(segment));
432 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
433 segment = FPDFPath_GetPathSegment(green_rect, 1);
434 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
435 EXPECT_EQ(100, x);
436 EXPECT_EQ(140, y);
437 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
438 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
439 segment = FPDFPath_GetPathSegment(green_rect, 2);
440 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
441 EXPECT_EQ(140, x);
442 EXPECT_EQ(140, y);
443 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
444 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
445 segment = FPDFPath_GetPathSegment(green_rect, 3);
446 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
447 EXPECT_EQ(140, x);
448 EXPECT_EQ(100, y);
449 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
450 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
451 segment = FPDFPath_GetPathSegment(green_rect, 4);
452 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
453 EXPECT_EQ(100, x);
454 EXPECT_EQ(100, y);
455 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
456 EXPECT_TRUE(FPDFPathSegment_GetClose(segment));
Miklos Vajna12abfd02017-09-15 07:49:03 +0200457
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500458 EXPECT_TRUE(FPDFPath_SetDrawMode(green_rect, FPDF_FILLMODE_WINDING, 0));
459 FPDFPage_InsertObject(page, green_rect);
Lei Zhang107fa7b2018-02-09 21:48:15 +0000460 {
Lei Zhang30ff2532019-01-31 21:37:55 +0000461 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Lei Zhang107fa7b2018-02-09 21:48:15 +0000462 CompareBitmap(page_bitmap.get(), 612, 792,
463 "7b0b87604594e773add528fae567a558");
464 }
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500465
466 // Add a black triangle.
467 FPDF_PAGEOBJECT black_path = FPDFPageObj_CreateNewPath(400, 100);
Lei Zhang3475b482019-05-13 18:30:57 +0000468 EXPECT_TRUE(FPDFPageObj_SetFillColor(black_path, 0, 0, 0, 200));
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500469 EXPECT_TRUE(FPDFPath_SetDrawMode(black_path, FPDF_FILLMODE_ALTERNATE, 0));
470 EXPECT_TRUE(FPDFPath_LineTo(black_path, 400, 200));
471 EXPECT_TRUE(FPDFPath_LineTo(black_path, 300, 100));
472 EXPECT_TRUE(FPDFPath_Close(black_path));
Miklos Vajna12abfd02017-09-15 07:49:03 +0200473
474 // Make sure the path has 3 points (1 FXPT_TYPE::MoveTo and 2
475 // FXPT_TYPE::LineTo).
Miklos Vajna0150a542017-09-21 21:46:56 +0200476 ASSERT_EQ(3, FPDFPath_CountSegments(black_path));
Miklos Vajna36eed872017-09-20 22:52:43 +0200477 // Verify actual coordinates.
478 segment = FPDFPath_GetPathSegment(black_path, 0);
479 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
480 EXPECT_EQ(400, x);
481 EXPECT_EQ(100, y);
482 EXPECT_EQ(FPDF_SEGMENT_MOVETO, FPDFPathSegment_GetType(segment));
483 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
484 segment = FPDFPath_GetPathSegment(black_path, 1);
485 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
486 EXPECT_EQ(400, x);
487 EXPECT_EQ(200, y);
488 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
489 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
490 segment = FPDFPath_GetPathSegment(black_path, 2);
491 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
492 EXPECT_EQ(300, x);
493 EXPECT_EQ(100, y);
494 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
495 EXPECT_TRUE(FPDFPathSegment_GetClose(segment));
496 // Make sure out of bounds index access fails properly.
497 EXPECT_EQ(nullptr, FPDFPath_GetPathSegment(black_path, 3));
Miklos Vajna12abfd02017-09-15 07:49:03 +0200498
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500499 FPDFPage_InsertObject(page, black_path);
Lei Zhang107fa7b2018-02-09 21:48:15 +0000500 {
Lei Zhang30ff2532019-01-31 21:37:55 +0000501 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Lei Zhang107fa7b2018-02-09 21:48:15 +0000502 CompareBitmap(page_bitmap.get(), 612, 792,
503 "eadc8020a14dfcf091da2688733d8806");
504 }
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500505
506 // Now add a more complex blue path.
507 FPDF_PAGEOBJECT blue_path = FPDFPageObj_CreateNewPath(200, 200);
Lei Zhang3475b482019-05-13 18:30:57 +0000508 EXPECT_TRUE(FPDFPageObj_SetFillColor(blue_path, 0, 0, 255, 255));
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500509 EXPECT_TRUE(FPDFPath_SetDrawMode(blue_path, FPDF_FILLMODE_WINDING, 0));
510 EXPECT_TRUE(FPDFPath_LineTo(blue_path, 230, 230));
511 EXPECT_TRUE(FPDFPath_BezierTo(blue_path, 250, 250, 280, 280, 300, 300));
512 EXPECT_TRUE(FPDFPath_LineTo(blue_path, 325, 325));
513 EXPECT_TRUE(FPDFPath_LineTo(blue_path, 350, 325));
514 EXPECT_TRUE(FPDFPath_BezierTo(blue_path, 375, 330, 390, 360, 400, 400));
515 EXPECT_TRUE(FPDFPath_Close(blue_path));
516 FPDFPage_InsertObject(page, blue_path);
Lei Zhang107fa7b2018-02-09 21:48:15 +0000517 const char kLastMD5[] = "9823e1a21bd9b72b6a442ba4f12af946";
518 {
Lei Zhang30ff2532019-01-31 21:37:55 +0000519 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Lei Zhang107fa7b2018-02-09 21:48:15 +0000520 CompareBitmap(page_bitmap.get(), 612, 792, kLastMD5);
521 }
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500522
523 // Now save the result, closing the page and document
Nicolas Pena207b7272017-05-26 17:37:06 -0400524 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Nicolas Penad03ca422017-03-06 13:54:33 -0500525 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500526 FPDF_ClosePage(page);
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500527
528 // Render the saved result
Lei Zhang107fa7b2018-02-09 21:48:15 +0000529 VerifySavedDocument(612, 792, kLastMD5);
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500530}
531
Miklos Vajna45501f32019-07-30 07:41:02 +0000532TEST_F(FPDFEditEmbedderTest, ClipPath) {
533 // Load document with a clipped rectangle.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +0000534 ASSERT_TRUE(OpenDocument("clip_path.pdf"));
Miklos Vajna45501f32019-07-30 07:41:02 +0000535 FPDF_PAGE page = LoadPage(0);
536 ASSERT_TRUE(page);
537
538 ASSERT_EQ(1, FPDFPage_CountObjects(page));
539
540 FPDF_PAGEOBJECT triangle = FPDFPage_GetObject(page, 0);
541 ASSERT_TRUE(triangle);
542
543 // Test that we got the expected triangle.
544 ASSERT_EQ(4, FPDFPath_CountSegments(triangle));
545
546 FPDF_PATHSEGMENT segment = FPDFPath_GetPathSegment(triangle, 0);
547 float x;
548 float y;
549 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
550 EXPECT_EQ(10, x);
551 EXPECT_EQ(10, y);
552 EXPECT_EQ(FPDF_SEGMENT_MOVETO, FPDFPathSegment_GetType(segment));
553 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
554
555 segment = FPDFPath_GetPathSegment(triangle, 1);
556 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
557 EXPECT_EQ(25, x);
558 EXPECT_EQ(40, y);
559 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
560 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
561
562 segment = FPDFPath_GetPathSegment(triangle, 2);
563 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
564 EXPECT_EQ(40, x);
565 EXPECT_EQ(10, y);
566 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
567 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
568
569 segment = FPDFPath_GetPathSegment(triangle, 3);
570 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
571 EXPECT_TRUE(FPDFPathSegment_GetClose(segment));
572
573 // Test FPDFPageObj_GetClipPath().
574 ASSERT_EQ(nullptr, FPDFPageObj_GetClipPath(nullptr));
575
576 FPDF_CLIPPATH clip_path = FPDFPageObj_GetClipPath(triangle);
577 ASSERT_TRUE(clip_path);
578
579 // Test FPDFClipPath_CountPaths().
580 ASSERT_EQ(-1, FPDFClipPath_CountPaths(nullptr));
581 ASSERT_EQ(1, FPDFClipPath_CountPaths(clip_path));
582
583 // Test FPDFClipPath_CountPathSegments().
584 ASSERT_EQ(-1, FPDFClipPath_CountPathSegments(nullptr, 0));
585 ASSERT_EQ(-1, FPDFClipPath_CountPathSegments(clip_path, -1));
586 ASSERT_EQ(-1, FPDFClipPath_CountPathSegments(clip_path, 1));
587 ASSERT_EQ(4, FPDFClipPath_CountPathSegments(clip_path, 0));
588
589 // FPDFClipPath_GetPathSegment() negative testing.
590 ASSERT_EQ(nullptr, FPDFClipPath_GetPathSegment(nullptr, 0, 0));
591 ASSERT_EQ(nullptr, FPDFClipPath_GetPathSegment(clip_path, -1, 0));
592 ASSERT_EQ(nullptr, FPDFClipPath_GetPathSegment(clip_path, 1, 0));
593 ASSERT_EQ(nullptr, FPDFClipPath_GetPathSegment(clip_path, 0, -1));
594 ASSERT_EQ(nullptr, FPDFClipPath_GetPathSegment(clip_path, 0, 4));
595
596 // FPDFClipPath_GetPathSegment() positive testing.
597 segment = FPDFClipPath_GetPathSegment(clip_path, 0, 0);
598 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
599 EXPECT_EQ(10, x);
600 EXPECT_EQ(15, y);
601 EXPECT_EQ(FPDF_SEGMENT_MOVETO, FPDFPathSegment_GetType(segment));
602 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
603
604 segment = FPDFClipPath_GetPathSegment(clip_path, 0, 1);
605 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
606 EXPECT_EQ(40, x);
607 EXPECT_EQ(15, y);
608 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
609 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
610
611 segment = FPDFClipPath_GetPathSegment(clip_path, 0, 2);
612 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
613 EXPECT_EQ(40, x);
614 EXPECT_EQ(35, y);
615 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
616 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
617
618 segment = FPDFClipPath_GetPathSegment(clip_path, 0, 3);
619 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
620 EXPECT_EQ(10, x);
621 EXPECT_EQ(35, y);
622 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
623 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
624
625 UnloadPage(page);
626}
627
Lei Zhang571f9322019-09-25 12:40:25 +0000628TEST_F(FPDFEditEmbedderTest, BUG_1399) {
629 // Load document with a clipped rectangle.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +0000630 ASSERT_TRUE(OpenDocument("bug_1399.pdf"));
Lei Zhang571f9322019-09-25 12:40:25 +0000631 FPDF_PAGE page = LoadPage(0);
632 ASSERT_TRUE(page);
633
634 ASSERT_EQ(7, FPDFPage_CountObjects(page));
635
636 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 0);
637 ASSERT_TRUE(obj);
638
639 ASSERT_EQ(2, FPDFPath_CountSegments(obj));
640
641 FPDF_PATHSEGMENT segment = FPDFPath_GetPathSegment(obj, 0);
642 float x;
643 float y;
644 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
645 EXPECT_FLOAT_EQ(107.718f, x);
646 EXPECT_FLOAT_EQ(719.922f, y);
647 EXPECT_EQ(FPDF_SEGMENT_MOVETO, FPDFPathSegment_GetType(segment));
648 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
649
650 segment = FPDFPath_GetPathSegment(obj, 1);
651 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
652 EXPECT_FLOAT_EQ(394.718f, x);
653 EXPECT_FLOAT_EQ(719.922f, y);
654 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
655 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
656
657 FPDF_CLIPPATH clip_path = FPDFPageObj_GetClipPath(obj);
658 ASSERT_TRUE(clip_path);
659
660 EXPECT_EQ(-1, FPDFClipPath_CountPaths(clip_path));
661 EXPECT_EQ(-1, FPDFClipPath_CountPathSegments(clip_path, 0));
662 EXPECT_FALSE(FPDFClipPath_GetPathSegment(clip_path, 0, 0));
663
664 UnloadPage(page);
665}
666
Hui Yingstcb4203d2020-06-09 01:48:47 +0000667TEST_F(FPDFEditEmbedderTest, SetText) {
Henrique Nakashima5ebfd642018-06-07 15:18:55 +0000668 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +0000669 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
Henrique Nakashima5ebfd642018-06-07 15:18:55 +0000670 FPDF_PAGE page = LoadPage(0);
671 ASSERT_TRUE(page);
672
673 // Get the "Hello, world!" text object and change it.
674 ASSERT_EQ(2, FPDFPage_CountObjects(page));
675 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0);
676 ASSERT_TRUE(page_object);
Lei Zhangf0f67682019-04-08 17:03:21 +0000677 ScopedFPDFWideString text1 = GetFPDFWideString(L"Changed for SetText test");
Henrique Nakashima5ebfd642018-06-07 15:18:55 +0000678 EXPECT_TRUE(FPDFText_SetText(page_object, text1.get()));
679
680 // Verify the "Hello, world!" text is gone and "Changed for SetText test" is
681 // now displayed.
682 ASSERT_EQ(2, FPDFPage_CountObjects(page));
Hui Yingstcb4203d2020-06-09 01:48:47 +0000683
684#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
685#if defined(OS_WIN)
686 const char kChangedChecksum[] = "119f828345f547c68e9454418fb7d80d";
Henrique Nakashima5ebfd642018-06-07 15:18:55 +0000687#else
Hui Yingstcb4203d2020-06-09 01:48:47 +0000688 const char kChangedChecksum[] = "6bc5171f4eb329474989c6ccfa3d6303";
689#endif // defined(OS_WIN)
690#else
691#if defined(OS_WIN)
692 const char kChangedChecksum[] = "3137fdb27962671f5c3963a5e965eff5";
Lei Zhang0c03d632020-07-30 17:05:36 +0000693#elif defined(OS_APPLE)
Hui Yingstcb4203d2020-06-09 01:48:47 +0000694 const char kChangedChecksum[] = "904132275a1144ea06b0694537c80b4c";
695#else
696 const char kChangedChecksum[] = "a0c4ea6620772991f66bf7130379b08a";
Henrique Nakashima5ebfd642018-06-07 15:18:55 +0000697#endif
Hui Yingstcb4203d2020-06-09 01:48:47 +0000698#endif // defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
Henrique Nakashima5ebfd642018-06-07 15:18:55 +0000699 {
Lei Zhang30ff2532019-01-31 21:37:55 +0000700 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstcb4203d2020-06-09 01:48:47 +0000701 CompareBitmap(page_bitmap.get(), 200, 200, kChangedChecksum);
Henrique Nakashima5ebfd642018-06-07 15:18:55 +0000702 }
703
704 // Now save the result.
705 EXPECT_TRUE(FPDFPage_GenerateContent(page));
706 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
707
708 UnloadPage(page);
709
710 // Re-open the file and check the changes were kept in the saved .pdf.
Lei Zhang0b494052019-01-31 21:41:15 +0000711 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashima5ebfd642018-06-07 15:18:55 +0000712 FPDF_PAGE saved_page = LoadSavedPage(0);
713 EXPECT_EQ(2, FPDFPage_CountObjects(saved_page));
714 {
Lei Zhang30ff2532019-01-31 21:37:55 +0000715 ScopedFPDFBitmap page_bitmap = RenderPage(saved_page);
Hui Yingstcb4203d2020-06-09 01:48:47 +0000716 CompareBitmap(page_bitmap.get(), 200, 200, kChangedChecksum);
Henrique Nakashima5ebfd642018-06-07 15:18:55 +0000717 }
718
719 CloseSavedPage(saved_page);
720 CloseSavedDocument();
721}
722
Hui Yingst5ce6a3f2020-11-11 02:29:09 +0000723TEST_F(FPDFEditEmbedderTest, SetTextKeepClippingPath) {
Daniel Hosseinianda651442020-07-17 23:30:22 +0000724 // Load document with some text, with parts clipped.
725 ASSERT_TRUE(OpenDocument("bug_1558.pdf"));
726 FPDF_PAGE page = LoadPage(0);
727 ASSERT_TRUE(page);
728
Hui Yingst5ce6a3f2020-11-11 02:29:09 +0000729#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
Daniel Hosseinianda651442020-07-17 23:30:22 +0000730 static constexpr char kOriginalChecksum[] =
Hui Yingst5ce6a3f2020-11-11 02:29:09 +0000731 "1e08d555f4863ff34a90f849c9464ed2";
732#else
Daniel Hosseinianda651442020-07-17 23:30:22 +0000733#if defined(OS_WIN)
Hui Yingst5ce6a3f2020-11-11 02:29:09 +0000734 static constexpr char kOriginalChecksum[] =
Daniel Hosseinianda651442020-07-17 23:30:22 +0000735 "220bf2086398fc46ac094952b244c8d9";
Lei Zhang0c03d632020-07-30 17:05:36 +0000736#elif defined(OS_APPLE)
Hui Yingst5ce6a3f2020-11-11 02:29:09 +0000737 static constexpr char kOriginalChecksum[] =
Daniel Hosseinianda651442020-07-17 23:30:22 +0000738 "53cbaad93551ef2ccc27ddd63f2ca2b3";
739#else
Hui Yingst5ce6a3f2020-11-11 02:29:09 +0000740 static constexpr char kOriginalChecksum[] =
Daniel Hosseinianda651442020-07-17 23:30:22 +0000741 "ba1936fa8ca1e8cca108da76ff3500a6";
742#endif
Hui Yingst5ce6a3f2020-11-11 02:29:09 +0000743#endif // defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
Daniel Hosseinianda651442020-07-17 23:30:22 +0000744 {
745 // When opened before any editing and saving, the clipping path is rendered.
746 ScopedFPDFBitmap original_bitmap = RenderPage(page);
747 CompareBitmap(original_bitmap.get(), 200, 200, kOriginalChecksum);
748 }
749
750 // "Change" the text in the objects to their current values to force them to
751 // regenerate when saving.
752 {
753 ScopedFPDFTextPage text_page(FPDFText_LoadPage(page));
754 ASSERT_TRUE(text_page);
755 const int obj_count = FPDFPage_CountObjects(page);
756 ASSERT_EQ(2, obj_count);
757 for (int i = 0; i < obj_count; ++i) {
758 FPDF_PAGEOBJECT text_obj = FPDFPage_GetObject(page, i);
759 ASSERT_EQ(FPDF_PAGEOBJ_TEXT, FPDFPageObj_GetType(text_obj));
760 unsigned long size =
761 FPDFTextObj_GetText(text_obj, text_page.get(),
762 /*buffer=*/nullptr, /*length=*/0);
763 ASSERT_GT(size, 0u);
764 std::vector<FPDF_WCHAR> buffer = GetFPDFWideStringBuffer(size);
765 ASSERT_EQ(size, FPDFTextObj_GetText(text_obj, text_page.get(),
766 buffer.data(), size));
767 EXPECT_TRUE(FPDFText_SetText(text_obj, buffer.data()));
768 }
769 }
770
771 {
772 // After editing but before saving, the clipping path is retained.
773 ScopedFPDFBitmap edited_bitmap = RenderPage(page);
774 CompareBitmap(edited_bitmap.get(), 200, 200, kOriginalChecksum);
775 }
776
777 // Save the file.
778 EXPECT_TRUE(FPDFPage_GenerateContent(page));
779 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
780 UnloadPage(page);
781
782 // Open the saved copy and render it.
783 ASSERT_TRUE(OpenSavedDocument());
784 FPDF_PAGE saved_page = LoadSavedPage(0);
785 ASSERT_TRUE(saved_page);
786
787 {
Daniel Hosseinianda651442020-07-17 23:30:22 +0000788 ScopedFPDFBitmap saved_bitmap = RenderSavedPage(saved_page);
Daniel Hosseinian75efd912020-07-21 08:07:28 +0000789 CompareBitmap(saved_bitmap.get(), 200, 200, kOriginalChecksum);
Daniel Hosseinianda651442020-07-17 23:30:22 +0000790 }
791
792 CloseSavedPage(saved_page);
793 CloseSavedDocument();
794}
795
Hui Yingst08c40712020-04-29 01:37:35 +0000796TEST_F(FPDFEditEmbedderTest, RemovePageObject) {
Henrique Nakashima35841fa2018-03-15 15:25:16 +0000797 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +0000798 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
Henrique Nakashima35841fa2018-03-15 15:25:16 +0000799 FPDF_PAGE page = LoadPage(0);
800 ASSERT_TRUE(page);
801
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000802 // Show what the original file looks like.
803 {
Lei Zhang30ff2532019-01-31 21:37:55 +0000804 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstb4baceb2020-04-28 23:46:10 +0000805 CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldChecksum);
Henrique Nakashima35841fa2018-03-15 15:25:16 +0000806 }
807
808 // Get the "Hello, world!" text object and remove it.
809 ASSERT_EQ(2, FPDFPage_CountObjects(page));
810 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0);
811 ASSERT_TRUE(page_object);
812 EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object));
813
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000814 // Verify the "Hello, world!" text is gone.
815 {
Lei Zhang30ff2532019-01-31 21:37:55 +0000816 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstb4baceb2020-04-28 23:46:10 +0000817 CompareBitmap(page_bitmap.get(), 200, 200, kFirstRemovedChecksum);
Henrique Nakashima35841fa2018-03-15 15:25:16 +0000818 }
819 ASSERT_EQ(1, FPDFPage_CountObjects(page));
820
821 UnloadPage(page);
822 FPDFPageObj_Destroy(page_object);
823}
824
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000825void CheckMarkCounts(FPDF_PAGE page,
826 int start_from,
827 int expected_object_count,
828 size_t expected_prime_count,
829 size_t expected_square_count,
830 size_t expected_greater_than_ten_count,
831 size_t expected_bounds_count) {
832 int object_count = FPDFPage_CountObjects(page);
833 ASSERT_EQ(expected_object_count, object_count);
834
835 size_t prime_count = 0;
836 size_t square_count = 0;
837 size_t greater_than_ten_count = 0;
838 size_t bounds_count = 0;
839 for (int i = 0; i < object_count; ++i) {
840 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i);
841
842 int mark_count = FPDFPageObj_CountMarks(page_object);
843 for (int j = 0; j < mark_count; ++j) {
844 FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, j);
845
846 char buffer[256];
Henrique Nakashimac3099d12018-09-18 18:08:15 +0000847 unsigned long name_len = 999u;
848 ASSERT_TRUE(
849 FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len));
850 EXPECT_GT(name_len, 0u);
851 EXPECT_NE(999u, name_len);
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000852 std::wstring name =
853 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer));
854 if (name == L"Prime") {
855 prime_count++;
856 } else if (name == L"Square") {
857 square_count++;
858 int expected_square = start_from + i;
859 EXPECT_EQ(1, FPDFPageObjMark_CountParams(mark));
860
Henrique Nakashimac3099d12018-09-18 18:08:15 +0000861 unsigned long get_param_key_return = 999u;
862 ASSERT_TRUE(FPDFPageObjMark_GetParamKey(mark, 0, buffer, sizeof(buffer),
863 &get_param_key_return));
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000864 EXPECT_EQ((6u + 1u) * 2u, get_param_key_return);
865 std::wstring key =
866 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer));
867 EXPECT_EQ(L"Factor", key);
868
869 EXPECT_EQ(FPDF_OBJECT_NUMBER,
Henrique Nakashima94230e52018-07-11 22:02:02 +0000870 FPDFPageObjMark_GetParamValueType(mark, "Factor"));
Henrique Nakashima140dead2018-07-11 21:40:03 +0000871 int square_root;
Henrique Nakashima94230e52018-07-11 22:02:02 +0000872 EXPECT_TRUE(
873 FPDFPageObjMark_GetParamIntValue(mark, "Factor", &square_root));
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000874 EXPECT_EQ(expected_square, square_root * square_root);
875 } else if (name == L"GreaterThanTen") {
876 greater_than_ten_count++;
877 } else if (name == L"Bounds") {
878 bounds_count++;
879 EXPECT_EQ(1, FPDFPageObjMark_CountParams(mark));
880
Henrique Nakashimac3099d12018-09-18 18:08:15 +0000881 unsigned long get_param_key_return = 999u;
882 ASSERT_TRUE(FPDFPageObjMark_GetParamKey(mark, 0, buffer, sizeof(buffer),
883 &get_param_key_return));
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000884 EXPECT_EQ((8u + 1u) * 2u, get_param_key_return);
885 std::wstring key =
886 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer));
887 EXPECT_EQ(L"Position", key);
888
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000889 EXPECT_EQ(FPDF_OBJECT_STRING,
Henrique Nakashima94230e52018-07-11 22:02:02 +0000890 FPDFPageObjMark_GetParamValueType(mark, "Position"));
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000891 unsigned long length;
Henrique Nakashimaa3406772018-07-13 19:10:53 +0000892 EXPECT_TRUE(FPDFPageObjMark_GetParamStringValue(
893 mark, "Position", buffer, sizeof(buffer), &length));
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000894 ASSERT_GT(length, 0u);
Henrique Nakashima140dead2018-07-11 21:40:03 +0000895 std::wstring value =
896 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer));
Henrique Nakashimad8df8c32018-07-12 22:15:09 +0000897
Henrique Nakashimaa3406772018-07-13 19:10:53 +0000898 // "Position" can be "First", "Last", or "End".
Henrique Nakashimad8df8c32018-07-12 22:15:09 +0000899 if (i == 0) {
900 EXPECT_EQ((5u + 1u) * 2u, length);
901 EXPECT_EQ(L"First", value);
902 } else if (i == object_count - 1) {
Henrique Nakashimaa3406772018-07-13 19:10:53 +0000903 if (length == (4u + 1u) * 2u) {
904 EXPECT_EQ(L"Last", value);
905 } else if (length == (3u + 1u) * 2u) {
906 EXPECT_EQ(L"End", value);
907 } else {
908 FAIL();
909 }
Henrique Nakashimad8df8c32018-07-12 22:15:09 +0000910 } else {
911 FAIL();
912 }
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000913 } else {
914 FAIL();
915 }
916 }
917 }
918
919 // Expect certain number of tagged objects. The test file contains strings
920 // from 1 to 19.
921 EXPECT_EQ(expected_prime_count, prime_count);
922 EXPECT_EQ(expected_square_count, square_count);
923 EXPECT_EQ(expected_greater_than_ten_count, greater_than_ten_count);
924 EXPECT_EQ(expected_bounds_count, bounds_count);
925}
926
Lei Zhangab41f252018-12-23 03:10:50 +0000927TEST_F(FPDFEditEmbedderTest, ReadMarkedObjectsIndirectDict) {
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000928 // Load document with some text marked with an indirect property.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +0000929 ASSERT_TRUE(OpenDocument("text_in_page_marked_indirect.pdf"));
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000930 FPDF_PAGE page = LoadPage(0);
931 ASSERT_TRUE(page);
932
933 CheckMarkCounts(page, 1, 19, 8, 4, 9, 1);
934
935 UnloadPage(page);
936}
937
Hui Yingstf26b5f82020-11-17 05:26:03 +0000938TEST_F(FPDFEditEmbedderTest, RemoveMarkedObjectsPrime) {
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000939 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +0000940 ASSERT_TRUE(OpenDocument("text_in_page_marked.pdf"));
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000941 FPDF_PAGE page = LoadPage(0);
942 ASSERT_TRUE(page);
943
944 // Show what the original file looks like.
945 {
Hui Yingstf26b5f82020-11-17 05:26:03 +0000946#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
947 static constexpr char kOriginalChecksum[] =
948 "748ed321a485d246ca6260b9e30dd200";
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000949#else
Hui Yingstf26b5f82020-11-17 05:26:03 +0000950#if defined(OS_WIN)
951 static constexpr char kOriginalChecksum[] =
952 "00542ee435b37749c4453be63bf7bdb6";
953#elif defined(OS_APPLE)
954 static constexpr char kOriginalChecksum[] =
955 "adf815e53c788a5272b4df07c610a1da";
956#else
957 static constexpr char kOriginalChecksum[] =
958 "41647268d5911d049801803b15c2dfb0";
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000959#endif
Hui Yingstf26b5f82020-11-17 05:26:03 +0000960#endif // defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
Lei Zhang30ff2532019-01-31 21:37:55 +0000961 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstf26b5f82020-11-17 05:26:03 +0000962 CompareBitmap(page_bitmap.get(), 200, 200, kOriginalChecksum);
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000963 }
964
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000965 constexpr int expected_object_count = 19;
966 CheckMarkCounts(page, 1, expected_object_count, 8, 4, 9, 1);
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000967
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000968 // Get all objects marked with "Prime"
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000969 std::vector<FPDF_PAGEOBJECT> primes;
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000970 for (int i = 0; i < expected_object_count; ++i) {
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000971 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i);
972
973 int mark_count = FPDFPageObj_CountMarks(page_object);
974 for (int j = 0; j < mark_count; ++j) {
975 FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, j);
976
977 char buffer[256];
Henrique Nakashimac3099d12018-09-18 18:08:15 +0000978 unsigned long name_len = 999u;
979 ASSERT_TRUE(
980 FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len));
981 EXPECT_GT(name_len, 0u);
982 EXPECT_NE(999u, name_len);
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000983 std::wstring name =
984 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer));
985 if (name == L"Prime") {
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000986 primes.push_back(page_object);
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000987 }
988 }
989 }
990
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000991 // Remove all objects marked with "Prime".
992 for (FPDF_PAGEOBJECT page_object : primes) {
993 EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object));
994 FPDFPageObj_Destroy(page_object);
995 }
996
997 EXPECT_EQ(11, FPDFPage_CountObjects(page));
Hui Yingstf26b5f82020-11-17 05:26:03 +0000998#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
999 static constexpr char kNonPrimesChecksum[] =
1000 "e2927fe2b7bbb595aca2a0e19ef3f1e8";
1001 static constexpr char kNonPrimesAfterSaveChecksum[] =
1002 "e2927fe2b7bbb595aca2a0e19ef3f1e8";
Henrique Nakashimac90adc52018-03-27 16:26:44 +00001003#else
Hui Yingstf26b5f82020-11-17 05:26:03 +00001004#if defined(OS_WIN)
1005 static constexpr char kNonPrimesChecksum[] =
1006 "86e371fdae30c2471f476631f3f93413";
1007 static constexpr char kNonPrimesAfterSaveChecksum[] =
1008 "86e371fdae30c2471f476631f3f93413";
1009#elif defined(OS_APPLE)
1010 static constexpr char kNonPrimesChecksum[] =
1011 "d29e2ddff56e0d12f340794d26796400";
1012 static constexpr char kNonPrimesAfterSaveChecksum[] =
1013 "10eff2cd0037b661496981779601fa6f";
1014#else
1015 static constexpr char kNonPrimesChecksum[] =
1016 "67ab13115d0cc34e99a1003c28047b40";
1017 static constexpr char kNonPrimesAfterSaveChecksum[] =
1018 "67ab13115d0cc34e99a1003c28047b40";
Henrique Nakashimac90adc52018-03-27 16:26:44 +00001019#endif
Hui Yingstf26b5f82020-11-17 05:26:03 +00001020#endif // defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
Henrique Nakashimaa5078b72018-09-10 17:09:42 +00001021 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001022 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstf26b5f82020-11-17 05:26:03 +00001023 CompareBitmap(page_bitmap.get(), 200, 200, kNonPrimesChecksum);
Henrique Nakashimac90adc52018-03-27 16:26:44 +00001024 }
1025
Henrique Nakashimaa5078b72018-09-10 17:09:42 +00001026 // Save the file.
1027 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1028 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
Henrique Nakashimac90adc52018-03-27 16:26:44 +00001029 UnloadPage(page);
Henrique Nakashimaa5078b72018-09-10 17:09:42 +00001030
1031 // Re-open the file and check the prime marks are not there anymore.
Lei Zhang0b494052019-01-31 21:41:15 +00001032 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashimaa5078b72018-09-10 17:09:42 +00001033 FPDF_PAGE saved_page = LoadSavedPage(0);
1034 EXPECT_EQ(11, FPDFPage_CountObjects(saved_page));
1035
1036 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001037 ScopedFPDFBitmap page_bitmap = RenderPage(saved_page);
Hui Yingstf26b5f82020-11-17 05:26:03 +00001038 CompareBitmap(page_bitmap.get(), 200, 200, kNonPrimesAfterSaveChecksum);
Henrique Nakashimaa5078b72018-09-10 17:09:42 +00001039 }
1040
1041 CloseSavedPage(saved_page);
1042 CloseSavedDocument();
Henrique Nakashimac90adc52018-03-27 16:26:44 +00001043}
1044
Lei Zhangab41f252018-12-23 03:10:50 +00001045TEST_F(FPDFEditEmbedderTest, RemoveMarks) {
Henrique Nakashimafed4adb2018-07-13 19:47:22 +00001046 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001047 ASSERT_TRUE(OpenDocument("text_in_page_marked.pdf"));
Henrique Nakashimafed4adb2018-07-13 19:47:22 +00001048 FPDF_PAGE page = LoadPage(0);
1049 ASSERT_TRUE(page);
1050
1051 constexpr int kExpectedObjectCount = 19;
1052 CheckMarkCounts(page, 1, kExpectedObjectCount, 8, 4, 9, 1);
1053
1054 // Remove all "Prime" content marks.
1055 for (int i = 0; i < kExpectedObjectCount; ++i) {
1056 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i);
1057
1058 int mark_count = FPDFPageObj_CountMarks(page_object);
1059 for (int j = mark_count - 1; j >= 0; --j) {
1060 FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, j);
1061
1062 char buffer[256];
Henrique Nakashimac3099d12018-09-18 18:08:15 +00001063 unsigned long name_len = 999u;
1064 ASSERT_TRUE(
1065 FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len));
1066 EXPECT_GT(name_len, 0u);
1067 EXPECT_NE(999u, name_len);
Henrique Nakashimafed4adb2018-07-13 19:47:22 +00001068 std::wstring name =
1069 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer));
1070 if (name == L"Prime") {
1071 // Remove mark.
1072 EXPECT_TRUE(FPDFPageObj_RemoveMark(page_object, mark));
1073
1074 // Verify there is now one fewer mark in the page object.
1075 EXPECT_EQ(mark_count - 1, FPDFPageObj_CountMarks(page_object));
1076 }
1077 }
1078 }
1079
1080 // Verify there are 0 "Prime" content marks now.
1081 CheckMarkCounts(page, 1, kExpectedObjectCount, 0, 4, 9, 1);
1082
1083 // Save the file.
1084 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1085 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1086 UnloadPage(page);
1087
1088 // Re-open the file and check the prime marks are not there anymore.
Lei Zhang0b494052019-01-31 21:41:15 +00001089 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashimafed4adb2018-07-13 19:47:22 +00001090 FPDF_PAGE saved_page = LoadSavedPage(0);
1091
1092 CheckMarkCounts(saved_page, 1, kExpectedObjectCount, 0, 4, 9, 1);
1093
1094 CloseSavedPage(saved_page);
1095 CloseSavedDocument();
1096}
1097
Lei Zhangab41f252018-12-23 03:10:50 +00001098TEST_F(FPDFEditEmbedderTest, RemoveMarkParam) {
Henrique Nakashimacf403ba2018-07-13 20:12:41 +00001099 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001100 ASSERT_TRUE(OpenDocument("text_in_page_marked.pdf"));
Henrique Nakashimacf403ba2018-07-13 20:12:41 +00001101 FPDF_PAGE page = LoadPage(0);
1102 ASSERT_TRUE(page);
1103
1104 constexpr int kExpectedObjectCount = 19;
1105 CheckMarkCounts(page, 1, kExpectedObjectCount, 8, 4, 9, 1);
1106
1107 // Remove all "Square" content marks parameters.
1108 for (int i = 0; i < kExpectedObjectCount; ++i) {
1109 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i);
1110
1111 int mark_count = FPDFPageObj_CountMarks(page_object);
1112 for (int j = 0; j < mark_count; ++j) {
1113 FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, j);
1114
1115 char buffer[256];
Henrique Nakashimac3099d12018-09-18 18:08:15 +00001116 unsigned long name_len = 999u;
1117 ASSERT_TRUE(
1118 FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len));
1119 EXPECT_GT(name_len, 0u);
1120 EXPECT_NE(999u, name_len);
Henrique Nakashimacf403ba2018-07-13 20:12:41 +00001121 std::wstring name =
1122 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer));
1123 if (name == L"Square") {
1124 // Show the mark has a "Factor" parameter.
1125 int out_value;
1126 EXPECT_TRUE(
1127 FPDFPageObjMark_GetParamIntValue(mark, "Factor", &out_value));
1128
1129 // Remove parameter.
1130 EXPECT_TRUE(FPDFPageObjMark_RemoveParam(page_object, mark, "Factor"));
1131
1132 // Verify the "Factor" parameter is gone.
1133 EXPECT_FALSE(
1134 FPDFPageObjMark_GetParamIntValue(mark, "Factor", &out_value));
1135 }
1136 }
1137 }
1138
1139 // Save the file.
1140 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1141 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1142 UnloadPage(page);
1143
1144 // Re-open the file and check the "Factor" parameters are still gone.
Lei Zhang0b494052019-01-31 21:41:15 +00001145 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashimacf403ba2018-07-13 20:12:41 +00001146 FPDF_PAGE saved_page = LoadSavedPage(0);
1147
1148 size_t square_count = 0;
1149 for (int i = 0; i < kExpectedObjectCount; ++i) {
1150 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(saved_page, i);
1151
1152 int mark_count = FPDFPageObj_CountMarks(page_object);
1153 for (int j = 0; j < mark_count; ++j) {
1154 FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, j);
1155
1156 char buffer[256];
Henrique Nakashimac3099d12018-09-18 18:08:15 +00001157 unsigned long name_len = 999u;
1158 ASSERT_TRUE(
1159 FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len));
1160 EXPECT_GT(name_len, 0u);
1161 EXPECT_NE(999u, name_len);
Henrique Nakashimacf403ba2018-07-13 20:12:41 +00001162 std::wstring name =
1163 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer));
1164 if (name == L"Square") {
1165 // Verify the "Factor" parameter is still gone.
1166 int out_value;
1167 EXPECT_FALSE(
1168 FPDFPageObjMark_GetParamIntValue(mark, "Factor", &out_value));
1169
1170 ++square_count;
1171 }
1172 }
1173 }
1174
1175 // Verify the parameters are gone, but the marks are not.
1176 EXPECT_EQ(4u, square_count);
1177
1178 CloseSavedPage(saved_page);
1179 CloseSavedDocument();
1180}
1181
Lei Zhangab41f252018-12-23 03:10:50 +00001182TEST_F(FPDFEditEmbedderTest, MaintainMarkedObjects) {
Henrique Nakashimab4bcf692018-07-11 21:19:22 +00001183 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001184 ASSERT_TRUE(OpenDocument("text_in_page_marked.pdf"));
Henrique Nakashimab4bcf692018-07-11 21:19:22 +00001185 FPDF_PAGE page = LoadPage(0);
1186 ASSERT_TRUE(page);
1187
1188 // Iterate over all objects, counting the number of times each content mark
1189 // name appears.
1190 CheckMarkCounts(page, 1, 19, 8, 4, 9, 1);
1191
1192 // Remove first page object.
1193 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0);
1194 EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object));
1195 FPDFPageObj_Destroy(page_object);
1196
1197 CheckMarkCounts(page, 2, 18, 8, 3, 9, 1);
1198
1199 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1200 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1201
1202 UnloadPage(page);
1203
Lei Zhang0b494052019-01-31 21:41:15 +00001204 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashimab4bcf692018-07-11 21:19:22 +00001205 FPDF_PAGE saved_page = LoadSavedPage(0);
1206
1207 CheckMarkCounts(saved_page, 2, 18, 8, 3, 9, 1);
1208
1209 CloseSavedPage(saved_page);
1210 CloseSavedDocument();
1211}
1212
Lei Zhangab41f252018-12-23 03:10:50 +00001213TEST_F(FPDFEditEmbedderTest, MaintainIndirectMarkedObjects) {
Henrique Nakashimab4bcf692018-07-11 21:19:22 +00001214 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001215 ASSERT_TRUE(OpenDocument("text_in_page_marked_indirect.pdf"));
Henrique Nakashimab4bcf692018-07-11 21:19:22 +00001216 FPDF_PAGE page = LoadPage(0);
1217 ASSERT_TRUE(page);
1218
1219 // Iterate over all objects, counting the number of times each content mark
1220 // name appears.
1221 CheckMarkCounts(page, 1, 19, 8, 4, 9, 1);
1222
1223 // Remove first page object.
1224 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0);
1225 EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object));
1226 FPDFPageObj_Destroy(page_object);
1227
1228 CheckMarkCounts(page, 2, 18, 8, 3, 9, 1);
1229
1230 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1231 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1232
1233 UnloadPage(page);
1234
Lei Zhang0b494052019-01-31 21:41:15 +00001235 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashimab4bcf692018-07-11 21:19:22 +00001236 FPDF_PAGE saved_page = LoadSavedPage(0);
1237
1238 CheckMarkCounts(saved_page, 2, 18, 8, 3, 9, 1);
1239
1240 CloseSavedPage(saved_page);
1241 CloseSavedDocument();
1242}
1243
Lei Zhangab41f252018-12-23 03:10:50 +00001244TEST_F(FPDFEditEmbedderTest, RemoveExistingPageObject) {
Henrique Nakashimac49e62e2018-04-16 20:58:47 +00001245 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001246 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
Henrique Nakashimac49e62e2018-04-16 20:58:47 +00001247 FPDF_PAGE page = LoadPage(0);
1248 ASSERT_TRUE(page);
1249
1250 // Get the "Hello, world!" text object and remove it.
1251 ASSERT_EQ(2, FPDFPage_CountObjects(page));
1252 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0);
1253 ASSERT_TRUE(page_object);
1254 EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object));
1255
1256 // Verify the "Hello, world!" text is gone.
1257 ASSERT_EQ(1, FPDFPage_CountObjects(page));
1258
1259 // Save the file
1260 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1261 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1262 UnloadPage(page);
1263 FPDFPageObj_Destroy(page_object);
1264
1265 // Re-open the file and check the page object count is still 1.
Lei Zhang0b494052019-01-31 21:41:15 +00001266 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashimac49e62e2018-04-16 20:58:47 +00001267 FPDF_PAGE saved_page = LoadSavedPage(0);
1268 EXPECT_EQ(1, FPDFPage_CountObjects(saved_page));
1269 CloseSavedPage(saved_page);
1270 CloseSavedDocument();
1271}
1272
Hui Yingstcb4203d2020-06-09 01:48:47 +00001273TEST_F(FPDFEditEmbedderTest, RemoveExistingPageObjectSplitStreamsNotLonely) {
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001274 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001275 ASSERT_TRUE(OpenDocument("hello_world_split_streams.pdf"));
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001276 FPDF_PAGE page = LoadPage(0);
1277 ASSERT_TRUE(page);
1278
1279 // Get the "Hello, world!" text object and remove it. There is another object
1280 // in the same stream that says "Goodbye, world!"
1281 ASSERT_EQ(3, FPDFPage_CountObjects(page));
1282 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0);
1283 ASSERT_TRUE(page_object);
1284 EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object));
1285
1286 // Verify the "Hello, world!" text is gone.
1287 ASSERT_EQ(2, FPDFPage_CountObjects(page));
Hui Yingstcb4203d2020-06-09 01:48:47 +00001288#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
1289#if defined(OS_WIN)
1290 const char kHelloRemovedChecksum[] = "e05c7837a5cad61305d157720637f969";
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001291#else
Hui Yingstcb4203d2020-06-09 01:48:47 +00001292 const char kHelloRemovedChecksum[] = "deed7dc2754dc80930f3b05e2ac86c94";
1293#endif // defined(OS_WIN)
1294#else
1295#if defined(OS_WIN)
1296 const char kHelloRemovedChecksum[] = "a97d4c72c969ba373c2dce675d277e65";
Lei Zhang0c03d632020-07-30 17:05:36 +00001297#elif defined(OS_APPLE)
Hui Yingstcb4203d2020-06-09 01:48:47 +00001298 const char kHelloRemovedChecksum[] = "3b3b27602a86dfe5996a33c42c59885b";
1299#else
1300 const char kHelloRemovedChecksum[] = "95b92950647a2190e1230911e7a1a0e9";
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001301#endif
Hui Yingstcb4203d2020-06-09 01:48:47 +00001302#endif // defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001303 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001304 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstcb4203d2020-06-09 01:48:47 +00001305 CompareBitmap(page_bitmap.get(), 200, 200, kHelloRemovedChecksum);
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001306 }
1307
1308 // Save the file
1309 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1310 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1311 UnloadPage(page);
1312 FPDFPageObj_Destroy(page_object);
1313
1314 // Re-open the file and check the page object count is still 2.
Lei Zhang0b494052019-01-31 21:41:15 +00001315 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001316 FPDF_PAGE saved_page = LoadSavedPage(0);
1317
1318 EXPECT_EQ(2, FPDFPage_CountObjects(saved_page));
1319 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001320 ScopedFPDFBitmap page_bitmap = RenderPage(saved_page);
Hui Yingstcb4203d2020-06-09 01:48:47 +00001321 CompareBitmap(page_bitmap.get(), 200, 200, kHelloRemovedChecksum);
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001322 }
1323
1324 CloseSavedPage(saved_page);
1325 CloseSavedDocument();
1326}
1327
Hui Yingst08c40712020-04-29 01:37:35 +00001328TEST_F(FPDFEditEmbedderTest, RemoveExistingPageObjectSplitStreamsLonely) {
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001329 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001330 ASSERT_TRUE(OpenDocument("hello_world_split_streams.pdf"));
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001331 FPDF_PAGE page = LoadPage(0);
1332 ASSERT_TRUE(page);
1333
1334 // Get the "Greetings, world!" text object and remove it. This is the only
1335 // object in the stream.
1336 ASSERT_EQ(3, FPDFPage_CountObjects(page));
1337 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 2);
1338 ASSERT_TRUE(page_object);
1339 EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object));
1340
1341 // Verify the "Greetings, world!" text is gone.
1342 ASSERT_EQ(2, FPDFPage_CountObjects(page));
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001343 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001344 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00001345 CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldChecksum);
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001346 }
1347
1348 // Save the file
1349 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1350 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1351 UnloadPage(page);
1352 FPDFPageObj_Destroy(page_object);
1353
1354 // Re-open the file and check the page object count is still 2.
Lei Zhang0b494052019-01-31 21:41:15 +00001355 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001356 FPDF_PAGE saved_page = LoadSavedPage(0);
1357
1358 EXPECT_EQ(2, FPDFPage_CountObjects(saved_page));
1359 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001360 ScopedFPDFBitmap page_bitmap = RenderPage(saved_page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00001361 CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldChecksum);
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001362 }
1363
1364 CloseSavedPage(saved_page);
1365 CloseSavedDocument();
1366}
1367
Lei Zhangab41f252018-12-23 03:10:50 +00001368TEST_F(FPDFEditEmbedderTest, GetContentStream) {
Henrique Nakashima6eb79392018-06-12 20:27:35 +00001369 // Load document with some text split across streams.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001370 ASSERT_TRUE(OpenDocument("split_streams.pdf"));
Henrique Nakashima6eb79392018-06-12 20:27:35 +00001371 FPDF_PAGE page = LoadPage(0);
1372 ASSERT_TRUE(page);
1373
1374 // Content stream 0: page objects 0-14.
1375 // Content stream 1: page objects 15-17.
1376 // Content stream 2: page object 18.
1377 ASSERT_EQ(19, FPDFPage_CountObjects(page));
1378 for (int i = 0; i < 19; i++) {
1379 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i);
1380 ASSERT_TRUE(page_object);
1381 CPDF_PageObject* cpdf_page_object =
1382 CPDFPageObjectFromFPDFPageObject(page_object);
1383 if (i < 15)
1384 EXPECT_EQ(0, cpdf_page_object->GetContentStream()) << i;
1385 else if (i < 18)
1386 EXPECT_EQ(1, cpdf_page_object->GetContentStream()) << i;
1387 else
1388 EXPECT_EQ(2, cpdf_page_object->GetContentStream()) << i;
1389 }
1390
1391 UnloadPage(page);
1392}
1393
Hui Yingst0d692fe2020-11-06 17:32:50 +00001394TEST_F(FPDFEditEmbedderTest, RemoveAllFromStream) {
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001395 // Load document with some text split across streams.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001396 ASSERT_TRUE(OpenDocument("split_streams.pdf"));
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001397 FPDF_PAGE page = LoadPage(0);
1398 ASSERT_TRUE(page);
1399
1400 // Content stream 0: page objects 0-14.
1401 // Content stream 1: page objects 15-17.
1402 // Content stream 2: page object 18.
1403 ASSERT_EQ(19, FPDFPage_CountObjects(page));
1404
1405 // Loop backwards because objects will being removed, which shifts the indexes
1406 // after the removed position.
1407 for (int i = 18; i >= 0; i--) {
1408 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i);
1409 ASSERT_TRUE(page_object);
1410 CPDF_PageObject* cpdf_page_object =
1411 CPDFPageObjectFromFPDFPageObject(page_object);
1412
1413 // Empty content stream 1.
1414 if (cpdf_page_object->GetContentStream() == 1) {
1415 EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object));
1416 FPDFPageObj_Destroy(page_object);
1417 }
1418 }
1419
1420 // Content stream 0: page objects 0-14.
1421 // Content stream 2: page object 15.
1422 ASSERT_EQ(16, FPDFPage_CountObjects(page));
1423 for (int i = 0; i < 16; i++) {
1424 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i);
1425 ASSERT_TRUE(page_object);
1426 CPDF_PageObject* cpdf_page_object =
1427 CPDFPageObjectFromFPDFPageObject(page_object);
1428 if (i < 15)
1429 EXPECT_EQ(0, cpdf_page_object->GetContentStream()) << i;
1430 else
1431 EXPECT_EQ(2, cpdf_page_object->GetContentStream()) << i;
1432 }
1433
1434 // Generate contents should remove the empty stream and update the page
1435 // objects' contents stream indexes.
1436 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1437
1438 // Content stream 0: page objects 0-14.
1439 // Content stream 1: page object 15.
1440 ASSERT_EQ(16, FPDFPage_CountObjects(page));
1441 for (int i = 0; i < 16; i++) {
1442 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i);
1443 ASSERT_TRUE(page_object);
1444 CPDF_PageObject* cpdf_page_object =
1445 CPDFPageObjectFromFPDFPageObject(page_object);
1446 if (i < 15)
1447 EXPECT_EQ(0, cpdf_page_object->GetContentStream()) << i;
1448 else
1449 EXPECT_EQ(1, cpdf_page_object->GetContentStream()) << i;
1450 }
1451
Hui Yingst0d692fe2020-11-06 17:32:50 +00001452#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
1453 const char kStream1RemovedChecksum[] = "89358c444a398b0b56b35738edd8fe43";
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001454#else
Hui Yingst0d692fe2020-11-06 17:32:50 +00001455#if defined(OS_WIN)
1456 const char kStream1RemovedChecksum[] = "b4140f203523e38793283a5943d8075b";
1457#elif defined(OS_APPLE)
1458 const char kStream1RemovedChecksum[] = "0e8856ca9abc7049412e64f9230c7c43";
1459#else
1460 const char kStream1RemovedChecksum[] = "e86a3efc160ede6cfcb1f59bcacf1105";
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001461#endif
Hui Yingst0d692fe2020-11-06 17:32:50 +00001462#endif // defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001463 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001464 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingst0d692fe2020-11-06 17:32:50 +00001465 CompareBitmap(page_bitmap.get(), 200, 200, kStream1RemovedChecksum);
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001466 }
1467
1468 // Save the file
1469 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1470 UnloadPage(page);
1471
1472 // Re-open the file and check the page object count is still 16, and that
1473 // content stream 1 was removed.
Lei Zhang0b494052019-01-31 21:41:15 +00001474 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001475 FPDF_PAGE saved_page = LoadSavedPage(0);
1476
1477 // Content stream 0: page objects 0-14.
1478 // Content stream 1: page object 15.
1479 EXPECT_EQ(16, FPDFPage_CountObjects(saved_page));
1480 for (int i = 0; i < 16; i++) {
1481 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(saved_page, i);
1482 ASSERT_TRUE(page_object);
1483 CPDF_PageObject* cpdf_page_object =
1484 CPDFPageObjectFromFPDFPageObject(page_object);
1485 if (i < 15)
1486 EXPECT_EQ(0, cpdf_page_object->GetContentStream()) << i;
1487 else
1488 EXPECT_EQ(1, cpdf_page_object->GetContentStream()) << i;
1489 }
1490
1491 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001492 ScopedFPDFBitmap page_bitmap = RenderPage(saved_page);
Hui Yingst0d692fe2020-11-06 17:32:50 +00001493 CompareBitmap(page_bitmap.get(), 200, 200, kStream1RemovedChecksum);
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001494 }
1495
1496 CloseSavedPage(saved_page);
1497 CloseSavedDocument();
1498}
1499
Lei Zhangab41f252018-12-23 03:10:50 +00001500TEST_F(FPDFEditEmbedderTest, RemoveAllFromSingleStream) {
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001501 // Load document with a single stream.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001502 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001503 FPDF_PAGE page = LoadPage(0);
1504 ASSERT_TRUE(page);
1505
1506 // Content stream 0: page objects 0-1.
1507 ASSERT_EQ(2, FPDFPage_CountObjects(page));
1508
1509 // Loop backwards because objects will being removed, which shifts the indexes
1510 // after the removed position.
1511 for (int i = 1; i >= 0; i--) {
1512 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i);
1513 ASSERT_TRUE(page_object);
1514 CPDF_PageObject* cpdf_page_object =
1515 CPDFPageObjectFromFPDFPageObject(page_object);
1516 ASSERT_EQ(0, cpdf_page_object->GetContentStream());
1517 ASSERT_TRUE(FPDFPage_RemoveObject(page, page_object));
1518 FPDFPageObj_Destroy(page_object);
1519 }
1520
1521 // No more objects in the stream
1522 ASSERT_EQ(0, FPDFPage_CountObjects(page));
1523
1524 // Generate contents should remove the empty stream and update the page
1525 // objects' contents stream indexes.
1526 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1527
1528 ASSERT_EQ(0, FPDFPage_CountObjects(page));
1529
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001530 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001531 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingst3b6136a2020-06-09 00:39:33 +00001532 CompareBitmap(page_bitmap.get(), 200, 200, kAllRemovedChecksum);
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001533 }
1534
1535 // Save the file
1536 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1537 UnloadPage(page);
1538
1539 // Re-open the file and check the page object count is still 0.
Lei Zhang0b494052019-01-31 21:41:15 +00001540 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001541 FPDF_PAGE saved_page = LoadSavedPage(0);
1542
1543 EXPECT_EQ(0, FPDFPage_CountObjects(saved_page));
1544 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001545 ScopedFPDFBitmap page_bitmap = RenderPage(saved_page);
Hui Yingst3b6136a2020-06-09 00:39:33 +00001546 CompareBitmap(page_bitmap.get(), 200, 200, kAllRemovedChecksum);
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001547 }
1548
1549 CloseSavedPage(saved_page);
1550 CloseSavedDocument();
1551}
1552
Hui Yingst08c40712020-04-29 01:37:35 +00001553TEST_F(FPDFEditEmbedderTest, RemoveFirstFromSingleStream) {
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001554 // Load document with a single stream.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001555 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001556 FPDF_PAGE page = LoadPage(0);
1557 ASSERT_TRUE(page);
1558
1559 // Content stream 0: page objects 0-1.
1560 ASSERT_EQ(2, FPDFPage_CountObjects(page));
1561
1562 // Remove first object.
1563 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0);
1564 ASSERT_TRUE(page_object);
1565 CPDF_PageObject* cpdf_page_object =
1566 CPDFPageObjectFromFPDFPageObject(page_object);
1567 ASSERT_EQ(0, cpdf_page_object->GetContentStream());
1568 ASSERT_TRUE(FPDFPage_RemoveObject(page, page_object));
1569 FPDFPageObj_Destroy(page_object);
1570
1571 // One object left in the stream.
1572 ASSERT_EQ(1, FPDFPage_CountObjects(page));
1573 page_object = FPDFPage_GetObject(page, 0);
1574 ASSERT_TRUE(page_object);
1575 cpdf_page_object = CPDFPageObjectFromFPDFPageObject(page_object);
1576 ASSERT_EQ(0, cpdf_page_object->GetContentStream());
1577
1578 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1579
1580 // Still one object left in the stream.
1581 ASSERT_EQ(1, FPDFPage_CountObjects(page));
1582 page_object = FPDFPage_GetObject(page, 0);
1583 ASSERT_TRUE(page_object);
1584 cpdf_page_object = CPDFPageObjectFromFPDFPageObject(page_object);
1585 ASSERT_EQ(0, cpdf_page_object->GetContentStream());
1586
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001587 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001588 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00001589 CompareBitmap(page_bitmap.get(), 200, 200, kFirstRemovedChecksum);
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001590 }
1591
1592 // Save the file
1593 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1594 UnloadPage(page);
1595
1596 // Re-open the file and check the page object count is still 0.
Lei Zhang0b494052019-01-31 21:41:15 +00001597 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001598 FPDF_PAGE saved_page = LoadSavedPage(0);
1599
1600 ASSERT_EQ(1, FPDFPage_CountObjects(saved_page));
1601 page_object = FPDFPage_GetObject(saved_page, 0);
1602 ASSERT_TRUE(page_object);
1603 cpdf_page_object = CPDFPageObjectFromFPDFPageObject(page_object);
1604 ASSERT_EQ(0, cpdf_page_object->GetContentStream());
1605 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001606 ScopedFPDFBitmap page_bitmap = RenderPage(saved_page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00001607 CompareBitmap(page_bitmap.get(), 200, 200, kFirstRemovedChecksum);
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001608 }
1609
1610 CloseSavedPage(saved_page);
1611 CloseSavedDocument();
1612}
1613
Hui Yingst08c40712020-04-29 01:37:35 +00001614TEST_F(FPDFEditEmbedderTest, RemoveLastFromSingleStream) {
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001615 // Load document with a single stream.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001616 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001617 FPDF_PAGE page = LoadPage(0);
1618 ASSERT_TRUE(page);
1619
1620 // Content stream 0: page objects 0-1.
1621 ASSERT_EQ(2, FPDFPage_CountObjects(page));
1622
1623 // Remove last object
1624 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 1);
1625 ASSERT_TRUE(page_object);
1626 CPDF_PageObject* cpdf_page_object =
1627 CPDFPageObjectFromFPDFPageObject(page_object);
1628 ASSERT_EQ(0, cpdf_page_object->GetContentStream());
1629 ASSERT_TRUE(FPDFPage_RemoveObject(page, page_object));
1630 FPDFPageObj_Destroy(page_object);
1631
1632 // One object left in the stream.
1633 ASSERT_EQ(1, FPDFPage_CountObjects(page));
1634 page_object = FPDFPage_GetObject(page, 0);
1635 ASSERT_TRUE(page_object);
1636 cpdf_page_object = CPDFPageObjectFromFPDFPageObject(page_object);
1637 ASSERT_EQ(0, cpdf_page_object->GetContentStream());
1638
1639 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1640
1641 // Still one object left in the stream.
1642 ASSERT_EQ(1, FPDFPage_CountObjects(page));
1643 page_object = FPDFPage_GetObject(page, 0);
1644 ASSERT_TRUE(page_object);
1645 cpdf_page_object = CPDFPageObjectFromFPDFPageObject(page_object);
1646 ASSERT_EQ(0, cpdf_page_object->GetContentStream());
1647
Hui Yingstb4baceb2020-04-28 23:46:10 +00001648 using pdfium::kHelloWorldRemovedChecksum;
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001649 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001650 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00001651 CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldRemovedChecksum);
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001652 }
1653
1654 // Save the file
1655 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1656 UnloadPage(page);
1657
1658 // Re-open the file and check the page object count is still 0.
Lei Zhang0b494052019-01-31 21:41:15 +00001659 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001660 FPDF_PAGE saved_page = LoadSavedPage(0);
1661
1662 ASSERT_EQ(1, FPDFPage_CountObjects(saved_page));
1663 page_object = FPDFPage_GetObject(saved_page, 0);
1664 ASSERT_TRUE(page_object);
1665 cpdf_page_object = CPDFPageObjectFromFPDFPageObject(page_object);
1666 ASSERT_EQ(0, cpdf_page_object->GetContentStream());
1667 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001668 ScopedFPDFBitmap page_bitmap = RenderPage(saved_page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00001669 CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldRemovedChecksum);
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001670 }
1671
1672 CloseSavedPage(saved_page);
1673 CloseSavedDocument();
1674}
1675
Lei Zhangab41f252018-12-23 03:10:50 +00001676TEST_F(FPDFEditEmbedderTest, RemoveAllFromMultipleStreams) {
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001677 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001678 ASSERT_TRUE(OpenDocument("hello_world_split_streams.pdf"));
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001679 FPDF_PAGE page = LoadPage(0);
1680 ASSERT_TRUE(page);
1681
1682 // Content stream 0: page objects 0-1.
1683 // Content stream 1: page object 2.
1684 ASSERT_EQ(3, FPDFPage_CountObjects(page));
1685
1686 // Loop backwards because objects will being removed, which shifts the indexes
1687 // after the removed position.
1688 for (int i = 2; i >= 0; i--) {
1689 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i);
1690 ASSERT_TRUE(page_object);
1691 ASSERT_TRUE(FPDFPage_RemoveObject(page, page_object));
1692 FPDFPageObj_Destroy(page_object);
1693 }
1694
1695 // No more objects in the page.
1696 ASSERT_EQ(0, FPDFPage_CountObjects(page));
1697
1698 // Generate contents should remove the empty streams and update the page
1699 // objects' contents stream indexes.
1700 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1701
1702 ASSERT_EQ(0, FPDFPage_CountObjects(page));
1703
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001704 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001705 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingst3b6136a2020-06-09 00:39:33 +00001706 CompareBitmap(page_bitmap.get(), 200, 200, kAllRemovedChecksum);
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001707 }
1708
1709 // Save the file
1710 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1711 UnloadPage(page);
1712
1713 // Re-open the file and check the page object count is still 0.
Lei Zhang0b494052019-01-31 21:41:15 +00001714 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001715 FPDF_PAGE saved_page = LoadSavedPage(0);
1716
1717 EXPECT_EQ(0, FPDFPage_CountObjects(saved_page));
1718 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001719 ScopedFPDFBitmap page_bitmap = RenderPage(saved_page);
Hui Yingst3b6136a2020-06-09 00:39:33 +00001720 CompareBitmap(page_bitmap.get(), 200, 200, kAllRemovedChecksum);
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001721 }
1722
1723 CloseSavedPage(saved_page);
1724 CloseSavedDocument();
1725}
1726
Lei Zhangab41f252018-12-23 03:10:50 +00001727TEST_F(FPDFEditEmbedderTest, InsertPageObjectAndSave) {
Henrique Nakashimac49e62e2018-04-16 20:58:47 +00001728 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001729 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
Henrique Nakashimac49e62e2018-04-16 20:58:47 +00001730 FPDF_PAGE page = LoadPage(0);
1731 ASSERT_TRUE(page);
1732
1733 // Add a red rectangle.
1734 ASSERT_EQ(2, FPDFPage_CountObjects(page));
1735 FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(20, 100, 50, 50);
Lei Zhang3475b482019-05-13 18:30:57 +00001736 EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 0, 0, 255));
Henrique Nakashimac49e62e2018-04-16 20:58:47 +00001737 EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0));
1738 FPDFPage_InsertObject(page, red_rect);
1739
1740 // Verify the red rectangle was added.
1741 ASSERT_EQ(3, FPDFPage_CountObjects(page));
1742
1743 // Save the file
1744 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1745 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1746 UnloadPage(page);
1747
1748 // Re-open the file and check the page object count is still 3.
Lei Zhang0b494052019-01-31 21:41:15 +00001749 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashimac49e62e2018-04-16 20:58:47 +00001750 FPDF_PAGE saved_page = LoadSavedPage(0);
1751 EXPECT_EQ(3, FPDFPage_CountObjects(saved_page));
1752 CloseSavedPage(saved_page);
1753 CloseSavedDocument();
1754}
1755
Lei Zhangab41f252018-12-23 03:10:50 +00001756TEST_F(FPDFEditEmbedderTest, InsertPageObjectEditAndSave) {
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001757 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001758 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001759 FPDF_PAGE page = LoadPage(0);
1760 ASSERT_TRUE(page);
1761
1762 // Add a red rectangle.
1763 ASSERT_EQ(2, FPDFPage_CountObjects(page));
1764 FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(20, 100, 50, 50);
Lei Zhang3475b482019-05-13 18:30:57 +00001765 EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 100, 100, 255));
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001766 EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0));
1767 FPDFPage_InsertObject(page, red_rect);
1768
1769 // Verify the red rectangle was added.
1770 ASSERT_EQ(3, FPDFPage_CountObjects(page));
1771
1772 // Generate content but change it again
1773 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Lei Zhang3475b482019-05-13 18:30:57 +00001774 EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 0, 0, 255));
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001775
1776 // Save the file
1777 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1778 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1779 UnloadPage(page);
1780
1781 // Re-open the file and check the page object count is still 3.
Lei Zhang0b494052019-01-31 21:41:15 +00001782 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001783 FPDF_PAGE saved_page = LoadSavedPage(0);
1784 EXPECT_EQ(3, FPDFPage_CountObjects(saved_page));
1785 CloseSavedPage(saved_page);
1786 CloseSavedDocument();
1787}
1788
Hui Yingst8b5dfb22020-07-21 17:08:30 +00001789TEST_F(FPDFEditEmbedderTest, InsertAndRemoveLargeFile) {
Henrique Nakashima657a1aa2018-09-12 16:19:22 +00001790 const int kOriginalObjectCount = 600;
1791
1792 // Load document with many objects.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001793 ASSERT_TRUE(OpenDocument("many_rectangles.pdf"));
Henrique Nakashima657a1aa2018-09-12 16:19:22 +00001794 FPDF_PAGE page = LoadPage(0);
1795 ASSERT_TRUE(page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00001796
1797 using pdfium::kManyRectanglesChecksum;
Henrique Nakashima657a1aa2018-09-12 16:19:22 +00001798 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001799 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00001800 CompareBitmap(page_bitmap.get(), 200, 300, kManyRectanglesChecksum);
Henrique Nakashima657a1aa2018-09-12 16:19:22 +00001801 }
1802
1803 // Add a black rectangle.
1804 ASSERT_EQ(kOriginalObjectCount, FPDFPage_CountObjects(page));
1805 FPDF_PAGEOBJECT black_rect = FPDFPageObj_CreateNewRect(20, 100, 50, 50);
Lei Zhang3475b482019-05-13 18:30:57 +00001806 EXPECT_TRUE(FPDFPageObj_SetFillColor(black_rect, 0, 0, 0, 255));
Henrique Nakashima657a1aa2018-09-12 16:19:22 +00001807 EXPECT_TRUE(FPDFPath_SetDrawMode(black_rect, FPDF_FILLMODE_ALTERNATE, 0));
1808 FPDFPage_InsertObject(page, black_rect);
1809
1810 // Verify the black rectangle was added.
1811 ASSERT_EQ(kOriginalObjectCount + 1, FPDFPage_CountObjects(page));
Hui Yingst8b5dfb22020-07-21 17:08:30 +00001812#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
1813 const char kPlusRectangleMD5[] = "0d3715fcfb9bd0dd25dcce60800bff47";
1814#else
Henrique Nakashima657a1aa2018-09-12 16:19:22 +00001815 const char kPlusRectangleMD5[] = "6b9396ab570754b32b04ca629e902f77";
Hui Yingst8b5dfb22020-07-21 17:08:30 +00001816#endif
Henrique Nakashima657a1aa2018-09-12 16:19:22 +00001817 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001818 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Henrique Nakashima657a1aa2018-09-12 16:19:22 +00001819 CompareBitmap(page_bitmap.get(), 200, 300, kPlusRectangleMD5);
1820 }
1821
1822 // Save the file.
1823 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1824 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1825 UnloadPage(page);
1826
1827 // Re-open the file and check the rectangle added is still there.
Lei Zhang0b494052019-01-31 21:41:15 +00001828 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashima657a1aa2018-09-12 16:19:22 +00001829 FPDF_PAGE saved_page = LoadSavedPage(0);
1830 EXPECT_EQ(kOriginalObjectCount + 1, FPDFPage_CountObjects(saved_page));
1831 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001832 ScopedFPDFBitmap page_bitmap = RenderPage(saved_page);
Henrique Nakashima657a1aa2018-09-12 16:19:22 +00001833 CompareBitmap(page_bitmap.get(), 200, 300, kPlusRectangleMD5);
1834 }
1835
1836 // Remove the added rectangle.
1837 FPDF_PAGEOBJECT added_object =
1838 FPDFPage_GetObject(saved_page, kOriginalObjectCount);
1839 EXPECT_TRUE(FPDFPage_RemoveObject(saved_page, added_object));
1840 FPDFPageObj_Destroy(added_object);
1841 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001842 ScopedFPDFBitmap page_bitmap = RenderPage(saved_page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00001843 CompareBitmap(page_bitmap.get(), 200, 300, kManyRectanglesChecksum);
Henrique Nakashima657a1aa2018-09-12 16:19:22 +00001844 }
1845 EXPECT_EQ(kOriginalObjectCount, FPDFPage_CountObjects(saved_page));
1846
1847 // Save the file again.
1848 EXPECT_TRUE(FPDFPage_GenerateContent(saved_page));
1849 EXPECT_TRUE(FPDF_SaveAsCopy(saved_document_, this, 0));
1850
1851 CloseSavedPage(saved_page);
1852 CloseSavedDocument();
1853
1854 // Re-open the file (again) and check the black rectangle was removed and the
1855 // rest is intact.
Lei Zhang0b494052019-01-31 21:41:15 +00001856 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashima657a1aa2018-09-12 16:19:22 +00001857 saved_page = LoadSavedPage(0);
1858 EXPECT_EQ(kOriginalObjectCount, FPDFPage_CountObjects(saved_page));
1859 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001860 ScopedFPDFBitmap page_bitmap = RenderPage(saved_page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00001861 CompareBitmap(page_bitmap.get(), 200, 300, kManyRectanglesChecksum);
Henrique Nakashima657a1aa2018-09-12 16:19:22 +00001862 }
1863
1864 CloseSavedPage(saved_page);
1865 CloseSavedDocument();
1866}
1867
Lei Zhangab41f252018-12-23 03:10:50 +00001868TEST_F(FPDFEditEmbedderTest, AddAndRemovePaths) {
Henrique Nakashima35841fa2018-03-15 15:25:16 +00001869 // Start with a blank page.
1870 FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792);
1871 ASSERT_TRUE(page);
1872
1873 // Render the blank page and verify it's a blank bitmap.
Hui Yingstb4baceb2020-04-28 23:46:10 +00001874 using pdfium::kBlankPage612By792Checksum;
Henrique Nakashima35841fa2018-03-15 15:25:16 +00001875 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001876 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00001877 CompareBitmap(page_bitmap.get(), 612, 792, kBlankPage612By792Checksum);
Henrique Nakashima35841fa2018-03-15 15:25:16 +00001878 }
1879 ASSERT_EQ(0, FPDFPage_CountObjects(page));
1880
1881 // Add a red rectangle.
1882 FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(10, 10, 20, 20);
1883 ASSERT_TRUE(red_rect);
Lei Zhang3475b482019-05-13 18:30:57 +00001884 EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 0, 0, 255));
Henrique Nakashima35841fa2018-03-15 15:25:16 +00001885 EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0));
1886 FPDFPage_InsertObject(page, red_rect);
Henrique Nakashima35841fa2018-03-15 15:25:16 +00001887 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001888 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00001889 CompareBitmap(page_bitmap.get(), 612, 792, kRedRectangleChecksum);
Henrique Nakashima35841fa2018-03-15 15:25:16 +00001890 }
1891 EXPECT_EQ(1, FPDFPage_CountObjects(page));
1892
1893 // Remove rectangle and verify it does not render anymore and the bitmap is
1894 // back to a blank one.
1895 EXPECT_TRUE(FPDFPage_RemoveObject(page, red_rect));
1896 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001897 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00001898 CompareBitmap(page_bitmap.get(), 612, 792, kBlankPage612By792Checksum);
Henrique Nakashima35841fa2018-03-15 15:25:16 +00001899 }
1900 EXPECT_EQ(0, FPDFPage_CountObjects(page));
1901
1902 // Trying to remove an object not in the page should return false.
1903 EXPECT_FALSE(FPDFPage_RemoveObject(page, red_rect));
1904
1905 FPDF_ClosePage(page);
1906 FPDFPageObj_Destroy(red_rect);
1907}
1908
Lei Zhangab41f252018-12-23 03:10:50 +00001909TEST_F(FPDFEditEmbedderTest, PathsPoints) {
Miklos Vajna12abfd02017-09-15 07:49:03 +02001910 CreateNewDocument();
1911 FPDF_PAGEOBJECT img = FPDFPageObj_NewImageObj(document_);
1912 // This should fail gracefully, even if img is not a path.
Miklos Vajna0150a542017-09-21 21:46:56 +02001913 ASSERT_EQ(-1, FPDFPath_CountSegments(img));
Miklos Vajna12abfd02017-09-15 07:49:03 +02001914
1915 // This should fail gracefully, even if path is NULL.
Miklos Vajna0150a542017-09-21 21:46:56 +02001916 ASSERT_EQ(-1, FPDFPath_CountSegments(nullptr));
Miklos Vajna12abfd02017-09-15 07:49:03 +02001917
Miklos Vajna36eed872017-09-20 22:52:43 +02001918 // FPDFPath_GetPathSegment() with a non-path.
1919 ASSERT_EQ(nullptr, FPDFPath_GetPathSegment(img, 0));
1920 // FPDFPath_GetPathSegment() with a NULL path.
1921 ASSERT_EQ(nullptr, FPDFPath_GetPathSegment(nullptr, 0));
1922 float x;
1923 float y;
1924 // FPDFPathSegment_GetPoint() with a NULL segment.
1925 EXPECT_FALSE(FPDFPathSegment_GetPoint(nullptr, &x, &y));
1926
1927 // FPDFPathSegment_GetType() with a NULL segment.
1928 ASSERT_EQ(FPDF_SEGMENT_UNKNOWN, FPDFPathSegment_GetType(nullptr));
1929
1930 // FPDFPathSegment_GetClose() with a NULL segment.
1931 EXPECT_FALSE(FPDFPathSegment_GetClose(nullptr));
1932
Miklos Vajna12abfd02017-09-15 07:49:03 +02001933 FPDFPageObj_Destroy(img);
1934}
1935
Hui Yingste95d5772020-08-24 23:58:26 +00001936TEST_F(FPDFEditEmbedderTest, PathOnTopOfText) {
Nicolas Pena0fc185e2017-02-08 12:13:20 -05001937 // Load document with some text
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001938 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
Nicolas Pena0fc185e2017-02-08 12:13:20 -05001939 FPDF_PAGE page = LoadPage(0);
Lei Zhang107fa7b2018-02-09 21:48:15 +00001940 ASSERT_TRUE(page);
Nicolas Pena0fc185e2017-02-08 12:13:20 -05001941
1942 // Add an opaque rectangle on top of some of the text.
1943 FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(20, 100, 50, 50);
Lei Zhang3475b482019-05-13 18:30:57 +00001944 EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 0, 0, 255));
Nicolas Pena0fc185e2017-02-08 12:13:20 -05001945 EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0));
1946 FPDFPage_InsertObject(page, red_rect);
Nicolas Pena0fc185e2017-02-08 12:13:20 -05001947
1948 // Add a transparent triangle on top of other part of the text.
1949 FPDF_PAGEOBJECT black_path = FPDFPageObj_CreateNewPath(20, 50);
Lei Zhang3475b482019-05-13 18:30:57 +00001950 EXPECT_TRUE(FPDFPageObj_SetFillColor(black_path, 0, 0, 0, 100));
Nicolas Pena0fc185e2017-02-08 12:13:20 -05001951 EXPECT_TRUE(FPDFPath_SetDrawMode(black_path, FPDF_FILLMODE_ALTERNATE, 0));
1952 EXPECT_TRUE(FPDFPath_LineTo(black_path, 30, 80));
1953 EXPECT_TRUE(FPDFPath_LineTo(black_path, 40, 10));
1954 EXPECT_TRUE(FPDFPath_Close(black_path));
1955 FPDFPage_InsertObject(page, black_path);
Nicolas Pena0fc185e2017-02-08 12:13:20 -05001956
Hui Yingste95d5772020-08-24 23:58:26 +00001957 // Render and check the result.
Tom Sepeze08d2b12018-04-25 18:49:32 +00001958 ScopedFPDFBitmap bitmap = RenderLoadedPage(page);
Hui Yingste95d5772020-08-24 23:58:26 +00001959#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
1960#if defined(OS_WIN)
1961 const char kChecksum[] = "e755a955696373e39dbebb5cb96e1338";
Nicolas Pena0fc185e2017-02-08 12:13:20 -05001962#else
Hui Yingste95d5772020-08-24 23:58:26 +00001963 const char kChecksum[] = "d082f9756c86bb47e1abbc2b1df7138a";
1964#endif // defined(OS_WIN)
1965#else
1966#if defined(OS_WIN)
1967 const char kChecksum[] = "74dd9c393b8b2578d2b7feb032b7daad";
1968#elif defined(OS_APPLE)
1969 const char kChecksum[] = "e55bcd1facb7243dc6e16dd5f912265b";
1970#else
1971 const char kChecksum[] = "aa71b09b93b55f467f1290e5111babee";
Nicolas Pena5bcd9a32017-03-22 11:04:35 -04001972#endif
Hui Yingste95d5772020-08-24 23:58:26 +00001973#endif // defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
1974 CompareBitmap(bitmap.get(), 200, 200, kChecksum);
Nicolas Pena0fc185e2017-02-08 12:13:20 -05001975 UnloadPage(page);
1976}
Nicolas Pena2eb1a702017-02-09 18:17:33 -05001977
Hui Yingst1f73c4f2020-07-29 00:46:25 +00001978TEST_F(FPDFEditEmbedderTest, EditOverExistingContent) {
wileyryae858aa42017-05-31 14:49:05 -05001979 // Load document with existing content
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001980 ASSERT_TRUE(OpenDocument("bug_717.pdf"));
wileyryae858aa42017-05-31 14:49:05 -05001981 FPDF_PAGE page = LoadPage(0);
Lei Zhang107fa7b2018-02-09 21:48:15 +00001982 ASSERT_TRUE(page);
wileyryae858aa42017-05-31 14:49:05 -05001983
1984 // Add a transparent rectangle on top of the existing content
1985 FPDF_PAGEOBJECT red_rect2 = FPDFPageObj_CreateNewRect(90, 700, 25, 50);
Lei Zhang3475b482019-05-13 18:30:57 +00001986 EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect2, 255, 0, 0, 100));
wileyryae858aa42017-05-31 14:49:05 -05001987 EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect2, FPDF_FILLMODE_ALTERNATE, 0));
1988 FPDFPage_InsertObject(page, red_rect2);
1989
1990 // Add an opaque rectangle on top of the existing content
1991 FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(115, 700, 25, 50);
Lei Zhang3475b482019-05-13 18:30:57 +00001992 EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 0, 0, 255));
wileyryae858aa42017-05-31 14:49:05 -05001993 EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0));
1994 FPDFPage_InsertObject(page, red_rect);
1995
Hui Yingst1f73c4f2020-07-29 00:46:25 +00001996#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
1997 const char kOriginalChecksum[] = "1e82fbdd21490cee9d3479fe6125af67";
1998#else
1999 const char kOriginalChecksum[] = "ad04e5bd0f471a9a564fb034bd0fb073";
2000#endif
Tom Sepeze08d2b12018-04-25 18:49:32 +00002001 ScopedFPDFBitmap bitmap = RenderLoadedPage(page);
Hui Yingst1f73c4f2020-07-29 00:46:25 +00002002 CompareBitmap(bitmap.get(), 612, 792, kOriginalChecksum);
wileyryae858aa42017-05-31 14:49:05 -05002003 EXPECT_TRUE(FPDFPage_GenerateContent(page));
2004
2005 // Now save the result, closing the page and document
2006 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
Nicolas Pena3ff54002017-07-05 11:55:35 -04002007 UnloadPage(page);
wileyryae858aa42017-05-31 14:49:05 -05002008
Lei Zhang0b494052019-01-31 21:41:15 +00002009 ASSERT_TRUE(OpenSavedDocument());
Lei Zhang107fa7b2018-02-09 21:48:15 +00002010 FPDF_PAGE saved_page = LoadSavedPage(0);
Hui Yingst1f73c4f2020-07-29 00:46:25 +00002011 VerifySavedRendering(saved_page, 612, 792, kOriginalChecksum);
wileyryae858aa42017-05-31 14:49:05 -05002012
2013 ClearString();
2014 // Add another opaque rectangle on top of the existing content
2015 FPDF_PAGEOBJECT green_rect = FPDFPageObj_CreateNewRect(150, 700, 25, 50);
Lei Zhang3475b482019-05-13 18:30:57 +00002016 EXPECT_TRUE(FPDFPageObj_SetFillColor(green_rect, 0, 255, 0, 255));
wileyryae858aa42017-05-31 14:49:05 -05002017 EXPECT_TRUE(FPDFPath_SetDrawMode(green_rect, FPDF_FILLMODE_ALTERNATE, 0));
Lei Zhang107fa7b2018-02-09 21:48:15 +00002018 FPDFPage_InsertObject(saved_page, green_rect);
wileyryae858aa42017-05-31 14:49:05 -05002019
2020 // Add another transparent rectangle on top of existing content
2021 FPDF_PAGEOBJECT green_rect2 = FPDFPageObj_CreateNewRect(175, 700, 25, 50);
Lei Zhang3475b482019-05-13 18:30:57 +00002022 EXPECT_TRUE(FPDFPageObj_SetFillColor(green_rect2, 0, 255, 0, 100));
wileyryae858aa42017-05-31 14:49:05 -05002023 EXPECT_TRUE(FPDFPath_SetDrawMode(green_rect2, FPDF_FILLMODE_ALTERNATE, 0));
Lei Zhang107fa7b2018-02-09 21:48:15 +00002024 FPDFPage_InsertObject(saved_page, green_rect2);
Hui Yingst1f73c4f2020-07-29 00:46:25 +00002025#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
2026 const char kLastChecksum[] = "8705d023e5fec3499d1e30cf2bcc5dc1";
2027#else
2028 const char kLastChecksum[] = "4b5b00f824620f8c9b8801ebb98e1cdd";
2029#endif
Lei Zhangc113c7a2018-02-12 14:58:44 +00002030 {
Tom Sepeze08d2b12018-04-25 18:49:32 +00002031 ScopedFPDFBitmap new_bitmap = RenderSavedPage(saved_page);
Hui Yingst1f73c4f2020-07-29 00:46:25 +00002032 CompareBitmap(new_bitmap.get(), 612, 792, kLastChecksum);
Lei Zhangc113c7a2018-02-12 14:58:44 +00002033 }
Lei Zhang107fa7b2018-02-09 21:48:15 +00002034 EXPECT_TRUE(FPDFPage_GenerateContent(saved_page));
wileyryae858aa42017-05-31 14:49:05 -05002035
2036 // Now save the result, closing the page and document
Lei Zhang0729be22018-02-05 21:13:51 +00002037 EXPECT_TRUE(FPDF_SaveAsCopy(saved_document_, this, 0));
Dan Sinclair04e4dc82017-10-18 12:17:14 -04002038
Lei Zhang107fa7b2018-02-09 21:48:15 +00002039 CloseSavedPage(saved_page);
Dan Sinclair04e4dc82017-10-18 12:17:14 -04002040 CloseSavedDocument();
wileyryae858aa42017-05-31 14:49:05 -05002041
2042 // Render the saved result
Hui Yingst1f73c4f2020-07-29 00:46:25 +00002043 VerifySavedDocument(612, 792, kLastChecksum);
wileyryae858aa42017-05-31 14:49:05 -05002044}
2045
Lei Zhang03e5e682019-09-16 19:45:55 +00002046// TODO(crbug.com/pdfium/11): Fix this test and enable.
2047#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
2048#define MAYBE_AddStrokedPaths DISABLED_AddStrokedPaths
2049#else
2050#define MAYBE_AddStrokedPaths AddStrokedPaths
2051#endif
2052TEST_F(FPDFEditEmbedderTest, MAYBE_AddStrokedPaths) {
Nicolas Pena2eb1a702017-02-09 18:17:33 -05002053 // Start with a blank page
Nicolas Penad03ca422017-03-06 13:54:33 -05002054 FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792);
Nicolas Pena2eb1a702017-02-09 18:17:33 -05002055
2056 // Add a large stroked rectangle (fill color should not affect it).
2057 FPDF_PAGEOBJECT rect = FPDFPageObj_CreateNewRect(20, 20, 200, 400);
Lei Zhang3475b482019-05-13 18:30:57 +00002058 EXPECT_TRUE(FPDFPageObj_SetFillColor(rect, 255, 0, 0, 255));
2059 EXPECT_TRUE(FPDFPageObj_SetStrokeColor(rect, 0, 255, 0, 255));
2060 EXPECT_TRUE(FPDFPageObj_SetStrokeWidth(rect, 15.0f));
Miklos Vajna366df7f2018-05-22 14:27:29 +00002061
2062 float width = 0;
2063 EXPECT_TRUE(FPDFPageObj_GetStrokeWidth(rect, &width));
2064 EXPECT_EQ(15.0f, width);
2065
Nicolas Pena2eb1a702017-02-09 18:17:33 -05002066 EXPECT_TRUE(FPDFPath_SetDrawMode(rect, 0, 1));
2067 FPDFPage_InsertObject(page, rect);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002068 {
Lei Zhang30ff2532019-01-31 21:37:55 +00002069 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002070 CompareBitmap(page_bitmap.get(), 612, 792,
2071 "64bd31f862a89e0a9e505a5af6efd506");
2072 }
Nicolas Pena2eb1a702017-02-09 18:17:33 -05002073
2074 // Add crossed-checkmark
2075 FPDF_PAGEOBJECT check = FPDFPageObj_CreateNewPath(300, 500);
2076 EXPECT_TRUE(FPDFPath_LineTo(check, 400, 400));
2077 EXPECT_TRUE(FPDFPath_LineTo(check, 600, 600));
2078 EXPECT_TRUE(FPDFPath_MoveTo(check, 400, 600));
2079 EXPECT_TRUE(FPDFPath_LineTo(check, 600, 400));
Lei Zhang3475b482019-05-13 18:30:57 +00002080 EXPECT_TRUE(FPDFPageObj_SetStrokeColor(check, 128, 128, 128, 180));
2081 EXPECT_TRUE(FPDFPageObj_SetStrokeWidth(check, 8.35f));
Nicolas Pena2eb1a702017-02-09 18:17:33 -05002082 EXPECT_TRUE(FPDFPath_SetDrawMode(check, 0, 1));
2083 FPDFPage_InsertObject(page, check);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002084 {
Lei Zhang30ff2532019-01-31 21:37:55 +00002085 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002086 CompareBitmap(page_bitmap.get(), 612, 792,
2087 "4b6f3b9d25c4e194821217d5016c3724");
2088 }
Nicolas Pena2eb1a702017-02-09 18:17:33 -05002089
2090 // Add stroked and filled oval-ish path.
2091 FPDF_PAGEOBJECT path = FPDFPageObj_CreateNewPath(250, 100);
2092 EXPECT_TRUE(FPDFPath_BezierTo(path, 180, 166, 180, 233, 250, 300));
2093 EXPECT_TRUE(FPDFPath_LineTo(path, 255, 305));
2094 EXPECT_TRUE(FPDFPath_BezierTo(path, 325, 233, 325, 166, 255, 105));
2095 EXPECT_TRUE(FPDFPath_Close(path));
Lei Zhang3475b482019-05-13 18:30:57 +00002096 EXPECT_TRUE(FPDFPageObj_SetFillColor(path, 200, 128, 128, 100));
2097 EXPECT_TRUE(FPDFPageObj_SetStrokeColor(path, 128, 200, 128, 150));
2098 EXPECT_TRUE(FPDFPageObj_SetStrokeWidth(path, 10.5f));
Nicolas Pena2eb1a702017-02-09 18:17:33 -05002099 EXPECT_TRUE(FPDFPath_SetDrawMode(path, FPDF_FILLMODE_ALTERNATE, 1));
2100 FPDFPage_InsertObject(page, path);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002101 {
Lei Zhang30ff2532019-01-31 21:37:55 +00002102 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002103 CompareBitmap(page_bitmap.get(), 612, 792,
2104 "ff3e6a22326754944cc6e56609acd73b");
2105 }
Nicolas Pena2eb1a702017-02-09 18:17:33 -05002106 FPDF_ClosePage(page);
Nicolas Pena2eb1a702017-02-09 18:17:33 -05002107}
Nicolas Pena49058402017-02-14 18:26:20 -05002108
Nicolas Pena4c48b102018-06-13 18:23:46 +00002109// Tests adding text from standard font using FPDFPageObj_NewTextObj.
Hui Yingst5ce6a3f2020-11-11 02:29:09 +00002110TEST_F(FPDFEditEmbedderTest, AddStandardFontText) {
Nicolas Pena49058402017-02-14 18:26:20 -05002111 // Start with a blank page
Nicolas Penad03ca422017-03-06 13:54:33 -05002112 FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792);
Nicolas Pena49058402017-02-14 18:26:20 -05002113
2114 // Add some text to the page
Nicolas Penab3161852017-05-02 14:12:50 -04002115 FPDF_PAGEOBJECT text_object1 =
2116 FPDFPageObj_NewTextObj(document(), "Arial", 12.0f);
2117 EXPECT_TRUE(text_object1);
Hui Yingst3b6136a2020-06-09 00:39:33 +00002118 ScopedFPDFWideString text1 = GetFPDFWideString(kBottomText);
Nicolas Penab3161852017-05-02 14:12:50 -04002119 EXPECT_TRUE(FPDFText_SetText(text_object1, text1.get()));
2120 FPDFPageObj_Transform(text_object1, 1, 0, 0, 1, 20, 20);
2121 FPDFPage_InsertObject(page, text_object1);
Lei Zhange039bab2019-03-18 19:57:56 +00002122 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Lei Zhang107fa7b2018-02-09 21:48:15 +00002123 {
Lei Zhang30ff2532019-01-31 21:37:55 +00002124 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingst3b6136a2020-06-09 00:39:33 +00002125 CompareBitmap(page_bitmap.get(), 612, 792, kBottomTextChecksum);
Lei Zhange039bab2019-03-18 19:57:56 +00002126
2127 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
Hui Yingst3b6136a2020-06-09 00:39:33 +00002128 VerifySavedDocument(612, 792, kBottomTextChecksum);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002129 }
Nicolas Pena49058402017-02-14 18:26:20 -05002130
2131 // Try another font
Nicolas Penab3161852017-05-02 14:12:50 -04002132 FPDF_PAGEOBJECT text_object2 =
Nicolas Penad03ca422017-03-06 13:54:33 -05002133 FPDFPageObj_NewTextObj(document(), "TimesNewRomanBold", 15.0f);
Nicolas Penab3161852017-05-02 14:12:50 -04002134 EXPECT_TRUE(text_object2);
Lei Zhangf0f67682019-04-08 17:03:21 +00002135 ScopedFPDFWideString text2 =
Nicolas Penab3161852017-05-02 14:12:50 -04002136 GetFPDFWideString(L"Hi, I'm Bold. Times New Roman Bold.");
2137 EXPECT_TRUE(FPDFText_SetText(text_object2, text2.get()));
2138 FPDFPageObj_Transform(text_object2, 1, 0, 0, 1, 100, 600);
2139 FPDFPage_InsertObject(page, text_object2);
Lei Zhange039bab2019-03-18 19:57:56 +00002140 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Lei Zhang107fa7b2018-02-09 21:48:15 +00002141 {
Lei Zhang30ff2532019-01-31 21:37:55 +00002142 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingst5ce6a3f2020-11-11 02:29:09 +00002143#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
2144 static constexpr char md5[] = "285cf09ca5600fc4ec061dc5ad5c6400";
Nicolas Pena49058402017-02-14 18:26:20 -05002145#else
Hui Yingst5ce6a3f2020-11-11 02:29:09 +00002146#if defined(OS_WIN)
2147 static constexpr char md5[] = "3755dd35abd4c605755369401ee85b2d";
2148#elif defined(OS_APPLE)
2149 static constexpr char md5[] = "26a516d923b0a18fbea0a24e3aca5562";
2150#else
2151 static constexpr char md5[] = "76fcc7d08aa15445efd2e2ceb7c6cc3b";
Nicolas Pena5bcd9a32017-03-22 11:04:35 -04002152#endif
Hui Yingst5ce6a3f2020-11-11 02:29:09 +00002153#endif // defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
Lei Zhange039bab2019-03-18 19:57:56 +00002154 CompareBitmap(page_bitmap.get(), 612, 792, md5);
2155
2156 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
2157 VerifySavedDocument(612, 792, md5);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002158 }
Nicolas Pena49058402017-02-14 18:26:20 -05002159
2160 // And some randomly transformed text
Nicolas Penab3161852017-05-02 14:12:50 -04002161 FPDF_PAGEOBJECT text_object3 =
Nicolas Penad03ca422017-03-06 13:54:33 -05002162 FPDFPageObj_NewTextObj(document(), "Courier-Bold", 20.0f);
Nicolas Penab3161852017-05-02 14:12:50 -04002163 EXPECT_TRUE(text_object3);
Lei Zhangf0f67682019-04-08 17:03:21 +00002164 ScopedFPDFWideString text3 = GetFPDFWideString(L"Can you read me? <:)>");
Nicolas Penab3161852017-05-02 14:12:50 -04002165 EXPECT_TRUE(FPDFText_SetText(text_object3, text3.get()));
2166 FPDFPageObj_Transform(text_object3, 1, 1.5, 2, 0.5, 200, 200);
2167 FPDFPage_InsertObject(page, text_object3);
Lei Zhange039bab2019-03-18 19:57:56 +00002168 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Lei Zhang107fa7b2018-02-09 21:48:15 +00002169 {
Lei Zhang30ff2532019-01-31 21:37:55 +00002170 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingst5ce6a3f2020-11-11 02:29:09 +00002171#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
2172#if defined(OS_WIN)
2173 static constexpr char md5[] = "03c4d98eae4fda51ca67743665ab61f4";
Nicolas Pena49058402017-02-14 18:26:20 -05002174#else
Hui Yingst5ce6a3f2020-11-11 02:29:09 +00002175 static constexpr char md5[] = "177285dd8cdaf476683173fce64034ea";
2176#endif // defined(OS_WIN)
2177#else
2178#if defined(OS_WIN)
2179 static constexpr char md5[] = "5ded49fe157f89627903553771431e3d";
2180#elif defined(OS_APPLE)
2181 static constexpr char md5[] = "532024c9ded47843313bb64a060118f3";
2182#else
2183 static constexpr char md5[] = "344534539aa7c5cc78404cfff4bde7fb";
Nicolas Pena5bcd9a32017-03-22 11:04:35 -04002184#endif
Hui Yingst5ce6a3f2020-11-11 02:29:09 +00002185#endif // defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
Lei Zhange039bab2019-03-18 19:57:56 +00002186 CompareBitmap(page_bitmap.get(), 612, 792, md5);
2187
2188 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
2189 VerifySavedDocument(612, 792, md5);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002190 }
Nicolas Pena49058402017-02-14 18:26:20 -05002191
Lei Zhang8da98232019-12-11 23:29:33 +00002192 FS_MATRIX matrix;
2193 EXPECT_FALSE(FPDFTextObj_GetMatrix(nullptr, &matrix));
2194 EXPECT_TRUE(FPDFTextObj_GetMatrix(text_object3, &matrix));
2195 EXPECT_FLOAT_EQ(1.0f, matrix.a);
2196 EXPECT_FLOAT_EQ(1.5f, matrix.b);
2197 EXPECT_FLOAT_EQ(2.0f, matrix.c);
2198 EXPECT_FLOAT_EQ(0.5f, matrix.d);
2199 EXPECT_FLOAT_EQ(200.0f, matrix.e);
2200 EXPECT_FLOAT_EQ(200.0f, matrix.f);
Miklos Vajnac765d2a2018-06-19 15:45:42 +00002201
Miklos Vajna8625d3b2018-06-26 15:12:48 +00002202 EXPECT_EQ(0, FPDFTextObj_GetFontSize(nullptr));
2203 EXPECT_EQ(20, FPDFTextObj_GetFontSize(text_object3));
2204
Nicolas Pena49058402017-02-14 18:26:20 -05002205 // TODO(npm): Why are there issues with text rotated by 90 degrees?
2206 // TODO(npm): FPDF_SaveAsCopy not giving the desired result after this.
2207 FPDF_ClosePage(page);
Nicolas Pena49058402017-02-14 18:26:20 -05002208}
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002209
Daniel Hosseinian8cb6a652019-12-18 00:50:41 +00002210TEST_F(FPDFEditEmbedderTest, GetTextRenderMode) {
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00002211 ASSERT_TRUE(OpenDocument("text_render_mode.pdf"));
Miklos Vajna1448cc12018-07-03 13:52:33 +00002212 FPDF_PAGE page = LoadPage(0);
2213 ASSERT_TRUE(page);
2214 ASSERT_EQ(2, FPDFPage_CountObjects(page));
2215
Daniel Hosseiniane83d8b42020-01-08 00:26:15 +00002216 EXPECT_EQ(FPDF_TEXTRENDERMODE_UNKNOWN,
Daniel Hosseinian8cb6a652019-12-18 00:50:41 +00002217 FPDFTextObj_GetTextRenderMode(nullptr));
Miklos Vajna1448cc12018-07-03 13:52:33 +00002218
2219 FPDF_PAGEOBJECT fill = FPDFPage_GetObject(page, 0);
Daniel Hosseiniane83d8b42020-01-08 00:26:15 +00002220 EXPECT_EQ(FPDF_TEXTRENDERMODE_FILL, FPDFTextObj_GetTextRenderMode(fill));
Miklos Vajna1448cc12018-07-03 13:52:33 +00002221
2222 FPDF_PAGEOBJECT stroke = FPDFPage_GetObject(page, 1);
Daniel Hosseiniane83d8b42020-01-08 00:26:15 +00002223 EXPECT_EQ(FPDF_TEXTRENDERMODE_STROKE, FPDFTextObj_GetTextRenderMode(stroke));
2224
2225 UnloadPage(page);
2226}
2227
Hui Yingst9d8e5ec2020-07-30 22:33:22 +00002228TEST_F(FPDFEditEmbedderTest, SetTextRenderMode) {
Hui Yingst609f7d62020-04-23 23:14:13 +00002229#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
Hui Yingst9d8e5ec2020-07-30 22:33:22 +00002230 const char kOriginalChecksum[] = "4960c720a6cdb7b2b16be809cce58aff";
2231 const char kStrokeChecksum[] = "4c099cf9abc1565806bd7cd1cca5ac1a";
Hui Yingst609f7d62020-04-23 23:14:13 +00002232#else
Hui Yingst9d8e5ec2020-07-30 22:33:22 +00002233#if defined(OS_WIN)
2234 const char kOriginalChecksum[] = "de6e86bad3e9fda753a8471a45cfbb58";
2235#elif defined(OS_APPLE)
2236 const char kOriginalChecksum[] = "2453a9a524ee3f5f525dd21345ec4d81";
Daniel Hosseiniane83d8b42020-01-08 00:26:15 +00002237#else
Hui Yingst9d8e5ec2020-07-30 22:33:22 +00002238 const char kOriginalChecksum[] = "5a012d2920ac075c39ffa9437ea42faa";
Daniel Hosseiniane83d8b42020-01-08 00:26:15 +00002239#endif
Hui Yingst9d8e5ec2020-07-30 22:33:22 +00002240 const char kStrokeChecksum[] = "412e52e621b46bd77baf2162e1fb1a1d";
2241#endif // defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
Daniel Hosseiniane83d8b42020-01-08 00:26:15 +00002242
Daniel Hosseiniane83d8b42020-01-08 00:26:15 +00002243 {
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00002244 ASSERT_TRUE(OpenDocument("text_render_mode.pdf"));
Daniel Hosseiniana933ebf2020-01-31 21:28:25 +00002245 FPDF_PAGE page = LoadPage(0);
2246 ASSERT_TRUE(page);
2247 ASSERT_EQ(2, FPDFPage_CountObjects(page));
2248
2249 // Check the bitmap
2250 {
2251 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingst9d8e5ec2020-07-30 22:33:22 +00002252 CompareBitmap(page_bitmap.get(), 612, 446, kOriginalChecksum);
Daniel Hosseiniana933ebf2020-01-31 21:28:25 +00002253 }
2254
2255 // Cannot set on a null object.
2256 EXPECT_FALSE(
2257 FPDFTextObj_SetTextRenderMode(nullptr, FPDF_TEXTRENDERMODE_UNKNOWN));
2258 EXPECT_FALSE(
2259 FPDFTextObj_SetTextRenderMode(nullptr, FPDF_TEXTRENDERMODE_INVISIBLE));
2260
2261 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0);
2262 ASSERT_TRUE(page_object);
2263 EXPECT_EQ(FPDF_TEXTRENDERMODE_FILL,
2264 FPDFTextObj_GetTextRenderMode(page_object));
2265
2266 // Cannot set UNKNOWN as a render mode.
2267 EXPECT_FALSE(FPDFTextObj_SetTextRenderMode(page_object,
2268 FPDF_TEXTRENDERMODE_UNKNOWN));
2269
2270 EXPECT_TRUE(
2271 FPDFTextObj_SetTextRenderMode(page_object, FPDF_TEXTRENDERMODE_STROKE));
2272 EXPECT_EQ(FPDF_TEXTRENDERMODE_STROKE,
2273 FPDFTextObj_GetTextRenderMode(page_object));
2274
2275 // Check that bitmap displays changed content
2276 {
2277 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingst9d8e5ec2020-07-30 22:33:22 +00002278 CompareBitmap(page_bitmap.get(), 612, 446, kStrokeChecksum);
Daniel Hosseiniana933ebf2020-01-31 21:28:25 +00002279 }
2280
Daniel Hosseinianafd27502020-07-01 04:59:03 +00002281 // Save a copy.
2282 EXPECT_TRUE(FPDFPage_GenerateContent(page));
2283 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
2284
Daniel Hosseiniana933ebf2020-01-31 21:28:25 +00002285 UnloadPage(page);
Daniel Hosseiniane83d8b42020-01-08 00:26:15 +00002286 }
Miklos Vajna1448cc12018-07-03 13:52:33 +00002287
Daniel Hosseiniana933ebf2020-01-31 21:28:25 +00002288 {
Daniel Hosseinianafd27502020-07-01 04:59:03 +00002289 // Open the saved copy and render it. Check that the changed text render
2290 // mode is kept in the saved copy.
Daniel Hosseiniana933ebf2020-01-31 21:28:25 +00002291 ASSERT_TRUE(OpenSavedDocument());
2292 FPDF_PAGE saved_page = LoadSavedPage(0);
2293 ASSERT_TRUE(saved_page);
2294
2295 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(saved_page, 0);
2296 EXPECT_TRUE(page_object);
Daniel Hosseinianafd27502020-07-01 04:59:03 +00002297 EXPECT_EQ(FPDF_TEXTRENDERMODE_STROKE,
Daniel Hosseiniana933ebf2020-01-31 21:28:25 +00002298 FPDFTextObj_GetTextRenderMode(page_object));
2299
2300 ScopedFPDFBitmap bitmap = RenderSavedPage(saved_page);
Hui Yingst9d8e5ec2020-07-30 22:33:22 +00002301 CompareBitmap(bitmap.get(), 612, 446, kStrokeChecksum);
Daniel Hosseiniana933ebf2020-01-31 21:28:25 +00002302
2303 CloseSavedPage(saved_page);
2304 CloseSavedDocument();
2305 }
Miklos Vajna1448cc12018-07-03 13:52:33 +00002306}
2307
Lei Zhangab41f252018-12-23 03:10:50 +00002308TEST_F(FPDFEditEmbedderTest, TestGetTextFontName) {
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00002309 ASSERT_TRUE(OpenDocument("text_font.pdf"));
Miklos Vajna53d4f0a2018-08-01 01:28:49 +00002310 FPDF_PAGE page = LoadPage(0);
2311 ASSERT_TRUE(page);
2312 ASSERT_EQ(1, FPDFPage_CountObjects(page));
2313
2314 // FPDFTextObj_GetFontName() positive testing.
2315 FPDF_PAGEOBJECT text = FPDFPage_GetObject(page, 0);
2316 unsigned long size = FPDFTextObj_GetFontName(text, nullptr, 0);
2317 const char kExpectedFontName[] = "Liberation Serif";
2318 ASSERT_EQ(sizeof(kExpectedFontName), size);
2319 std::vector<char> font_name(size);
2320 ASSERT_EQ(size, FPDFTextObj_GetFontName(text, font_name.data(), size));
2321 ASSERT_STREQ(kExpectedFontName, font_name.data());
2322
2323 // FPDFTextObj_GetFontName() negative testing.
2324 ASSERT_EQ(0U, FPDFTextObj_GetFontName(nullptr, nullptr, 0));
2325
2326 font_name.resize(2);
2327 font_name[0] = 'x';
2328 font_name[1] = '\0';
2329 size = FPDFTextObj_GetFontName(text, font_name.data(), font_name.size());
2330 ASSERT_EQ(sizeof(kExpectedFontName), size);
2331 ASSERT_EQ(std::string("x"), std::string(font_name.data()));
2332
2333 UnloadPage(page);
2334}
2335
Lei Zhangab41f252018-12-23 03:10:50 +00002336TEST_F(FPDFEditEmbedderTest, TestFormGetObjects) {
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00002337 ASSERT_TRUE(OpenDocument("form_object.pdf"));
Miklos Vajnab66077d2018-07-11 13:25:02 +00002338 FPDF_PAGE page = LoadPage(0);
2339 ASSERT_TRUE(page);
2340 ASSERT_EQ(1, FPDFPage_CountObjects(page));
2341
2342 FPDF_PAGEOBJECT form = FPDFPage_GetObject(page, 0);
2343 EXPECT_EQ(FPDF_PAGEOBJ_FORM, FPDFPageObj_GetType(form));
2344 ASSERT_EQ(-1, FPDFFormObj_CountObjects(nullptr));
2345 ASSERT_EQ(2, FPDFFormObj_CountObjects(form));
2346
Miklos Vajna1d273f12018-07-16 19:20:36 +00002347 // FPDFFormObj_GetObject() positive testing.
2348 FPDF_PAGEOBJECT text1 = FPDFFormObj_GetObject(form, 0);
2349 ASSERT_TRUE(text1);
2350 float left = 0;
2351 float bottom = 0;
2352 float right = 0;
2353 float top = 0;
2354 ASSERT_TRUE(FPDFPageObj_GetBounds(text1, &left, &bottom, &right, &top));
2355 ASSERT_EQ(271, static_cast<int>(top));
2356
2357 FPDF_PAGEOBJECT text2 = FPDFFormObj_GetObject(form, 1);
2358 ASSERT_TRUE(text2);
2359 ASSERT_TRUE(FPDFPageObj_GetBounds(text2, &left, &bottom, &right, &top));
2360 ASSERT_EQ(221, static_cast<int>(top));
2361
2362 // FPDFFormObj_GetObject() negative testing.
2363 ASSERT_EQ(nullptr, FPDFFormObj_GetObject(nullptr, 0));
2364 ASSERT_EQ(nullptr, FPDFFormObj_GetObject(form, -1));
2365 ASSERT_EQ(nullptr, FPDFFormObj_GetObject(form, 2));
2366
Miklos Vajna46b43732018-08-14 19:15:43 +00002367 // Reset the form object matrix to identity.
2368 auto* pPageObj = CPDFPageObjectFromFPDFPageObject(form);
2369 CPDF_FormObject* pFormObj = pPageObj->AsForm();
2370 pFormObj->Transform(pFormObj->form_matrix().GetInverse());
2371
2372 // FPDFFormObj_GetMatrix() positive testing.
Lei Zhang8da98232019-12-11 23:29:33 +00002373 static constexpr FS_MATRIX kMatrix = {1.0f, 1.5f, 2.0f, 2.5f, 100.0f, 200.0f};
2374 pFormObj->Transform(CFXMatrixFromFSMatrix(kMatrix));
Miklos Vajna46b43732018-08-14 19:15:43 +00002375
Lei Zhang8da98232019-12-11 23:29:33 +00002376 FS_MATRIX matrix;
2377 EXPECT_TRUE(FPDFFormObj_GetMatrix(form, &matrix));
2378 EXPECT_FLOAT_EQ(kMatrix.a, matrix.a);
2379 EXPECT_FLOAT_EQ(kMatrix.b, matrix.b);
2380 EXPECT_FLOAT_EQ(kMatrix.c, matrix.c);
2381 EXPECT_FLOAT_EQ(kMatrix.d, matrix.d);
2382 EXPECT_FLOAT_EQ(kMatrix.e, matrix.e);
2383 EXPECT_FLOAT_EQ(kMatrix.f, matrix.f);
Miklos Vajna46b43732018-08-14 19:15:43 +00002384
2385 // FPDFFormObj_GetMatrix() negative testing.
Lei Zhang8da98232019-12-11 23:29:33 +00002386 EXPECT_FALSE(FPDFFormObj_GetMatrix(nullptr, &matrix));
2387 EXPECT_FALSE(FPDFFormObj_GetMatrix(form, nullptr));
2388 EXPECT_FALSE(FPDFFormObj_GetMatrix(nullptr, nullptr));
Miklos Vajna46b43732018-08-14 19:15:43 +00002389
Miklos Vajnab66077d2018-07-11 13:25:02 +00002390 UnloadPage(page);
2391}
2392
Nicolas Pena4c48b102018-06-13 18:23:46 +00002393// Tests adding text from standard font using FPDFText_LoadStandardFont.
Hui Yingst5ce6a3f2020-11-11 02:29:09 +00002394TEST_F(FPDFEditEmbedderTest, AddStandardFontText2) {
Nicolas Pena4c48b102018-06-13 18:23:46 +00002395 // Start with a blank page
2396 ScopedFPDFPage page(FPDFPage_New(CreateNewDocument(), 0, 612, 792));
2397
2398 // Load a standard font.
Tom Sepezffff6c52019-07-30 21:56:19 +00002399 ScopedFPDFFont font(FPDFText_LoadStandardFont(document(), "Helvetica"));
Nicolas Pena4c48b102018-06-13 18:23:46 +00002400 ASSERT_TRUE(font);
2401
2402 // Add some text to the page.
2403 FPDF_PAGEOBJECT text_object =
Tom Sepezffff6c52019-07-30 21:56:19 +00002404 FPDFPageObj_CreateTextObj(document(), font.get(), 12.0f);
Nicolas Pena4c48b102018-06-13 18:23:46 +00002405 EXPECT_TRUE(text_object);
Hui Yingst3b6136a2020-06-09 00:39:33 +00002406 ScopedFPDFWideString text = GetFPDFWideString(kBottomText);
Nicolas Pena4c48b102018-06-13 18:23:46 +00002407 EXPECT_TRUE(FPDFText_SetText(text_object, text.get()));
2408 FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 20, 20);
2409 FPDFPage_InsertObject(page.get(), text_object);
Lei Zhang30ff2532019-01-31 21:37:55 +00002410 ScopedFPDFBitmap page_bitmap = RenderPage(page.get());
Hui Yingst3b6136a2020-06-09 00:39:33 +00002411 CompareBitmap(page_bitmap.get(), 612, 792, kBottomTextChecksum);
Nicolas Pena4c48b102018-06-13 18:23:46 +00002412}
2413
Lei Zhangab41f252018-12-23 03:10:50 +00002414TEST_F(FPDFEditEmbedderTest, LoadStandardFonts) {
Nicolas Pena4c48b102018-06-13 18:23:46 +00002415 CreateNewDocument();
Lei Zhang590f4242019-05-15 20:57:26 +00002416 static constexpr const char* kStandardFontNames[] = {
Lei Zhangd72fd582018-07-27 19:37:27 +00002417 "Arial",
2418 "Arial-Bold",
2419 "Arial-BoldItalic",
2420 "Arial-Italic",
2421 "Courier",
2422 "Courier-BoldOblique",
2423 "Courier-Oblique",
2424 "Courier-Bold",
2425 "CourierNew",
2426 "CourierNew-Bold",
2427 "CourierNew-BoldItalic",
2428 "CourierNew-Italic",
2429 "Helvetica",
2430 "Helvetica-Bold",
2431 "Helvetica-BoldOblique",
2432 "Helvetica-Oblique",
2433 "Symbol",
2434 "TimesNewRoman",
2435 "TimesNewRoman-Bold",
2436 "TimesNewRoman-BoldItalic",
2437 "TimesNewRoman-Italic",
2438 "ZapfDingbats"};
Lei Zhang590f4242019-05-15 20:57:26 +00002439 for (const char* font_name : kStandardFontNames) {
Tom Sepezffff6c52019-07-30 21:56:19 +00002440 ScopedFPDFFont font(FPDFText_LoadStandardFont(document(), font_name));
Nicolas Pena4c48b102018-06-13 18:23:46 +00002441 EXPECT_TRUE(font) << font_name << " should be considered a standard font.";
2442 }
Lei Zhang590f4242019-05-15 20:57:26 +00002443 static constexpr const char* kNotStandardFontNames[] = {
Nicolas Pena4c48b102018-06-13 18:23:46 +00002444 "Abcdefg", "ArialB", "Arial-Style",
2445 "Font Name", "FontArial", "NotAStandardFontName",
2446 "TestFontName", "Quack", "Symbol-Italic",
2447 "Zapf"};
Lei Zhang590f4242019-05-15 20:57:26 +00002448 for (const char* font_name : kNotStandardFontNames) {
Tom Sepezffff6c52019-07-30 21:56:19 +00002449 ScopedFPDFFont font(FPDFText_LoadStandardFont(document(), font_name));
Nicolas Pena4c48b102018-06-13 18:23:46 +00002450 EXPECT_FALSE(font) << font_name
2451 << " should not be considered a standard font.";
2452 }
2453}
2454
Lei Zhangab41f252018-12-23 03:10:50 +00002455TEST_F(FPDFEditEmbedderTest, GraphicsData) {
Nicolas Pena603a31d2017-06-14 11:41:18 -04002456 // New page
Tom Sepeze08d2b12018-04-25 18:49:32 +00002457 ScopedFPDFPage page(FPDFPage_New(CreateNewDocument(), 0, 612, 792));
Nicolas Pena603a31d2017-06-14 11:41:18 -04002458
2459 // Create a rect with nontrivial graphics
2460 FPDF_PAGEOBJECT rect1 = FPDFPageObj_CreateNewRect(10, 10, 100, 100);
2461 FPDFPageObj_SetBlendMode(rect1, "Color");
2462 FPDFPage_InsertObject(page.get(), rect1);
2463 EXPECT_TRUE(FPDFPage_GenerateContent(page.get()));
2464
2465 // Check that the ExtGState was created
Lei Zhang107fa7b2018-02-09 21:48:15 +00002466 CPDF_Page* cpage = CPDFPageFromFPDFPage(page.get());
2467 CPDF_Dictionary* graphics_dict = cpage->m_pResources->GetDictFor("ExtGState");
Nicolas Pena603a31d2017-06-14 11:41:18 -04002468 ASSERT_TRUE(graphics_dict);
Lei Zhangf40380f2018-10-12 18:31:51 +00002469 EXPECT_EQ(2u, graphics_dict->size());
Nicolas Pena603a31d2017-06-14 11:41:18 -04002470
2471 // Add a text object causing no change to the graphics dictionary
2472 FPDF_PAGEOBJECT text1 = FPDFPageObj_NewTextObj(document(), "Arial", 12.0f);
2473 // Only alpha, the last component, matters for the graphics dictionary. And
2474 // the default value is 255.
Lei Zhang3475b482019-05-13 18:30:57 +00002475 EXPECT_TRUE(FPDFPageObj_SetFillColor(text1, 100, 100, 100, 255));
Nicolas Pena603a31d2017-06-14 11:41:18 -04002476 FPDFPage_InsertObject(page.get(), text1);
2477 EXPECT_TRUE(FPDFPage_GenerateContent(page.get()));
Lei Zhangf40380f2018-10-12 18:31:51 +00002478 EXPECT_EQ(2u, graphics_dict->size());
Nicolas Pena603a31d2017-06-14 11:41:18 -04002479
2480 // Add a text object increasing the size of the graphics dictionary
2481 FPDF_PAGEOBJECT text2 =
2482 FPDFPageObj_NewTextObj(document(), "Times-Roman", 12.0f);
2483 FPDFPage_InsertObject(page.get(), text2);
2484 FPDFPageObj_SetBlendMode(text2, "Darken");
Lei Zhang3475b482019-05-13 18:30:57 +00002485 EXPECT_TRUE(FPDFPageObj_SetFillColor(text2, 0, 0, 255, 150));
Nicolas Pena603a31d2017-06-14 11:41:18 -04002486 EXPECT_TRUE(FPDFPage_GenerateContent(page.get()));
Lei Zhangf40380f2018-10-12 18:31:51 +00002487 EXPECT_EQ(3u, graphics_dict->size());
Nicolas Pena603a31d2017-06-14 11:41:18 -04002488
2489 // Add a path that should reuse graphics
Nicolas Penace67be42017-06-14 14:52:49 -04002490 FPDF_PAGEOBJECT path = FPDFPageObj_CreateNewPath(400, 100);
Nicolas Pena603a31d2017-06-14 11:41:18 -04002491 FPDFPageObj_SetBlendMode(path, "Darken");
Lei Zhang3475b482019-05-13 18:30:57 +00002492 EXPECT_TRUE(FPDFPageObj_SetFillColor(path, 200, 200, 100, 150));
Nicolas Pena603a31d2017-06-14 11:41:18 -04002493 FPDFPage_InsertObject(page.get(), path);
2494 EXPECT_TRUE(FPDFPage_GenerateContent(page.get()));
Lei Zhangf40380f2018-10-12 18:31:51 +00002495 EXPECT_EQ(3u, graphics_dict->size());
Nicolas Pena603a31d2017-06-14 11:41:18 -04002496
2497 // Add a rect increasing the size of the graphics dictionary
2498 FPDF_PAGEOBJECT rect2 = FPDFPageObj_CreateNewRect(10, 10, 100, 100);
2499 FPDFPageObj_SetBlendMode(rect2, "Darken");
Lei Zhang3475b482019-05-13 18:30:57 +00002500 EXPECT_TRUE(FPDFPageObj_SetFillColor(rect2, 0, 0, 255, 150));
2501 EXPECT_TRUE(FPDFPageObj_SetStrokeColor(rect2, 0, 0, 0, 200));
Nicolas Pena603a31d2017-06-14 11:41:18 -04002502 FPDFPage_InsertObject(page.get(), rect2);
2503 EXPECT_TRUE(FPDFPage_GenerateContent(page.get()));
Lei Zhangf40380f2018-10-12 18:31:51 +00002504 EXPECT_EQ(4u, graphics_dict->size());
Nicolas Pena603a31d2017-06-14 11:41:18 -04002505}
2506
Lei Zhangab41f252018-12-23 03:10:50 +00002507TEST_F(FPDFEditEmbedderTest, DoubleGenerating) {
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002508 // Start with a blank page
Nicolas Penad03ca422017-03-06 13:54:33 -05002509 FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792);
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002510
2511 // Add a red rectangle with some non-default alpha
2512 FPDF_PAGEOBJECT rect = FPDFPageObj_CreateNewRect(10, 10, 100, 100);
Lei Zhang3475b482019-05-13 18:30:57 +00002513 EXPECT_TRUE(FPDFPageObj_SetFillColor(rect, 255, 0, 0, 128));
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002514 EXPECT_TRUE(FPDFPath_SetDrawMode(rect, FPDF_FILLMODE_WINDING, 0));
2515 FPDFPage_InsertObject(page, rect);
2516 EXPECT_TRUE(FPDFPage_GenerateContent(page));
2517
2518 // Check the ExtGState
Lei Zhang107fa7b2018-02-09 21:48:15 +00002519 CPDF_Page* cpage = CPDFPageFromFPDFPage(page);
2520 CPDF_Dictionary* graphics_dict = cpage->m_pResources->GetDictFor("ExtGState");
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002521 ASSERT_TRUE(graphics_dict);
Lei Zhangf40380f2018-10-12 18:31:51 +00002522 EXPECT_EQ(2u, graphics_dict->size());
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002523
2524 // Check the bitmap
Lei Zhang107fa7b2018-02-09 21:48:15 +00002525 {
Lei Zhang30ff2532019-01-31 21:37:55 +00002526 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002527 CompareBitmap(page_bitmap.get(), 612, 792,
2528 "5384da3406d62360ffb5cac4476fff1c");
2529 }
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002530
2531 // Never mind, my new favorite color is blue, increase alpha
Lei Zhang3475b482019-05-13 18:30:57 +00002532 EXPECT_TRUE(FPDFPageObj_SetFillColor(rect, 0, 0, 255, 180));
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002533 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Lei Zhangf40380f2018-10-12 18:31:51 +00002534 EXPECT_EQ(3u, graphics_dict->size());
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002535
2536 // Check that bitmap displays changed content
Lei Zhang107fa7b2018-02-09 21:48:15 +00002537 {
Lei Zhang30ff2532019-01-31 21:37:55 +00002538 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002539 CompareBitmap(page_bitmap.get(), 612, 792,
2540 "2e51656f5073b0bee611d9cd086aa09c");
2541 }
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002542
2543 // And now generate, without changes
2544 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Lei Zhangf40380f2018-10-12 18:31:51 +00002545 EXPECT_EQ(3u, graphics_dict->size());
Lei Zhang107fa7b2018-02-09 21:48:15 +00002546 {
Lei Zhang30ff2532019-01-31 21:37:55 +00002547 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002548 CompareBitmap(page_bitmap.get(), 612, 792,
2549 "2e51656f5073b0bee611d9cd086aa09c");
2550 }
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002551
2552 // Add some text to the page
Nicolas Penab3161852017-05-02 14:12:50 -04002553 FPDF_PAGEOBJECT text_object =
2554 FPDFPageObj_NewTextObj(document(), "Arial", 12.0f);
Lei Zhangf0f67682019-04-08 17:03:21 +00002555 ScopedFPDFWideString text =
Nicolas Penab3161852017-05-02 14:12:50 -04002556 GetFPDFWideString(L"Something something #text# something");
2557 EXPECT_TRUE(FPDFText_SetText(text_object, text.get()));
2558 FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 300, 300);
2559 FPDFPage_InsertObject(page, text_object);
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002560 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Lei Zhang107fa7b2018-02-09 21:48:15 +00002561 CPDF_Dictionary* font_dict = cpage->m_pResources->GetDictFor("Font");
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002562 ASSERT_TRUE(font_dict);
Lei Zhangf40380f2018-10-12 18:31:51 +00002563 EXPECT_EQ(1u, font_dict->size());
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002564
2565 // Generate yet again, check dicts are reasonably sized
2566 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Lei Zhangf40380f2018-10-12 18:31:51 +00002567 EXPECT_EQ(3u, graphics_dict->size());
2568 EXPECT_EQ(1u, font_dict->size());
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002569 FPDF_ClosePage(page);
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002570}
Nicolas Penabe90aae2017-02-27 10:41:41 -05002571
Lei Zhangab41f252018-12-23 03:10:50 +00002572TEST_F(FPDFEditEmbedderTest, LoadSimpleType1Font) {
Nicolas Penad03ca422017-03-06 13:54:33 -05002573 CreateNewDocument();
2574 // TODO(npm): use other fonts after disallowing loading any font as any type
Tom Sepezffff6c52019-07-30 21:56:19 +00002575 RetainPtr<CPDF_Font> stock_font =
Nicolas Penad03ca422017-03-06 13:54:33 -05002576 CPDF_Font::GetStockFont(cpdf_doc(), "Times-Bold");
Tom Sepez20c41a52018-08-29 23:53:53 +00002577 pdfium::span<const uint8_t> span = stock_font->GetFont()->GetFontSpan();
2578 ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(),
2579 FPDF_FONT_TYPE1, false));
Nicolas Penab3161852017-05-02 14:12:50 -04002580 ASSERT_TRUE(font.get());
Tom Sepez525147a2018-05-03 17:19:53 +00002581 CPDF_Font* typed_font = CPDFFontFromFPDFFont(font.get());
Nicolas Penad03ca422017-03-06 13:54:33 -05002582 EXPECT_TRUE(typed_font->IsType1Font());
Nicolas Penabe90aae2017-02-27 10:41:41 -05002583
Lei Zhang710fa992018-05-25 16:24:48 +00002584 const CPDF_Dictionary* font_dict = typed_font->GetFontDict();
Lei Zhanga228ff32020-06-24 17:39:33 +00002585 EXPECT_EQ("Font", font_dict->GetNameFor("Type"));
2586 EXPECT_EQ("Type1", font_dict->GetNameFor("Subtype"));
2587 EXPECT_EQ("TimesNewRomanPS-BoldMT", font_dict->GetNameFor("BaseFont"));
Nicolas Penabe90aae2017-02-27 10:41:41 -05002588 ASSERT_TRUE(font_dict->KeyExist("FirstChar"));
2589 ASSERT_TRUE(font_dict->KeyExist("LastChar"));
2590 EXPECT_EQ(32, font_dict->GetIntegerFor("FirstChar"));
Nicolas Penad03ca422017-03-06 13:54:33 -05002591 EXPECT_EQ(255, font_dict->GetIntegerFor("LastChar"));
2592
Lei Zhangde579ab2018-05-25 21:49:49 +00002593 const CPDF_Array* widths_array = font_dict->GetArrayFor("Widths");
Nicolas Penad03ca422017-03-06 13:54:33 -05002594 ASSERT_TRUE(widths_array);
Lei Zhangf40380f2018-10-12 18:31:51 +00002595 ASSERT_EQ(224u, widths_array->size());
Nicolas Penabe90aae2017-02-27 10:41:41 -05002596 EXPECT_EQ(250, widths_array->GetNumberAt(0));
Nicolas Penad03ca422017-03-06 13:54:33 -05002597 EXPECT_EQ(569, widths_array->GetNumberAt(11));
2598 EXPECT_EQ(500, widths_array->GetNumberAt(223));
Tom Sepez20c41a52018-08-29 23:53:53 +00002599 CheckFontDescriptor(font_dict, FPDF_FONT_TYPE1, true, false, span);
Nicolas Penad03ca422017-03-06 13:54:33 -05002600}
Nicolas Penabe90aae2017-02-27 10:41:41 -05002601
Lei Zhangab41f252018-12-23 03:10:50 +00002602TEST_F(FPDFEditEmbedderTest, LoadSimpleTrueTypeFont) {
Nicolas Penad03ca422017-03-06 13:54:33 -05002603 CreateNewDocument();
Tom Sepezffff6c52019-07-30 21:56:19 +00002604 RetainPtr<CPDF_Font> stock_font =
2605 CPDF_Font::GetStockFont(cpdf_doc(), "Courier");
Tom Sepez20c41a52018-08-29 23:53:53 +00002606 pdfium::span<const uint8_t> span = stock_font->GetFont()->GetFontSpan();
2607 ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(),
2608 FPDF_FONT_TRUETYPE, false));
Nicolas Penab3161852017-05-02 14:12:50 -04002609 ASSERT_TRUE(font.get());
Tom Sepez525147a2018-05-03 17:19:53 +00002610 CPDF_Font* typed_font = CPDFFontFromFPDFFont(font.get());
Nicolas Penad03ca422017-03-06 13:54:33 -05002611 EXPECT_TRUE(typed_font->IsTrueTypeFont());
Nicolas Penabe90aae2017-02-27 10:41:41 -05002612
Lei Zhang710fa992018-05-25 16:24:48 +00002613 const CPDF_Dictionary* font_dict = typed_font->GetFontDict();
Lei Zhanga228ff32020-06-24 17:39:33 +00002614 EXPECT_EQ("Font", font_dict->GetNameFor("Type"));
2615 EXPECT_EQ("TrueType", font_dict->GetNameFor("Subtype"));
2616 EXPECT_EQ("CourierNewPSMT", font_dict->GetNameFor("BaseFont"));
Nicolas Penad03ca422017-03-06 13:54:33 -05002617 ASSERT_TRUE(font_dict->KeyExist("FirstChar"));
2618 ASSERT_TRUE(font_dict->KeyExist("LastChar"));
2619 EXPECT_EQ(32, font_dict->GetIntegerFor("FirstChar"));
2620 EXPECT_EQ(255, font_dict->GetIntegerFor("LastChar"));
Nicolas Penabe90aae2017-02-27 10:41:41 -05002621
Lei Zhangde579ab2018-05-25 21:49:49 +00002622 const CPDF_Array* widths_array = font_dict->GetArrayFor("Widths");
Nicolas Penad03ca422017-03-06 13:54:33 -05002623 ASSERT_TRUE(widths_array);
Lei Zhangf40380f2018-10-12 18:31:51 +00002624 ASSERT_EQ(224u, widths_array->size());
Nicolas Penad03ca422017-03-06 13:54:33 -05002625 EXPECT_EQ(600, widths_array->GetNumberAt(33));
2626 EXPECT_EQ(600, widths_array->GetNumberAt(74));
2627 EXPECT_EQ(600, widths_array->GetNumberAt(223));
Tom Sepez20c41a52018-08-29 23:53:53 +00002628 CheckFontDescriptor(font_dict, FPDF_FONT_TRUETYPE, false, false, span);
Nicolas Penad03ca422017-03-06 13:54:33 -05002629}
2630
Lei Zhangab41f252018-12-23 03:10:50 +00002631TEST_F(FPDFEditEmbedderTest, LoadCIDType0Font) {
Nicolas Penad03ca422017-03-06 13:54:33 -05002632 CreateNewDocument();
Tom Sepezffff6c52019-07-30 21:56:19 +00002633 RetainPtr<CPDF_Font> stock_font =
Nicolas Penad03ca422017-03-06 13:54:33 -05002634 CPDF_Font::GetStockFont(cpdf_doc(), "Times-Roman");
Tom Sepez20c41a52018-08-29 23:53:53 +00002635 pdfium::span<const uint8_t> span = stock_font->GetFont()->GetFontSpan();
2636 ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(),
2637 FPDF_FONT_TYPE1, 1));
Nicolas Penab3161852017-05-02 14:12:50 -04002638 ASSERT_TRUE(font.get());
Tom Sepez525147a2018-05-03 17:19:53 +00002639 CPDF_Font* typed_font = CPDFFontFromFPDFFont(font.get());
Nicolas Penad03ca422017-03-06 13:54:33 -05002640 EXPECT_TRUE(typed_font->IsCIDFont());
2641
2642 // Check font dictionary entries
Lei Zhang710fa992018-05-25 16:24:48 +00002643 const CPDF_Dictionary* font_dict = typed_font->GetFontDict();
Lei Zhanga228ff32020-06-24 17:39:33 +00002644 EXPECT_EQ("Font", font_dict->GetNameFor("Type"));
2645 EXPECT_EQ("Type0", font_dict->GetNameFor("Subtype"));
2646 EXPECT_EQ("TimesNewRomanPSMT-Identity-H", font_dict->GetNameFor("BaseFont"));
2647 EXPECT_EQ("Identity-H", font_dict->GetNameFor("Encoding"));
Lei Zhangde579ab2018-05-25 21:49:49 +00002648 const CPDF_Array* descendant_array =
2649 font_dict->GetArrayFor("DescendantFonts");
Nicolas Penad03ca422017-03-06 13:54:33 -05002650 ASSERT_TRUE(descendant_array);
Lei Zhangf40380f2018-10-12 18:31:51 +00002651 EXPECT_EQ(1u, descendant_array->size());
Nicolas Penad03ca422017-03-06 13:54:33 -05002652
2653 // Check the CIDFontDict
Lei Zhangde579ab2018-05-25 21:49:49 +00002654 const CPDF_Dictionary* cidfont_dict = descendant_array->GetDictAt(0);
Lei Zhanga228ff32020-06-24 17:39:33 +00002655 EXPECT_EQ("Font", cidfont_dict->GetNameFor("Type"));
2656 EXPECT_EQ("CIDFontType0", cidfont_dict->GetNameFor("Subtype"));
2657 EXPECT_EQ("TimesNewRomanPSMT", cidfont_dict->GetNameFor("BaseFont"));
Lei Zhangb1ec2802018-05-25 21:55:24 +00002658 const CPDF_Dictionary* cidinfo_dict =
2659 cidfont_dict->GetDictFor("CIDSystemInfo");
Nicolas Penad03ca422017-03-06 13:54:33 -05002660 ASSERT_TRUE(cidinfo_dict);
Lei Zhang9c950b12019-01-16 19:06:37 +00002661 const CPDF_Object* registry = cidinfo_dict->GetObjectFor("Registry");
2662 ASSERT_TRUE(registry);
KDr28da0e1b2019-01-17 03:44:29 +00002663 EXPECT_EQ(CPDF_Object::kString, registry->GetType());
Lei Zhang9c950b12019-01-16 19:06:37 +00002664 EXPECT_EQ("Adobe", registry->GetString());
2665 const CPDF_Object* ordering = cidinfo_dict->GetObjectFor("Ordering");
2666 ASSERT_TRUE(ordering);
KDr28da0e1b2019-01-17 03:44:29 +00002667 EXPECT_EQ(CPDF_Object::kString, ordering->GetType());
Lei Zhang9c950b12019-01-16 19:06:37 +00002668 EXPECT_EQ("Identity", ordering->GetString());
Nicolas Penad03ca422017-03-06 13:54:33 -05002669 EXPECT_EQ(0, cidinfo_dict->GetNumberFor("Supplement"));
Tom Sepez20c41a52018-08-29 23:53:53 +00002670 CheckFontDescriptor(cidfont_dict, FPDF_FONT_TYPE1, false, false, span);
Nicolas Penad03ca422017-03-06 13:54:33 -05002671
2672 // Check widths
Lei Zhangde579ab2018-05-25 21:49:49 +00002673 const CPDF_Array* widths_array = cidfont_dict->GetArrayFor("W");
Nicolas Penad03ca422017-03-06 13:54:33 -05002674 ASSERT_TRUE(widths_array);
Lei Zhangf40380f2018-10-12 18:31:51 +00002675 EXPECT_GT(widths_array->size(), 1u);
Nicolas Penad03ca422017-03-06 13:54:33 -05002676 CheckCompositeFontWidths(widths_array, typed_font);
2677}
2678
Lei Zhangab41f252018-12-23 03:10:50 +00002679TEST_F(FPDFEditEmbedderTest, LoadCIDType2Font) {
Nicolas Penad03ca422017-03-06 13:54:33 -05002680 CreateNewDocument();
Tom Sepezffff6c52019-07-30 21:56:19 +00002681 RetainPtr<CPDF_Font> stock_font =
Nicolas Penad03ca422017-03-06 13:54:33 -05002682 CPDF_Font::GetStockFont(cpdf_doc(), "Helvetica-Oblique");
Tom Sepez20c41a52018-08-29 23:53:53 +00002683 pdfium::span<const uint8_t> span = stock_font->GetFont()->GetFontSpan();
2684 ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(),
2685 FPDF_FONT_TRUETYPE, 1));
Nicolas Penab3161852017-05-02 14:12:50 -04002686 ASSERT_TRUE(font.get());
Tom Sepez525147a2018-05-03 17:19:53 +00002687 CPDF_Font* typed_font = CPDFFontFromFPDFFont(font.get());
Nicolas Penad03ca422017-03-06 13:54:33 -05002688 EXPECT_TRUE(typed_font->IsCIDFont());
2689
2690 // Check font dictionary entries
Lei Zhang710fa992018-05-25 16:24:48 +00002691 const CPDF_Dictionary* font_dict = typed_font->GetFontDict();
Lei Zhanga228ff32020-06-24 17:39:33 +00002692 EXPECT_EQ("Font", font_dict->GetNameFor("Type"));
2693 EXPECT_EQ("Type0", font_dict->GetNameFor("Subtype"));
2694 EXPECT_EQ("Arial-ItalicMT", font_dict->GetNameFor("BaseFont"));
2695 EXPECT_EQ("Identity-H", font_dict->GetNameFor("Encoding"));
Lei Zhangde579ab2018-05-25 21:49:49 +00002696 const CPDF_Array* descendant_array =
2697 font_dict->GetArrayFor("DescendantFonts");
Nicolas Penad03ca422017-03-06 13:54:33 -05002698 ASSERT_TRUE(descendant_array);
Lei Zhangf40380f2018-10-12 18:31:51 +00002699 EXPECT_EQ(1u, descendant_array->size());
Nicolas Penad03ca422017-03-06 13:54:33 -05002700
2701 // Check the CIDFontDict
Lei Zhangde579ab2018-05-25 21:49:49 +00002702 const CPDF_Dictionary* cidfont_dict = descendant_array->GetDictAt(0);
Lei Zhanga228ff32020-06-24 17:39:33 +00002703 EXPECT_EQ("Font", cidfont_dict->GetNameFor("Type"));
2704 EXPECT_EQ("CIDFontType2", cidfont_dict->GetNameFor("Subtype"));
2705 EXPECT_EQ("Arial-ItalicMT", cidfont_dict->GetNameFor("BaseFont"));
Lei Zhangb1ec2802018-05-25 21:55:24 +00002706 const CPDF_Dictionary* cidinfo_dict =
2707 cidfont_dict->GetDictFor("CIDSystemInfo");
Nicolas Penad03ca422017-03-06 13:54:33 -05002708 ASSERT_TRUE(cidinfo_dict);
2709 EXPECT_EQ("Adobe", cidinfo_dict->GetStringFor("Registry"));
2710 EXPECT_EQ("Identity", cidinfo_dict->GetStringFor("Ordering"));
2711 EXPECT_EQ(0, cidinfo_dict->GetNumberFor("Supplement"));
Tom Sepez20c41a52018-08-29 23:53:53 +00002712 CheckFontDescriptor(cidfont_dict, FPDF_FONT_TRUETYPE, false, true, span);
Nicolas Penad03ca422017-03-06 13:54:33 -05002713
2714 // Check widths
Lei Zhangde579ab2018-05-25 21:49:49 +00002715 const CPDF_Array* widths_array = cidfont_dict->GetArrayFor("W");
Nicolas Penad03ca422017-03-06 13:54:33 -05002716 ASSERT_TRUE(widths_array);
2717 CheckCompositeFontWidths(widths_array, typed_font);
Nicolas Penabe90aae2017-02-27 10:41:41 -05002718}
rbpotterce8e51e2017-04-28 12:42:47 -07002719
Lei Zhangab41f252018-12-23 03:10:50 +00002720TEST_F(FPDFEditEmbedderTest, NormalizeNegativeRotation) {
rbpotterce8e51e2017-04-28 12:42:47 -07002721 // Load document with a -90 degree rotation
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00002722 ASSERT_TRUE(OpenDocument("bug_713197.pdf"));
rbpotterce8e51e2017-04-28 12:42:47 -07002723 FPDF_PAGE page = LoadPage(0);
2724 EXPECT_NE(nullptr, page);
2725
2726 EXPECT_EQ(3, FPDFPage_GetRotation(page));
2727 UnloadPage(page);
2728}
Nicolas Penab3161852017-05-02 14:12:50 -04002729
Hui Yingst9d6ec8f2020-11-05 23:35:38 +00002730TEST_F(FPDFEditEmbedderTest, AddTrueTypeFontText) {
Nicolas Penab3161852017-05-02 14:12:50 -04002731 // Start with a blank page
2732 FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792);
2733 {
Tom Sepezffff6c52019-07-30 21:56:19 +00002734 RetainPtr<CPDF_Font> stock_font =
2735 CPDF_Font::GetStockFont(cpdf_doc(), "Arial");
Tom Sepez20c41a52018-08-29 23:53:53 +00002736 pdfium::span<const uint8_t> span = stock_font->GetFont()->GetFontSpan();
2737 ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(),
2738 FPDF_FONT_TRUETYPE, 0));
Nicolas Penab3161852017-05-02 14:12:50 -04002739 ASSERT_TRUE(font.get());
2740
2741 // Add some text to the page
2742 FPDF_PAGEOBJECT text_object =
2743 FPDFPageObj_CreateTextObj(document(), font.get(), 12.0f);
2744 EXPECT_TRUE(text_object);
Hui Yingst3b6136a2020-06-09 00:39:33 +00002745 ScopedFPDFWideString text = GetFPDFWideString(kLoadedFontText);
Nicolas Penab3161852017-05-02 14:12:50 -04002746 EXPECT_TRUE(FPDFText_SetText(text_object, text.get()));
2747 FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 400, 400);
2748 FPDFPage_InsertObject(page, text_object);
Lei Zhang30ff2532019-01-31 21:37:55 +00002749 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingst3b6136a2020-06-09 00:39:33 +00002750 CompareBitmap(page_bitmap.get(), 612, 792, kLoadedFontTextChecksum);
Nicolas Penab3161852017-05-02 14:12:50 -04002751
2752 // Add some more text, same font
2753 FPDF_PAGEOBJECT text_object2 =
2754 FPDFPageObj_CreateTextObj(document(), font.get(), 15.0f);
Lei Zhangf0f67682019-04-08 17:03:21 +00002755 ScopedFPDFWideString text2 = GetFPDFWideString(L"Bigger font size");
Nicolas Penab3161852017-05-02 14:12:50 -04002756 EXPECT_TRUE(FPDFText_SetText(text_object2, text2.get()));
2757 FPDFPageObj_Transform(text_object2, 1, 0, 0, 1, 200, 200);
2758 FPDFPage_InsertObject(page, text_object2);
Nicolas Penab3161852017-05-02 14:12:50 -04002759 }
Lei Zhang30ff2532019-01-31 21:37:55 +00002760 ScopedFPDFBitmap page_bitmap2 = RenderPage(page);
Hui Yingst9d6ec8f2020-11-05 23:35:38 +00002761#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
2762#if defined(OS_WIN)
2763 const char kInsertTrueTypeChecksum[] = "c80411cb051a9d45c4b7a8ec8a72637d";
2764#else
2765 const char kInsertTrueTypeChecksum[] = "f2ee263957a5584f3c72424e8683ac8c";
2766#endif // defined(OS_WIN)
2767#else
Hui Yingst3b6136a2020-06-09 00:39:33 +00002768#if defined(OS_WIN)
2769 const char kInsertTrueTypeChecksum[] = "2199b579c49ab5f80c246a586a80ee90";
Lei Zhang0c03d632020-07-30 17:05:36 +00002770#elif defined(OS_APPLE)
Hui Yingst3b6136a2020-06-09 00:39:33 +00002771 const char kInsertTrueTypeChecksum[] = "9a1a7dfebe659513691aadd0d95b8d50";
Nicolas Penab3161852017-05-02 14:12:50 -04002772#else
Hui Yingst3b6136a2020-06-09 00:39:33 +00002773 const char kInsertTrueTypeChecksum[] = "c1d10cce1761c4a998a16b2562030568";
Lei Zhange4cdac52019-04-30 16:45:57 +00002774#endif
Hui Yingst9d6ec8f2020-11-05 23:35:38 +00002775#endif // defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
Hui Yingst3b6136a2020-06-09 00:39:33 +00002776 CompareBitmap(page_bitmap2.get(), 612, 792, kInsertTrueTypeChecksum);
Nicolas Penab3161852017-05-02 14:12:50 -04002777
Nicolas Pena207b7272017-05-26 17:37:06 -04002778 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Nicolas Penab3161852017-05-02 14:12:50 -04002779 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
2780 FPDF_ClosePage(page);
Dan Sinclair04e4dc82017-10-18 12:17:14 -04002781
Hui Yingst3b6136a2020-06-09 00:39:33 +00002782 VerifySavedDocument(612, 792, kInsertTrueTypeChecksum);
Nicolas Penab3161852017-05-02 14:12:50 -04002783}
Nicolas Penaf45ade32017-05-03 10:23:49 -04002784
Lei Zhangab41f252018-12-23 03:10:50 +00002785TEST_F(FPDFEditEmbedderTest, TransformAnnot) {
Jane Liueda65252017-06-07 11:31:27 -04002786 // Open a file with one annotation and load its first page.
2787 ASSERT_TRUE(OpenDocument("annotation_highlight_long_content.pdf"));
Lei Zhang75c81712018-02-08 17:22:39 +00002788 FPDF_PAGE page = LoadPage(0);
Jane Liueda65252017-06-07 11:31:27 -04002789 ASSERT_TRUE(page);
2790
Lei Zhanga21d5932018-02-05 18:28:38 +00002791 {
2792 // Add an underline annotation to the page without specifying its rectangle.
Tom Sepeze08d2b12018-04-25 18:49:32 +00002793 ScopedFPDFAnnotation annot(
Lei Zhanga21d5932018-02-05 18:28:38 +00002794 FPDFPage_CreateAnnot(page, FPDF_ANNOT_UNDERLINE));
2795 ASSERT_TRUE(annot);
Jane Liueda65252017-06-07 11:31:27 -04002796
Lei Zhanga21d5932018-02-05 18:28:38 +00002797 // FPDFPage_TransformAnnots() should run without errors when modifying
2798 // annotation rectangles.
2799 FPDFPage_TransformAnnots(page, 1, 2, 3, 4, 5, 6);
2800 }
Jane Liueda65252017-06-07 11:31:27 -04002801 UnloadPage(page);
2802}
2803
Nicolas Penaf45ade32017-05-03 10:23:49 -04002804// TODO(npm): Add tests using Japanese fonts in other OS.
Lei Zhangfe8e7582020-08-05 19:23:16 +00002805#if defined(OS_LINUX) || defined(OS_CHROMEOS)
Hui Yingst5ce6a3f2020-11-11 02:29:09 +00002806TEST_F(FPDFEditEmbedderTest, AddCIDFontText) {
Nicolas Penaf45ade32017-05-03 10:23:49 -04002807 // Start with a blank page
2808 FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792);
2809 CFX_Font CIDfont;
2810 {
2811 // First, get the data from the font
2812 CIDfont.LoadSubst("IPAGothic", 1, 0, 400, 0, 932, 0);
2813 EXPECT_EQ("IPAGothic", CIDfont.GetFaceName());
Tom Sepez20c41a52018-08-29 23:53:53 +00002814 pdfium::span<const uint8_t> span = CIDfont.GetFontSpan();
Nicolas Penaf45ade32017-05-03 10:23:49 -04002815
2816 // Load the data into a FPDF_Font.
Tom Sepez20c41a52018-08-29 23:53:53 +00002817 ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(),
2818 FPDF_FONT_TRUETYPE, 1));
Nicolas Penaf45ade32017-05-03 10:23:49 -04002819 ASSERT_TRUE(font.get());
2820
2821 // Add some text to the page
2822 FPDF_PAGEOBJECT text_object =
2823 FPDFPageObj_CreateTextObj(document(), font.get(), 12.0f);
2824 ASSERT_TRUE(text_object);
2825 std::wstring wstr = L"ABCDEFGhijklmnop.";
Lei Zhangf0f67682019-04-08 17:03:21 +00002826 ScopedFPDFWideString text = GetFPDFWideString(wstr);
Nicolas Penaf45ade32017-05-03 10:23:49 -04002827 EXPECT_TRUE(FPDFText_SetText(text_object, text.get()));
2828 FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 200, 200);
2829 FPDFPage_InsertObject(page, text_object);
2830
2831 // And add some Japanese characters
2832 FPDF_PAGEOBJECT text_object2 =
2833 FPDFPageObj_CreateTextObj(document(), font.get(), 18.0f);
2834 ASSERT_TRUE(text_object2);
2835 std::wstring wstr2 =
2836 L"\u3053\u3093\u306B\u3061\u306f\u4e16\u754C\u3002\u3053\u3053\u306B1"
2837 L"\u756A";
Lei Zhangf0f67682019-04-08 17:03:21 +00002838 ScopedFPDFWideString text2 = GetFPDFWideString(wstr2);
Nicolas Penaf45ade32017-05-03 10:23:49 -04002839 EXPECT_TRUE(FPDFText_SetText(text_object2, text2.get()));
2840 FPDFPageObj_Transform(text_object2, 1, 0, 0, 1, 100, 500);
2841 FPDFPage_InsertObject(page, text_object2);
2842 }
2843
Nicolas Pena207b7272017-05-26 17:37:06 -04002844 // Check that the text renders properly.
Hui Yingst5ce6a3f2020-11-11 02:29:09 +00002845#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
2846 static constexpr char md5[] = "683eefd6c31206de23b7f709b66e6daf";
2847#else
2848 static constexpr char md5[] = "5159a72903fe57bf0cf645c894de8a74";
2849#endif
Lei Zhang107fa7b2018-02-09 21:48:15 +00002850 {
Lei Zhang30ff2532019-01-31 21:37:55 +00002851 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002852 CompareBitmap(page_bitmap.get(), 612, 792, md5);
2853 }
Nicolas Penaf45ade32017-05-03 10:23:49 -04002854
2855 // Save the document, close the page.
Nicolas Pena207b7272017-05-26 17:37:06 -04002856 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Nicolas Penaf45ade32017-05-03 10:23:49 -04002857 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
2858 FPDF_ClosePage(page);
Dan Sinclair04e4dc82017-10-18 12:17:14 -04002859
2860 VerifySavedDocument(612, 792, md5);
Nicolas Penaf45ade32017-05-03 10:23:49 -04002861}
Lei Zhangfe8e7582020-08-05 19:23:16 +00002862#endif // defined(OS_LINUX) || defined(OS_CHROMEOS)
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -04002863
Lei Zhang03e5e682019-09-16 19:45:55 +00002864// TODO(crbug.com/pdfium/11): Fix this test and enable.
2865#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
2866#define MAYBE_SaveAndRender DISABLED_SaveAndRender
2867#else
2868#define MAYBE_SaveAndRender SaveAndRender
2869#endif
2870TEST_F(FPDFEditEmbedderTest, MAYBE_SaveAndRender) {
Nicolas Penaa0b48aa2017-06-29 11:01:46 -04002871 const char md5[] = "3c20472b0552c0c22b88ab1ed8c6202b";
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -04002872 {
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00002873 ASSERT_TRUE(OpenDocument("bug_779.pdf"));
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -04002874 FPDF_PAGE page = LoadPage(0);
2875 ASSERT_NE(nullptr, page);
2876
2877 // Now add a more complex blue path.
2878 FPDF_PAGEOBJECT green_path = FPDFPageObj_CreateNewPath(20, 20);
Lei Zhang3475b482019-05-13 18:30:57 +00002879 EXPECT_TRUE(FPDFPageObj_SetFillColor(green_path, 0, 255, 0, 200));
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -04002880 // TODO(npm): stroking will cause the MD5s to differ.
2881 EXPECT_TRUE(FPDFPath_SetDrawMode(green_path, FPDF_FILLMODE_WINDING, 0));
2882 EXPECT_TRUE(FPDFPath_LineTo(green_path, 20, 63));
2883 EXPECT_TRUE(FPDFPath_BezierTo(green_path, 55, 55, 78, 78, 90, 90));
2884 EXPECT_TRUE(FPDFPath_LineTo(green_path, 133, 133));
2885 EXPECT_TRUE(FPDFPath_LineTo(green_path, 133, 33));
2886 EXPECT_TRUE(FPDFPath_BezierTo(green_path, 38, 33, 39, 36, 40, 40));
2887 EXPECT_TRUE(FPDFPath_Close(green_path));
2888 FPDFPage_InsertObject(page, green_path);
Tom Sepeze08d2b12018-04-25 18:49:32 +00002889 ScopedFPDFBitmap page_bitmap = RenderLoadedPage(page);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002890 CompareBitmap(page_bitmap.get(), 612, 792, md5);
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -04002891
2892 // Now save the result, closing the page and document
2893 EXPECT_TRUE(FPDFPage_GenerateContent(page));
2894 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
2895 UnloadPage(page);
2896 }
Dan Sinclair04e4dc82017-10-18 12:17:14 -04002897
2898 VerifySavedDocument(612, 792, md5);
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -04002899}
Jane Liu28fb7ba2017-08-02 21:45:57 -04002900
Lei Zhangab41f252018-12-23 03:10:50 +00002901TEST_F(FPDFEditEmbedderTest, AddMark) {
Henrique Nakashimad8df8c32018-07-12 22:15:09 +00002902 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00002903 ASSERT_TRUE(OpenDocument("text_in_page_marked.pdf"));
Henrique Nakashimad8df8c32018-07-12 22:15:09 +00002904 FPDF_PAGE page = LoadPage(0);
2905 ASSERT_TRUE(page);
2906
Lei Zhang2697cb12019-07-03 18:14:29 +00002907 CheckMarkCounts(page, 1, 19, 8, 4, 9, 1);
Henrique Nakashimad8df8c32018-07-12 22:15:09 +00002908
2909 // Add to the first page object a "Bounds" mark with "Position": "First".
2910 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0);
2911 FPDF_PAGEOBJECTMARK mark = FPDFPageObj_AddMark(page_object, "Bounds");
2912 EXPECT_TRUE(mark);
Henrique Nakashimaa3406772018-07-13 19:10:53 +00002913 EXPECT_TRUE(FPDFPageObjMark_SetStringParam(document(), page_object, mark,
2914 "Position", "First"));
Henrique Nakashimad8df8c32018-07-12 22:15:09 +00002915
Lei Zhang2697cb12019-07-03 18:14:29 +00002916 CheckMarkCounts(page, 1, 19, 8, 4, 9, 2);
Henrique Nakashimad8df8c32018-07-12 22:15:09 +00002917
2918 // Save the file
2919 EXPECT_TRUE(FPDFPage_GenerateContent(page));
2920 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
2921 UnloadPage(page);
2922
2923 // Re-open the file and check the new mark is present.
Lei Zhang0b494052019-01-31 21:41:15 +00002924 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashimad8df8c32018-07-12 22:15:09 +00002925 FPDF_PAGE saved_page = LoadSavedPage(0);
2926
Lei Zhang2697cb12019-07-03 18:14:29 +00002927 CheckMarkCounts(saved_page, 1, 19, 8, 4, 9, 2);
2928
2929 CloseSavedPage(saved_page);
2930 CloseSavedDocument();
2931}
2932
Hui Yingst04440af2020-07-27 22:39:02 +00002933TEST_F(FPDFEditEmbedderTest, AddMarkCompressedStream) {
Lei Zhang2697cb12019-07-03 18:14:29 +00002934 // Load document with some text in a compressed stream.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00002935 ASSERT_TRUE(OpenDocument("hello_world_compressed_stream.pdf"));
Lei Zhang2697cb12019-07-03 18:14:29 +00002936 FPDF_PAGE page = LoadPage(0);
2937 ASSERT_TRUE(page);
2938
2939 // Render and check there are no marks.
2940 {
2941 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00002942 CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldChecksum);
Lei Zhang2697cb12019-07-03 18:14:29 +00002943 }
2944 CheckMarkCounts(page, 0, 2, 0, 0, 0, 0);
2945
2946 // Add to the first page object a "Bounds" mark with "Position": "First".
2947 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0);
2948 FPDF_PAGEOBJECTMARK mark = FPDFPageObj_AddMark(page_object, "Bounds");
2949 EXPECT_TRUE(mark);
2950 EXPECT_TRUE(FPDFPageObjMark_SetStringParam(document(), page_object, mark,
2951 "Position", "First"));
2952
2953 // Render and check there is 1 mark.
2954 {
2955 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00002956 CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldChecksum);
Lei Zhang2697cb12019-07-03 18:14:29 +00002957 }
2958 CheckMarkCounts(page, 0, 2, 0, 0, 0, 1);
2959
2960 // Save the file.
2961 EXPECT_TRUE(FPDFPage_GenerateContent(page));
2962 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
2963 UnloadPage(page);
2964
2965 // Re-open the file and check the new mark is present.
2966 ASSERT_TRUE(OpenSavedDocument());
2967 FPDF_PAGE saved_page = LoadSavedPage(0);
2968
2969 {
2970 ScopedFPDFBitmap page_bitmap = RenderPage(saved_page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00002971 CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldChecksum);
Lei Zhang2697cb12019-07-03 18:14:29 +00002972 }
2973 CheckMarkCounts(saved_page, 0, 2, 0, 0, 0, 1);
Henrique Nakashimad8df8c32018-07-12 22:15:09 +00002974
2975 CloseSavedPage(saved_page);
2976 CloseSavedDocument();
2977}
2978
Lei Zhangab41f252018-12-23 03:10:50 +00002979TEST_F(FPDFEditEmbedderTest, SetMarkParam) {
Henrique Nakashimaa3406772018-07-13 19:10:53 +00002980 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00002981 ASSERT_TRUE(OpenDocument("text_in_page_marked.pdf"));
Henrique Nakashimaa3406772018-07-13 19:10:53 +00002982 FPDF_PAGE page = LoadPage(0);
2983 ASSERT_TRUE(page);
2984
2985 constexpr int kExpectedObjectCount = 19;
2986 CheckMarkCounts(page, 1, kExpectedObjectCount, 8, 4, 9, 1);
2987
2988 // Check the "Bounds" mark's "Position" param is "Last".
2989 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 18);
2990 FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, 1);
2991 ASSERT_TRUE(mark);
2992 char buffer[256];
Henrique Nakashimac3099d12018-09-18 18:08:15 +00002993 unsigned long name_len = 999u;
2994 ASSERT_TRUE(FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len));
2995 EXPECT_EQ((6u + 1u) * 2u, name_len);
Henrique Nakashimaa3406772018-07-13 19:10:53 +00002996 ASSERT_EQ(L"Bounds",
2997 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)));
2998 unsigned long out_buffer_len;
2999 ASSERT_TRUE(FPDFPageObjMark_GetParamStringValue(
3000 mark, "Position", buffer, sizeof(buffer), &out_buffer_len));
3001 ASSERT_EQ(L"Last",
3002 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)));
3003
3004 // Set is to "End".
3005 EXPECT_TRUE(FPDFPageObjMark_SetStringParam(document(), page_object, mark,
3006 "Position", "End"));
3007
3008 // Verify the object passed must correspond to the mark passed.
3009 FPDF_PAGEOBJECT another_page_object = FPDFPage_GetObject(page, 17);
3010 EXPECT_FALSE(FPDFPageObjMark_SetStringParam(document(), another_page_object,
3011 mark, "Position", "End"));
3012
3013 // Verify nothing else changed.
3014 CheckMarkCounts(page, 1, kExpectedObjectCount, 8, 4, 9, 1);
3015
3016 // Verify "Position" now maps to "End".
3017 EXPECT_TRUE(FPDFPageObjMark_GetParamStringValue(
3018 mark, "Position", buffer, sizeof(buffer), &out_buffer_len));
3019 EXPECT_EQ(L"End",
3020 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)));
3021
3022 // Save the file
3023 EXPECT_TRUE(FPDFPage_GenerateContent(page));
3024 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
3025 UnloadPage(page);
3026
3027 // Re-open the file and cerify "Position" still maps to "End".
Lei Zhang0b494052019-01-31 21:41:15 +00003028 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashimaa3406772018-07-13 19:10:53 +00003029 FPDF_PAGE saved_page = LoadSavedPage(0);
3030
3031 CheckMarkCounts(saved_page, 1, kExpectedObjectCount, 8, 4, 9, 1);
3032 page_object = FPDFPage_GetObject(saved_page, 18);
3033 mark = FPDFPageObj_GetMark(page_object, 1);
3034 EXPECT_TRUE(FPDFPageObjMark_GetParamStringValue(
3035 mark, "Position", buffer, sizeof(buffer), &out_buffer_len));
3036 EXPECT_EQ(L"End",
3037 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)));
3038
3039 CloseSavedPage(saved_page);
3040 CloseSavedDocument();
3041}
3042
Hui Yingst9d6ec8f2020-11-05 23:35:38 +00003043TEST_F(FPDFEditEmbedderTest, AddMarkedText) {
Henrique Nakashima144107d2018-07-10 21:04:05 +00003044 // Start with a blank page.
3045 FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792);
3046
Tom Sepezffff6c52019-07-30 21:56:19 +00003047 RetainPtr<CPDF_Font> stock_font =
3048 CPDF_Font::GetStockFont(cpdf_doc(), "Arial");
Tom Sepez20c41a52018-08-29 23:53:53 +00003049 pdfium::span<const uint8_t> span = stock_font->GetFont()->GetFontSpan();
3050 ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(),
3051 FPDF_FONT_TRUETYPE, 0));
Henrique Nakashima144107d2018-07-10 21:04:05 +00003052 ASSERT_TRUE(font.get());
3053
3054 // Add some text to the page.
3055 FPDF_PAGEOBJECT text_object =
3056 FPDFPageObj_CreateTextObj(document(), font.get(), 12.0f);
3057
3058 EXPECT_TRUE(text_object);
Hui Yingst3b6136a2020-06-09 00:39:33 +00003059 ScopedFPDFWideString text1 = GetFPDFWideString(kLoadedFontText);
Henrique Nakashima144107d2018-07-10 21:04:05 +00003060 EXPECT_TRUE(FPDFText_SetText(text_object, text1.get()));
3061 FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 400, 400);
3062 FPDFPage_InsertObject(page, text_object);
3063
3064 // Add a mark with the tag "TestMarkName" to that text.
3065 EXPECT_EQ(0, FPDFPageObj_CountMarks(text_object));
Henrique Nakashima6fc8d872018-09-18 16:48:31 +00003066 FPDF_PAGEOBJECTMARK mark = FPDFPageObj_AddMark(text_object, "Test Mark Name");
Henrique Nakashima144107d2018-07-10 21:04:05 +00003067 EXPECT_TRUE(mark);
3068 EXPECT_EQ(1, FPDFPageObj_CountMarks(text_object));
3069 EXPECT_EQ(mark, FPDFPageObj_GetMark(text_object, 0));
3070 char buffer[256];
Henrique Nakashimac3099d12018-09-18 18:08:15 +00003071 unsigned long name_len = 999u;
3072 ASSERT_TRUE(FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len));
3073 EXPECT_EQ((14u + 1u) * 2, name_len);
Henrique Nakashima144107d2018-07-10 21:04:05 +00003074 std::wstring name =
3075 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer));
Henrique Nakashima6fc8d872018-09-18 16:48:31 +00003076 EXPECT_EQ(L"Test Mark Name", name);
Henrique Nakashima144107d2018-07-10 21:04:05 +00003077
3078 // Add parameters:
3079 // - int "IntKey" : 42
3080 // - string "StringKey": "StringValue"
Henrique Nakashima07520f62018-07-12 19:45:29 +00003081 // - blob "BlobKey": "\x01\x02\x03\0BlobValue1\0\0\0BlobValue2\0"
Lei Zhang590f4242019-05-15 20:57:26 +00003082 constexpr size_t kBlobLen = 28;
Lei Zhangd3b028b2018-11-30 22:22:00 +00003083 char block_value[kBlobLen];
3084 memcpy(block_value, "\x01\x02\x03\0BlobValue1\0\0\0BlobValue2\0", kBlobLen);
Henrique Nakashima144107d2018-07-10 21:04:05 +00003085 EXPECT_EQ(0, FPDFPageObjMark_CountParams(mark));
Henrique Nakashimaa3406772018-07-13 19:10:53 +00003086 EXPECT_TRUE(
3087 FPDFPageObjMark_SetIntParam(document(), text_object, mark, "IntKey", 42));
3088 EXPECT_TRUE(FPDFPageObjMark_SetStringParam(document(), text_object, mark,
3089 "StringKey", "StringValue"));
3090 EXPECT_TRUE(FPDFPageObjMark_SetBlobParam(document(), text_object, mark,
Lei Zhangd3b028b2018-11-30 22:22:00 +00003091 "BlobKey", block_value, kBlobLen));
Henrique Nakashima07520f62018-07-12 19:45:29 +00003092 EXPECT_EQ(3, FPDFPageObjMark_CountParams(mark));
Henrique Nakashima144107d2018-07-10 21:04:05 +00003093
3094 // Check the two parameters can be retrieved.
3095 EXPECT_EQ(FPDF_OBJECT_NUMBER,
Henrique Nakashima94230e52018-07-11 22:02:02 +00003096 FPDFPageObjMark_GetParamValueType(mark, "IntKey"));
Henrique Nakashima144107d2018-07-10 21:04:05 +00003097 int int_value;
Henrique Nakashima94230e52018-07-11 22:02:02 +00003098 EXPECT_TRUE(FPDFPageObjMark_GetParamIntValue(mark, "IntKey", &int_value));
Henrique Nakashima144107d2018-07-10 21:04:05 +00003099 EXPECT_EQ(42, int_value);
3100
3101 EXPECT_EQ(FPDF_OBJECT_STRING,
Henrique Nakashima94230e52018-07-11 22:02:02 +00003102 FPDFPageObjMark_GetParamValueType(mark, "StringKey"));
Henrique Nakashimac3099d12018-09-18 18:08:15 +00003103 unsigned long out_buffer_len = 999u;
Henrique Nakashimaa3406772018-07-13 19:10:53 +00003104 EXPECT_TRUE(FPDFPageObjMark_GetParamStringValue(
3105 mark, "StringKey", buffer, sizeof(buffer), &out_buffer_len));
Henrique Nakashima144107d2018-07-10 21:04:05 +00003106 EXPECT_GT(out_buffer_len, 0u);
Henrique Nakashimac3099d12018-09-18 18:08:15 +00003107 EXPECT_NE(999u, out_buffer_len);
Henrique Nakashima144107d2018-07-10 21:04:05 +00003108 name = GetPlatformWString(reinterpret_cast<unsigned short*>(buffer));
3109 EXPECT_EQ(L"StringValue", name);
3110
Henrique Nakashima07520f62018-07-12 19:45:29 +00003111 EXPECT_EQ(FPDF_OBJECT_STRING,
3112 FPDFPageObjMark_GetParamValueType(mark, "BlobKey"));
3113 out_buffer_len = 0;
Henrique Nakashimaa3406772018-07-13 19:10:53 +00003114 EXPECT_TRUE(FPDFPageObjMark_GetParamBlobValue(
3115 mark, "BlobKey", buffer, sizeof(buffer), &out_buffer_len));
Henrique Nakashima07520f62018-07-12 19:45:29 +00003116 EXPECT_EQ(kBlobLen, out_buffer_len);
Lei Zhangd3b028b2018-11-30 22:22:00 +00003117 EXPECT_EQ(0, memcmp(block_value, buffer, kBlobLen));
Henrique Nakashima07520f62018-07-12 19:45:29 +00003118
Lei Zhangf16376e2020-06-18 19:19:32 +00003119 // Render and check the bitmap is the expected one.
Henrique Nakashima144107d2018-07-10 21:04:05 +00003120 {
Lei Zhang30ff2532019-01-31 21:37:55 +00003121 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingst3b6136a2020-06-09 00:39:33 +00003122 CompareBitmap(page_bitmap.get(), 612, 792, kLoadedFontTextChecksum);
Henrique Nakashima144107d2018-07-10 21:04:05 +00003123 }
3124
Henrique Nakashimab4bcf692018-07-11 21:19:22 +00003125 // Now save the result.
3126 EXPECT_EQ(1, FPDFPage_CountObjects(page));
3127 EXPECT_TRUE(FPDFPage_GenerateContent(page));
3128 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
3129
Henrique Nakashima144107d2018-07-10 21:04:05 +00003130 FPDF_ClosePage(page);
3131
Henrique Nakashimab4bcf692018-07-11 21:19:22 +00003132 // Re-open the file and check the changes were kept in the saved .pdf.
Lei Zhang0b494052019-01-31 21:41:15 +00003133 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashimab4bcf692018-07-11 21:19:22 +00003134 FPDF_PAGE saved_page = LoadSavedPage(0);
3135 EXPECT_EQ(1, FPDFPage_CountObjects(saved_page));
3136
3137 text_object = FPDFPage_GetObject(saved_page, 0);
3138 EXPECT_TRUE(text_object);
3139 EXPECT_EQ(1, FPDFPageObj_CountMarks(text_object));
3140 mark = FPDFPageObj_GetMark(text_object, 0);
3141 EXPECT_TRUE(mark);
Henrique Nakashimac3099d12018-09-18 18:08:15 +00003142
3143 name_len = 999u;
3144 ASSERT_TRUE(FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len));
3145 EXPECT_EQ((14u + 1u) * 2, name_len);
Henrique Nakashimab4bcf692018-07-11 21:19:22 +00003146 name = GetPlatformWString(reinterpret_cast<unsigned short*>(buffer));
Henrique Nakashima6fc8d872018-09-18 16:48:31 +00003147 EXPECT_EQ(L"Test Mark Name", name);
Henrique Nakashimab4bcf692018-07-11 21:19:22 +00003148
3149 CloseSavedPage(saved_page);
3150 CloseSavedDocument();
Henrique Nakashima144107d2018-07-10 21:04:05 +00003151}
3152
Lei Zhangab41f252018-12-23 03:10:50 +00003153TEST_F(FPDFEditEmbedderTest, MarkGetName) {
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00003154 ASSERT_TRUE(OpenDocument("text_in_page_marked.pdf"));
Henrique Nakashimac3099d12018-09-18 18:08:15 +00003155 FPDF_PAGE page = LoadPage(0);
3156 ASSERT_TRUE(page);
3157 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 18);
3158 FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, 1);
3159 ASSERT_TRUE(mark);
3160
3161 char buffer[256];
3162 unsigned long out_len;
3163
3164 // Show the positive cases of FPDFPageObjMark_GetName.
3165 out_len = 999u;
3166 EXPECT_TRUE(FPDFPageObjMark_GetName(mark, nullptr, 0, &out_len));
3167 EXPECT_EQ((6u + 1u) * 2u, out_len);
3168
3169 out_len = 999u;
3170 EXPECT_TRUE(FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &out_len));
3171 EXPECT_EQ(L"Bounds",
3172 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)));
3173 EXPECT_EQ((6u + 1u) * 2u, out_len);
3174
3175 // Show the negative cases of FPDFPageObjMark_GetName.
3176 out_len = 999u;
3177 EXPECT_FALSE(
3178 FPDFPageObjMark_GetName(nullptr, buffer, sizeof(buffer), &out_len));
3179 EXPECT_EQ(999u, out_len);
3180
3181 EXPECT_FALSE(FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), nullptr));
3182
3183 UnloadPage(page);
3184}
3185
Lei Zhangab41f252018-12-23 03:10:50 +00003186TEST_F(FPDFEditEmbedderTest, MarkGetParamKey) {
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00003187 ASSERT_TRUE(OpenDocument("text_in_page_marked.pdf"));
Henrique Nakashimac3099d12018-09-18 18:08:15 +00003188 FPDF_PAGE page = LoadPage(0);
3189 ASSERT_TRUE(page);
3190 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 18);
3191 FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, 1);
3192 ASSERT_TRUE(mark);
3193
3194 char buffer[256];
3195 unsigned long out_len;
3196
3197 // Show the positive cases of FPDFPageObjMark_GetParamKey.
3198 out_len = 999u;
3199 EXPECT_TRUE(FPDFPageObjMark_GetParamKey(mark, 0, nullptr, 0, &out_len));
3200 EXPECT_EQ((8u + 1u) * 2u, out_len);
3201
3202 out_len = 999u;
3203 EXPECT_TRUE(
3204 FPDFPageObjMark_GetParamKey(mark, 0, buffer, sizeof(buffer), &out_len));
3205 EXPECT_EQ(L"Position",
3206 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)));
3207 EXPECT_EQ((8u + 1u) * 2u, out_len);
3208
3209 // Show the negative cases of FPDFPageObjMark_GetParamKey.
3210 out_len = 999u;
3211 EXPECT_FALSE(FPDFPageObjMark_GetParamKey(nullptr, 0, buffer, sizeof(buffer),
3212 &out_len));
3213 EXPECT_EQ(999u, out_len);
3214
3215 out_len = 999u;
3216 EXPECT_FALSE(
3217 FPDFPageObjMark_GetParamKey(mark, 1, buffer, sizeof(buffer), &out_len));
3218 EXPECT_EQ(999u, out_len);
3219
3220 EXPECT_FALSE(
3221 FPDFPageObjMark_GetParamKey(mark, 0, buffer, sizeof(buffer), nullptr));
3222
3223 UnloadPage(page);
3224}
3225
Lei Zhangab41f252018-12-23 03:10:50 +00003226TEST_F(FPDFEditEmbedderTest, MarkGetIntParam) {
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00003227 ASSERT_TRUE(OpenDocument("text_in_page_marked.pdf"));
Henrique Nakashimac3099d12018-09-18 18:08:15 +00003228 FPDF_PAGE page = LoadPage(0);
3229 ASSERT_TRUE(page);
3230 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 8);
3231 FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, 0);
3232 ASSERT_TRUE(mark);
3233
3234 int out_value;
3235
3236 // Show the positive cases of FPDFPageObjMark_GetParamIntValue.
3237 out_value = 999;
3238 EXPECT_TRUE(FPDFPageObjMark_GetParamIntValue(mark, "Factor", &out_value));
3239 EXPECT_EQ(3, out_value);
3240
3241 // Show the negative cases of FPDFPageObjMark_GetParamIntValue.
3242 out_value = 999;
3243 EXPECT_FALSE(FPDFPageObjMark_GetParamIntValue(nullptr, "Factor", &out_value));
3244 EXPECT_EQ(999, out_value);
3245
3246 out_value = 999;
3247 EXPECT_FALSE(FPDFPageObjMark_GetParamIntValue(mark, "ParamThatDoesNotExist",
3248 &out_value));
3249 EXPECT_EQ(999, out_value);
3250
3251 EXPECT_FALSE(FPDFPageObjMark_GetParamIntValue(mark, "Factor", nullptr));
3252
3253 page_object = FPDFPage_GetObject(page, 18);
3254 mark = FPDFPageObj_GetMark(page_object, 1);
3255 out_value = 999;
3256 EXPECT_FALSE(FPDFPageObjMark_GetParamIntValue(mark, "Position", &out_value));
3257 EXPECT_EQ(999, out_value);
3258
3259 UnloadPage(page);
3260}
3261
Lei Zhangab41f252018-12-23 03:10:50 +00003262TEST_F(FPDFEditEmbedderTest, MarkGetStringParam) {
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00003263 ASSERT_TRUE(OpenDocument("text_in_page_marked.pdf"));
Henrique Nakashimac3099d12018-09-18 18:08:15 +00003264 FPDF_PAGE page = LoadPage(0);
3265 ASSERT_TRUE(page);
3266 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 18);
3267 FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, 1);
3268 ASSERT_TRUE(mark);
3269
3270 char buffer[256];
3271 unsigned long out_len;
3272
3273 // Show the positive cases of FPDFPageObjMark_GetParamStringValue.
3274 out_len = 999u;
3275 EXPECT_TRUE(FPDFPageObjMark_GetParamStringValue(mark, "Position", nullptr, 0,
3276 &out_len));
3277 EXPECT_EQ((4u + 1u) * 2u, out_len);
3278
3279 out_len = 999u;
3280 EXPECT_TRUE(FPDFPageObjMark_GetParamStringValue(mark, "Position", buffer,
3281 sizeof(buffer), &out_len));
3282 EXPECT_EQ(L"Last",
3283 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)));
3284 EXPECT_EQ((4u + 1u) * 2u, out_len);
3285
3286 // Show the negative cases of FPDFPageObjMark_GetParamStringValue.
3287 out_len = 999u;
3288 EXPECT_FALSE(FPDFPageObjMark_GetParamStringValue(nullptr, "Position", buffer,
3289 sizeof(buffer), &out_len));
3290 EXPECT_EQ(999u, out_len);
3291
3292 out_len = 999u;
3293 EXPECT_FALSE(FPDFPageObjMark_GetParamStringValue(
3294 mark, "ParamThatDoesNotExist", buffer, sizeof(buffer), &out_len));
3295 EXPECT_EQ(999u, out_len);
3296
3297 EXPECT_FALSE(FPDFPageObjMark_GetParamStringValue(mark, "Position", buffer,
3298 sizeof(buffer), nullptr));
3299
3300 page_object = FPDFPage_GetObject(page, 8);
3301 mark = FPDFPageObj_GetMark(page_object, 0);
3302 out_len = 999u;
3303 EXPECT_FALSE(FPDFPageObjMark_GetParamStringValue(mark, "Factor", buffer,
3304 sizeof(buffer), &out_len));
3305 EXPECT_EQ(999u, out_len);
3306
3307 UnloadPage(page);
3308}
3309
Lei Zhang0ac7b4e2020-07-18 00:35:53 +00003310TEST_F(FPDFEditEmbedderTest, GetBitmap) {
Jane Liu28fb7ba2017-08-02 21:45:57 -04003311 ASSERT_TRUE(OpenDocument("embedded_images.pdf"));
3312 FPDF_PAGE page = LoadPage(0);
3313 ASSERT_TRUE(page);
Miklos Vajna92627612017-09-25 12:59:29 +02003314 ASSERT_EQ(39, FPDFPage_CountObjects(page));
Jane Liu28fb7ba2017-08-02 21:45:57 -04003315
3316 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 32);
3317 EXPECT_NE(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3318 EXPECT_FALSE(FPDFImageObj_GetBitmap(obj));
3319
Lei Zhang89ebf5c2020-07-14 20:59:05 +00003320 {
3321 obj = FPDFPage_GetObject(page, 33);
3322 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3323 ScopedFPDFBitmap bitmap(FPDFImageObj_GetBitmap(obj));
3324 EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap.get()));
Lei Zhang85166512020-07-14 23:28:56 +00003325 CompareBitmap(bitmap.get(), 109, 88, kEmbeddedImage33Checksum);
Lei Zhang89ebf5c2020-07-14 20:59:05 +00003326 }
Jane Liu28fb7ba2017-08-02 21:45:57 -04003327
Lei Zhang89ebf5c2020-07-14 20:59:05 +00003328 {
3329 obj = FPDFPage_GetObject(page, 34);
3330 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3331 ScopedFPDFBitmap bitmap(FPDFImageObj_GetBitmap(obj));
3332 EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap.get()));
3333 CompareBitmap(bitmap.get(), 103, 75, "c8d51fa6821ceb2a67f08446ff236c40");
3334 }
Jane Liu28fb7ba2017-08-02 21:45:57 -04003335
Lei Zhang89ebf5c2020-07-14 20:59:05 +00003336 {
3337 obj = FPDFPage_GetObject(page, 35);
3338 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3339 ScopedFPDFBitmap bitmap(FPDFImageObj_GetBitmap(obj));
3340 EXPECT_EQ(FPDFBitmap_Gray, FPDFBitmap_GetFormat(bitmap.get()));
3341 CompareBitmap(bitmap.get(), 92, 68, "9c6d76cb1e37ef8514f9455d759391f3");
3342 }
Jane Liu28fb7ba2017-08-02 21:45:57 -04003343
Lei Zhang89ebf5c2020-07-14 20:59:05 +00003344 {
3345 obj = FPDFPage_GetObject(page, 36);
3346 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3347 ScopedFPDFBitmap bitmap(FPDFImageObj_GetBitmap(obj));
3348 EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap.get()));
3349 CompareBitmap(bitmap.get(), 79, 60, "f4e72fb783a01c7b4614cdc25eaa98ac");
3350 }
Jane Liu28fb7ba2017-08-02 21:45:57 -04003351
Lei Zhang89ebf5c2020-07-14 20:59:05 +00003352 {
3353 obj = FPDFPage_GetObject(page, 37);
3354 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3355 ScopedFPDFBitmap bitmap(FPDFImageObj_GetBitmap(obj));
3356 EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap.get()));
3357 CompareBitmap(bitmap.get(), 126, 106, "2cf9e66414c72461f4ccbf9cdebdfa1b");
3358 }
Jane Liu28fb7ba2017-08-02 21:45:57 -04003359
Lei Zhang89ebf5c2020-07-14 20:59:05 +00003360 {
3361 obj = FPDFPage_GetObject(page, 38);
3362 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3363 ScopedFPDFBitmap bitmap(FPDFImageObj_GetBitmap(obj));
3364 EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap.get()));
3365 CompareBitmap(bitmap.get(), 194, 119, "a8f3a126cec274dab8242fd2ccdc1b8b");
3366 }
3367
Jane Liu28fb7ba2017-08-02 21:45:57 -04003368 UnloadPage(page);
3369}
Jane Liu548334e2017-08-03 16:33:40 -04003370
Lei Zhang0ac7b4e2020-07-18 00:35:53 +00003371TEST_F(FPDFEditEmbedderTest, GetBitmapIgnoresSetMatrix) {
Lei Zhang85166512020-07-14 23:28:56 +00003372 ASSERT_TRUE(OpenDocument("embedded_images.pdf"));
3373 FPDF_PAGE page = LoadPage(0);
3374 ASSERT_TRUE(page);
3375 ASSERT_EQ(39, FPDFPage_CountObjects(page));
3376
3377 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 33);
3378 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3379
3380 {
3381 // Render |obj| as is.
3382 ScopedFPDFBitmap bitmap(FPDFImageObj_GetBitmap(obj));
3383 EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap.get()));
3384 CompareBitmap(bitmap.get(), 109, 88, kEmbeddedImage33Checksum);
3385 }
3386
3387 // Check the matrix for |obj|.
3388 double a;
3389 double b;
3390 double c;
3391 double d;
3392 double e;
3393 double f;
3394 EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f));
3395 EXPECT_DOUBLE_EQ(53.0, a);
3396 EXPECT_DOUBLE_EQ(0.0, b);
3397 EXPECT_DOUBLE_EQ(0.0, c);
3398 EXPECT_DOUBLE_EQ(43.0, d);
3399 EXPECT_DOUBLE_EQ(72.0, e);
3400 EXPECT_DOUBLE_EQ(646.510009765625, f);
3401
3402 // Modify the matrix for |obj|.
3403 a = 120.0;
3404 EXPECT_TRUE(FPDFImageObj_SetMatrix(obj, a, b, c, d, e, f));
3405
3406 // Make sure the matrix modification took place.
3407 EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f));
3408 EXPECT_DOUBLE_EQ(120.0, a);
3409 EXPECT_DOUBLE_EQ(0.0, b);
3410 EXPECT_DOUBLE_EQ(0.0, c);
3411 EXPECT_DOUBLE_EQ(43.0, d);
3412 EXPECT_DOUBLE_EQ(72.0, e);
3413 EXPECT_DOUBLE_EQ(646.510009765625, f);
3414
3415 {
3416 // Render |obj| again. Note that the FPDFImageObj_SetMatrix() call has no
3417 // effect.
3418 ScopedFPDFBitmap bitmap(FPDFImageObj_GetBitmap(obj));
3419 EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap.get()));
3420 CompareBitmap(bitmap.get(), 109, 88, kEmbeddedImage33Checksum);
3421 }
3422
3423 UnloadPage(page);
3424}
3425
Lei Zhang0ac7b4e2020-07-18 00:35:53 +00003426TEST_F(FPDFEditEmbedderTest, GetBitmapForJBigImage) {
Lei Zhang53341dd2018-03-01 15:42:47 +00003427 ASSERT_TRUE(OpenDocument("bug_631912.pdf"));
3428 FPDF_PAGE page = LoadPage(0);
3429 ASSERT_TRUE(page);
3430 ASSERT_EQ(1, FPDFPage_CountObjects(page));
3431
3432 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 0);
3433 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3434 {
Tom Sepeze08d2b12018-04-25 18:49:32 +00003435 ScopedFPDFBitmap bitmap(FPDFImageObj_GetBitmap(obj));
Lei Zhang1330ebb2018-03-05 15:16:37 +00003436 ASSERT_TRUE(bitmap);
3437 EXPECT_EQ(FPDFBitmap_Gray, FPDFBitmap_GetFormat(bitmap.get()));
3438 CompareBitmap(bitmap.get(), 1152, 720, "3f6a48e2b3e91b799bf34567f55cb4de");
Lei Zhang53341dd2018-03-01 15:42:47 +00003439 }
3440
3441 UnloadPage(page);
3442}
3443
Lei Zhang0ac7b4e2020-07-18 00:35:53 +00003444TEST_F(FPDFEditEmbedderTest, GetBitmapIgnoresSMask) {
Lei Zhang2e1b4f22020-07-08 23:40:18 +00003445 ASSERT_TRUE(OpenDocument("matte.pdf"));
3446 FPDF_PAGE page = LoadPage(0);
3447 ASSERT_TRUE(page);
3448
3449 constexpr int kExpectedObjects = 4;
3450 ASSERT_EQ(kExpectedObjects, FPDFPage_CountObjects(page));
3451
3452 for (int i = 0; i < kExpectedObjects; ++i) {
3453 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, i);
3454 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3455 ScopedFPDFBitmap bitmap(FPDFImageObj_GetBitmap(obj));
3456 ASSERT_TRUE(bitmap);
3457 EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap.get()));
3458 CompareBitmap(bitmap.get(), 50, 50, "46c9a1dbe0b44765ce46017ad629a2fe");
3459 }
3460
3461 UnloadPage(page);
3462}
3463
Hui Yingst822fb8d2020-10-05 22:10:35 +00003464// TODO(crbug.com/pdfium/11): Fix this test and enable.
3465#if defined(_SKIA_SUPPORT_)
3466#define MAYBE_GetRenderedBitmapHandlesSetMatrix \
3467 DISABLED_GetRenderedBitmapHandlesSetMatrix
3468#else
3469#define MAYBE_GetRenderedBitmapHandlesSetMatrix \
3470 GetRenderedBitmapHandlesSetMatrix
3471#endif
3472TEST_F(FPDFEditEmbedderTest, MAYBE_GetRenderedBitmapHandlesSetMatrix) {
Lei Zhang0ac7b4e2020-07-18 00:35:53 +00003473 ASSERT_TRUE(OpenDocument("embedded_images.pdf"));
3474 FPDF_PAGE page = LoadPage(0);
3475 ASSERT_TRUE(page);
3476 ASSERT_EQ(39, FPDFPage_CountObjects(page));
3477
3478 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 33);
3479 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3480
3481 {
3482 // Render |obj| as is.
3483 ScopedFPDFBitmap bitmap(
3484 FPDFImageObj_GetRenderedBitmap(document(), page, obj));
3485 EXPECT_EQ(FPDFBitmap_BGRA, FPDFBitmap_GetFormat(bitmap.get()));
3486 CompareBitmap(bitmap.get(), 53, 43, "90fa16c2fb2bf8ad3654c2258417664c");
3487 }
3488
3489 // Check the matrix for |obj|.
3490 double a;
3491 double b;
3492 double c;
3493 double d;
3494 double e;
3495 double f;
3496 EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f));
3497 EXPECT_DOUBLE_EQ(53.0, a);
3498 EXPECT_DOUBLE_EQ(0.0, b);
3499 EXPECT_DOUBLE_EQ(0.0, c);
3500 EXPECT_DOUBLE_EQ(43.0, d);
3501 EXPECT_DOUBLE_EQ(72.0, e);
3502 EXPECT_DOUBLE_EQ(646.510009765625, f);
3503
3504 // Modify the matrix for |obj|.
3505 a = 120.0;
3506 EXPECT_TRUE(FPDFImageObj_SetMatrix(obj, a, b, c, d, e, f));
3507
3508 // Make sure the matrix modification took place.
3509 EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f));
3510 EXPECT_DOUBLE_EQ(120.0, a);
3511 EXPECT_DOUBLE_EQ(0.0, b);
3512 EXPECT_DOUBLE_EQ(0.0, c);
3513 EXPECT_DOUBLE_EQ(43.0, d);
3514 EXPECT_DOUBLE_EQ(72.0, e);
3515 EXPECT_DOUBLE_EQ(646.510009765625, f);
3516
3517 {
3518 // Render |obj| again. Note that the FPDFImageObj_SetMatrix() call has an
3519 // effect.
3520 ScopedFPDFBitmap bitmap(
3521 FPDFImageObj_GetRenderedBitmap(document(), page, obj));
3522 EXPECT_EQ(FPDFBitmap_BGRA, FPDFBitmap_GetFormat(bitmap.get()));
3523 CompareBitmap(bitmap.get(), 120, 43, "57ed8e15daa535490ff0c8b7640a36b4");
3524 }
3525
3526 UnloadPage(page);
3527}
3528
Hui Yingst822fb8d2020-10-05 22:10:35 +00003529// TODO(crbug.com/pdfium/11): Fix this test and enable.
3530#if defined(_SKIA_SUPPORT_)
3531#define MAYBE_GetRenderedBitmapHandlesSMask \
3532 DISABLED_GetRenderedBitmapHandlesSMask
3533#else
3534#define MAYBE_GetRenderedBitmapHandlesSMask GetRenderedBitmapHandlesSMask
3535#endif
3536TEST_F(FPDFEditEmbedderTest, MAYBE_GetRenderedBitmapHandlesSMask) {
Lei Zhang0ac7b4e2020-07-18 00:35:53 +00003537 ASSERT_TRUE(OpenDocument("matte.pdf"));
3538 FPDF_PAGE page = LoadPage(0);
3539 ASSERT_TRUE(page);
3540
3541 constexpr int kExpectedObjects = 4;
3542 ASSERT_EQ(kExpectedObjects, FPDFPage_CountObjects(page));
3543
3544 for (int i = 0; i < kExpectedObjects; ++i) {
3545 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, i);
3546 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3547 ScopedFPDFBitmap bitmap(
3548 FPDFImageObj_GetRenderedBitmap(document(), page, obj));
3549 ASSERT_TRUE(bitmap);
3550 EXPECT_EQ(FPDFBitmap_BGRA, FPDFBitmap_GetFormat(bitmap.get()));
3551 if (i == 0)
3552 CompareBitmap(bitmap.get(), 40, 60, "5a3ae4a660ce919e29c42ec2258142f1");
3553 else
3554 CompareBitmap(bitmap.get(), 40, 60, "67504e83f5d78214ea00efc19082c5c1");
3555 }
3556
3557 UnloadPage(page);
3558}
3559
3560TEST_F(FPDFEditEmbedderTest, GetRenderedBitmapBadParams) {
3561 ASSERT_TRUE(OpenDocument("embedded_images.pdf"));
3562 FPDF_PAGE page = LoadPage(0);
3563 ASSERT_TRUE(page);
3564
3565 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 33);
3566 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3567
3568 // Test various null parameters.
3569 EXPECT_FALSE(FPDFImageObj_GetRenderedBitmap(nullptr, nullptr, nullptr));
3570 EXPECT_FALSE(FPDFImageObj_GetRenderedBitmap(document(), nullptr, nullptr));
3571 EXPECT_FALSE(FPDFImageObj_GetRenderedBitmap(nullptr, page, nullptr));
3572 EXPECT_FALSE(FPDFImageObj_GetRenderedBitmap(nullptr, nullptr, obj));
3573 EXPECT_FALSE(FPDFImageObj_GetRenderedBitmap(document(), page, nullptr));
3574 EXPECT_FALSE(FPDFImageObj_GetRenderedBitmap(nullptr, page, obj));
3575
3576 // Test mismatch between document and page parameters.
3577 ScopedFPDFDocument new_document(FPDF_CreateNewDocument());
3578 EXPECT_FALSE(FPDFImageObj_GetRenderedBitmap(new_document.get(), page, obj));
3579
3580 UnloadPage(page);
3581}
3582
Lei Zhangab41f252018-12-23 03:10:50 +00003583TEST_F(FPDFEditEmbedderTest, GetImageData) {
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00003584 ASSERT_TRUE(OpenDocument("embedded_images.pdf"));
Jane Liu548334e2017-08-03 16:33:40 -04003585 FPDF_PAGE page = LoadPage(0);
3586 ASSERT_TRUE(page);
Miklos Vajna92627612017-09-25 12:59:29 +02003587 ASSERT_EQ(39, FPDFPage_CountObjects(page));
Jane Liu548334e2017-08-03 16:33:40 -04003588
3589 // Retrieve an image object with flate-encoded data stream.
3590 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 33);
3591 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3592
3593 // Check that the raw image data has the correct length and hash value.
3594 unsigned long len = FPDFImageObj_GetImageDataRaw(obj, nullptr, 0);
3595 std::vector<char> buf(len);
3596 EXPECT_EQ(4091u, FPDFImageObj_GetImageDataRaw(obj, buf.data(), len));
3597 EXPECT_EQ("f73802327d2e88e890f653961bcda81a",
3598 GenerateMD5Base16(reinterpret_cast<uint8_t*>(buf.data()), len));
3599
3600 // Check that the decoded image data has the correct length and hash value.
3601 len = FPDFImageObj_GetImageDataDecoded(obj, nullptr, 0);
3602 buf.clear();
3603 buf.resize(len);
3604 EXPECT_EQ(28776u, FPDFImageObj_GetImageDataDecoded(obj, buf.data(), len));
Lei Zhang85166512020-07-14 23:28:56 +00003605 EXPECT_EQ(kEmbeddedImage33Checksum,
Jane Liu548334e2017-08-03 16:33:40 -04003606 GenerateMD5Base16(reinterpret_cast<uint8_t*>(buf.data()), len));
3607
Lei Zhang351e8b02018-12-20 01:10:06 +00003608 // Retrieve an image object with DCTDecode-encoded data stream.
Jane Liu548334e2017-08-03 16:33:40 -04003609 obj = FPDFPage_GetObject(page, 37);
3610 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3611
3612 // Check that the raw image data has the correct length and hash value.
3613 len = FPDFImageObj_GetImageDataRaw(obj, nullptr, 0);
3614 buf.clear();
3615 buf.resize(len);
3616 EXPECT_EQ(4370u, FPDFImageObj_GetImageDataRaw(obj, buf.data(), len));
3617 EXPECT_EQ("6aae1f3710335023a9e12191be66b64b",
3618 GenerateMD5Base16(reinterpret_cast<uint8_t*>(buf.data()), len));
3619
3620 // Check that the decoded image data has the correct length and hash value,
3621 // which should be the same as those of the raw data, since this image is
3622 // encoded by a single DCTDecode filter and decoding is a noop.
3623 len = FPDFImageObj_GetImageDataDecoded(obj, nullptr, 0);
3624 buf.clear();
3625 buf.resize(len);
3626 EXPECT_EQ(4370u, FPDFImageObj_GetImageDataDecoded(obj, buf.data(), len));
3627 EXPECT_EQ("6aae1f3710335023a9e12191be66b64b",
3628 GenerateMD5Base16(reinterpret_cast<uint8_t*>(buf.data()), len));
3629
3630 UnloadPage(page);
3631}
Jane Liu2e5f0ae2017-08-08 15:23:27 -04003632
Lei Zhangab41f252018-12-23 03:10:50 +00003633TEST_F(FPDFEditEmbedderTest, GetImageMatrix) {
Lei Zhang866d6882018-10-24 17:31:01 +00003634 ASSERT_TRUE(OpenDocument("embedded_images.pdf"));
3635 FPDF_PAGE page = LoadPage(0);
3636 ASSERT_TRUE(page);
3637 ASSERT_EQ(39, FPDFPage_CountObjects(page));
3638
3639 FPDF_PAGEOBJECT obj;
3640 double a;
3641 double b;
3642 double c;
3643 double d;
3644 double e;
3645 double f;
3646
3647 obj = FPDFPage_GetObject(page, 33);
3648 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3649 EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f));
3650 EXPECT_DOUBLE_EQ(53.0, a);
3651 EXPECT_DOUBLE_EQ(0.0, b);
3652 EXPECT_DOUBLE_EQ(0.0, c);
3653 EXPECT_DOUBLE_EQ(43.0, d);
3654 EXPECT_DOUBLE_EQ(72.0, e);
3655 EXPECT_DOUBLE_EQ(646.510009765625, f);
3656
3657 obj = FPDFPage_GetObject(page, 34);
3658 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3659 EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f));
3660 EXPECT_DOUBLE_EQ(70.0, a);
3661 EXPECT_DOUBLE_EQ(0.0, b);
3662 EXPECT_DOUBLE_EQ(0.0, c);
3663 EXPECT_DOUBLE_EQ(51.0, d);
3664 EXPECT_DOUBLE_EQ(216.0, e);
3665 EXPECT_DOUBLE_EQ(646.510009765625, f);
3666
3667 obj = FPDFPage_GetObject(page, 35);
3668 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3669 EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f));
3670 EXPECT_DOUBLE_EQ(69.0, a);
3671 EXPECT_DOUBLE_EQ(0.0, b);
3672 EXPECT_DOUBLE_EQ(0.0, c);
3673 EXPECT_DOUBLE_EQ(51.0, d);
3674 EXPECT_DOUBLE_EQ(360.0, e);
3675 EXPECT_DOUBLE_EQ(646.510009765625, f);
3676
3677 obj = FPDFPage_GetObject(page, 36);
3678 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3679 EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f));
3680 EXPECT_DOUBLE_EQ(59.0, a);
3681 EXPECT_DOUBLE_EQ(0.0, b);
3682 EXPECT_DOUBLE_EQ(0.0, c);
3683 EXPECT_DOUBLE_EQ(45.0, d);
3684 EXPECT_DOUBLE_EQ(72.0, e);
3685 EXPECT_DOUBLE_EQ(553.510009765625, f);
3686
3687 obj = FPDFPage_GetObject(page, 37);
3688 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3689 EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f));
3690 EXPECT_DOUBLE_EQ(55.94000244140625, a);
3691 EXPECT_DOUBLE_EQ(0.0, b);
3692 EXPECT_DOUBLE_EQ(0.0, c);
3693 EXPECT_DOUBLE_EQ(46.950000762939453, d);
3694 EXPECT_DOUBLE_EQ(216.0, e);
3695 EXPECT_DOUBLE_EQ(552.510009765625, f);
3696
3697 obj = FPDFPage_GetObject(page, 38);
3698 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3699 EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f));
3700 EXPECT_DOUBLE_EQ(70.528999328613281, a);
3701 EXPECT_DOUBLE_EQ(0.0, b);
3702 EXPECT_DOUBLE_EQ(0.0, c);
3703 EXPECT_DOUBLE_EQ(43.149997711181641, d);
3704 EXPECT_DOUBLE_EQ(360.0, e);
3705 EXPECT_DOUBLE_EQ(553.3599853515625, f);
3706
3707 UnloadPage(page);
3708}
3709
Lei Zhangab41f252018-12-23 03:10:50 +00003710TEST_F(FPDFEditEmbedderTest, DestroyPageObject) {
Jane Liu2e5f0ae2017-08-08 15:23:27 -04003711 FPDF_PAGEOBJECT rect = FPDFPageObj_CreateNewRect(10, 10, 20, 20);
3712 ASSERT_TRUE(rect);
3713
3714 // There should be no memory leaks with a call to FPDFPageObj_Destroy().
3715 FPDFPageObj_Destroy(rect);
3716}
Jane Liube63ab92017-08-09 14:09:34 -04003717
Lei Zhangab41f252018-12-23 03:10:50 +00003718TEST_F(FPDFEditEmbedderTest, GetImageFilters) {
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00003719 ASSERT_TRUE(OpenDocument("embedded_images.pdf"));
Jane Liube63ab92017-08-09 14:09:34 -04003720 FPDF_PAGE page = LoadPage(0);
3721 ASSERT_TRUE(page);
3722
3723 // Verify that retrieving the filter of a non-image object would fail.
3724 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 32);
3725 ASSERT_NE(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3726 ASSERT_EQ(0, FPDFImageObj_GetImageFilterCount(obj));
3727 EXPECT_EQ(0u, FPDFImageObj_GetImageFilter(obj, 0, nullptr, 0));
3728
3729 // Verify the returned filter string for an image object with a single filter.
3730 obj = FPDFPage_GetObject(page, 33);
3731 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3732 ASSERT_EQ(1, FPDFImageObj_GetImageFilterCount(obj));
3733 unsigned long len = FPDFImageObj_GetImageFilter(obj, 0, nullptr, 0);
3734 std::vector<char> buf(len);
Lei Zhang0733a1b2017-08-31 12:36:31 -07003735 static constexpr char kFlateDecode[] = "FlateDecode";
3736 EXPECT_EQ(sizeof(kFlateDecode),
3737 FPDFImageObj_GetImageFilter(obj, 0, buf.data(), len));
3738 EXPECT_STREQ(kFlateDecode, buf.data());
Jane Liube63ab92017-08-09 14:09:34 -04003739 EXPECT_EQ(0u, FPDFImageObj_GetImageFilter(obj, 1, nullptr, 0));
3740
3741 // Verify all the filters for an image object with a list of filters.
3742 obj = FPDFPage_GetObject(page, 38);
3743 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3744 ASSERT_EQ(2, FPDFImageObj_GetImageFilterCount(obj));
3745 len = FPDFImageObj_GetImageFilter(obj, 0, nullptr, 0);
3746 buf.clear();
3747 buf.resize(len);
Lei Zhang0733a1b2017-08-31 12:36:31 -07003748 static constexpr char kASCIIHexDecode[] = "ASCIIHexDecode";
3749 EXPECT_EQ(sizeof(kASCIIHexDecode),
3750 FPDFImageObj_GetImageFilter(obj, 0, buf.data(), len));
3751 EXPECT_STREQ(kASCIIHexDecode, buf.data());
Jane Liube63ab92017-08-09 14:09:34 -04003752
3753 len = FPDFImageObj_GetImageFilter(obj, 1, nullptr, 0);
3754 buf.clear();
3755 buf.resize(len);
Lei Zhang0733a1b2017-08-31 12:36:31 -07003756 static constexpr char kDCTDecode[] = "DCTDecode";
3757 EXPECT_EQ(sizeof(kDCTDecode),
3758 FPDFImageObj_GetImageFilter(obj, 1, buf.data(), len));
3759 EXPECT_STREQ(kDCTDecode, buf.data());
Jane Liube63ab92017-08-09 14:09:34 -04003760
3761 UnloadPage(page);
3762}
Jane Liuca898292017-08-16 11:25:35 -04003763
Lei Zhangab41f252018-12-23 03:10:50 +00003764TEST_F(FPDFEditEmbedderTest, GetImageMetadata) {
Jane Liuca898292017-08-16 11:25:35 -04003765 ASSERT_TRUE(OpenDocument("embedded_images.pdf"));
3766 FPDF_PAGE page = LoadPage(0);
3767 ASSERT_TRUE(page);
3768
3769 // Check that getting the metadata of a null object would fail.
3770 FPDF_IMAGEOBJ_METADATA metadata;
3771 EXPECT_FALSE(FPDFImageObj_GetImageMetadata(nullptr, page, &metadata));
3772
3773 // Check that receiving the metadata with a null metadata object would fail.
3774 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 35);
3775 EXPECT_FALSE(FPDFImageObj_GetImageMetadata(obj, page, nullptr));
3776
3777 // Check that when retrieving an image object's metadata without passing in
3778 // |page|, all values are correct, with the last two being default values.
3779 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3780 ASSERT_TRUE(FPDFImageObj_GetImageMetadata(obj, nullptr, &metadata));
Julian Lungerecd063e2017-12-27 10:18:50 -05003781 EXPECT_EQ(7, metadata.marked_content_id);
Jane Liuca898292017-08-16 11:25:35 -04003782 EXPECT_EQ(92u, metadata.width);
3783 EXPECT_EQ(68u, metadata.height);
Lei Zhang351e8b02018-12-20 01:10:06 +00003784 EXPECT_FLOAT_EQ(96.0f, metadata.horizontal_dpi);
3785 EXPECT_FLOAT_EQ(96.0f, metadata.vertical_dpi);
Jane Liuca898292017-08-16 11:25:35 -04003786 EXPECT_EQ(0u, metadata.bits_per_pixel);
3787 EXPECT_EQ(FPDF_COLORSPACE_UNKNOWN, metadata.colorspace);
3788
3789 // Verify the metadata of a bitmap image with indexed colorspace.
3790 ASSERT_TRUE(FPDFImageObj_GetImageMetadata(obj, page, &metadata));
Julian Lungerecd063e2017-12-27 10:18:50 -05003791 EXPECT_EQ(7, metadata.marked_content_id);
Jane Liuca898292017-08-16 11:25:35 -04003792 EXPECT_EQ(92u, metadata.width);
3793 EXPECT_EQ(68u, metadata.height);
Lei Zhang351e8b02018-12-20 01:10:06 +00003794 EXPECT_FLOAT_EQ(96.0f, metadata.horizontal_dpi);
3795 EXPECT_FLOAT_EQ(96.0f, metadata.vertical_dpi);
Jane Liuca898292017-08-16 11:25:35 -04003796 EXPECT_EQ(1u, metadata.bits_per_pixel);
3797 EXPECT_EQ(FPDF_COLORSPACE_INDEXED, metadata.colorspace);
3798
3799 // Verify the metadata of an image with RGB colorspace.
3800 obj = FPDFPage_GetObject(page, 37);
3801 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3802 ASSERT_TRUE(FPDFImageObj_GetImageMetadata(obj, page, &metadata));
Julian Lungerecd063e2017-12-27 10:18:50 -05003803 EXPECT_EQ(9, metadata.marked_content_id);
Jane Liuca898292017-08-16 11:25:35 -04003804 EXPECT_EQ(126u, metadata.width);
3805 EXPECT_EQ(106u, metadata.height);
Lei Zhang351e8b02018-12-20 01:10:06 +00003806 EXPECT_FLOAT_EQ(162.173752f, metadata.horizontal_dpi);
3807 EXPECT_FLOAT_EQ(162.555878f, metadata.vertical_dpi);
Jane Liuca898292017-08-16 11:25:35 -04003808 EXPECT_EQ(24u, metadata.bits_per_pixel);
3809 EXPECT_EQ(FPDF_COLORSPACE_DEVICERGB, metadata.colorspace);
3810
3811 UnloadPage(page);
3812}
Lei Zhangdc1b7392020-05-14 21:15:53 +00003813
3814TEST_F(FPDFEditEmbedderTest, GetImageMetadataJpxLzw) {
3815 ASSERT_TRUE(OpenDocument("jpx_lzw.pdf"));
3816 FPDF_PAGE page = LoadPage(0);
3817 ASSERT_TRUE(page);
3818
3819 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 0);
3820 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3821
3822 FPDF_IMAGEOBJ_METADATA metadata;
3823 ASSERT_TRUE(FPDFImageObj_GetImageMetadata(obj, page, &metadata));
3824 EXPECT_EQ(-1, metadata.marked_content_id);
3825 EXPECT_EQ(612u, metadata.width);
3826 EXPECT_EQ(792u, metadata.height);
3827 EXPECT_FLOAT_EQ(72.0f, metadata.horizontal_dpi);
3828 EXPECT_FLOAT_EQ(72.0f, metadata.vertical_dpi);
3829 EXPECT_EQ(24u, metadata.bits_per_pixel);
3830 EXPECT_EQ(FPDF_COLORSPACE_UNKNOWN, metadata.colorspace);
3831
3832 UnloadPage(page);
3833}