blob: 98f85cec470923fa49ccf39925718c52c76c7ee4 [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
44#if defined(OS_WIN)
45const char kBottomTextChecksum[] = "08d1ff3e5a42801bee6077fd366bef00";
Lei Zhang0c03d632020-07-30 17:05:36 +000046#elif defined(OS_APPLE)
Hui Yingst3b6136a2020-06-09 00:39:33 +000047const char kBottomTextChecksum[] = "324e1db8164a040cf6104538baa95ba6";
48#else
49const char kBottomTextChecksum[] = "eacaa24573b8ce997b3882595f096f00";
50#endif
Hui Yingstb4baceb2020-04-28 23:46:10 +000051
Hui Yingst08c40712020-04-29 01:37:35 +000052#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
53#if defined(OS_WIN)
54const char kFirstRemovedChecksum[] = "d468228cc55071a9e838f66afc0f58ff";
55#else
56const char kFirstRemovedChecksum[] = "0c79dc1065b1d06991e3ac4aaa35d25c";
57#endif // defined(OS_WIN)
58#else
Hui Yingstb4baceb2020-04-28 23:46:10 +000059#if defined(OS_WIN)
60const char kFirstRemovedChecksum[] = "aae6c5334721f90ec30d3d59f4ef7deb";
Lei Zhang0c03d632020-07-30 17:05:36 +000061#elif defined(OS_APPLE)
Hui Yingstb4baceb2020-04-28 23:46:10 +000062const char kFirstRemovedChecksum[] = "17ca3778fd8bb395b46532f1fa17f702";
63#else
64const char kFirstRemovedChecksum[] = "b76df015fe88009c3c342395df96abf1";
65#endif
Hui Yingst08c40712020-04-29 01:37:35 +000066#endif // defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
Hui Yingstb4baceb2020-04-28 23:46:10 +000067
Hui Yingst3b6136a2020-06-09 00:39:33 +000068const wchar_t kLoadedFontText[] = L"I am testing my loaded font, WEE.";
69
70#if defined(OS_WIN)
71const char kLoadedFontTextChecksum[] = "d60ba39f9698e32360d99e727dd93165";
Lei Zhang0c03d632020-07-30 17:05:36 +000072#elif defined(OS_APPLE)
Hui Yingst3b6136a2020-06-09 00:39:33 +000073const char kLoadedFontTextChecksum[] = "fc921c0bbdde73986ac13c15a85db4c3";
74#else
75const char kLoadedFontTextChecksum[] = "70592859010ffbf532a2237b8118bcc4";
76#endif
77
78const char kRedRectangleChecksum[] = "66d02eaa6181e2c069ce2ea99beda497";
79
Lei Zhang85166512020-07-14 23:28:56 +000080// In embedded_images.pdf.
81const char kEmbeddedImage33Checksum[] = "cb3637934bb3b95a6e4ae1ea9eb9e56e";
82
Hui Yingstb4baceb2020-04-28 23:46:10 +000083} // namespace
84
Lei Zhangab41f252018-12-23 03:10:50 +000085class FPDFEditEmbedderTest : public EmbedderTest {
Nicolas Penad03ca422017-03-06 13:54:33 -050086 protected:
87 FPDF_DOCUMENT CreateNewDocument() {
88 document_ = FPDF_CreateNewDocument();
Tom Sepez41066c12017-05-18 09:28:49 -070089 cpdf_doc_ = CPDFDocumentFromFPDFDocument(document_);
Nicolas Penad03ca422017-03-06 13:54:33 -050090 return document_;
91 }
92
Lei Zhang710fa992018-05-25 16:24:48 +000093 void CheckFontDescriptor(const CPDF_Dictionary* font_dict,
Nicolas Penad03ca422017-03-06 13:54:33 -050094 int font_type,
95 bool bold,
96 bool italic,
Tom Sepez20c41a52018-08-29 23:53:53 +000097 pdfium::span<const uint8_t> span) {
Lei Zhangb1ec2802018-05-25 21:55:24 +000098 const CPDF_Dictionary* font_desc = font_dict->GetDictFor("FontDescriptor");
Nicolas Penad03ca422017-03-06 13:54:33 -050099 ASSERT_TRUE(font_desc);
Lei Zhanga228ff32020-06-24 17:39:33 +0000100 EXPECT_EQ("FontDescriptor", font_desc->GetNameFor("Type"));
101 ByteString font_name = font_desc->GetNameFor("FontName");
102 EXPECT_FALSE(font_name.IsEmpty());
103 EXPECT_EQ(font_dict->GetNameFor("BaseFont"), font_name);
Nicolas Penad03ca422017-03-06 13:54:33 -0500104
105 // Check that the font descriptor has the required keys according to spec
106 // 1.7 Table 5.19
107 ASSERT_TRUE(font_desc->KeyExist("Flags"));
Dan Sinclair10e1f052017-09-28 15:59:42 -0400108
Nicolas Penad03ca422017-03-06 13:54:33 -0500109 int font_flags = font_desc->GetIntegerFor("Flags");
Lei Zhang44d03832019-10-18 18:45:31 +0000110 EXPECT_EQ(bold, FontStyleIsForceBold(font_flags));
Dan Sinclair10e1f052017-09-28 15:59:42 -0400111 EXPECT_EQ(italic, FontStyleIsItalic(font_flags));
112 EXPECT_TRUE(FontStyleIsNonSymbolic(font_flags));
Nicolas Penad03ca422017-03-06 13:54:33 -0500113 ASSERT_TRUE(font_desc->KeyExist("FontBBox"));
Nicolás Peña5f95f362017-09-28 13:00:45 +0900114
Lei Zhangb1ec2802018-05-25 21:55:24 +0000115 const CPDF_Array* fontBBox = font_desc->GetArrayFor("FontBBox");
Nicolás Peña5f95f362017-09-28 13:00:45 +0900116 ASSERT_TRUE(fontBBox);
Lei Zhangf40380f2018-10-12 18:31:51 +0000117 EXPECT_EQ(4u, fontBBox->size());
Nicolás Peña5f95f362017-09-28 13:00:45 +0900118 // Check that the coordinates are in the preferred order according to spec
119 // 1.7 Section 3.8.4
120 EXPECT_TRUE(fontBBox->GetIntegerAt(0) < fontBBox->GetIntegerAt(2));
121 EXPECT_TRUE(fontBBox->GetIntegerAt(1) < fontBBox->GetIntegerAt(3));
122
Nicolas Penad03ca422017-03-06 13:54:33 -0500123 EXPECT_TRUE(font_desc->KeyExist("ItalicAngle"));
124 EXPECT_TRUE(font_desc->KeyExist("Ascent"));
125 EXPECT_TRUE(font_desc->KeyExist("Descent"));
126 EXPECT_TRUE(font_desc->KeyExist("CapHeight"));
127 EXPECT_TRUE(font_desc->KeyExist("StemV"));
Ryan Harrison275e2602017-09-18 14:23:18 -0400128 ByteString present("FontFile");
129 ByteString absent("FontFile2");
Nicolas Penad03ca422017-03-06 13:54:33 -0500130 if (font_type == FPDF_FONT_TRUETYPE)
131 std::swap(present, absent);
132 EXPECT_TRUE(font_desc->KeyExist(present));
133 EXPECT_FALSE(font_desc->KeyExist(absent));
134
Artem Strygineababa12018-06-06 12:31:18 +0000135 auto streamAcc =
136 pdfium::MakeRetain<CPDF_StreamAcc>(font_desc->GetStreamFor(present));
137 streamAcc->LoadAllDataRaw();
138
Nicolas Penad03ca422017-03-06 13:54:33 -0500139 // Check that the font stream is the one that was provided
Tom Sepez20c41a52018-08-29 23:53:53 +0000140 ASSERT_EQ(span.size(), streamAcc->GetSize());
Nicolás Peña79eab232017-09-28 13:29:05 +0900141 if (font_type == FPDF_FONT_TRUETYPE) {
Tom Sepez20c41a52018-08-29 23:53:53 +0000142 ASSERT_EQ(static_cast<int>(span.size()),
Artem Strygineababa12018-06-06 12:31:18 +0000143 streamAcc->GetDict()->GetIntegerFor("Length1"));
Nicolás Peña79eab232017-09-28 13:29:05 +0900144 }
Artem Strygineababa12018-06-06 12:31:18 +0000145
146 const uint8_t* stream_data = streamAcc->GetData();
Tom Sepez20c41a52018-08-29 23:53:53 +0000147 for (size_t j = 0; j < span.size(); j++)
148 EXPECT_EQ(span[j], stream_data[j]) << " at byte " << j;
Nicolas Penad03ca422017-03-06 13:54:33 -0500149 }
150
Lei Zhangde579ab2018-05-25 21:49:49 +0000151 void CheckCompositeFontWidths(const CPDF_Array* widths_array,
Nicolas Penad03ca422017-03-06 13:54:33 -0500152 CPDF_Font* typed_font) {
153 // Check that W array is in a format that conforms to PDF spec 1.7 section
154 // "Glyph Metrics in CIDFonts" (these checks are not
155 // implementation-specific).
Lei Zhangf40380f2018-10-12 18:31:51 +0000156 EXPECT_GT(widths_array->size(), 1u);
Nicolas Penad03ca422017-03-06 13:54:33 -0500157 int num_cids_checked = 0;
158 int cur_cid = 0;
Lei Zhangf40380f2018-10-12 18:31:51 +0000159 for (size_t idx = 0; idx < widths_array->size(); idx++) {
Nicolas Penad03ca422017-03-06 13:54:33 -0500160 int cid = widths_array->GetNumberAt(idx);
161 EXPECT_GE(cid, cur_cid);
Lei Zhangf40380f2018-10-12 18:31:51 +0000162 ASSERT_FALSE(++idx == widths_array->size());
Lei Zhangde579ab2018-05-25 21:49:49 +0000163 const CPDF_Object* next = widths_array->GetObjectAt(idx);
Nicolas Penad03ca422017-03-06 13:54:33 -0500164 if (next->IsArray()) {
165 // We are in the c [w1 w2 ...] case
Lei Zhangde579ab2018-05-25 21:49:49 +0000166 const CPDF_Array* arr = next->AsArray();
Lei Zhangf40380f2018-10-12 18:31:51 +0000167 int cnt = static_cast<int>(arr->size());
Nicolas Penad03ca422017-03-06 13:54:33 -0500168 size_t inner_idx = 0;
169 for (cur_cid = cid; cur_cid < cid + cnt; cur_cid++) {
Lei Zhang70e66192020-09-18 20:50:51 +0000170 int width = arr->GetNumberAt(inner_idx++);
Dan Sinclair971a6742018-03-28 19:23:25 +0000171 EXPECT_EQ(width, typed_font->GetCharWidthF(cur_cid))
172 << " at cid " << cur_cid;
Nicolas Penad03ca422017-03-06 13:54:33 -0500173 }
174 num_cids_checked += cnt;
175 continue;
176 }
177 // Otherwise, are in the c_first c_last w case.
178 ASSERT_TRUE(next->IsNumber());
179 int last_cid = next->AsNumber()->GetInteger();
Lei Zhangf40380f2018-10-12 18:31:51 +0000180 ASSERT_FALSE(++idx == widths_array->size());
Lei Zhang70e66192020-09-18 20:50:51 +0000181 int width = widths_array->GetNumberAt(idx);
Nicolas Penad03ca422017-03-06 13:54:33 -0500182 for (cur_cid = cid; cur_cid <= last_cid; cur_cid++) {
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 += last_cid - cid + 1;
187 }
188 // Make sure we have a good amount of cids described
189 EXPECT_GT(num_cids_checked, 900);
190 }
191 CPDF_Document* cpdf_doc() { return cpdf_doc_; }
192
193 private:
194 CPDF_Document* cpdf_doc_;
195};
Tom Sepezd483eb42016-01-06 10:03:59 -0800196
etienneb7712c262016-04-26 08:13:45 -0700197namespace {
thestigdc7ec032016-11-21 15:32:52 -0800198
etienneb7712c262016-04-26 08:13:45 -0700199const char kExpectedPDF[] =
200 "%PDF-1.7\r\n"
201 "%\xA1\xB3\xC5\xD7\r\n"
202 "1 0 obj\r\n"
203 "<</Pages 2 0 R /Type/Catalog>>\r\n"
204 "endobj\r\n"
205 "2 0 obj\r\n"
206 "<</Count 1/Kids\\[ 4 0 R \\]/Type/Pages>>\r\n"
207 "endobj\r\n"
208 "3 0 obj\r\n"
209 "<</CreationDate\\(D:.*\\)/Creator\\(PDFium\\)>>\r\n"
210 "endobj\r\n"
211 "4 0 obj\r\n"
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400212 "<</MediaBox\\[ 0 0 640 480\\]/Parent 2 0 R "
213 "/Resources<</ExtGState<</FXE1 5 0 R >>>>"
Nicolas Penad9d6c292017-06-06 16:12:10 -0400214 "/Rotate 0/Type/Page"
etienneb7712c262016-04-26 08:13:45 -0700215 ">>\r\n"
216 "endobj\r\n"
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400217 "5 0 obj\r\n"
218 "<</BM/Normal/CA 1/ca 1>>\r\n"
219 "endobj\r\n"
etienneb7712c262016-04-26 08:13:45 -0700220 "xref\r\n"
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400221 "0 6\r\n"
etienneb7712c262016-04-26 08:13:45 -0700222 "0000000000 65535 f\r\n"
223 "0000000017 00000 n\r\n"
224 "0000000066 00000 n\r\n"
225 "0000000122 00000 n\r\n"
226 "0000000192 00000 n\r\n"
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400227 "0000000311 00000 n\r\n"
etienneb7712c262016-04-26 08:13:45 -0700228 "trailer\r\n"
229 "<<\r\n"
230 "/Root 1 0 R\r\n"
231 "/Info 3 0 R\r\n"
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400232 "/Size 6/ID\\[<.*><.*>\\]>>\r\n"
etienneb7712c262016-04-26 08:13:45 -0700233 "startxref\r\n"
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400234 "354\r\n"
etienneb7712c262016-04-26 08:13:45 -0700235 "%%EOF\r\n";
thestigdc7ec032016-11-21 15:32:52 -0800236
etienneb7712c262016-04-26 08:13:45 -0700237} // namespace
238
Hui Yingst8e88e382020-11-02 18:19:56 +0000239TEST_F(FPDFEditEmbedderTest, EmbedNotoSansSCFont) {
240 EXPECT_TRUE(CreateEmptyDocument());
241 ScopedFPDFPage page(FPDFPage_New(document(), 0, 400, 400));
242 std::string font_path;
243 ASSERT_TRUE(PathService::GetTestFilePath(
244 "fonts/third_party/NotoSansSC/NotoSansSC-Regular.subset.otf",
245 &font_path));
246
247 size_t file_length = 0;
248 std::unique_ptr<char, pdfium::FreeDeleter> font_data =
249 GetFileContents(font_path.c_str(), &file_length);
250 ASSERT(font_data);
251
252 ScopedFPDFFont font(FPDFText_LoadFont(
253 document(), reinterpret_cast<const uint8_t*>(font_data.get()),
254 file_length, FPDF_FONT_TRUETYPE, /*cid=*/true));
255 FPDF_PAGEOBJECT text_object =
256 FPDFPageObj_CreateTextObj(document(), font.get(), 20.0f);
257 EXPECT_TRUE(text_object);
258
259 // Test the characters which are either mapped to one single unicode or
260 // multiple unicodes in the embedded font.
261 ScopedFPDFWideString text = GetFPDFWideString(L"这是第一句。 这是第二行。");
262 EXPECT_TRUE(FPDFText_SetText(text_object, text.get()));
263
264 FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 50, 200);
265 FPDFPage_InsertObject(page.get(), text_object);
266 EXPECT_TRUE(FPDFPage_GenerateContent(page.get()));
267
268#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
269#if defined(OS_APPLE)
270 const char kChecksum[] = "9a31fb87d1c6d2346bba22d1196041cd";
271#else
272 const char kChecksum[] = "5bb65e15fc0a685934cd5006dec08a76";
273#endif // defined(OS_APPLE)
274#else
275#if defined(OS_WIN)
276 const char kChecksum[] = "89e8eef5d6ad18c542a92a0519954d0f";
277#else
278 const char kChecksum[] = "9a31fb87d1c6d2346bba22d1196041cd";
279#endif // defined(OS_WIN)
280#endif // defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
281 ScopedFPDFBitmap page_bitmap = RenderPage(page.get());
282 CompareBitmap(page_bitmap.get(), 400, 400, kChecksum);
283
284 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
285 VerifySavedDocument(400, 400, kChecksum);
286}
287
Lei Zhangab41f252018-12-23 03:10:50 +0000288TEST_F(FPDFEditEmbedderTest, EmptyCreation) {
Tom Sepezd483eb42016-01-06 10:03:59 -0800289 EXPECT_TRUE(CreateEmptyDocument());
weili9b777de2016-08-19 16:19:46 -0700290 FPDF_PAGE page = FPDFPage_New(document(), 0, 640.0, 480.0);
Tom Sepezd483eb42016-01-06 10:03:59 -0800291 EXPECT_NE(nullptr, page);
Nicolas Penad9d6c292017-06-06 16:12:10 -0400292 // The FPDFPage_GenerateContent call should do nothing.
Tom Sepezd483eb42016-01-06 10:03:59 -0800293 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Tom Sepez0aec19b2016-01-07 12:22:44 -0800294 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
etienneb7712c262016-04-26 08:13:45 -0700295
Nicolas Penad9d6c292017-06-06 16:12:10 -0400296 EXPECT_THAT(GetString(), testing::MatchesRegex(std::string(
297 kExpectedPDF, sizeof(kExpectedPDF))));
weili9b777de2016-08-19 16:19:46 -0700298 FPDF_ClosePage(page);
Tom Sepezd483eb42016-01-06 10:03:59 -0800299}
thestigdc7ec032016-11-21 15:32:52 -0800300
301// Regression test for https://crbug.com/667012
Lei Zhangab41f252018-12-23 03:10:50 +0000302TEST_F(FPDFEditEmbedderTest, RasterizePDF) {
thestigdc7ec032016-11-21 15:32:52 -0800303 const char kAllBlackMd5sum[] = "5708fc5c4a8bd0abde99c8e8f0390615";
304
Lei Zhangf0542892019-01-17 18:46:27 +0000305 // Get the bitmap for the original document.
Tom Sepeze08d2b12018-04-25 18:49:32 +0000306 ScopedFPDFBitmap orig_bitmap;
thestigdc7ec032016-11-21 15:32:52 -0800307 {
Daniel Hosseinian5af51b62020-07-18 00:53:43 +0000308 ASSERT_TRUE(OpenDocument("black.pdf"));
thestigdc7ec032016-11-21 15:32:52 -0800309 FPDF_PAGE orig_page = LoadPage(0);
Lei Zhang107fa7b2018-02-09 21:48:15 +0000310 ASSERT_TRUE(orig_page);
311 orig_bitmap = RenderLoadedPage(orig_page);
312 CompareBitmap(orig_bitmap.get(), 612, 792, kAllBlackMd5sum);
thestigdc7ec032016-11-21 15:32:52 -0800313 UnloadPage(orig_page);
314 }
315
316 // Create a new document from |orig_bitmap| and save it.
317 {
318 FPDF_DOCUMENT temp_doc = FPDF_CreateNewDocument();
319 FPDF_PAGE temp_page = FPDFPage_New(temp_doc, 0, 612, 792);
320
321 // Add the bitmap to an image object and add the image object to the output
322 // page.
Lei Zhangcbd89572017-03-15 17:35:47 -0700323 FPDF_PAGEOBJECT temp_img = FPDFPageObj_NewImageObj(temp_doc);
Lei Zhang107fa7b2018-02-09 21:48:15 +0000324 EXPECT_TRUE(
325 FPDFImageObj_SetBitmap(&temp_page, 1, temp_img, orig_bitmap.get()));
thestigdc7ec032016-11-21 15:32:52 -0800326 EXPECT_TRUE(FPDFImageObj_SetMatrix(temp_img, 612, 0, 0, 792, 0, 0));
327 FPDFPage_InsertObject(temp_page, temp_img);
328 EXPECT_TRUE(FPDFPage_GenerateContent(temp_page));
329 EXPECT_TRUE(FPDF_SaveAsCopy(temp_doc, this, 0));
330 FPDF_ClosePage(temp_page);
331 FPDF_CloseDocument(temp_doc);
332 }
thestigdc7ec032016-11-21 15:32:52 -0800333
334 // Get the generated content. Make sure it is at least as big as the original
335 // PDF.
Lei Zhangd72fd582018-07-27 19:37:27 +0000336 EXPECT_GT(GetString().size(), 923u);
Dan Sinclair04e4dc82017-10-18 12:17:14 -0400337 VerifySavedDocument(612, 792, kAllBlackMd5sum);
thestigdc7ec032016-11-21 15:32:52 -0800338}
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500339
Lei Zhang03e5e682019-09-16 19:45:55 +0000340// TODO(crbug.com/pdfium/11): Fix this test and enable.
341#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
342#define MAYBE_AddPaths DISABLED_AddPaths
343#else
344#define MAYBE_AddPaths AddPaths
345#endif
346TEST_F(FPDFEditEmbedderTest, MAYBE_AddPaths) {
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500347 // Start with a blank page
Nicolas Penad03ca422017-03-06 13:54:33 -0500348 FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792);
Lei Zhang107fa7b2018-02-09 21:48:15 +0000349 ASSERT_TRUE(page);
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500350
351 // We will first add a red rectangle
352 FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(10, 10, 20, 20);
Lei Zhang107fa7b2018-02-09 21:48:15 +0000353 ASSERT_TRUE(red_rect);
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500354 // Expect false when trying to set colors out of range
Lei Zhang3475b482019-05-13 18:30:57 +0000355 EXPECT_FALSE(FPDFPageObj_SetStrokeColor(red_rect, 100, 100, 100, 300));
356 EXPECT_FALSE(FPDFPageObj_SetFillColor(red_rect, 200, 256, 200, 0));
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500357
358 // Fill rectangle with red and insert to the page
Lei Zhang3475b482019-05-13 18:30:57 +0000359 EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 0, 0, 255));
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500360 EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0));
Miklos Vajna491112b2018-05-30 13:30:10 +0000361
362 int fillmode = FPDF_FILLMODE_NONE;
363 FPDF_BOOL stroke = true;
364 EXPECT_TRUE(FPDFPath_GetDrawMode(red_rect, &fillmode, &stroke));
365 EXPECT_EQ(FPDF_FILLMODE_ALTERNATE, fillmode);
366 EXPECT_FALSE(stroke);
367
Lei Zhang8da98232019-12-11 23:29:33 +0000368 static const FS_MATRIX kMatrix = {1, 2, 3, 4, 5, 6};
369 EXPECT_FALSE(FPDFPath_SetMatrix(nullptr, &kMatrix));
370 EXPECT_TRUE(FPDFPath_SetMatrix(red_rect, &kMatrix));
371
372 FS_MATRIX matrix;
373 EXPECT_FALSE(FPDFPath_GetMatrix(nullptr, &matrix));
374 EXPECT_TRUE(FPDFPath_GetMatrix(red_rect, &matrix));
375 EXPECT_FLOAT_EQ(1.0f, matrix.a);
376 EXPECT_FLOAT_EQ(2.0f, matrix.b);
377 EXPECT_FLOAT_EQ(3.0f, matrix.c);
378 EXPECT_FLOAT_EQ(4.0f, matrix.d);
379 EXPECT_FLOAT_EQ(5.0f, matrix.e);
380 EXPECT_FLOAT_EQ(6.0f, matrix.f);
381
382 // Set back the identity matrix.
383 matrix = {1.0f, 0.0f, 0.0f, 1.0f, 0.0f, 0.0f};
384 EXPECT_TRUE(FPDFPath_SetMatrix(red_rect, &matrix));
Miklos Vajna97f4d672018-06-04 14:47:17 +0000385
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500386 FPDFPage_InsertObject(page, red_rect);
Lei Zhang107fa7b2018-02-09 21:48:15 +0000387 {
Lei Zhang30ff2532019-01-31 21:37:55 +0000388 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstb4baceb2020-04-28 23:46:10 +0000389 CompareBitmap(page_bitmap.get(), 612, 792, kRedRectangleChecksum);
Lei Zhang107fa7b2018-02-09 21:48:15 +0000390 }
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500391
392 // Now add to that a green rectangle with some medium alpha
393 FPDF_PAGEOBJECT green_rect = FPDFPageObj_CreateNewRect(100, 100, 40, 40);
Lei Zhang3475b482019-05-13 18:30:57 +0000394 EXPECT_TRUE(FPDFPageObj_SetFillColor(green_rect, 0, 255, 0, 128));
Miklos Vajnaed4705b2017-04-05 09:24:50 +0200395
Miklos Vajna1ef04c92017-05-08 18:14:19 +0200396 // Make sure the type of the rectangle is a path.
397 EXPECT_EQ(FPDF_PAGEOBJ_PATH, FPDFPageObj_GetType(green_rect));
398
Miklos Vajnaed4705b2017-04-05 09:24:50 +0200399 // Make sure we get back the same color we set previously.
400 unsigned int R;
401 unsigned int G;
402 unsigned int B;
403 unsigned int A;
Lei Zhang3475b482019-05-13 18:30:57 +0000404 EXPECT_TRUE(FPDFPageObj_GetFillColor(green_rect, &R, &G, &B, &A));
Lei Zhangd72fd582018-07-27 19:37:27 +0000405 EXPECT_EQ(0u, R);
406 EXPECT_EQ(255u, G);
407 EXPECT_EQ(0u, B);
408 EXPECT_EQ(128u, A);
Miklos Vajnaed4705b2017-04-05 09:24:50 +0200409
Miklos Vajna12abfd02017-09-15 07:49:03 +0200410 // Make sure the path has 5 points (1 FXPT_TYPE::MoveTo and 4
411 // FXPT_TYPE::LineTo).
Miklos Vajna0150a542017-09-21 21:46:56 +0200412 ASSERT_EQ(5, FPDFPath_CountSegments(green_rect));
Miklos Vajna36eed872017-09-20 22:52:43 +0200413 // Verify actual coordinates.
414 FPDF_PATHSEGMENT segment = FPDFPath_GetPathSegment(green_rect, 0);
415 float x;
416 float y;
417 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
418 EXPECT_EQ(100, x);
419 EXPECT_EQ(100, y);
420 EXPECT_EQ(FPDF_SEGMENT_MOVETO, FPDFPathSegment_GetType(segment));
421 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
422 segment = FPDFPath_GetPathSegment(green_rect, 1);
423 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
424 EXPECT_EQ(100, x);
425 EXPECT_EQ(140, y);
426 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
427 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
428 segment = FPDFPath_GetPathSegment(green_rect, 2);
429 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
430 EXPECT_EQ(140, x);
431 EXPECT_EQ(140, y);
432 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
433 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
434 segment = FPDFPath_GetPathSegment(green_rect, 3);
435 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
436 EXPECT_EQ(140, x);
437 EXPECT_EQ(100, y);
438 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
439 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
440 segment = FPDFPath_GetPathSegment(green_rect, 4);
441 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
442 EXPECT_EQ(100, x);
443 EXPECT_EQ(100, y);
444 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
445 EXPECT_TRUE(FPDFPathSegment_GetClose(segment));
Miklos Vajna12abfd02017-09-15 07:49:03 +0200446
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500447 EXPECT_TRUE(FPDFPath_SetDrawMode(green_rect, FPDF_FILLMODE_WINDING, 0));
448 FPDFPage_InsertObject(page, green_rect);
Lei Zhang107fa7b2018-02-09 21:48:15 +0000449 {
Lei Zhang30ff2532019-01-31 21:37:55 +0000450 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Lei Zhang107fa7b2018-02-09 21:48:15 +0000451 CompareBitmap(page_bitmap.get(), 612, 792,
452 "7b0b87604594e773add528fae567a558");
453 }
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500454
455 // Add a black triangle.
456 FPDF_PAGEOBJECT black_path = FPDFPageObj_CreateNewPath(400, 100);
Lei Zhang3475b482019-05-13 18:30:57 +0000457 EXPECT_TRUE(FPDFPageObj_SetFillColor(black_path, 0, 0, 0, 200));
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500458 EXPECT_TRUE(FPDFPath_SetDrawMode(black_path, FPDF_FILLMODE_ALTERNATE, 0));
459 EXPECT_TRUE(FPDFPath_LineTo(black_path, 400, 200));
460 EXPECT_TRUE(FPDFPath_LineTo(black_path, 300, 100));
461 EXPECT_TRUE(FPDFPath_Close(black_path));
Miklos Vajna12abfd02017-09-15 07:49:03 +0200462
463 // Make sure the path has 3 points (1 FXPT_TYPE::MoveTo and 2
464 // FXPT_TYPE::LineTo).
Miklos Vajna0150a542017-09-21 21:46:56 +0200465 ASSERT_EQ(3, FPDFPath_CountSegments(black_path));
Miklos Vajna36eed872017-09-20 22:52:43 +0200466 // Verify actual coordinates.
467 segment = FPDFPath_GetPathSegment(black_path, 0);
468 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
469 EXPECT_EQ(400, x);
470 EXPECT_EQ(100, y);
471 EXPECT_EQ(FPDF_SEGMENT_MOVETO, FPDFPathSegment_GetType(segment));
472 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
473 segment = FPDFPath_GetPathSegment(black_path, 1);
474 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
475 EXPECT_EQ(400, x);
476 EXPECT_EQ(200, y);
477 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
478 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
479 segment = FPDFPath_GetPathSegment(black_path, 2);
480 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
481 EXPECT_EQ(300, x);
482 EXPECT_EQ(100, y);
483 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
484 EXPECT_TRUE(FPDFPathSegment_GetClose(segment));
485 // Make sure out of bounds index access fails properly.
486 EXPECT_EQ(nullptr, FPDFPath_GetPathSegment(black_path, 3));
Miklos Vajna12abfd02017-09-15 07:49:03 +0200487
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500488 FPDFPage_InsertObject(page, black_path);
Lei Zhang107fa7b2018-02-09 21:48:15 +0000489 {
Lei Zhang30ff2532019-01-31 21:37:55 +0000490 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Lei Zhang107fa7b2018-02-09 21:48:15 +0000491 CompareBitmap(page_bitmap.get(), 612, 792,
492 "eadc8020a14dfcf091da2688733d8806");
493 }
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500494
495 // Now add a more complex blue path.
496 FPDF_PAGEOBJECT blue_path = FPDFPageObj_CreateNewPath(200, 200);
Lei Zhang3475b482019-05-13 18:30:57 +0000497 EXPECT_TRUE(FPDFPageObj_SetFillColor(blue_path, 0, 0, 255, 255));
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500498 EXPECT_TRUE(FPDFPath_SetDrawMode(blue_path, FPDF_FILLMODE_WINDING, 0));
499 EXPECT_TRUE(FPDFPath_LineTo(blue_path, 230, 230));
500 EXPECT_TRUE(FPDFPath_BezierTo(blue_path, 250, 250, 280, 280, 300, 300));
501 EXPECT_TRUE(FPDFPath_LineTo(blue_path, 325, 325));
502 EXPECT_TRUE(FPDFPath_LineTo(blue_path, 350, 325));
503 EXPECT_TRUE(FPDFPath_BezierTo(blue_path, 375, 330, 390, 360, 400, 400));
504 EXPECT_TRUE(FPDFPath_Close(blue_path));
505 FPDFPage_InsertObject(page, blue_path);
Lei Zhang107fa7b2018-02-09 21:48:15 +0000506 const char kLastMD5[] = "9823e1a21bd9b72b6a442ba4f12af946";
507 {
Lei Zhang30ff2532019-01-31 21:37:55 +0000508 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Lei Zhang107fa7b2018-02-09 21:48:15 +0000509 CompareBitmap(page_bitmap.get(), 612, 792, kLastMD5);
510 }
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500511
512 // Now save the result, closing the page and document
Nicolas Pena207b7272017-05-26 17:37:06 -0400513 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Nicolas Penad03ca422017-03-06 13:54:33 -0500514 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500515 FPDF_ClosePage(page);
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500516
517 // Render the saved result
Lei Zhang107fa7b2018-02-09 21:48:15 +0000518 VerifySavedDocument(612, 792, kLastMD5);
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500519}
520
Miklos Vajna45501f32019-07-30 07:41:02 +0000521TEST_F(FPDFEditEmbedderTest, ClipPath) {
522 // Load document with a clipped rectangle.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +0000523 ASSERT_TRUE(OpenDocument("clip_path.pdf"));
Miklos Vajna45501f32019-07-30 07:41:02 +0000524 FPDF_PAGE page = LoadPage(0);
525 ASSERT_TRUE(page);
526
527 ASSERT_EQ(1, FPDFPage_CountObjects(page));
528
529 FPDF_PAGEOBJECT triangle = FPDFPage_GetObject(page, 0);
530 ASSERT_TRUE(triangle);
531
532 // Test that we got the expected triangle.
533 ASSERT_EQ(4, FPDFPath_CountSegments(triangle));
534
535 FPDF_PATHSEGMENT segment = FPDFPath_GetPathSegment(triangle, 0);
536 float x;
537 float y;
538 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
539 EXPECT_EQ(10, x);
540 EXPECT_EQ(10, y);
541 EXPECT_EQ(FPDF_SEGMENT_MOVETO, FPDFPathSegment_GetType(segment));
542 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
543
544 segment = FPDFPath_GetPathSegment(triangle, 1);
545 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
546 EXPECT_EQ(25, x);
547 EXPECT_EQ(40, y);
548 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
549 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
550
551 segment = FPDFPath_GetPathSegment(triangle, 2);
552 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
553 EXPECT_EQ(40, x);
554 EXPECT_EQ(10, y);
555 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
556 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
557
558 segment = FPDFPath_GetPathSegment(triangle, 3);
559 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
560 EXPECT_TRUE(FPDFPathSegment_GetClose(segment));
561
562 // Test FPDFPageObj_GetClipPath().
563 ASSERT_EQ(nullptr, FPDFPageObj_GetClipPath(nullptr));
564
565 FPDF_CLIPPATH clip_path = FPDFPageObj_GetClipPath(triangle);
566 ASSERT_TRUE(clip_path);
567
568 // Test FPDFClipPath_CountPaths().
569 ASSERT_EQ(-1, FPDFClipPath_CountPaths(nullptr));
570 ASSERT_EQ(1, FPDFClipPath_CountPaths(clip_path));
571
572 // Test FPDFClipPath_CountPathSegments().
573 ASSERT_EQ(-1, FPDFClipPath_CountPathSegments(nullptr, 0));
574 ASSERT_EQ(-1, FPDFClipPath_CountPathSegments(clip_path, -1));
575 ASSERT_EQ(-1, FPDFClipPath_CountPathSegments(clip_path, 1));
576 ASSERT_EQ(4, FPDFClipPath_CountPathSegments(clip_path, 0));
577
578 // FPDFClipPath_GetPathSegment() negative testing.
579 ASSERT_EQ(nullptr, FPDFClipPath_GetPathSegment(nullptr, 0, 0));
580 ASSERT_EQ(nullptr, FPDFClipPath_GetPathSegment(clip_path, -1, 0));
581 ASSERT_EQ(nullptr, FPDFClipPath_GetPathSegment(clip_path, 1, 0));
582 ASSERT_EQ(nullptr, FPDFClipPath_GetPathSegment(clip_path, 0, -1));
583 ASSERT_EQ(nullptr, FPDFClipPath_GetPathSegment(clip_path, 0, 4));
584
585 // FPDFClipPath_GetPathSegment() positive testing.
586 segment = FPDFClipPath_GetPathSegment(clip_path, 0, 0);
587 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
588 EXPECT_EQ(10, x);
589 EXPECT_EQ(15, y);
590 EXPECT_EQ(FPDF_SEGMENT_MOVETO, FPDFPathSegment_GetType(segment));
591 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
592
593 segment = FPDFClipPath_GetPathSegment(clip_path, 0, 1);
594 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
595 EXPECT_EQ(40, x);
596 EXPECT_EQ(15, y);
597 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
598 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
599
600 segment = FPDFClipPath_GetPathSegment(clip_path, 0, 2);
601 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
602 EXPECT_EQ(40, x);
603 EXPECT_EQ(35, y);
604 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
605 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
606
607 segment = FPDFClipPath_GetPathSegment(clip_path, 0, 3);
608 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
609 EXPECT_EQ(10, x);
610 EXPECT_EQ(35, y);
611 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
612 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
613
614 UnloadPage(page);
615}
616
Lei Zhang571f9322019-09-25 12:40:25 +0000617TEST_F(FPDFEditEmbedderTest, BUG_1399) {
618 // Load document with a clipped rectangle.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +0000619 ASSERT_TRUE(OpenDocument("bug_1399.pdf"));
Lei Zhang571f9322019-09-25 12:40:25 +0000620 FPDF_PAGE page = LoadPage(0);
621 ASSERT_TRUE(page);
622
623 ASSERT_EQ(7, FPDFPage_CountObjects(page));
624
625 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 0);
626 ASSERT_TRUE(obj);
627
628 ASSERT_EQ(2, FPDFPath_CountSegments(obj));
629
630 FPDF_PATHSEGMENT segment = FPDFPath_GetPathSegment(obj, 0);
631 float x;
632 float y;
633 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
634 EXPECT_FLOAT_EQ(107.718f, x);
635 EXPECT_FLOAT_EQ(719.922f, y);
636 EXPECT_EQ(FPDF_SEGMENT_MOVETO, FPDFPathSegment_GetType(segment));
637 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
638
639 segment = FPDFPath_GetPathSegment(obj, 1);
640 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
641 EXPECT_FLOAT_EQ(394.718f, x);
642 EXPECT_FLOAT_EQ(719.922f, y);
643 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
644 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
645
646 FPDF_CLIPPATH clip_path = FPDFPageObj_GetClipPath(obj);
647 ASSERT_TRUE(clip_path);
648
649 EXPECT_EQ(-1, FPDFClipPath_CountPaths(clip_path));
650 EXPECT_EQ(-1, FPDFClipPath_CountPathSegments(clip_path, 0));
651 EXPECT_FALSE(FPDFClipPath_GetPathSegment(clip_path, 0, 0));
652
653 UnloadPage(page);
654}
655
Hui Yingstcb4203d2020-06-09 01:48:47 +0000656TEST_F(FPDFEditEmbedderTest, SetText) {
Henrique Nakashima5ebfd642018-06-07 15:18:55 +0000657 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +0000658 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
Henrique Nakashima5ebfd642018-06-07 15:18:55 +0000659 FPDF_PAGE page = LoadPage(0);
660 ASSERT_TRUE(page);
661
662 // Get the "Hello, world!" text object and change it.
663 ASSERT_EQ(2, FPDFPage_CountObjects(page));
664 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0);
665 ASSERT_TRUE(page_object);
Lei Zhangf0f67682019-04-08 17:03:21 +0000666 ScopedFPDFWideString text1 = GetFPDFWideString(L"Changed for SetText test");
Henrique Nakashima5ebfd642018-06-07 15:18:55 +0000667 EXPECT_TRUE(FPDFText_SetText(page_object, text1.get()));
668
669 // Verify the "Hello, world!" text is gone and "Changed for SetText test" is
670 // now displayed.
671 ASSERT_EQ(2, FPDFPage_CountObjects(page));
Hui Yingstcb4203d2020-06-09 01:48:47 +0000672
673#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
674#if defined(OS_WIN)
675 const char kChangedChecksum[] = "119f828345f547c68e9454418fb7d80d";
Henrique Nakashima5ebfd642018-06-07 15:18:55 +0000676#else
Hui Yingstcb4203d2020-06-09 01:48:47 +0000677 const char kChangedChecksum[] = "6bc5171f4eb329474989c6ccfa3d6303";
678#endif // defined(OS_WIN)
679#else
680#if defined(OS_WIN)
681 const char kChangedChecksum[] = "3137fdb27962671f5c3963a5e965eff5";
Lei Zhang0c03d632020-07-30 17:05:36 +0000682#elif defined(OS_APPLE)
Hui Yingstcb4203d2020-06-09 01:48:47 +0000683 const char kChangedChecksum[] = "904132275a1144ea06b0694537c80b4c";
684#else
685 const char kChangedChecksum[] = "a0c4ea6620772991f66bf7130379b08a";
Henrique Nakashima5ebfd642018-06-07 15:18:55 +0000686#endif
Hui Yingstcb4203d2020-06-09 01:48:47 +0000687#endif // defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
Henrique Nakashima5ebfd642018-06-07 15:18:55 +0000688 {
Lei Zhang30ff2532019-01-31 21:37:55 +0000689 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstcb4203d2020-06-09 01:48:47 +0000690 CompareBitmap(page_bitmap.get(), 200, 200, kChangedChecksum);
Henrique Nakashima5ebfd642018-06-07 15:18:55 +0000691 }
692
693 // Now save the result.
694 EXPECT_TRUE(FPDFPage_GenerateContent(page));
695 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
696
697 UnloadPage(page);
698
699 // Re-open the file and check the changes were kept in the saved .pdf.
Lei Zhang0b494052019-01-31 21:41:15 +0000700 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashima5ebfd642018-06-07 15:18:55 +0000701 FPDF_PAGE saved_page = LoadSavedPage(0);
702 EXPECT_EQ(2, FPDFPage_CountObjects(saved_page));
703 {
Lei Zhang30ff2532019-01-31 21:37:55 +0000704 ScopedFPDFBitmap page_bitmap = RenderPage(saved_page);
Hui Yingstcb4203d2020-06-09 01:48:47 +0000705 CompareBitmap(page_bitmap.get(), 200, 200, kChangedChecksum);
Henrique Nakashima5ebfd642018-06-07 15:18:55 +0000706 }
707
708 CloseSavedPage(saved_page);
709 CloseSavedDocument();
710}
711
Hui Yingst822fb8d2020-10-05 22:10:35 +0000712// TODO(crbug.com/pdfium/11): Fix this test and enable.
713#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
714#define MAYBE_SetTextKeepClippingPath DISABLED_SetTextKeepClippingPath
715#else
716#define MAYBE_SetTextKeepClippingPath SetTextKeepClippingPath
717#endif
718TEST_F(FPDFEditEmbedderTest, MAYBE_SetTextKeepClippingPath) {
Daniel Hosseinianda651442020-07-17 23:30:22 +0000719 // Load document with some text, with parts clipped.
720 ASSERT_TRUE(OpenDocument("bug_1558.pdf"));
721 FPDF_PAGE page = LoadPage(0);
722 ASSERT_TRUE(page);
723
724 static constexpr char kOriginalChecksum[] =
725#if defined(OS_WIN)
726 "220bf2086398fc46ac094952b244c8d9";
Lei Zhang0c03d632020-07-30 17:05:36 +0000727#elif defined(OS_APPLE)
Daniel Hosseinianda651442020-07-17 23:30:22 +0000728 "53cbaad93551ef2ccc27ddd63f2ca2b3";
729#else
730 "ba1936fa8ca1e8cca108da76ff3500a6";
731#endif
732 {
733 // When opened before any editing and saving, the clipping path is rendered.
734 ScopedFPDFBitmap original_bitmap = RenderPage(page);
735 CompareBitmap(original_bitmap.get(), 200, 200, kOriginalChecksum);
736 }
737
738 // "Change" the text in the objects to their current values to force them to
739 // regenerate when saving.
740 {
741 ScopedFPDFTextPage text_page(FPDFText_LoadPage(page));
742 ASSERT_TRUE(text_page);
743 const int obj_count = FPDFPage_CountObjects(page);
744 ASSERT_EQ(2, obj_count);
745 for (int i = 0; i < obj_count; ++i) {
746 FPDF_PAGEOBJECT text_obj = FPDFPage_GetObject(page, i);
747 ASSERT_EQ(FPDF_PAGEOBJ_TEXT, FPDFPageObj_GetType(text_obj));
748 unsigned long size =
749 FPDFTextObj_GetText(text_obj, text_page.get(),
750 /*buffer=*/nullptr, /*length=*/0);
751 ASSERT_GT(size, 0u);
752 std::vector<FPDF_WCHAR> buffer = GetFPDFWideStringBuffer(size);
753 ASSERT_EQ(size, FPDFTextObj_GetText(text_obj, text_page.get(),
754 buffer.data(), size));
755 EXPECT_TRUE(FPDFText_SetText(text_obj, buffer.data()));
756 }
757 }
758
759 {
760 // After editing but before saving, the clipping path is retained.
761 ScopedFPDFBitmap edited_bitmap = RenderPage(page);
762 CompareBitmap(edited_bitmap.get(), 200, 200, kOriginalChecksum);
763 }
764
765 // Save the file.
766 EXPECT_TRUE(FPDFPage_GenerateContent(page));
767 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
768 UnloadPage(page);
769
770 // Open the saved copy and render it.
771 ASSERT_TRUE(OpenSavedDocument());
772 FPDF_PAGE saved_page = LoadSavedPage(0);
773 ASSERT_TRUE(saved_page);
774
775 {
Daniel Hosseinianda651442020-07-17 23:30:22 +0000776 ScopedFPDFBitmap saved_bitmap = RenderSavedPage(saved_page);
Daniel Hosseinian75efd912020-07-21 08:07:28 +0000777 CompareBitmap(saved_bitmap.get(), 200, 200, kOriginalChecksum);
Daniel Hosseinianda651442020-07-17 23:30:22 +0000778 }
779
780 CloseSavedPage(saved_page);
781 CloseSavedDocument();
782}
783
Hui Yingst08c40712020-04-29 01:37:35 +0000784TEST_F(FPDFEditEmbedderTest, RemovePageObject) {
Henrique Nakashima35841fa2018-03-15 15:25:16 +0000785 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +0000786 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
Henrique Nakashima35841fa2018-03-15 15:25:16 +0000787 FPDF_PAGE page = LoadPage(0);
788 ASSERT_TRUE(page);
789
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000790 // Show what the original file looks like.
791 {
Lei Zhang30ff2532019-01-31 21:37:55 +0000792 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstb4baceb2020-04-28 23:46:10 +0000793 CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldChecksum);
Henrique Nakashima35841fa2018-03-15 15:25:16 +0000794 }
795
796 // Get the "Hello, world!" text object and remove it.
797 ASSERT_EQ(2, FPDFPage_CountObjects(page));
798 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0);
799 ASSERT_TRUE(page_object);
800 EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object));
801
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000802 // Verify the "Hello, world!" text is gone.
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, kFirstRemovedChecksum);
Henrique Nakashima35841fa2018-03-15 15:25:16 +0000806 }
807 ASSERT_EQ(1, FPDFPage_CountObjects(page));
808
809 UnloadPage(page);
810 FPDFPageObj_Destroy(page_object);
811}
812
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000813void CheckMarkCounts(FPDF_PAGE page,
814 int start_from,
815 int expected_object_count,
816 size_t expected_prime_count,
817 size_t expected_square_count,
818 size_t expected_greater_than_ten_count,
819 size_t expected_bounds_count) {
820 int object_count = FPDFPage_CountObjects(page);
821 ASSERT_EQ(expected_object_count, object_count);
822
823 size_t prime_count = 0;
824 size_t square_count = 0;
825 size_t greater_than_ten_count = 0;
826 size_t bounds_count = 0;
827 for (int i = 0; i < object_count; ++i) {
828 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i);
829
830 int mark_count = FPDFPageObj_CountMarks(page_object);
831 for (int j = 0; j < mark_count; ++j) {
832 FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, j);
833
834 char buffer[256];
Henrique Nakashimac3099d12018-09-18 18:08:15 +0000835 unsigned long name_len = 999u;
836 ASSERT_TRUE(
837 FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len));
838 EXPECT_GT(name_len, 0u);
839 EXPECT_NE(999u, name_len);
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000840 std::wstring name =
841 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer));
842 if (name == L"Prime") {
843 prime_count++;
844 } else if (name == L"Square") {
845 square_count++;
846 int expected_square = start_from + i;
847 EXPECT_EQ(1, FPDFPageObjMark_CountParams(mark));
848
Henrique Nakashimac3099d12018-09-18 18:08:15 +0000849 unsigned long get_param_key_return = 999u;
850 ASSERT_TRUE(FPDFPageObjMark_GetParamKey(mark, 0, buffer, sizeof(buffer),
851 &get_param_key_return));
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000852 EXPECT_EQ((6u + 1u) * 2u, get_param_key_return);
853 std::wstring key =
854 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer));
855 EXPECT_EQ(L"Factor", key);
856
857 EXPECT_EQ(FPDF_OBJECT_NUMBER,
Henrique Nakashima94230e52018-07-11 22:02:02 +0000858 FPDFPageObjMark_GetParamValueType(mark, "Factor"));
Henrique Nakashima140dead2018-07-11 21:40:03 +0000859 int square_root;
Henrique Nakashima94230e52018-07-11 22:02:02 +0000860 EXPECT_TRUE(
861 FPDFPageObjMark_GetParamIntValue(mark, "Factor", &square_root));
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000862 EXPECT_EQ(expected_square, square_root * square_root);
863 } else if (name == L"GreaterThanTen") {
864 greater_than_ten_count++;
865 } else if (name == L"Bounds") {
866 bounds_count++;
867 EXPECT_EQ(1, FPDFPageObjMark_CountParams(mark));
868
Henrique Nakashimac3099d12018-09-18 18:08:15 +0000869 unsigned long get_param_key_return = 999u;
870 ASSERT_TRUE(FPDFPageObjMark_GetParamKey(mark, 0, buffer, sizeof(buffer),
871 &get_param_key_return));
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000872 EXPECT_EQ((8u + 1u) * 2u, get_param_key_return);
873 std::wstring key =
874 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer));
875 EXPECT_EQ(L"Position", key);
876
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000877 EXPECT_EQ(FPDF_OBJECT_STRING,
Henrique Nakashima94230e52018-07-11 22:02:02 +0000878 FPDFPageObjMark_GetParamValueType(mark, "Position"));
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000879 unsigned long length;
Henrique Nakashimaa3406772018-07-13 19:10:53 +0000880 EXPECT_TRUE(FPDFPageObjMark_GetParamStringValue(
881 mark, "Position", buffer, sizeof(buffer), &length));
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000882 ASSERT_GT(length, 0u);
Henrique Nakashima140dead2018-07-11 21:40:03 +0000883 std::wstring value =
884 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer));
Henrique Nakashimad8df8c32018-07-12 22:15:09 +0000885
Henrique Nakashimaa3406772018-07-13 19:10:53 +0000886 // "Position" can be "First", "Last", or "End".
Henrique Nakashimad8df8c32018-07-12 22:15:09 +0000887 if (i == 0) {
888 EXPECT_EQ((5u + 1u) * 2u, length);
889 EXPECT_EQ(L"First", value);
890 } else if (i == object_count - 1) {
Henrique Nakashimaa3406772018-07-13 19:10:53 +0000891 if (length == (4u + 1u) * 2u) {
892 EXPECT_EQ(L"Last", value);
893 } else if (length == (3u + 1u) * 2u) {
894 EXPECT_EQ(L"End", value);
895 } else {
896 FAIL();
897 }
Henrique Nakashimad8df8c32018-07-12 22:15:09 +0000898 } else {
899 FAIL();
900 }
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000901 } else {
902 FAIL();
903 }
904 }
905 }
906
907 // Expect certain number of tagged objects. The test file contains strings
908 // from 1 to 19.
909 EXPECT_EQ(expected_prime_count, prime_count);
910 EXPECT_EQ(expected_square_count, square_count);
911 EXPECT_EQ(expected_greater_than_ten_count, greater_than_ten_count);
912 EXPECT_EQ(expected_bounds_count, bounds_count);
913}
914
Lei Zhangab41f252018-12-23 03:10:50 +0000915TEST_F(FPDFEditEmbedderTest, ReadMarkedObjectsIndirectDict) {
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000916 // Load document with some text marked with an indirect property.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +0000917 ASSERT_TRUE(OpenDocument("text_in_page_marked_indirect.pdf"));
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000918 FPDF_PAGE page = LoadPage(0);
919 ASSERT_TRUE(page);
920
921 CheckMarkCounts(page, 1, 19, 8, 4, 9, 1);
922
923 UnloadPage(page);
924}
925
Lei Zhang03e5e682019-09-16 19:45:55 +0000926// TODO(crbug.com/pdfium/11): Fix this test and enable.
927#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
928#define MAYBE_RemoveMarkedObjectsPrime DISABLED_RemoveMarkedObjectsPrime
929#else
930#define MAYBE_RemoveMarkedObjectsPrime RemoveMarkedObjectsPrime
931#endif
932TEST_F(FPDFEditEmbedderTest, MAYBE_RemoveMarkedObjectsPrime) {
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000933 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +0000934 ASSERT_TRUE(OpenDocument("text_in_page_marked.pdf"));
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000935 FPDF_PAGE page = LoadPage(0);
936 ASSERT_TRUE(page);
937
938 // Show what the original file looks like.
939 {
Lei Zhang0c03d632020-07-30 17:05:36 +0000940#if defined(OS_APPLE)
Lei Zhang0f6b4342020-02-25 20:00:39 +0000941 const char kOriginalMD5[] = "adf815e53c788a5272b4df07c610a1da";
Lei Zhange67bcc72019-04-30 18:55:58 +0000942#elif defined(OS_WIN)
Lei Zhanga2b70732019-06-25 08:34:22 +0000943 const char kOriginalMD5[] = "00542ee435b37749c4453be63bf7bdb6";
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000944#else
Lei Zhanga2b70732019-06-25 08:34:22 +0000945 const char kOriginalMD5[] = "41647268d5911d049801803b15c2dfb0";
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000946#endif
Lei Zhang30ff2532019-01-31 21:37:55 +0000947 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000948 CompareBitmap(page_bitmap.get(), 200, 200, kOriginalMD5);
949 }
950
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000951 constexpr int expected_object_count = 19;
952 CheckMarkCounts(page, 1, expected_object_count, 8, 4, 9, 1);
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000953
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000954 // Get all objects marked with "Prime"
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000955 std::vector<FPDF_PAGEOBJECT> primes;
Henrique Nakashimacbed9492018-07-10 21:54:26 +0000956 for (int i = 0; i < expected_object_count; ++i) {
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000957 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i);
958
959 int mark_count = FPDFPageObj_CountMarks(page_object);
960 for (int j = 0; j < mark_count; ++j) {
961 FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, j);
962
963 char buffer[256];
Henrique Nakashimac3099d12018-09-18 18:08:15 +0000964 unsigned long name_len = 999u;
965 ASSERT_TRUE(
966 FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len));
967 EXPECT_GT(name_len, 0u);
968 EXPECT_NE(999u, name_len);
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000969 std::wstring name =
970 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer));
971 if (name == L"Prime") {
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000972 primes.push_back(page_object);
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000973 }
974 }
975 }
976
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000977 // Remove all objects marked with "Prime".
978 for (FPDF_PAGEOBJECT page_object : primes) {
979 EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object));
980 FPDFPageObj_Destroy(page_object);
981 }
982
983 EXPECT_EQ(11, FPDFPage_CountObjects(page));
984
Lei Zhang0c03d632020-07-30 17:05:36 +0000985#if defined(OS_APPLE)
Lei Zhang0f6b4342020-02-25 20:00:39 +0000986 const char kNonPrimesMD5[] = "d29e2ddff56e0d12f340794d26796400";
987 const char kNonPrimesAfterSaveMD5[] = "10eff2cd0037b661496981779601fa6f";
Lei Zhange67bcc72019-04-30 18:55:58 +0000988#elif defined(OS_WIN)
Lei Zhanga2b70732019-06-25 08:34:22 +0000989 const char kNonPrimesMD5[] = "86e371fdae30c2471f476631f3f93413";
990 const char kNonPrimesAfterSaveMD5[] = "86e371fdae30c2471f476631f3f93413";
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000991#else
Lei Zhanga2b70732019-06-25 08:34:22 +0000992 const char kNonPrimesMD5[] = "67ab13115d0cc34e99a1003c28047b40";
993 const char kNonPrimesAfterSaveMD5[] = "67ab13115d0cc34e99a1003c28047b40";
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000994#endif
Henrique Nakashimaa5078b72018-09-10 17:09:42 +0000995 {
Lei Zhang30ff2532019-01-31 21:37:55 +0000996 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Henrique Nakashimac90adc52018-03-27 16:26:44 +0000997 CompareBitmap(page_bitmap.get(), 200, 200, kNonPrimesMD5);
998 }
999
Henrique Nakashimaa5078b72018-09-10 17:09:42 +00001000 // Save the file.
1001 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1002 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
Henrique Nakashimac90adc52018-03-27 16:26:44 +00001003 UnloadPage(page);
Henrique Nakashimaa5078b72018-09-10 17:09:42 +00001004
1005 // Re-open the file and check the prime marks are not there anymore.
Lei Zhang0b494052019-01-31 21:41:15 +00001006 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashimaa5078b72018-09-10 17:09:42 +00001007 FPDF_PAGE saved_page = LoadSavedPage(0);
1008 EXPECT_EQ(11, FPDFPage_CountObjects(saved_page));
1009
1010 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001011 ScopedFPDFBitmap page_bitmap = RenderPage(saved_page);
Manoj Biswas7daf7a22019-03-22 23:02:34 +00001012 CompareBitmap(page_bitmap.get(), 200, 200, kNonPrimesAfterSaveMD5);
Henrique Nakashimaa5078b72018-09-10 17:09:42 +00001013 }
1014
1015 CloseSavedPage(saved_page);
1016 CloseSavedDocument();
Henrique Nakashimac90adc52018-03-27 16:26:44 +00001017}
1018
Lei Zhangab41f252018-12-23 03:10:50 +00001019TEST_F(FPDFEditEmbedderTest, RemoveMarks) {
Henrique Nakashimafed4adb2018-07-13 19:47:22 +00001020 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001021 ASSERT_TRUE(OpenDocument("text_in_page_marked.pdf"));
Henrique Nakashimafed4adb2018-07-13 19:47:22 +00001022 FPDF_PAGE page = LoadPage(0);
1023 ASSERT_TRUE(page);
1024
1025 constexpr int kExpectedObjectCount = 19;
1026 CheckMarkCounts(page, 1, kExpectedObjectCount, 8, 4, 9, 1);
1027
1028 // Remove all "Prime" content marks.
1029 for (int i = 0; i < kExpectedObjectCount; ++i) {
1030 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i);
1031
1032 int mark_count = FPDFPageObj_CountMarks(page_object);
1033 for (int j = mark_count - 1; j >= 0; --j) {
1034 FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, j);
1035
1036 char buffer[256];
Henrique Nakashimac3099d12018-09-18 18:08:15 +00001037 unsigned long name_len = 999u;
1038 ASSERT_TRUE(
1039 FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len));
1040 EXPECT_GT(name_len, 0u);
1041 EXPECT_NE(999u, name_len);
Henrique Nakashimafed4adb2018-07-13 19:47:22 +00001042 std::wstring name =
1043 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer));
1044 if (name == L"Prime") {
1045 // Remove mark.
1046 EXPECT_TRUE(FPDFPageObj_RemoveMark(page_object, mark));
1047
1048 // Verify there is now one fewer mark in the page object.
1049 EXPECT_EQ(mark_count - 1, FPDFPageObj_CountMarks(page_object));
1050 }
1051 }
1052 }
1053
1054 // Verify there are 0 "Prime" content marks now.
1055 CheckMarkCounts(page, 1, kExpectedObjectCount, 0, 4, 9, 1);
1056
1057 // Save the file.
1058 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1059 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1060 UnloadPage(page);
1061
1062 // Re-open the file and check the prime marks are not there anymore.
Lei Zhang0b494052019-01-31 21:41:15 +00001063 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashimafed4adb2018-07-13 19:47:22 +00001064 FPDF_PAGE saved_page = LoadSavedPage(0);
1065
1066 CheckMarkCounts(saved_page, 1, kExpectedObjectCount, 0, 4, 9, 1);
1067
1068 CloseSavedPage(saved_page);
1069 CloseSavedDocument();
1070}
1071
Lei Zhangab41f252018-12-23 03:10:50 +00001072TEST_F(FPDFEditEmbedderTest, RemoveMarkParam) {
Henrique Nakashimacf403ba2018-07-13 20:12:41 +00001073 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001074 ASSERT_TRUE(OpenDocument("text_in_page_marked.pdf"));
Henrique Nakashimacf403ba2018-07-13 20:12:41 +00001075 FPDF_PAGE page = LoadPage(0);
1076 ASSERT_TRUE(page);
1077
1078 constexpr int kExpectedObjectCount = 19;
1079 CheckMarkCounts(page, 1, kExpectedObjectCount, 8, 4, 9, 1);
1080
1081 // Remove all "Square" content marks parameters.
1082 for (int i = 0; i < kExpectedObjectCount; ++i) {
1083 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i);
1084
1085 int mark_count = FPDFPageObj_CountMarks(page_object);
1086 for (int j = 0; j < mark_count; ++j) {
1087 FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, j);
1088
1089 char buffer[256];
Henrique Nakashimac3099d12018-09-18 18:08:15 +00001090 unsigned long name_len = 999u;
1091 ASSERT_TRUE(
1092 FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len));
1093 EXPECT_GT(name_len, 0u);
1094 EXPECT_NE(999u, name_len);
Henrique Nakashimacf403ba2018-07-13 20:12:41 +00001095 std::wstring name =
1096 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer));
1097 if (name == L"Square") {
1098 // Show the mark has a "Factor" parameter.
1099 int out_value;
1100 EXPECT_TRUE(
1101 FPDFPageObjMark_GetParamIntValue(mark, "Factor", &out_value));
1102
1103 // Remove parameter.
1104 EXPECT_TRUE(FPDFPageObjMark_RemoveParam(page_object, mark, "Factor"));
1105
1106 // Verify the "Factor" parameter is gone.
1107 EXPECT_FALSE(
1108 FPDFPageObjMark_GetParamIntValue(mark, "Factor", &out_value));
1109 }
1110 }
1111 }
1112
1113 // Save the file.
1114 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1115 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1116 UnloadPage(page);
1117
1118 // Re-open the file and check the "Factor" parameters are still gone.
Lei Zhang0b494052019-01-31 21:41:15 +00001119 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashimacf403ba2018-07-13 20:12:41 +00001120 FPDF_PAGE saved_page = LoadSavedPage(0);
1121
1122 size_t square_count = 0;
1123 for (int i = 0; i < kExpectedObjectCount; ++i) {
1124 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(saved_page, i);
1125
1126 int mark_count = FPDFPageObj_CountMarks(page_object);
1127 for (int j = 0; j < mark_count; ++j) {
1128 FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, j);
1129
1130 char buffer[256];
Henrique Nakashimac3099d12018-09-18 18:08:15 +00001131 unsigned long name_len = 999u;
1132 ASSERT_TRUE(
1133 FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len));
1134 EXPECT_GT(name_len, 0u);
1135 EXPECT_NE(999u, name_len);
Henrique Nakashimacf403ba2018-07-13 20:12:41 +00001136 std::wstring name =
1137 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer));
1138 if (name == L"Square") {
1139 // Verify the "Factor" parameter is still gone.
1140 int out_value;
1141 EXPECT_FALSE(
1142 FPDFPageObjMark_GetParamIntValue(mark, "Factor", &out_value));
1143
1144 ++square_count;
1145 }
1146 }
1147 }
1148
1149 // Verify the parameters are gone, but the marks are not.
1150 EXPECT_EQ(4u, square_count);
1151
1152 CloseSavedPage(saved_page);
1153 CloseSavedDocument();
1154}
1155
Lei Zhangab41f252018-12-23 03:10:50 +00001156TEST_F(FPDFEditEmbedderTest, MaintainMarkedObjects) {
Henrique Nakashimab4bcf692018-07-11 21:19:22 +00001157 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001158 ASSERT_TRUE(OpenDocument("text_in_page_marked.pdf"));
Henrique Nakashimab4bcf692018-07-11 21:19:22 +00001159 FPDF_PAGE page = LoadPage(0);
1160 ASSERT_TRUE(page);
1161
1162 // Iterate over all objects, counting the number of times each content mark
1163 // name appears.
1164 CheckMarkCounts(page, 1, 19, 8, 4, 9, 1);
1165
1166 // Remove first page object.
1167 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0);
1168 EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object));
1169 FPDFPageObj_Destroy(page_object);
1170
1171 CheckMarkCounts(page, 2, 18, 8, 3, 9, 1);
1172
1173 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1174 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1175
1176 UnloadPage(page);
1177
Lei Zhang0b494052019-01-31 21:41:15 +00001178 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashimab4bcf692018-07-11 21:19:22 +00001179 FPDF_PAGE saved_page = LoadSavedPage(0);
1180
1181 CheckMarkCounts(saved_page, 2, 18, 8, 3, 9, 1);
1182
1183 CloseSavedPage(saved_page);
1184 CloseSavedDocument();
1185}
1186
Lei Zhangab41f252018-12-23 03:10:50 +00001187TEST_F(FPDFEditEmbedderTest, MaintainIndirectMarkedObjects) {
Henrique Nakashimab4bcf692018-07-11 21:19:22 +00001188 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001189 ASSERT_TRUE(OpenDocument("text_in_page_marked_indirect.pdf"));
Henrique Nakashimab4bcf692018-07-11 21:19:22 +00001190 FPDF_PAGE page = LoadPage(0);
1191 ASSERT_TRUE(page);
1192
1193 // Iterate over all objects, counting the number of times each content mark
1194 // name appears.
1195 CheckMarkCounts(page, 1, 19, 8, 4, 9, 1);
1196
1197 // Remove first page object.
1198 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0);
1199 EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object));
1200 FPDFPageObj_Destroy(page_object);
1201
1202 CheckMarkCounts(page, 2, 18, 8, 3, 9, 1);
1203
1204 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1205 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1206
1207 UnloadPage(page);
1208
Lei Zhang0b494052019-01-31 21:41:15 +00001209 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashimab4bcf692018-07-11 21:19:22 +00001210 FPDF_PAGE saved_page = LoadSavedPage(0);
1211
1212 CheckMarkCounts(saved_page, 2, 18, 8, 3, 9, 1);
1213
1214 CloseSavedPage(saved_page);
1215 CloseSavedDocument();
1216}
1217
Lei Zhangab41f252018-12-23 03:10:50 +00001218TEST_F(FPDFEditEmbedderTest, RemoveExistingPageObject) {
Henrique Nakashimac49e62e2018-04-16 20:58:47 +00001219 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001220 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
Henrique Nakashimac49e62e2018-04-16 20:58:47 +00001221 FPDF_PAGE page = LoadPage(0);
1222 ASSERT_TRUE(page);
1223
1224 // Get the "Hello, world!" text object and remove it.
1225 ASSERT_EQ(2, FPDFPage_CountObjects(page));
1226 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0);
1227 ASSERT_TRUE(page_object);
1228 EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object));
1229
1230 // Verify the "Hello, world!" text is gone.
1231 ASSERT_EQ(1, FPDFPage_CountObjects(page));
1232
1233 // Save the file
1234 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1235 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1236 UnloadPage(page);
1237 FPDFPageObj_Destroy(page_object);
1238
1239 // Re-open the file and check the page object count is still 1.
Lei Zhang0b494052019-01-31 21:41:15 +00001240 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashimac49e62e2018-04-16 20:58:47 +00001241 FPDF_PAGE saved_page = LoadSavedPage(0);
1242 EXPECT_EQ(1, FPDFPage_CountObjects(saved_page));
1243 CloseSavedPage(saved_page);
1244 CloseSavedDocument();
1245}
1246
Hui Yingstcb4203d2020-06-09 01:48:47 +00001247TEST_F(FPDFEditEmbedderTest, RemoveExistingPageObjectSplitStreamsNotLonely) {
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001248 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001249 ASSERT_TRUE(OpenDocument("hello_world_split_streams.pdf"));
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001250 FPDF_PAGE page = LoadPage(0);
1251 ASSERT_TRUE(page);
1252
1253 // Get the "Hello, world!" text object and remove it. There is another object
1254 // in the same stream that says "Goodbye, world!"
1255 ASSERT_EQ(3, FPDFPage_CountObjects(page));
1256 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0);
1257 ASSERT_TRUE(page_object);
1258 EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object));
1259
1260 // Verify the "Hello, world!" text is gone.
1261 ASSERT_EQ(2, FPDFPage_CountObjects(page));
Hui Yingstcb4203d2020-06-09 01:48:47 +00001262#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
1263#if defined(OS_WIN)
1264 const char kHelloRemovedChecksum[] = "e05c7837a5cad61305d157720637f969";
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001265#else
Hui Yingstcb4203d2020-06-09 01:48:47 +00001266 const char kHelloRemovedChecksum[] = "deed7dc2754dc80930f3b05e2ac86c94";
1267#endif // defined(OS_WIN)
1268#else
1269#if defined(OS_WIN)
1270 const char kHelloRemovedChecksum[] = "a97d4c72c969ba373c2dce675d277e65";
Lei Zhang0c03d632020-07-30 17:05:36 +00001271#elif defined(OS_APPLE)
Hui Yingstcb4203d2020-06-09 01:48:47 +00001272 const char kHelloRemovedChecksum[] = "3b3b27602a86dfe5996a33c42c59885b";
1273#else
1274 const char kHelloRemovedChecksum[] = "95b92950647a2190e1230911e7a1a0e9";
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001275#endif
Hui Yingstcb4203d2020-06-09 01:48:47 +00001276#endif // defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001277 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001278 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstcb4203d2020-06-09 01:48:47 +00001279 CompareBitmap(page_bitmap.get(), 200, 200, kHelloRemovedChecksum);
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001280 }
1281
1282 // Save the file
1283 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1284 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1285 UnloadPage(page);
1286 FPDFPageObj_Destroy(page_object);
1287
1288 // Re-open the file and check the page object count is still 2.
Lei Zhang0b494052019-01-31 21:41:15 +00001289 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001290 FPDF_PAGE saved_page = LoadSavedPage(0);
1291
1292 EXPECT_EQ(2, FPDFPage_CountObjects(saved_page));
1293 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001294 ScopedFPDFBitmap page_bitmap = RenderPage(saved_page);
Hui Yingstcb4203d2020-06-09 01:48:47 +00001295 CompareBitmap(page_bitmap.get(), 200, 200, kHelloRemovedChecksum);
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001296 }
1297
1298 CloseSavedPage(saved_page);
1299 CloseSavedDocument();
1300}
1301
Hui Yingst08c40712020-04-29 01:37:35 +00001302TEST_F(FPDFEditEmbedderTest, RemoveExistingPageObjectSplitStreamsLonely) {
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001303 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001304 ASSERT_TRUE(OpenDocument("hello_world_split_streams.pdf"));
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001305 FPDF_PAGE page = LoadPage(0);
1306 ASSERT_TRUE(page);
1307
1308 // Get the "Greetings, world!" text object and remove it. This is the only
1309 // object in the stream.
1310 ASSERT_EQ(3, FPDFPage_CountObjects(page));
1311 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 2);
1312 ASSERT_TRUE(page_object);
1313 EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object));
1314
1315 // Verify the "Greetings, world!" text is gone.
1316 ASSERT_EQ(2, FPDFPage_CountObjects(page));
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001317 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001318 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00001319 CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldChecksum);
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001320 }
1321
1322 // Save the file
1323 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1324 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1325 UnloadPage(page);
1326 FPDFPageObj_Destroy(page_object);
1327
1328 // Re-open the file and check the page object count is still 2.
Lei Zhang0b494052019-01-31 21:41:15 +00001329 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001330 FPDF_PAGE saved_page = LoadSavedPage(0);
1331
1332 EXPECT_EQ(2, FPDFPage_CountObjects(saved_page));
1333 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001334 ScopedFPDFBitmap page_bitmap = RenderPage(saved_page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00001335 CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldChecksum);
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001336 }
1337
1338 CloseSavedPage(saved_page);
1339 CloseSavedDocument();
1340}
1341
Lei Zhangab41f252018-12-23 03:10:50 +00001342TEST_F(FPDFEditEmbedderTest, GetContentStream) {
Henrique Nakashima6eb79392018-06-12 20:27:35 +00001343 // Load document with some text split across streams.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001344 ASSERT_TRUE(OpenDocument("split_streams.pdf"));
Henrique Nakashima6eb79392018-06-12 20:27:35 +00001345 FPDF_PAGE page = LoadPage(0);
1346 ASSERT_TRUE(page);
1347
1348 // Content stream 0: page objects 0-14.
1349 // Content stream 1: page objects 15-17.
1350 // Content stream 2: page object 18.
1351 ASSERT_EQ(19, FPDFPage_CountObjects(page));
1352 for (int i = 0; i < 19; i++) {
1353 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i);
1354 ASSERT_TRUE(page_object);
1355 CPDF_PageObject* cpdf_page_object =
1356 CPDFPageObjectFromFPDFPageObject(page_object);
1357 if (i < 15)
1358 EXPECT_EQ(0, cpdf_page_object->GetContentStream()) << i;
1359 else if (i < 18)
1360 EXPECT_EQ(1, cpdf_page_object->GetContentStream()) << i;
1361 else
1362 EXPECT_EQ(2, cpdf_page_object->GetContentStream()) << i;
1363 }
1364
1365 UnloadPage(page);
1366}
1367
Lei Zhang03e5e682019-09-16 19:45:55 +00001368// TODO(crbug.com/pdfium/11): Fix this test and enable.
1369#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
1370#define MAYBE_RemoveAllFromStream DISABLED_RemoveAllFromStream
1371#else
1372#define MAYBE_RemoveAllFromStream RemoveAllFromStream
1373#endif
1374TEST_F(FPDFEditEmbedderTest, MAYBE_RemoveAllFromStream) {
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001375 // Load document with some text split across streams.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001376 ASSERT_TRUE(OpenDocument("split_streams.pdf"));
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001377 FPDF_PAGE page = LoadPage(0);
1378 ASSERT_TRUE(page);
1379
1380 // Content stream 0: page objects 0-14.
1381 // Content stream 1: page objects 15-17.
1382 // Content stream 2: page object 18.
1383 ASSERT_EQ(19, FPDFPage_CountObjects(page));
1384
1385 // Loop backwards because objects will being removed, which shifts the indexes
1386 // after the removed position.
1387 for (int i = 18; i >= 0; i--) {
1388 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i);
1389 ASSERT_TRUE(page_object);
1390 CPDF_PageObject* cpdf_page_object =
1391 CPDFPageObjectFromFPDFPageObject(page_object);
1392
1393 // Empty content stream 1.
1394 if (cpdf_page_object->GetContentStream() == 1) {
1395 EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object));
1396 FPDFPageObj_Destroy(page_object);
1397 }
1398 }
1399
1400 // Content stream 0: page objects 0-14.
1401 // Content stream 2: page object 15.
1402 ASSERT_EQ(16, FPDFPage_CountObjects(page));
1403 for (int i = 0; i < 16; i++) {
1404 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i);
1405 ASSERT_TRUE(page_object);
1406 CPDF_PageObject* cpdf_page_object =
1407 CPDFPageObjectFromFPDFPageObject(page_object);
1408 if (i < 15)
1409 EXPECT_EQ(0, cpdf_page_object->GetContentStream()) << i;
1410 else
1411 EXPECT_EQ(2, cpdf_page_object->GetContentStream()) << i;
1412 }
1413
1414 // Generate contents should remove the empty stream and update the page
1415 // objects' contents stream indexes.
1416 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1417
1418 // Content stream 0: page objects 0-14.
1419 // Content stream 1: page object 15.
1420 ASSERT_EQ(16, FPDFPage_CountObjects(page));
1421 for (int i = 0; i < 16; i++) {
1422 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i);
1423 ASSERT_TRUE(page_object);
1424 CPDF_PageObject* cpdf_page_object =
1425 CPDFPageObjectFromFPDFPageObject(page_object);
1426 if (i < 15)
1427 EXPECT_EQ(0, cpdf_page_object->GetContentStream()) << i;
1428 else
1429 EXPECT_EQ(1, cpdf_page_object->GetContentStream()) << i;
1430 }
1431
Lei Zhang0c03d632020-07-30 17:05:36 +00001432#if defined(OS_APPLE)
Lei Zhang0f6b4342020-02-25 20:00:39 +00001433 const char kStream1RemovedMD5[] = "0e8856ca9abc7049412e64f9230c7c43";
Lei Zhange67bcc72019-04-30 18:55:58 +00001434#elif defined(OS_WIN)
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001435 const char kStream1RemovedMD5[] = "b4140f203523e38793283a5943d8075b";
1436#else
1437 const char kStream1RemovedMD5[] = "e86a3efc160ede6cfcb1f59bcacf1105";
1438#endif
1439 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001440 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001441 CompareBitmap(page_bitmap.get(), 200, 200, kStream1RemovedMD5);
1442 }
1443
1444 // Save the file
1445 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1446 UnloadPage(page);
1447
1448 // Re-open the file and check the page object count is still 16, and that
1449 // content stream 1 was removed.
Lei Zhang0b494052019-01-31 21:41:15 +00001450 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001451 FPDF_PAGE saved_page = LoadSavedPage(0);
1452
1453 // Content stream 0: page objects 0-14.
1454 // Content stream 1: page object 15.
1455 EXPECT_EQ(16, FPDFPage_CountObjects(saved_page));
1456 for (int i = 0; i < 16; i++) {
1457 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(saved_page, i);
1458 ASSERT_TRUE(page_object);
1459 CPDF_PageObject* cpdf_page_object =
1460 CPDFPageObjectFromFPDFPageObject(page_object);
1461 if (i < 15)
1462 EXPECT_EQ(0, cpdf_page_object->GetContentStream()) << i;
1463 else
1464 EXPECT_EQ(1, cpdf_page_object->GetContentStream()) << i;
1465 }
1466
1467 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001468 ScopedFPDFBitmap page_bitmap = RenderPage(saved_page);
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001469 CompareBitmap(page_bitmap.get(), 200, 200, kStream1RemovedMD5);
1470 }
1471
1472 CloseSavedPage(saved_page);
1473 CloseSavedDocument();
1474}
1475
Lei Zhangab41f252018-12-23 03:10:50 +00001476TEST_F(FPDFEditEmbedderTest, RemoveAllFromSingleStream) {
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001477 // Load document with a single stream.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001478 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001479 FPDF_PAGE page = LoadPage(0);
1480 ASSERT_TRUE(page);
1481
1482 // Content stream 0: page objects 0-1.
1483 ASSERT_EQ(2, FPDFPage_CountObjects(page));
1484
1485 // Loop backwards because objects will being removed, which shifts the indexes
1486 // after the removed position.
1487 for (int i = 1; i >= 0; i--) {
1488 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i);
1489 ASSERT_TRUE(page_object);
1490 CPDF_PageObject* cpdf_page_object =
1491 CPDFPageObjectFromFPDFPageObject(page_object);
1492 ASSERT_EQ(0, cpdf_page_object->GetContentStream());
1493 ASSERT_TRUE(FPDFPage_RemoveObject(page, page_object));
1494 FPDFPageObj_Destroy(page_object);
1495 }
1496
1497 // No more objects in the stream
1498 ASSERT_EQ(0, FPDFPage_CountObjects(page));
1499
1500 // Generate contents should remove the empty stream and update the page
1501 // objects' contents stream indexes.
1502 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1503
1504 ASSERT_EQ(0, FPDFPage_CountObjects(page));
1505
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001506 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001507 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingst3b6136a2020-06-09 00:39:33 +00001508 CompareBitmap(page_bitmap.get(), 200, 200, kAllRemovedChecksum);
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001509 }
1510
1511 // Save the file
1512 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1513 UnloadPage(page);
1514
1515 // Re-open the file and check the page object count is still 0.
Lei Zhang0b494052019-01-31 21:41:15 +00001516 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001517 FPDF_PAGE saved_page = LoadSavedPage(0);
1518
1519 EXPECT_EQ(0, FPDFPage_CountObjects(saved_page));
1520 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001521 ScopedFPDFBitmap page_bitmap = RenderPage(saved_page);
Hui Yingst3b6136a2020-06-09 00:39:33 +00001522 CompareBitmap(page_bitmap.get(), 200, 200, kAllRemovedChecksum);
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001523 }
1524
1525 CloseSavedPage(saved_page);
1526 CloseSavedDocument();
1527}
1528
Hui Yingst08c40712020-04-29 01:37:35 +00001529TEST_F(FPDFEditEmbedderTest, RemoveFirstFromSingleStream) {
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001530 // Load document with a single stream.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001531 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001532 FPDF_PAGE page = LoadPage(0);
1533 ASSERT_TRUE(page);
1534
1535 // Content stream 0: page objects 0-1.
1536 ASSERT_EQ(2, FPDFPage_CountObjects(page));
1537
1538 // Remove first object.
1539 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0);
1540 ASSERT_TRUE(page_object);
1541 CPDF_PageObject* cpdf_page_object =
1542 CPDFPageObjectFromFPDFPageObject(page_object);
1543 ASSERT_EQ(0, cpdf_page_object->GetContentStream());
1544 ASSERT_TRUE(FPDFPage_RemoveObject(page, page_object));
1545 FPDFPageObj_Destroy(page_object);
1546
1547 // One object left in the stream.
1548 ASSERT_EQ(1, FPDFPage_CountObjects(page));
1549 page_object = FPDFPage_GetObject(page, 0);
1550 ASSERT_TRUE(page_object);
1551 cpdf_page_object = CPDFPageObjectFromFPDFPageObject(page_object);
1552 ASSERT_EQ(0, cpdf_page_object->GetContentStream());
1553
1554 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1555
1556 // Still one object left in the stream.
1557 ASSERT_EQ(1, FPDFPage_CountObjects(page));
1558 page_object = FPDFPage_GetObject(page, 0);
1559 ASSERT_TRUE(page_object);
1560 cpdf_page_object = CPDFPageObjectFromFPDFPageObject(page_object);
1561 ASSERT_EQ(0, cpdf_page_object->GetContentStream());
1562
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001563 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001564 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00001565 CompareBitmap(page_bitmap.get(), 200, 200, kFirstRemovedChecksum);
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001566 }
1567
1568 // Save the file
1569 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1570 UnloadPage(page);
1571
1572 // Re-open the file and check the page object count is still 0.
Lei Zhang0b494052019-01-31 21:41:15 +00001573 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001574 FPDF_PAGE saved_page = LoadSavedPage(0);
1575
1576 ASSERT_EQ(1, FPDFPage_CountObjects(saved_page));
1577 page_object = FPDFPage_GetObject(saved_page, 0);
1578 ASSERT_TRUE(page_object);
1579 cpdf_page_object = CPDFPageObjectFromFPDFPageObject(page_object);
1580 ASSERT_EQ(0, cpdf_page_object->GetContentStream());
1581 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001582 ScopedFPDFBitmap page_bitmap = RenderPage(saved_page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00001583 CompareBitmap(page_bitmap.get(), 200, 200, kFirstRemovedChecksum);
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001584 }
1585
1586 CloseSavedPage(saved_page);
1587 CloseSavedDocument();
1588}
1589
Hui Yingst08c40712020-04-29 01:37:35 +00001590TEST_F(FPDFEditEmbedderTest, RemoveLastFromSingleStream) {
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001591 // Load document with a single stream.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001592 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001593 FPDF_PAGE page = LoadPage(0);
1594 ASSERT_TRUE(page);
1595
1596 // Content stream 0: page objects 0-1.
1597 ASSERT_EQ(2, FPDFPage_CountObjects(page));
1598
1599 // Remove last object
1600 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 1);
1601 ASSERT_TRUE(page_object);
1602 CPDF_PageObject* cpdf_page_object =
1603 CPDFPageObjectFromFPDFPageObject(page_object);
1604 ASSERT_EQ(0, cpdf_page_object->GetContentStream());
1605 ASSERT_TRUE(FPDFPage_RemoveObject(page, page_object));
1606 FPDFPageObj_Destroy(page_object);
1607
1608 // One object left in the stream.
1609 ASSERT_EQ(1, FPDFPage_CountObjects(page));
1610 page_object = FPDFPage_GetObject(page, 0);
1611 ASSERT_TRUE(page_object);
1612 cpdf_page_object = CPDFPageObjectFromFPDFPageObject(page_object);
1613 ASSERT_EQ(0, cpdf_page_object->GetContentStream());
1614
1615 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1616
1617 // Still one object left in the stream.
1618 ASSERT_EQ(1, FPDFPage_CountObjects(page));
1619 page_object = FPDFPage_GetObject(page, 0);
1620 ASSERT_TRUE(page_object);
1621 cpdf_page_object = CPDFPageObjectFromFPDFPageObject(page_object);
1622 ASSERT_EQ(0, cpdf_page_object->GetContentStream());
1623
Hui Yingstb4baceb2020-04-28 23:46:10 +00001624 using pdfium::kHelloWorldRemovedChecksum;
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001625 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001626 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00001627 CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldRemovedChecksum);
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001628 }
1629
1630 // Save the file
1631 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1632 UnloadPage(page);
1633
1634 // Re-open the file and check the page object count is still 0.
Lei Zhang0b494052019-01-31 21:41:15 +00001635 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001636 FPDF_PAGE saved_page = LoadSavedPage(0);
1637
1638 ASSERT_EQ(1, FPDFPage_CountObjects(saved_page));
1639 page_object = FPDFPage_GetObject(saved_page, 0);
1640 ASSERT_TRUE(page_object);
1641 cpdf_page_object = CPDFPageObjectFromFPDFPageObject(page_object);
1642 ASSERT_EQ(0, cpdf_page_object->GetContentStream());
1643 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001644 ScopedFPDFBitmap page_bitmap = RenderPage(saved_page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00001645 CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldRemovedChecksum);
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001646 }
1647
1648 CloseSavedPage(saved_page);
1649 CloseSavedDocument();
1650}
1651
Lei Zhangab41f252018-12-23 03:10:50 +00001652TEST_F(FPDFEditEmbedderTest, RemoveAllFromMultipleStreams) {
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001653 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001654 ASSERT_TRUE(OpenDocument("hello_world_split_streams.pdf"));
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001655 FPDF_PAGE page = LoadPage(0);
1656 ASSERT_TRUE(page);
1657
1658 // Content stream 0: page objects 0-1.
1659 // Content stream 1: page object 2.
1660 ASSERT_EQ(3, FPDFPage_CountObjects(page));
1661
1662 // Loop backwards because objects will being removed, which shifts the indexes
1663 // after the removed position.
1664 for (int i = 2; i >= 0; i--) {
1665 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i);
1666 ASSERT_TRUE(page_object);
1667 ASSERT_TRUE(FPDFPage_RemoveObject(page, page_object));
1668 FPDFPageObj_Destroy(page_object);
1669 }
1670
1671 // No more objects in the page.
1672 ASSERT_EQ(0, FPDFPage_CountObjects(page));
1673
1674 // Generate contents should remove the empty streams and update the page
1675 // objects' contents stream indexes.
1676 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1677
1678 ASSERT_EQ(0, FPDFPage_CountObjects(page));
1679
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001680 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001681 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingst3b6136a2020-06-09 00:39:33 +00001682 CompareBitmap(page_bitmap.get(), 200, 200, kAllRemovedChecksum);
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001683 }
1684
1685 // Save the file
1686 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1687 UnloadPage(page);
1688
1689 // Re-open the file and check the page object count is still 0.
Lei Zhang0b494052019-01-31 21:41:15 +00001690 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001691 FPDF_PAGE saved_page = LoadSavedPage(0);
1692
1693 EXPECT_EQ(0, FPDFPage_CountObjects(saved_page));
1694 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001695 ScopedFPDFBitmap page_bitmap = RenderPage(saved_page);
Hui Yingst3b6136a2020-06-09 00:39:33 +00001696 CompareBitmap(page_bitmap.get(), 200, 200, kAllRemovedChecksum);
Henrique Nakashima0dcf1f42018-06-21 18:51:15 +00001697 }
1698
1699 CloseSavedPage(saved_page);
1700 CloseSavedDocument();
1701}
1702
Lei Zhangab41f252018-12-23 03:10:50 +00001703TEST_F(FPDFEditEmbedderTest, InsertPageObjectAndSave) {
Henrique Nakashimac49e62e2018-04-16 20:58:47 +00001704 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001705 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
Henrique Nakashimac49e62e2018-04-16 20:58:47 +00001706 FPDF_PAGE page = LoadPage(0);
1707 ASSERT_TRUE(page);
1708
1709 // Add a red rectangle.
1710 ASSERT_EQ(2, FPDFPage_CountObjects(page));
1711 FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(20, 100, 50, 50);
Lei Zhang3475b482019-05-13 18:30:57 +00001712 EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 0, 0, 255));
Henrique Nakashimac49e62e2018-04-16 20:58:47 +00001713 EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0));
1714 FPDFPage_InsertObject(page, red_rect);
1715
1716 // Verify the red rectangle was added.
1717 ASSERT_EQ(3, FPDFPage_CountObjects(page));
1718
1719 // Save the file
1720 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1721 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1722 UnloadPage(page);
1723
1724 // Re-open the file and check the page object count is still 3.
Lei Zhang0b494052019-01-31 21:41:15 +00001725 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashimac49e62e2018-04-16 20:58:47 +00001726 FPDF_PAGE saved_page = LoadSavedPage(0);
1727 EXPECT_EQ(3, FPDFPage_CountObjects(saved_page));
1728 CloseSavedPage(saved_page);
1729 CloseSavedDocument();
1730}
1731
Lei Zhangab41f252018-12-23 03:10:50 +00001732TEST_F(FPDFEditEmbedderTest, InsertPageObjectEditAndSave) {
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001733 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001734 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001735 FPDF_PAGE page = LoadPage(0);
1736 ASSERT_TRUE(page);
1737
1738 // Add a red rectangle.
1739 ASSERT_EQ(2, FPDFPage_CountObjects(page));
1740 FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(20, 100, 50, 50);
Lei Zhang3475b482019-05-13 18:30:57 +00001741 EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 100, 100, 255));
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001742 EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0));
1743 FPDFPage_InsertObject(page, red_rect);
1744
1745 // Verify the red rectangle was added.
1746 ASSERT_EQ(3, FPDFPage_CountObjects(page));
1747
1748 // Generate content but change it again
1749 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Lei Zhang3475b482019-05-13 18:30:57 +00001750 EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 0, 0, 255));
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001751
1752 // Save the file
1753 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1754 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1755 UnloadPage(page);
1756
1757 // Re-open the file and check the page object count is still 3.
Lei Zhang0b494052019-01-31 21:41:15 +00001758 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashima27cf78d2018-06-14 16:22:30 +00001759 FPDF_PAGE saved_page = LoadSavedPage(0);
1760 EXPECT_EQ(3, FPDFPage_CountObjects(saved_page));
1761 CloseSavedPage(saved_page);
1762 CloseSavedDocument();
1763}
1764
Hui Yingst8b5dfb22020-07-21 17:08:30 +00001765TEST_F(FPDFEditEmbedderTest, InsertAndRemoveLargeFile) {
Henrique Nakashima657a1aa2018-09-12 16:19:22 +00001766 const int kOriginalObjectCount = 600;
1767
1768 // Load document with many objects.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001769 ASSERT_TRUE(OpenDocument("many_rectangles.pdf"));
Henrique Nakashima657a1aa2018-09-12 16:19:22 +00001770 FPDF_PAGE page = LoadPage(0);
1771 ASSERT_TRUE(page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00001772
1773 using pdfium::kManyRectanglesChecksum;
Henrique Nakashima657a1aa2018-09-12 16:19:22 +00001774 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001775 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00001776 CompareBitmap(page_bitmap.get(), 200, 300, kManyRectanglesChecksum);
Henrique Nakashima657a1aa2018-09-12 16:19:22 +00001777 }
1778
1779 // Add a black rectangle.
1780 ASSERT_EQ(kOriginalObjectCount, FPDFPage_CountObjects(page));
1781 FPDF_PAGEOBJECT black_rect = FPDFPageObj_CreateNewRect(20, 100, 50, 50);
Lei Zhang3475b482019-05-13 18:30:57 +00001782 EXPECT_TRUE(FPDFPageObj_SetFillColor(black_rect, 0, 0, 0, 255));
Henrique Nakashima657a1aa2018-09-12 16:19:22 +00001783 EXPECT_TRUE(FPDFPath_SetDrawMode(black_rect, FPDF_FILLMODE_ALTERNATE, 0));
1784 FPDFPage_InsertObject(page, black_rect);
1785
1786 // Verify the black rectangle was added.
1787 ASSERT_EQ(kOriginalObjectCount + 1, FPDFPage_CountObjects(page));
Hui Yingst8b5dfb22020-07-21 17:08:30 +00001788#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
1789 const char kPlusRectangleMD5[] = "0d3715fcfb9bd0dd25dcce60800bff47";
1790#else
Henrique Nakashima657a1aa2018-09-12 16:19:22 +00001791 const char kPlusRectangleMD5[] = "6b9396ab570754b32b04ca629e902f77";
Hui Yingst8b5dfb22020-07-21 17:08:30 +00001792#endif
Henrique Nakashima657a1aa2018-09-12 16:19:22 +00001793 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001794 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Henrique Nakashima657a1aa2018-09-12 16:19:22 +00001795 CompareBitmap(page_bitmap.get(), 200, 300, kPlusRectangleMD5);
1796 }
1797
1798 // Save the file.
1799 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1800 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1801 UnloadPage(page);
1802
1803 // Re-open the file and check the rectangle added is still there.
Lei Zhang0b494052019-01-31 21:41:15 +00001804 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashima657a1aa2018-09-12 16:19:22 +00001805 FPDF_PAGE saved_page = LoadSavedPage(0);
1806 EXPECT_EQ(kOriginalObjectCount + 1, FPDFPage_CountObjects(saved_page));
1807 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001808 ScopedFPDFBitmap page_bitmap = RenderPage(saved_page);
Henrique Nakashima657a1aa2018-09-12 16:19:22 +00001809 CompareBitmap(page_bitmap.get(), 200, 300, kPlusRectangleMD5);
1810 }
1811
1812 // Remove the added rectangle.
1813 FPDF_PAGEOBJECT added_object =
1814 FPDFPage_GetObject(saved_page, kOriginalObjectCount);
1815 EXPECT_TRUE(FPDFPage_RemoveObject(saved_page, added_object));
1816 FPDFPageObj_Destroy(added_object);
1817 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001818 ScopedFPDFBitmap page_bitmap = RenderPage(saved_page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00001819 CompareBitmap(page_bitmap.get(), 200, 300, kManyRectanglesChecksum);
Henrique Nakashima657a1aa2018-09-12 16:19:22 +00001820 }
1821 EXPECT_EQ(kOriginalObjectCount, FPDFPage_CountObjects(saved_page));
1822
1823 // Save the file again.
1824 EXPECT_TRUE(FPDFPage_GenerateContent(saved_page));
1825 EXPECT_TRUE(FPDF_SaveAsCopy(saved_document_, this, 0));
1826
1827 CloseSavedPage(saved_page);
1828 CloseSavedDocument();
1829
1830 // Re-open the file (again) and check the black rectangle was removed and the
1831 // rest is intact.
Lei Zhang0b494052019-01-31 21:41:15 +00001832 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashima657a1aa2018-09-12 16:19:22 +00001833 saved_page = LoadSavedPage(0);
1834 EXPECT_EQ(kOriginalObjectCount, FPDFPage_CountObjects(saved_page));
1835 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001836 ScopedFPDFBitmap page_bitmap = RenderPage(saved_page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00001837 CompareBitmap(page_bitmap.get(), 200, 300, kManyRectanglesChecksum);
Henrique Nakashima657a1aa2018-09-12 16:19:22 +00001838 }
1839
1840 CloseSavedPage(saved_page);
1841 CloseSavedDocument();
1842}
1843
Lei Zhangab41f252018-12-23 03:10:50 +00001844TEST_F(FPDFEditEmbedderTest, AddAndRemovePaths) {
Henrique Nakashima35841fa2018-03-15 15:25:16 +00001845 // Start with a blank page.
1846 FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792);
1847 ASSERT_TRUE(page);
1848
1849 // Render the blank page and verify it's a blank bitmap.
Hui Yingstb4baceb2020-04-28 23:46:10 +00001850 using pdfium::kBlankPage612By792Checksum;
Henrique Nakashima35841fa2018-03-15 15:25:16 +00001851 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001852 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00001853 CompareBitmap(page_bitmap.get(), 612, 792, kBlankPage612By792Checksum);
Henrique Nakashima35841fa2018-03-15 15:25:16 +00001854 }
1855 ASSERT_EQ(0, FPDFPage_CountObjects(page));
1856
1857 // Add a red rectangle.
1858 FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(10, 10, 20, 20);
1859 ASSERT_TRUE(red_rect);
Lei Zhang3475b482019-05-13 18:30:57 +00001860 EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 0, 0, 255));
Henrique Nakashima35841fa2018-03-15 15:25:16 +00001861 EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0));
1862 FPDFPage_InsertObject(page, red_rect);
Henrique Nakashima35841fa2018-03-15 15:25:16 +00001863 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001864 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00001865 CompareBitmap(page_bitmap.get(), 612, 792, kRedRectangleChecksum);
Henrique Nakashima35841fa2018-03-15 15:25:16 +00001866 }
1867 EXPECT_EQ(1, FPDFPage_CountObjects(page));
1868
1869 // Remove rectangle and verify it does not render anymore and the bitmap is
1870 // back to a blank one.
1871 EXPECT_TRUE(FPDFPage_RemoveObject(page, red_rect));
1872 {
Lei Zhang30ff2532019-01-31 21:37:55 +00001873 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00001874 CompareBitmap(page_bitmap.get(), 612, 792, kBlankPage612By792Checksum);
Henrique Nakashima35841fa2018-03-15 15:25:16 +00001875 }
1876 EXPECT_EQ(0, FPDFPage_CountObjects(page));
1877
1878 // Trying to remove an object not in the page should return false.
1879 EXPECT_FALSE(FPDFPage_RemoveObject(page, red_rect));
1880
1881 FPDF_ClosePage(page);
1882 FPDFPageObj_Destroy(red_rect);
1883}
1884
Lei Zhangab41f252018-12-23 03:10:50 +00001885TEST_F(FPDFEditEmbedderTest, PathsPoints) {
Miklos Vajna12abfd02017-09-15 07:49:03 +02001886 CreateNewDocument();
1887 FPDF_PAGEOBJECT img = FPDFPageObj_NewImageObj(document_);
1888 // This should fail gracefully, even if img is not a path.
Miklos Vajna0150a542017-09-21 21:46:56 +02001889 ASSERT_EQ(-1, FPDFPath_CountSegments(img));
Miklos Vajna12abfd02017-09-15 07:49:03 +02001890
1891 // This should fail gracefully, even if path is NULL.
Miklos Vajna0150a542017-09-21 21:46:56 +02001892 ASSERT_EQ(-1, FPDFPath_CountSegments(nullptr));
Miklos Vajna12abfd02017-09-15 07:49:03 +02001893
Miklos Vajna36eed872017-09-20 22:52:43 +02001894 // FPDFPath_GetPathSegment() with a non-path.
1895 ASSERT_EQ(nullptr, FPDFPath_GetPathSegment(img, 0));
1896 // FPDFPath_GetPathSegment() with a NULL path.
1897 ASSERT_EQ(nullptr, FPDFPath_GetPathSegment(nullptr, 0));
1898 float x;
1899 float y;
1900 // FPDFPathSegment_GetPoint() with a NULL segment.
1901 EXPECT_FALSE(FPDFPathSegment_GetPoint(nullptr, &x, &y));
1902
1903 // FPDFPathSegment_GetType() with a NULL segment.
1904 ASSERT_EQ(FPDF_SEGMENT_UNKNOWN, FPDFPathSegment_GetType(nullptr));
1905
1906 // FPDFPathSegment_GetClose() with a NULL segment.
1907 EXPECT_FALSE(FPDFPathSegment_GetClose(nullptr));
1908
Miklos Vajna12abfd02017-09-15 07:49:03 +02001909 FPDFPageObj_Destroy(img);
1910}
1911
Hui Yingste95d5772020-08-24 23:58:26 +00001912TEST_F(FPDFEditEmbedderTest, PathOnTopOfText) {
Nicolas Pena0fc185e2017-02-08 12:13:20 -05001913 // Load document with some text
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001914 ASSERT_TRUE(OpenDocument("hello_world.pdf"));
Nicolas Pena0fc185e2017-02-08 12:13:20 -05001915 FPDF_PAGE page = LoadPage(0);
Lei Zhang107fa7b2018-02-09 21:48:15 +00001916 ASSERT_TRUE(page);
Nicolas Pena0fc185e2017-02-08 12:13:20 -05001917
1918 // Add an opaque rectangle on top of some of the text.
1919 FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(20, 100, 50, 50);
Lei Zhang3475b482019-05-13 18:30:57 +00001920 EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 0, 0, 255));
Nicolas Pena0fc185e2017-02-08 12:13:20 -05001921 EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0));
1922 FPDFPage_InsertObject(page, red_rect);
Nicolas Pena0fc185e2017-02-08 12:13:20 -05001923
1924 // Add a transparent triangle on top of other part of the text.
1925 FPDF_PAGEOBJECT black_path = FPDFPageObj_CreateNewPath(20, 50);
Lei Zhang3475b482019-05-13 18:30:57 +00001926 EXPECT_TRUE(FPDFPageObj_SetFillColor(black_path, 0, 0, 0, 100));
Nicolas Pena0fc185e2017-02-08 12:13:20 -05001927 EXPECT_TRUE(FPDFPath_SetDrawMode(black_path, FPDF_FILLMODE_ALTERNATE, 0));
1928 EXPECT_TRUE(FPDFPath_LineTo(black_path, 30, 80));
1929 EXPECT_TRUE(FPDFPath_LineTo(black_path, 40, 10));
1930 EXPECT_TRUE(FPDFPath_Close(black_path));
1931 FPDFPage_InsertObject(page, black_path);
Nicolas Pena0fc185e2017-02-08 12:13:20 -05001932
Hui Yingste95d5772020-08-24 23:58:26 +00001933 // Render and check the result.
Tom Sepeze08d2b12018-04-25 18:49:32 +00001934 ScopedFPDFBitmap bitmap = RenderLoadedPage(page);
Hui Yingste95d5772020-08-24 23:58:26 +00001935#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
1936#if defined(OS_WIN)
1937 const char kChecksum[] = "e755a955696373e39dbebb5cb96e1338";
Nicolas Pena0fc185e2017-02-08 12:13:20 -05001938#else
Hui Yingste95d5772020-08-24 23:58:26 +00001939 const char kChecksum[] = "d082f9756c86bb47e1abbc2b1df7138a";
1940#endif // defined(OS_WIN)
1941#else
1942#if defined(OS_WIN)
1943 const char kChecksum[] = "74dd9c393b8b2578d2b7feb032b7daad";
1944#elif defined(OS_APPLE)
1945 const char kChecksum[] = "e55bcd1facb7243dc6e16dd5f912265b";
1946#else
1947 const char kChecksum[] = "aa71b09b93b55f467f1290e5111babee";
Nicolas Pena5bcd9a32017-03-22 11:04:35 -04001948#endif
Hui Yingste95d5772020-08-24 23:58:26 +00001949#endif // defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
1950 CompareBitmap(bitmap.get(), 200, 200, kChecksum);
Nicolas Pena0fc185e2017-02-08 12:13:20 -05001951 UnloadPage(page);
1952}
Nicolas Pena2eb1a702017-02-09 18:17:33 -05001953
Hui Yingst1f73c4f2020-07-29 00:46:25 +00001954TEST_F(FPDFEditEmbedderTest, EditOverExistingContent) {
wileyryae858aa42017-05-31 14:49:05 -05001955 // Load document with existing content
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00001956 ASSERT_TRUE(OpenDocument("bug_717.pdf"));
wileyryae858aa42017-05-31 14:49:05 -05001957 FPDF_PAGE page = LoadPage(0);
Lei Zhang107fa7b2018-02-09 21:48:15 +00001958 ASSERT_TRUE(page);
wileyryae858aa42017-05-31 14:49:05 -05001959
1960 // Add a transparent rectangle on top of the existing content
1961 FPDF_PAGEOBJECT red_rect2 = FPDFPageObj_CreateNewRect(90, 700, 25, 50);
Lei Zhang3475b482019-05-13 18:30:57 +00001962 EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect2, 255, 0, 0, 100));
wileyryae858aa42017-05-31 14:49:05 -05001963 EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect2, FPDF_FILLMODE_ALTERNATE, 0));
1964 FPDFPage_InsertObject(page, red_rect2);
1965
1966 // Add an opaque rectangle on top of the existing content
1967 FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(115, 700, 25, 50);
Lei Zhang3475b482019-05-13 18:30:57 +00001968 EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 0, 0, 255));
wileyryae858aa42017-05-31 14:49:05 -05001969 EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0));
1970 FPDFPage_InsertObject(page, red_rect);
1971
Hui Yingst1f73c4f2020-07-29 00:46:25 +00001972#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
1973 const char kOriginalChecksum[] = "1e82fbdd21490cee9d3479fe6125af67";
1974#else
1975 const char kOriginalChecksum[] = "ad04e5bd0f471a9a564fb034bd0fb073";
1976#endif
Tom Sepeze08d2b12018-04-25 18:49:32 +00001977 ScopedFPDFBitmap bitmap = RenderLoadedPage(page);
Hui Yingst1f73c4f2020-07-29 00:46:25 +00001978 CompareBitmap(bitmap.get(), 612, 792, kOriginalChecksum);
wileyryae858aa42017-05-31 14:49:05 -05001979 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1980
1981 // Now save the result, closing the page and document
1982 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
Nicolas Pena3ff54002017-07-05 11:55:35 -04001983 UnloadPage(page);
wileyryae858aa42017-05-31 14:49:05 -05001984
Lei Zhang0b494052019-01-31 21:41:15 +00001985 ASSERT_TRUE(OpenSavedDocument());
Lei Zhang107fa7b2018-02-09 21:48:15 +00001986 FPDF_PAGE saved_page = LoadSavedPage(0);
Hui Yingst1f73c4f2020-07-29 00:46:25 +00001987 VerifySavedRendering(saved_page, 612, 792, kOriginalChecksum);
wileyryae858aa42017-05-31 14:49:05 -05001988
1989 ClearString();
1990 // Add another opaque rectangle on top of the existing content
1991 FPDF_PAGEOBJECT green_rect = FPDFPageObj_CreateNewRect(150, 700, 25, 50);
Lei Zhang3475b482019-05-13 18:30:57 +00001992 EXPECT_TRUE(FPDFPageObj_SetFillColor(green_rect, 0, 255, 0, 255));
wileyryae858aa42017-05-31 14:49:05 -05001993 EXPECT_TRUE(FPDFPath_SetDrawMode(green_rect, FPDF_FILLMODE_ALTERNATE, 0));
Lei Zhang107fa7b2018-02-09 21:48:15 +00001994 FPDFPage_InsertObject(saved_page, green_rect);
wileyryae858aa42017-05-31 14:49:05 -05001995
1996 // Add another transparent rectangle on top of existing content
1997 FPDF_PAGEOBJECT green_rect2 = FPDFPageObj_CreateNewRect(175, 700, 25, 50);
Lei Zhang3475b482019-05-13 18:30:57 +00001998 EXPECT_TRUE(FPDFPageObj_SetFillColor(green_rect2, 0, 255, 0, 100));
wileyryae858aa42017-05-31 14:49:05 -05001999 EXPECT_TRUE(FPDFPath_SetDrawMode(green_rect2, FPDF_FILLMODE_ALTERNATE, 0));
Lei Zhang107fa7b2018-02-09 21:48:15 +00002000 FPDFPage_InsertObject(saved_page, green_rect2);
Hui Yingst1f73c4f2020-07-29 00:46:25 +00002001#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
2002 const char kLastChecksum[] = "8705d023e5fec3499d1e30cf2bcc5dc1";
2003#else
2004 const char kLastChecksum[] = "4b5b00f824620f8c9b8801ebb98e1cdd";
2005#endif
Lei Zhangc113c7a2018-02-12 14:58:44 +00002006 {
Tom Sepeze08d2b12018-04-25 18:49:32 +00002007 ScopedFPDFBitmap new_bitmap = RenderSavedPage(saved_page);
Hui Yingst1f73c4f2020-07-29 00:46:25 +00002008 CompareBitmap(new_bitmap.get(), 612, 792, kLastChecksum);
Lei Zhangc113c7a2018-02-12 14:58:44 +00002009 }
Lei Zhang107fa7b2018-02-09 21:48:15 +00002010 EXPECT_TRUE(FPDFPage_GenerateContent(saved_page));
wileyryae858aa42017-05-31 14:49:05 -05002011
2012 // Now save the result, closing the page and document
Lei Zhang0729be22018-02-05 21:13:51 +00002013 EXPECT_TRUE(FPDF_SaveAsCopy(saved_document_, this, 0));
Dan Sinclair04e4dc82017-10-18 12:17:14 -04002014
Lei Zhang107fa7b2018-02-09 21:48:15 +00002015 CloseSavedPage(saved_page);
Dan Sinclair04e4dc82017-10-18 12:17:14 -04002016 CloseSavedDocument();
wileyryae858aa42017-05-31 14:49:05 -05002017
2018 // Render the saved result
Hui Yingst1f73c4f2020-07-29 00:46:25 +00002019 VerifySavedDocument(612, 792, kLastChecksum);
wileyryae858aa42017-05-31 14:49:05 -05002020}
2021
Lei Zhang03e5e682019-09-16 19:45:55 +00002022// TODO(crbug.com/pdfium/11): Fix this test and enable.
2023#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
2024#define MAYBE_AddStrokedPaths DISABLED_AddStrokedPaths
2025#else
2026#define MAYBE_AddStrokedPaths AddStrokedPaths
2027#endif
2028TEST_F(FPDFEditEmbedderTest, MAYBE_AddStrokedPaths) {
Nicolas Pena2eb1a702017-02-09 18:17:33 -05002029 // Start with a blank page
Nicolas Penad03ca422017-03-06 13:54:33 -05002030 FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792);
Nicolas Pena2eb1a702017-02-09 18:17:33 -05002031
2032 // Add a large stroked rectangle (fill color should not affect it).
2033 FPDF_PAGEOBJECT rect = FPDFPageObj_CreateNewRect(20, 20, 200, 400);
Lei Zhang3475b482019-05-13 18:30:57 +00002034 EXPECT_TRUE(FPDFPageObj_SetFillColor(rect, 255, 0, 0, 255));
2035 EXPECT_TRUE(FPDFPageObj_SetStrokeColor(rect, 0, 255, 0, 255));
2036 EXPECT_TRUE(FPDFPageObj_SetStrokeWidth(rect, 15.0f));
Miklos Vajna366df7f2018-05-22 14:27:29 +00002037
2038 float width = 0;
2039 EXPECT_TRUE(FPDFPageObj_GetStrokeWidth(rect, &width));
2040 EXPECT_EQ(15.0f, width);
2041
Nicolas Pena2eb1a702017-02-09 18:17:33 -05002042 EXPECT_TRUE(FPDFPath_SetDrawMode(rect, 0, 1));
2043 FPDFPage_InsertObject(page, rect);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002044 {
Lei Zhang30ff2532019-01-31 21:37:55 +00002045 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002046 CompareBitmap(page_bitmap.get(), 612, 792,
2047 "64bd31f862a89e0a9e505a5af6efd506");
2048 }
Nicolas Pena2eb1a702017-02-09 18:17:33 -05002049
2050 // Add crossed-checkmark
2051 FPDF_PAGEOBJECT check = FPDFPageObj_CreateNewPath(300, 500);
2052 EXPECT_TRUE(FPDFPath_LineTo(check, 400, 400));
2053 EXPECT_TRUE(FPDFPath_LineTo(check, 600, 600));
2054 EXPECT_TRUE(FPDFPath_MoveTo(check, 400, 600));
2055 EXPECT_TRUE(FPDFPath_LineTo(check, 600, 400));
Lei Zhang3475b482019-05-13 18:30:57 +00002056 EXPECT_TRUE(FPDFPageObj_SetStrokeColor(check, 128, 128, 128, 180));
2057 EXPECT_TRUE(FPDFPageObj_SetStrokeWidth(check, 8.35f));
Nicolas Pena2eb1a702017-02-09 18:17:33 -05002058 EXPECT_TRUE(FPDFPath_SetDrawMode(check, 0, 1));
2059 FPDFPage_InsertObject(page, check);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002060 {
Lei Zhang30ff2532019-01-31 21:37:55 +00002061 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002062 CompareBitmap(page_bitmap.get(), 612, 792,
2063 "4b6f3b9d25c4e194821217d5016c3724");
2064 }
Nicolas Pena2eb1a702017-02-09 18:17:33 -05002065
2066 // Add stroked and filled oval-ish path.
2067 FPDF_PAGEOBJECT path = FPDFPageObj_CreateNewPath(250, 100);
2068 EXPECT_TRUE(FPDFPath_BezierTo(path, 180, 166, 180, 233, 250, 300));
2069 EXPECT_TRUE(FPDFPath_LineTo(path, 255, 305));
2070 EXPECT_TRUE(FPDFPath_BezierTo(path, 325, 233, 325, 166, 255, 105));
2071 EXPECT_TRUE(FPDFPath_Close(path));
Lei Zhang3475b482019-05-13 18:30:57 +00002072 EXPECT_TRUE(FPDFPageObj_SetFillColor(path, 200, 128, 128, 100));
2073 EXPECT_TRUE(FPDFPageObj_SetStrokeColor(path, 128, 200, 128, 150));
2074 EXPECT_TRUE(FPDFPageObj_SetStrokeWidth(path, 10.5f));
Nicolas Pena2eb1a702017-02-09 18:17:33 -05002075 EXPECT_TRUE(FPDFPath_SetDrawMode(path, FPDF_FILLMODE_ALTERNATE, 1));
2076 FPDFPage_InsertObject(page, path);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002077 {
Lei Zhang30ff2532019-01-31 21:37:55 +00002078 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002079 CompareBitmap(page_bitmap.get(), 612, 792,
2080 "ff3e6a22326754944cc6e56609acd73b");
2081 }
Nicolas Pena2eb1a702017-02-09 18:17:33 -05002082 FPDF_ClosePage(page);
Nicolas Pena2eb1a702017-02-09 18:17:33 -05002083}
Nicolas Pena49058402017-02-14 18:26:20 -05002084
Nicolas Pena4c48b102018-06-13 18:23:46 +00002085// Tests adding text from standard font using FPDFPageObj_NewTextObj.
Lei Zhang03e5e682019-09-16 19:45:55 +00002086// TODO(crbug.com/pdfium/11): Fix this test and enable.
2087#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
2088#define MAYBE_AddStandardFontText DISABLED_AddStandardFontText
2089#else
2090#define MAYBE_AddStandardFontText AddStandardFontText
2091#endif
2092TEST_F(FPDFEditEmbedderTest, MAYBE_AddStandardFontText) {
Nicolas Pena49058402017-02-14 18:26:20 -05002093 // Start with a blank page
Nicolas Penad03ca422017-03-06 13:54:33 -05002094 FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792);
Nicolas Pena49058402017-02-14 18:26:20 -05002095
2096 // Add some text to the page
Nicolas Penab3161852017-05-02 14:12:50 -04002097 FPDF_PAGEOBJECT text_object1 =
2098 FPDFPageObj_NewTextObj(document(), "Arial", 12.0f);
2099 EXPECT_TRUE(text_object1);
Hui Yingst3b6136a2020-06-09 00:39:33 +00002100 ScopedFPDFWideString text1 = GetFPDFWideString(kBottomText);
Nicolas Penab3161852017-05-02 14:12:50 -04002101 EXPECT_TRUE(FPDFText_SetText(text_object1, text1.get()));
2102 FPDFPageObj_Transform(text_object1, 1, 0, 0, 1, 20, 20);
2103 FPDFPage_InsertObject(page, text_object1);
Lei Zhange039bab2019-03-18 19:57:56 +00002104 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Lei Zhang107fa7b2018-02-09 21:48:15 +00002105 {
Lei Zhang30ff2532019-01-31 21:37:55 +00002106 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingst3b6136a2020-06-09 00:39:33 +00002107 CompareBitmap(page_bitmap.get(), 612, 792, kBottomTextChecksum);
Lei Zhange039bab2019-03-18 19:57:56 +00002108
2109 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
Hui Yingst3b6136a2020-06-09 00:39:33 +00002110 VerifySavedDocument(612, 792, kBottomTextChecksum);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002111 }
Nicolas Pena49058402017-02-14 18:26:20 -05002112
2113 // Try another font
Nicolas Penab3161852017-05-02 14:12:50 -04002114 FPDF_PAGEOBJECT text_object2 =
Nicolas Penad03ca422017-03-06 13:54:33 -05002115 FPDFPageObj_NewTextObj(document(), "TimesNewRomanBold", 15.0f);
Nicolas Penab3161852017-05-02 14:12:50 -04002116 EXPECT_TRUE(text_object2);
Lei Zhangf0f67682019-04-08 17:03:21 +00002117 ScopedFPDFWideString text2 =
Nicolas Penab3161852017-05-02 14:12:50 -04002118 GetFPDFWideString(L"Hi, I'm Bold. Times New Roman Bold.");
2119 EXPECT_TRUE(FPDFText_SetText(text_object2, text2.get()));
2120 FPDFPageObj_Transform(text_object2, 1, 0, 0, 1, 100, 600);
2121 FPDFPage_InsertObject(page, text_object2);
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);
Lei Zhang0c03d632020-07-30 17:05:36 +00002125#if defined(OS_APPLE)
Lei Zhang0f6b4342020-02-25 20:00:39 +00002126 const char md5[] = "26a516d923b0a18fbea0a24e3aca5562";
Lei Zhange67bcc72019-04-30 18:55:58 +00002127#elif defined(OS_WIN)
Lei Zhange039bab2019-03-18 19:57:56 +00002128 const char md5[] = "3755dd35abd4c605755369401ee85b2d";
Nicolas Pena49058402017-02-14 18:26:20 -05002129#else
Lei Zhange039bab2019-03-18 19:57:56 +00002130 const char md5[] = "76fcc7d08aa15445efd2e2ceb7c6cc3b";
Nicolas Pena5bcd9a32017-03-22 11:04:35 -04002131#endif
Lei Zhange039bab2019-03-18 19:57:56 +00002132 CompareBitmap(page_bitmap.get(), 612, 792, md5);
2133
2134 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
2135 VerifySavedDocument(612, 792, md5);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002136 }
Nicolas Pena49058402017-02-14 18:26:20 -05002137
2138 // And some randomly transformed text
Nicolas Penab3161852017-05-02 14:12:50 -04002139 FPDF_PAGEOBJECT text_object3 =
Nicolas Penad03ca422017-03-06 13:54:33 -05002140 FPDFPageObj_NewTextObj(document(), "Courier-Bold", 20.0f);
Nicolas Penab3161852017-05-02 14:12:50 -04002141 EXPECT_TRUE(text_object3);
Lei Zhangf0f67682019-04-08 17:03:21 +00002142 ScopedFPDFWideString text3 = GetFPDFWideString(L"Can you read me? <:)>");
Nicolas Penab3161852017-05-02 14:12:50 -04002143 EXPECT_TRUE(FPDFText_SetText(text_object3, text3.get()));
2144 FPDFPageObj_Transform(text_object3, 1, 1.5, 2, 0.5, 200, 200);
2145 FPDFPage_InsertObject(page, text_object3);
Lei Zhange039bab2019-03-18 19:57:56 +00002146 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Lei Zhang107fa7b2018-02-09 21:48:15 +00002147 {
Lei Zhang30ff2532019-01-31 21:37:55 +00002148 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Lei Zhang0c03d632020-07-30 17:05:36 +00002149#if defined(OS_APPLE)
Lei Zhang0f6b4342020-02-25 20:00:39 +00002150 const char md5[] = "532024c9ded47843313bb64a060118f3";
Lei Zhange67bcc72019-04-30 18:55:58 +00002151#elif defined(OS_WIN)
Lei Zhanga2b70732019-06-25 08:34:22 +00002152 const char md5[] = "5ded49fe157f89627903553771431e3d";
Nicolas Pena49058402017-02-14 18:26:20 -05002153#else
Lei Zhanga2b70732019-06-25 08:34:22 +00002154 const char md5[] = "344534539aa7c5cc78404cfff4bde7fb";
Nicolas Pena5bcd9a32017-03-22 11:04:35 -04002155#endif
Lei Zhange039bab2019-03-18 19:57:56 +00002156 CompareBitmap(page_bitmap.get(), 612, 792, md5);
2157
2158 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
2159 VerifySavedDocument(612, 792, md5);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002160 }
Nicolas Pena49058402017-02-14 18:26:20 -05002161
Lei Zhang8da98232019-12-11 23:29:33 +00002162 FS_MATRIX matrix;
2163 EXPECT_FALSE(FPDFTextObj_GetMatrix(nullptr, &matrix));
2164 EXPECT_TRUE(FPDFTextObj_GetMatrix(text_object3, &matrix));
2165 EXPECT_FLOAT_EQ(1.0f, matrix.a);
2166 EXPECT_FLOAT_EQ(1.5f, matrix.b);
2167 EXPECT_FLOAT_EQ(2.0f, matrix.c);
2168 EXPECT_FLOAT_EQ(0.5f, matrix.d);
2169 EXPECT_FLOAT_EQ(200.0f, matrix.e);
2170 EXPECT_FLOAT_EQ(200.0f, matrix.f);
Miklos Vajnac765d2a2018-06-19 15:45:42 +00002171
Miklos Vajna8625d3b2018-06-26 15:12:48 +00002172 EXPECT_EQ(0, FPDFTextObj_GetFontSize(nullptr));
2173 EXPECT_EQ(20, FPDFTextObj_GetFontSize(text_object3));
2174
Nicolas Pena49058402017-02-14 18:26:20 -05002175 // TODO(npm): Why are there issues with text rotated by 90 degrees?
2176 // TODO(npm): FPDF_SaveAsCopy not giving the desired result after this.
2177 FPDF_ClosePage(page);
Nicolas Pena49058402017-02-14 18:26:20 -05002178}
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002179
Daniel Hosseinian8cb6a652019-12-18 00:50:41 +00002180TEST_F(FPDFEditEmbedderTest, GetTextRenderMode) {
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00002181 ASSERT_TRUE(OpenDocument("text_render_mode.pdf"));
Miklos Vajna1448cc12018-07-03 13:52:33 +00002182 FPDF_PAGE page = LoadPage(0);
2183 ASSERT_TRUE(page);
2184 ASSERT_EQ(2, FPDFPage_CountObjects(page));
2185
Daniel Hosseiniane83d8b42020-01-08 00:26:15 +00002186 EXPECT_EQ(FPDF_TEXTRENDERMODE_UNKNOWN,
Daniel Hosseinian8cb6a652019-12-18 00:50:41 +00002187 FPDFTextObj_GetTextRenderMode(nullptr));
Miklos Vajna1448cc12018-07-03 13:52:33 +00002188
2189 FPDF_PAGEOBJECT fill = FPDFPage_GetObject(page, 0);
Daniel Hosseiniane83d8b42020-01-08 00:26:15 +00002190 EXPECT_EQ(FPDF_TEXTRENDERMODE_FILL, FPDFTextObj_GetTextRenderMode(fill));
Miklos Vajna1448cc12018-07-03 13:52:33 +00002191
2192 FPDF_PAGEOBJECT stroke = FPDFPage_GetObject(page, 1);
Daniel Hosseiniane83d8b42020-01-08 00:26:15 +00002193 EXPECT_EQ(FPDF_TEXTRENDERMODE_STROKE, FPDFTextObj_GetTextRenderMode(stroke));
2194
2195 UnloadPage(page);
2196}
2197
Hui Yingst9d8e5ec2020-07-30 22:33:22 +00002198TEST_F(FPDFEditEmbedderTest, SetTextRenderMode) {
Hui Yingst609f7d62020-04-23 23:14:13 +00002199#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
Hui Yingst9d8e5ec2020-07-30 22:33:22 +00002200 const char kOriginalChecksum[] = "4960c720a6cdb7b2b16be809cce58aff";
2201 const char kStrokeChecksum[] = "4c099cf9abc1565806bd7cd1cca5ac1a";
Hui Yingst609f7d62020-04-23 23:14:13 +00002202#else
Hui Yingst9d8e5ec2020-07-30 22:33:22 +00002203#if defined(OS_WIN)
2204 const char kOriginalChecksum[] = "de6e86bad3e9fda753a8471a45cfbb58";
2205#elif defined(OS_APPLE)
2206 const char kOriginalChecksum[] = "2453a9a524ee3f5f525dd21345ec4d81";
Daniel Hosseiniane83d8b42020-01-08 00:26:15 +00002207#else
Hui Yingst9d8e5ec2020-07-30 22:33:22 +00002208 const char kOriginalChecksum[] = "5a012d2920ac075c39ffa9437ea42faa";
Daniel Hosseiniane83d8b42020-01-08 00:26:15 +00002209#endif
Hui Yingst9d8e5ec2020-07-30 22:33:22 +00002210 const char kStrokeChecksum[] = "412e52e621b46bd77baf2162e1fb1a1d";
2211#endif // defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
Daniel Hosseiniane83d8b42020-01-08 00:26:15 +00002212
Daniel Hosseiniane83d8b42020-01-08 00:26:15 +00002213 {
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00002214 ASSERT_TRUE(OpenDocument("text_render_mode.pdf"));
Daniel Hosseiniana933ebf2020-01-31 21:28:25 +00002215 FPDF_PAGE page = LoadPage(0);
2216 ASSERT_TRUE(page);
2217 ASSERT_EQ(2, FPDFPage_CountObjects(page));
2218
2219 // Check the bitmap
2220 {
2221 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingst9d8e5ec2020-07-30 22:33:22 +00002222 CompareBitmap(page_bitmap.get(), 612, 446, kOriginalChecksum);
Daniel Hosseiniana933ebf2020-01-31 21:28:25 +00002223 }
2224
2225 // Cannot set on a null object.
2226 EXPECT_FALSE(
2227 FPDFTextObj_SetTextRenderMode(nullptr, FPDF_TEXTRENDERMODE_UNKNOWN));
2228 EXPECT_FALSE(
2229 FPDFTextObj_SetTextRenderMode(nullptr, FPDF_TEXTRENDERMODE_INVISIBLE));
2230
2231 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0);
2232 ASSERT_TRUE(page_object);
2233 EXPECT_EQ(FPDF_TEXTRENDERMODE_FILL,
2234 FPDFTextObj_GetTextRenderMode(page_object));
2235
2236 // Cannot set UNKNOWN as a render mode.
2237 EXPECT_FALSE(FPDFTextObj_SetTextRenderMode(page_object,
2238 FPDF_TEXTRENDERMODE_UNKNOWN));
2239
2240 EXPECT_TRUE(
2241 FPDFTextObj_SetTextRenderMode(page_object, FPDF_TEXTRENDERMODE_STROKE));
2242 EXPECT_EQ(FPDF_TEXTRENDERMODE_STROKE,
2243 FPDFTextObj_GetTextRenderMode(page_object));
2244
2245 // Check that bitmap displays changed content
2246 {
2247 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingst9d8e5ec2020-07-30 22:33:22 +00002248 CompareBitmap(page_bitmap.get(), 612, 446, kStrokeChecksum);
Daniel Hosseiniana933ebf2020-01-31 21:28:25 +00002249 }
2250
Daniel Hosseinianafd27502020-07-01 04:59:03 +00002251 // Save a copy.
2252 EXPECT_TRUE(FPDFPage_GenerateContent(page));
2253 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
2254
Daniel Hosseiniana933ebf2020-01-31 21:28:25 +00002255 UnloadPage(page);
Daniel Hosseiniane83d8b42020-01-08 00:26:15 +00002256 }
Miklos Vajna1448cc12018-07-03 13:52:33 +00002257
Daniel Hosseiniana933ebf2020-01-31 21:28:25 +00002258 {
Daniel Hosseinianafd27502020-07-01 04:59:03 +00002259 // Open the saved copy and render it. Check that the changed text render
2260 // mode is kept in the saved copy.
Daniel Hosseiniana933ebf2020-01-31 21:28:25 +00002261 ASSERT_TRUE(OpenSavedDocument());
2262 FPDF_PAGE saved_page = LoadSavedPage(0);
2263 ASSERT_TRUE(saved_page);
2264
2265 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(saved_page, 0);
2266 EXPECT_TRUE(page_object);
Daniel Hosseinianafd27502020-07-01 04:59:03 +00002267 EXPECT_EQ(FPDF_TEXTRENDERMODE_STROKE,
Daniel Hosseiniana933ebf2020-01-31 21:28:25 +00002268 FPDFTextObj_GetTextRenderMode(page_object));
2269
2270 ScopedFPDFBitmap bitmap = RenderSavedPage(saved_page);
Hui Yingst9d8e5ec2020-07-30 22:33:22 +00002271 CompareBitmap(bitmap.get(), 612, 446, kStrokeChecksum);
Daniel Hosseiniana933ebf2020-01-31 21:28:25 +00002272
2273 CloseSavedPage(saved_page);
2274 CloseSavedDocument();
2275 }
Miklos Vajna1448cc12018-07-03 13:52:33 +00002276}
2277
Lei Zhangab41f252018-12-23 03:10:50 +00002278TEST_F(FPDFEditEmbedderTest, TestGetTextFontName) {
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00002279 ASSERT_TRUE(OpenDocument("text_font.pdf"));
Miklos Vajna53d4f0a2018-08-01 01:28:49 +00002280 FPDF_PAGE page = LoadPage(0);
2281 ASSERT_TRUE(page);
2282 ASSERT_EQ(1, FPDFPage_CountObjects(page));
2283
2284 // FPDFTextObj_GetFontName() positive testing.
2285 FPDF_PAGEOBJECT text = FPDFPage_GetObject(page, 0);
2286 unsigned long size = FPDFTextObj_GetFontName(text, nullptr, 0);
2287 const char kExpectedFontName[] = "Liberation Serif";
2288 ASSERT_EQ(sizeof(kExpectedFontName), size);
2289 std::vector<char> font_name(size);
2290 ASSERT_EQ(size, FPDFTextObj_GetFontName(text, font_name.data(), size));
2291 ASSERT_STREQ(kExpectedFontName, font_name.data());
2292
2293 // FPDFTextObj_GetFontName() negative testing.
2294 ASSERT_EQ(0U, FPDFTextObj_GetFontName(nullptr, nullptr, 0));
2295
2296 font_name.resize(2);
2297 font_name[0] = 'x';
2298 font_name[1] = '\0';
2299 size = FPDFTextObj_GetFontName(text, font_name.data(), font_name.size());
2300 ASSERT_EQ(sizeof(kExpectedFontName), size);
2301 ASSERT_EQ(std::string("x"), std::string(font_name.data()));
2302
2303 UnloadPage(page);
2304}
2305
Lei Zhangab41f252018-12-23 03:10:50 +00002306TEST_F(FPDFEditEmbedderTest, TestFormGetObjects) {
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00002307 ASSERT_TRUE(OpenDocument("form_object.pdf"));
Miklos Vajnab66077d2018-07-11 13:25:02 +00002308 FPDF_PAGE page = LoadPage(0);
2309 ASSERT_TRUE(page);
2310 ASSERT_EQ(1, FPDFPage_CountObjects(page));
2311
2312 FPDF_PAGEOBJECT form = FPDFPage_GetObject(page, 0);
2313 EXPECT_EQ(FPDF_PAGEOBJ_FORM, FPDFPageObj_GetType(form));
2314 ASSERT_EQ(-1, FPDFFormObj_CountObjects(nullptr));
2315 ASSERT_EQ(2, FPDFFormObj_CountObjects(form));
2316
Miklos Vajna1d273f12018-07-16 19:20:36 +00002317 // FPDFFormObj_GetObject() positive testing.
2318 FPDF_PAGEOBJECT text1 = FPDFFormObj_GetObject(form, 0);
2319 ASSERT_TRUE(text1);
2320 float left = 0;
2321 float bottom = 0;
2322 float right = 0;
2323 float top = 0;
2324 ASSERT_TRUE(FPDFPageObj_GetBounds(text1, &left, &bottom, &right, &top));
2325 ASSERT_EQ(271, static_cast<int>(top));
2326
2327 FPDF_PAGEOBJECT text2 = FPDFFormObj_GetObject(form, 1);
2328 ASSERT_TRUE(text2);
2329 ASSERT_TRUE(FPDFPageObj_GetBounds(text2, &left, &bottom, &right, &top));
2330 ASSERT_EQ(221, static_cast<int>(top));
2331
2332 // FPDFFormObj_GetObject() negative testing.
2333 ASSERT_EQ(nullptr, FPDFFormObj_GetObject(nullptr, 0));
2334 ASSERT_EQ(nullptr, FPDFFormObj_GetObject(form, -1));
2335 ASSERT_EQ(nullptr, FPDFFormObj_GetObject(form, 2));
2336
Miklos Vajna46b43732018-08-14 19:15:43 +00002337 // Reset the form object matrix to identity.
2338 auto* pPageObj = CPDFPageObjectFromFPDFPageObject(form);
2339 CPDF_FormObject* pFormObj = pPageObj->AsForm();
2340 pFormObj->Transform(pFormObj->form_matrix().GetInverse());
2341
2342 // FPDFFormObj_GetMatrix() positive testing.
Lei Zhang8da98232019-12-11 23:29:33 +00002343 static constexpr FS_MATRIX kMatrix = {1.0f, 1.5f, 2.0f, 2.5f, 100.0f, 200.0f};
2344 pFormObj->Transform(CFXMatrixFromFSMatrix(kMatrix));
Miklos Vajna46b43732018-08-14 19:15:43 +00002345
Lei Zhang8da98232019-12-11 23:29:33 +00002346 FS_MATRIX matrix;
2347 EXPECT_TRUE(FPDFFormObj_GetMatrix(form, &matrix));
2348 EXPECT_FLOAT_EQ(kMatrix.a, matrix.a);
2349 EXPECT_FLOAT_EQ(kMatrix.b, matrix.b);
2350 EXPECT_FLOAT_EQ(kMatrix.c, matrix.c);
2351 EXPECT_FLOAT_EQ(kMatrix.d, matrix.d);
2352 EXPECT_FLOAT_EQ(kMatrix.e, matrix.e);
2353 EXPECT_FLOAT_EQ(kMatrix.f, matrix.f);
Miklos Vajna46b43732018-08-14 19:15:43 +00002354
2355 // FPDFFormObj_GetMatrix() negative testing.
Lei Zhang8da98232019-12-11 23:29:33 +00002356 EXPECT_FALSE(FPDFFormObj_GetMatrix(nullptr, &matrix));
2357 EXPECT_FALSE(FPDFFormObj_GetMatrix(form, nullptr));
2358 EXPECT_FALSE(FPDFFormObj_GetMatrix(nullptr, nullptr));
Miklos Vajna46b43732018-08-14 19:15:43 +00002359
Miklos Vajnab66077d2018-07-11 13:25:02 +00002360 UnloadPage(page);
2361}
2362
Nicolas Pena4c48b102018-06-13 18:23:46 +00002363// Tests adding text from standard font using FPDFText_LoadStandardFont.
Lei Zhang03e5e682019-09-16 19:45:55 +00002364// TODO(crbug.com/pdfium/11): Fix this test and enable.
2365#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
2366#define MAYBE_AddStandardFontText2 DISABLED_AddStandardFontText2
2367#else
2368#define MAYBE_AddStandardFontText2 AddStandardFontText2
2369#endif
2370TEST_F(FPDFEditEmbedderTest, MAYBE_AddStandardFontText2) {
Nicolas Pena4c48b102018-06-13 18:23:46 +00002371 // Start with a blank page
2372 ScopedFPDFPage page(FPDFPage_New(CreateNewDocument(), 0, 612, 792));
2373
2374 // Load a standard font.
Tom Sepezffff6c52019-07-30 21:56:19 +00002375 ScopedFPDFFont font(FPDFText_LoadStandardFont(document(), "Helvetica"));
Nicolas Pena4c48b102018-06-13 18:23:46 +00002376 ASSERT_TRUE(font);
2377
2378 // Add some text to the page.
2379 FPDF_PAGEOBJECT text_object =
Tom Sepezffff6c52019-07-30 21:56:19 +00002380 FPDFPageObj_CreateTextObj(document(), font.get(), 12.0f);
Nicolas Pena4c48b102018-06-13 18:23:46 +00002381 EXPECT_TRUE(text_object);
Hui Yingst3b6136a2020-06-09 00:39:33 +00002382 ScopedFPDFWideString text = GetFPDFWideString(kBottomText);
Nicolas Pena4c48b102018-06-13 18:23:46 +00002383 EXPECT_TRUE(FPDFText_SetText(text_object, text.get()));
2384 FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 20, 20);
2385 FPDFPage_InsertObject(page.get(), text_object);
Lei Zhang30ff2532019-01-31 21:37:55 +00002386 ScopedFPDFBitmap page_bitmap = RenderPage(page.get());
Hui Yingst3b6136a2020-06-09 00:39:33 +00002387 CompareBitmap(page_bitmap.get(), 612, 792, kBottomTextChecksum);
Nicolas Pena4c48b102018-06-13 18:23:46 +00002388}
2389
Lei Zhangab41f252018-12-23 03:10:50 +00002390TEST_F(FPDFEditEmbedderTest, LoadStandardFonts) {
Nicolas Pena4c48b102018-06-13 18:23:46 +00002391 CreateNewDocument();
Lei Zhang590f4242019-05-15 20:57:26 +00002392 static constexpr const char* kStandardFontNames[] = {
Lei Zhangd72fd582018-07-27 19:37:27 +00002393 "Arial",
2394 "Arial-Bold",
2395 "Arial-BoldItalic",
2396 "Arial-Italic",
2397 "Courier",
2398 "Courier-BoldOblique",
2399 "Courier-Oblique",
2400 "Courier-Bold",
2401 "CourierNew",
2402 "CourierNew-Bold",
2403 "CourierNew-BoldItalic",
2404 "CourierNew-Italic",
2405 "Helvetica",
2406 "Helvetica-Bold",
2407 "Helvetica-BoldOblique",
2408 "Helvetica-Oblique",
2409 "Symbol",
2410 "TimesNewRoman",
2411 "TimesNewRoman-Bold",
2412 "TimesNewRoman-BoldItalic",
2413 "TimesNewRoman-Italic",
2414 "ZapfDingbats"};
Lei Zhang590f4242019-05-15 20:57:26 +00002415 for (const char* font_name : kStandardFontNames) {
Tom Sepezffff6c52019-07-30 21:56:19 +00002416 ScopedFPDFFont font(FPDFText_LoadStandardFont(document(), font_name));
Nicolas Pena4c48b102018-06-13 18:23:46 +00002417 EXPECT_TRUE(font) << font_name << " should be considered a standard font.";
2418 }
Lei Zhang590f4242019-05-15 20:57:26 +00002419 static constexpr const char* kNotStandardFontNames[] = {
Nicolas Pena4c48b102018-06-13 18:23:46 +00002420 "Abcdefg", "ArialB", "Arial-Style",
2421 "Font Name", "FontArial", "NotAStandardFontName",
2422 "TestFontName", "Quack", "Symbol-Italic",
2423 "Zapf"};
Lei Zhang590f4242019-05-15 20:57:26 +00002424 for (const char* font_name : kNotStandardFontNames) {
Tom Sepezffff6c52019-07-30 21:56:19 +00002425 ScopedFPDFFont font(FPDFText_LoadStandardFont(document(), font_name));
Nicolas Pena4c48b102018-06-13 18:23:46 +00002426 EXPECT_FALSE(font) << font_name
2427 << " should not be considered a standard font.";
2428 }
2429}
2430
Lei Zhangab41f252018-12-23 03:10:50 +00002431TEST_F(FPDFEditEmbedderTest, GraphicsData) {
Nicolas Pena603a31d2017-06-14 11:41:18 -04002432 // New page
Tom Sepeze08d2b12018-04-25 18:49:32 +00002433 ScopedFPDFPage page(FPDFPage_New(CreateNewDocument(), 0, 612, 792));
Nicolas Pena603a31d2017-06-14 11:41:18 -04002434
2435 // Create a rect with nontrivial graphics
2436 FPDF_PAGEOBJECT rect1 = FPDFPageObj_CreateNewRect(10, 10, 100, 100);
2437 FPDFPageObj_SetBlendMode(rect1, "Color");
2438 FPDFPage_InsertObject(page.get(), rect1);
2439 EXPECT_TRUE(FPDFPage_GenerateContent(page.get()));
2440
2441 // Check that the ExtGState was created
Lei Zhang107fa7b2018-02-09 21:48:15 +00002442 CPDF_Page* cpage = CPDFPageFromFPDFPage(page.get());
2443 CPDF_Dictionary* graphics_dict = cpage->m_pResources->GetDictFor("ExtGState");
Nicolas Pena603a31d2017-06-14 11:41:18 -04002444 ASSERT_TRUE(graphics_dict);
Lei Zhangf40380f2018-10-12 18:31:51 +00002445 EXPECT_EQ(2u, graphics_dict->size());
Nicolas Pena603a31d2017-06-14 11:41:18 -04002446
2447 // Add a text object causing no change to the graphics dictionary
2448 FPDF_PAGEOBJECT text1 = FPDFPageObj_NewTextObj(document(), "Arial", 12.0f);
2449 // Only alpha, the last component, matters for the graphics dictionary. And
2450 // the default value is 255.
Lei Zhang3475b482019-05-13 18:30:57 +00002451 EXPECT_TRUE(FPDFPageObj_SetFillColor(text1, 100, 100, 100, 255));
Nicolas Pena603a31d2017-06-14 11:41:18 -04002452 FPDFPage_InsertObject(page.get(), text1);
2453 EXPECT_TRUE(FPDFPage_GenerateContent(page.get()));
Lei Zhangf40380f2018-10-12 18:31:51 +00002454 EXPECT_EQ(2u, graphics_dict->size());
Nicolas Pena603a31d2017-06-14 11:41:18 -04002455
2456 // Add a text object increasing the size of the graphics dictionary
2457 FPDF_PAGEOBJECT text2 =
2458 FPDFPageObj_NewTextObj(document(), "Times-Roman", 12.0f);
2459 FPDFPage_InsertObject(page.get(), text2);
2460 FPDFPageObj_SetBlendMode(text2, "Darken");
Lei Zhang3475b482019-05-13 18:30:57 +00002461 EXPECT_TRUE(FPDFPageObj_SetFillColor(text2, 0, 0, 255, 150));
Nicolas Pena603a31d2017-06-14 11:41:18 -04002462 EXPECT_TRUE(FPDFPage_GenerateContent(page.get()));
Lei Zhangf40380f2018-10-12 18:31:51 +00002463 EXPECT_EQ(3u, graphics_dict->size());
Nicolas Pena603a31d2017-06-14 11:41:18 -04002464
2465 // Add a path that should reuse graphics
Nicolas Penace67be42017-06-14 14:52:49 -04002466 FPDF_PAGEOBJECT path = FPDFPageObj_CreateNewPath(400, 100);
Nicolas Pena603a31d2017-06-14 11:41:18 -04002467 FPDFPageObj_SetBlendMode(path, "Darken");
Lei Zhang3475b482019-05-13 18:30:57 +00002468 EXPECT_TRUE(FPDFPageObj_SetFillColor(path, 200, 200, 100, 150));
Nicolas Pena603a31d2017-06-14 11:41:18 -04002469 FPDFPage_InsertObject(page.get(), path);
2470 EXPECT_TRUE(FPDFPage_GenerateContent(page.get()));
Lei Zhangf40380f2018-10-12 18:31:51 +00002471 EXPECT_EQ(3u, graphics_dict->size());
Nicolas Pena603a31d2017-06-14 11:41:18 -04002472
2473 // Add a rect increasing the size of the graphics dictionary
2474 FPDF_PAGEOBJECT rect2 = FPDFPageObj_CreateNewRect(10, 10, 100, 100);
2475 FPDFPageObj_SetBlendMode(rect2, "Darken");
Lei Zhang3475b482019-05-13 18:30:57 +00002476 EXPECT_TRUE(FPDFPageObj_SetFillColor(rect2, 0, 0, 255, 150));
2477 EXPECT_TRUE(FPDFPageObj_SetStrokeColor(rect2, 0, 0, 0, 200));
Nicolas Pena603a31d2017-06-14 11:41:18 -04002478 FPDFPage_InsertObject(page.get(), rect2);
2479 EXPECT_TRUE(FPDFPage_GenerateContent(page.get()));
Lei Zhangf40380f2018-10-12 18:31:51 +00002480 EXPECT_EQ(4u, graphics_dict->size());
Nicolas Pena603a31d2017-06-14 11:41:18 -04002481}
2482
Lei Zhangab41f252018-12-23 03:10:50 +00002483TEST_F(FPDFEditEmbedderTest, DoubleGenerating) {
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002484 // Start with a blank page
Nicolas Penad03ca422017-03-06 13:54:33 -05002485 FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792);
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002486
2487 // Add a red rectangle with some non-default alpha
2488 FPDF_PAGEOBJECT rect = FPDFPageObj_CreateNewRect(10, 10, 100, 100);
Lei Zhang3475b482019-05-13 18:30:57 +00002489 EXPECT_TRUE(FPDFPageObj_SetFillColor(rect, 255, 0, 0, 128));
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002490 EXPECT_TRUE(FPDFPath_SetDrawMode(rect, FPDF_FILLMODE_WINDING, 0));
2491 FPDFPage_InsertObject(page, rect);
2492 EXPECT_TRUE(FPDFPage_GenerateContent(page));
2493
2494 // Check the ExtGState
Lei Zhang107fa7b2018-02-09 21:48:15 +00002495 CPDF_Page* cpage = CPDFPageFromFPDFPage(page);
2496 CPDF_Dictionary* graphics_dict = cpage->m_pResources->GetDictFor("ExtGState");
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002497 ASSERT_TRUE(graphics_dict);
Lei Zhangf40380f2018-10-12 18:31:51 +00002498 EXPECT_EQ(2u, graphics_dict->size());
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002499
2500 // Check the bitmap
Lei Zhang107fa7b2018-02-09 21:48:15 +00002501 {
Lei Zhang30ff2532019-01-31 21:37:55 +00002502 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002503 CompareBitmap(page_bitmap.get(), 612, 792,
2504 "5384da3406d62360ffb5cac4476fff1c");
2505 }
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002506
2507 // Never mind, my new favorite color is blue, increase alpha
Lei Zhang3475b482019-05-13 18:30:57 +00002508 EXPECT_TRUE(FPDFPageObj_SetFillColor(rect, 0, 0, 255, 180));
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002509 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Lei Zhangf40380f2018-10-12 18:31:51 +00002510 EXPECT_EQ(3u, graphics_dict->size());
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002511
2512 // Check that bitmap displays changed content
Lei Zhang107fa7b2018-02-09 21:48:15 +00002513 {
Lei Zhang30ff2532019-01-31 21:37:55 +00002514 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002515 CompareBitmap(page_bitmap.get(), 612, 792,
2516 "2e51656f5073b0bee611d9cd086aa09c");
2517 }
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002518
2519 // And now generate, without changes
2520 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Lei Zhangf40380f2018-10-12 18:31:51 +00002521 EXPECT_EQ(3u, graphics_dict->size());
Lei Zhang107fa7b2018-02-09 21:48:15 +00002522 {
Lei Zhang30ff2532019-01-31 21:37:55 +00002523 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002524 CompareBitmap(page_bitmap.get(), 612, 792,
2525 "2e51656f5073b0bee611d9cd086aa09c");
2526 }
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002527
2528 // Add some text to the page
Nicolas Penab3161852017-05-02 14:12:50 -04002529 FPDF_PAGEOBJECT text_object =
2530 FPDFPageObj_NewTextObj(document(), "Arial", 12.0f);
Lei Zhangf0f67682019-04-08 17:03:21 +00002531 ScopedFPDFWideString text =
Nicolas Penab3161852017-05-02 14:12:50 -04002532 GetFPDFWideString(L"Something something #text# something");
2533 EXPECT_TRUE(FPDFText_SetText(text_object, text.get()));
2534 FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 300, 300);
2535 FPDFPage_InsertObject(page, text_object);
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002536 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Lei Zhang107fa7b2018-02-09 21:48:15 +00002537 CPDF_Dictionary* font_dict = cpage->m_pResources->GetDictFor("Font");
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002538 ASSERT_TRUE(font_dict);
Lei Zhangf40380f2018-10-12 18:31:51 +00002539 EXPECT_EQ(1u, font_dict->size());
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002540
2541 // Generate yet again, check dicts are reasonably sized
2542 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Lei Zhangf40380f2018-10-12 18:31:51 +00002543 EXPECT_EQ(3u, graphics_dict->size());
2544 EXPECT_EQ(1u, font_dict->size());
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002545 FPDF_ClosePage(page);
Nicolas Penaa4ad01f2017-02-15 16:26:48 -05002546}
Nicolas Penabe90aae2017-02-27 10:41:41 -05002547
Lei Zhangab41f252018-12-23 03:10:50 +00002548TEST_F(FPDFEditEmbedderTest, LoadSimpleType1Font) {
Nicolas Penad03ca422017-03-06 13:54:33 -05002549 CreateNewDocument();
2550 // TODO(npm): use other fonts after disallowing loading any font as any type
Tom Sepezffff6c52019-07-30 21:56:19 +00002551 RetainPtr<CPDF_Font> stock_font =
Nicolas Penad03ca422017-03-06 13:54:33 -05002552 CPDF_Font::GetStockFont(cpdf_doc(), "Times-Bold");
Tom Sepez20c41a52018-08-29 23:53:53 +00002553 pdfium::span<const uint8_t> span = stock_font->GetFont()->GetFontSpan();
2554 ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(),
2555 FPDF_FONT_TYPE1, false));
Nicolas Penab3161852017-05-02 14:12:50 -04002556 ASSERT_TRUE(font.get());
Tom Sepez525147a2018-05-03 17:19:53 +00002557 CPDF_Font* typed_font = CPDFFontFromFPDFFont(font.get());
Nicolas Penad03ca422017-03-06 13:54:33 -05002558 EXPECT_TRUE(typed_font->IsType1Font());
Nicolas Penabe90aae2017-02-27 10:41:41 -05002559
Lei Zhang710fa992018-05-25 16:24:48 +00002560 const CPDF_Dictionary* font_dict = typed_font->GetFontDict();
Lei Zhanga228ff32020-06-24 17:39:33 +00002561 EXPECT_EQ("Font", font_dict->GetNameFor("Type"));
2562 EXPECT_EQ("Type1", font_dict->GetNameFor("Subtype"));
2563 EXPECT_EQ("TimesNewRomanPS-BoldMT", font_dict->GetNameFor("BaseFont"));
Nicolas Penabe90aae2017-02-27 10:41:41 -05002564 ASSERT_TRUE(font_dict->KeyExist("FirstChar"));
2565 ASSERT_TRUE(font_dict->KeyExist("LastChar"));
2566 EXPECT_EQ(32, font_dict->GetIntegerFor("FirstChar"));
Nicolas Penad03ca422017-03-06 13:54:33 -05002567 EXPECT_EQ(255, font_dict->GetIntegerFor("LastChar"));
2568
Lei Zhangde579ab2018-05-25 21:49:49 +00002569 const CPDF_Array* widths_array = font_dict->GetArrayFor("Widths");
Nicolas Penad03ca422017-03-06 13:54:33 -05002570 ASSERT_TRUE(widths_array);
Lei Zhangf40380f2018-10-12 18:31:51 +00002571 ASSERT_EQ(224u, widths_array->size());
Nicolas Penabe90aae2017-02-27 10:41:41 -05002572 EXPECT_EQ(250, widths_array->GetNumberAt(0));
Nicolas Penad03ca422017-03-06 13:54:33 -05002573 EXPECT_EQ(569, widths_array->GetNumberAt(11));
2574 EXPECT_EQ(500, widths_array->GetNumberAt(223));
Tom Sepez20c41a52018-08-29 23:53:53 +00002575 CheckFontDescriptor(font_dict, FPDF_FONT_TYPE1, true, false, span);
Nicolas Penad03ca422017-03-06 13:54:33 -05002576}
Nicolas Penabe90aae2017-02-27 10:41:41 -05002577
Lei Zhangab41f252018-12-23 03:10:50 +00002578TEST_F(FPDFEditEmbedderTest, LoadSimpleTrueTypeFont) {
Nicolas Penad03ca422017-03-06 13:54:33 -05002579 CreateNewDocument();
Tom Sepezffff6c52019-07-30 21:56:19 +00002580 RetainPtr<CPDF_Font> stock_font =
2581 CPDF_Font::GetStockFont(cpdf_doc(), "Courier");
Tom Sepez20c41a52018-08-29 23:53:53 +00002582 pdfium::span<const uint8_t> span = stock_font->GetFont()->GetFontSpan();
2583 ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(),
2584 FPDF_FONT_TRUETYPE, false));
Nicolas Penab3161852017-05-02 14:12:50 -04002585 ASSERT_TRUE(font.get());
Tom Sepez525147a2018-05-03 17:19:53 +00002586 CPDF_Font* typed_font = CPDFFontFromFPDFFont(font.get());
Nicolas Penad03ca422017-03-06 13:54:33 -05002587 EXPECT_TRUE(typed_font->IsTrueTypeFont());
Nicolas Penabe90aae2017-02-27 10:41:41 -05002588
Lei Zhang710fa992018-05-25 16:24:48 +00002589 const CPDF_Dictionary* font_dict = typed_font->GetFontDict();
Lei Zhanga228ff32020-06-24 17:39:33 +00002590 EXPECT_EQ("Font", font_dict->GetNameFor("Type"));
2591 EXPECT_EQ("TrueType", font_dict->GetNameFor("Subtype"));
2592 EXPECT_EQ("CourierNewPSMT", font_dict->GetNameFor("BaseFont"));
Nicolas Penad03ca422017-03-06 13:54:33 -05002593 ASSERT_TRUE(font_dict->KeyExist("FirstChar"));
2594 ASSERT_TRUE(font_dict->KeyExist("LastChar"));
2595 EXPECT_EQ(32, font_dict->GetIntegerFor("FirstChar"));
2596 EXPECT_EQ(255, font_dict->GetIntegerFor("LastChar"));
Nicolas Penabe90aae2017-02-27 10:41:41 -05002597
Lei Zhangde579ab2018-05-25 21:49:49 +00002598 const CPDF_Array* widths_array = font_dict->GetArrayFor("Widths");
Nicolas Penad03ca422017-03-06 13:54:33 -05002599 ASSERT_TRUE(widths_array);
Lei Zhangf40380f2018-10-12 18:31:51 +00002600 ASSERT_EQ(224u, widths_array->size());
Nicolas Penad03ca422017-03-06 13:54:33 -05002601 EXPECT_EQ(600, widths_array->GetNumberAt(33));
2602 EXPECT_EQ(600, widths_array->GetNumberAt(74));
2603 EXPECT_EQ(600, widths_array->GetNumberAt(223));
Tom Sepez20c41a52018-08-29 23:53:53 +00002604 CheckFontDescriptor(font_dict, FPDF_FONT_TRUETYPE, false, false, span);
Nicolas Penad03ca422017-03-06 13:54:33 -05002605}
2606
Lei Zhangab41f252018-12-23 03:10:50 +00002607TEST_F(FPDFEditEmbedderTest, LoadCIDType0Font) {
Nicolas Penad03ca422017-03-06 13:54:33 -05002608 CreateNewDocument();
Tom Sepezffff6c52019-07-30 21:56:19 +00002609 RetainPtr<CPDF_Font> stock_font =
Nicolas Penad03ca422017-03-06 13:54:33 -05002610 CPDF_Font::GetStockFont(cpdf_doc(), "Times-Roman");
Tom Sepez20c41a52018-08-29 23:53:53 +00002611 pdfium::span<const uint8_t> span = stock_font->GetFont()->GetFontSpan();
2612 ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(),
2613 FPDF_FONT_TYPE1, 1));
Nicolas Penab3161852017-05-02 14:12:50 -04002614 ASSERT_TRUE(font.get());
Tom Sepez525147a2018-05-03 17:19:53 +00002615 CPDF_Font* typed_font = CPDFFontFromFPDFFont(font.get());
Nicolas Penad03ca422017-03-06 13:54:33 -05002616 EXPECT_TRUE(typed_font->IsCIDFont());
2617
2618 // Check font dictionary entries
Lei Zhang710fa992018-05-25 16:24:48 +00002619 const CPDF_Dictionary* font_dict = typed_font->GetFontDict();
Lei Zhanga228ff32020-06-24 17:39:33 +00002620 EXPECT_EQ("Font", font_dict->GetNameFor("Type"));
2621 EXPECT_EQ("Type0", font_dict->GetNameFor("Subtype"));
2622 EXPECT_EQ("TimesNewRomanPSMT-Identity-H", font_dict->GetNameFor("BaseFont"));
2623 EXPECT_EQ("Identity-H", font_dict->GetNameFor("Encoding"));
Lei Zhangde579ab2018-05-25 21:49:49 +00002624 const CPDF_Array* descendant_array =
2625 font_dict->GetArrayFor("DescendantFonts");
Nicolas Penad03ca422017-03-06 13:54:33 -05002626 ASSERT_TRUE(descendant_array);
Lei Zhangf40380f2018-10-12 18:31:51 +00002627 EXPECT_EQ(1u, descendant_array->size());
Nicolas Penad03ca422017-03-06 13:54:33 -05002628
2629 // Check the CIDFontDict
Lei Zhangde579ab2018-05-25 21:49:49 +00002630 const CPDF_Dictionary* cidfont_dict = descendant_array->GetDictAt(0);
Lei Zhanga228ff32020-06-24 17:39:33 +00002631 EXPECT_EQ("Font", cidfont_dict->GetNameFor("Type"));
2632 EXPECT_EQ("CIDFontType0", cidfont_dict->GetNameFor("Subtype"));
2633 EXPECT_EQ("TimesNewRomanPSMT", cidfont_dict->GetNameFor("BaseFont"));
Lei Zhangb1ec2802018-05-25 21:55:24 +00002634 const CPDF_Dictionary* cidinfo_dict =
2635 cidfont_dict->GetDictFor("CIDSystemInfo");
Nicolas Penad03ca422017-03-06 13:54:33 -05002636 ASSERT_TRUE(cidinfo_dict);
Lei Zhang9c950b12019-01-16 19:06:37 +00002637 const CPDF_Object* registry = cidinfo_dict->GetObjectFor("Registry");
2638 ASSERT_TRUE(registry);
KDr28da0e1b2019-01-17 03:44:29 +00002639 EXPECT_EQ(CPDF_Object::kString, registry->GetType());
Lei Zhang9c950b12019-01-16 19:06:37 +00002640 EXPECT_EQ("Adobe", registry->GetString());
2641 const CPDF_Object* ordering = cidinfo_dict->GetObjectFor("Ordering");
2642 ASSERT_TRUE(ordering);
KDr28da0e1b2019-01-17 03:44:29 +00002643 EXPECT_EQ(CPDF_Object::kString, ordering->GetType());
Lei Zhang9c950b12019-01-16 19:06:37 +00002644 EXPECT_EQ("Identity", ordering->GetString());
Nicolas Penad03ca422017-03-06 13:54:33 -05002645 EXPECT_EQ(0, cidinfo_dict->GetNumberFor("Supplement"));
Tom Sepez20c41a52018-08-29 23:53:53 +00002646 CheckFontDescriptor(cidfont_dict, FPDF_FONT_TYPE1, false, false, span);
Nicolas Penad03ca422017-03-06 13:54:33 -05002647
2648 // Check widths
Lei Zhangde579ab2018-05-25 21:49:49 +00002649 const CPDF_Array* widths_array = cidfont_dict->GetArrayFor("W");
Nicolas Penad03ca422017-03-06 13:54:33 -05002650 ASSERT_TRUE(widths_array);
Lei Zhangf40380f2018-10-12 18:31:51 +00002651 EXPECT_GT(widths_array->size(), 1u);
Nicolas Penad03ca422017-03-06 13:54:33 -05002652 CheckCompositeFontWidths(widths_array, typed_font);
2653}
2654
Lei Zhangab41f252018-12-23 03:10:50 +00002655TEST_F(FPDFEditEmbedderTest, LoadCIDType2Font) {
Nicolas Penad03ca422017-03-06 13:54:33 -05002656 CreateNewDocument();
Tom Sepezffff6c52019-07-30 21:56:19 +00002657 RetainPtr<CPDF_Font> stock_font =
Nicolas Penad03ca422017-03-06 13:54:33 -05002658 CPDF_Font::GetStockFont(cpdf_doc(), "Helvetica-Oblique");
Tom Sepez20c41a52018-08-29 23:53:53 +00002659 pdfium::span<const uint8_t> span = stock_font->GetFont()->GetFontSpan();
2660 ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(),
2661 FPDF_FONT_TRUETYPE, 1));
Nicolas Penab3161852017-05-02 14:12:50 -04002662 ASSERT_TRUE(font.get());
Tom Sepez525147a2018-05-03 17:19:53 +00002663 CPDF_Font* typed_font = CPDFFontFromFPDFFont(font.get());
Nicolas Penad03ca422017-03-06 13:54:33 -05002664 EXPECT_TRUE(typed_font->IsCIDFont());
2665
2666 // Check font dictionary entries
Lei Zhang710fa992018-05-25 16:24:48 +00002667 const CPDF_Dictionary* font_dict = typed_font->GetFontDict();
Lei Zhanga228ff32020-06-24 17:39:33 +00002668 EXPECT_EQ("Font", font_dict->GetNameFor("Type"));
2669 EXPECT_EQ("Type0", font_dict->GetNameFor("Subtype"));
2670 EXPECT_EQ("Arial-ItalicMT", font_dict->GetNameFor("BaseFont"));
2671 EXPECT_EQ("Identity-H", font_dict->GetNameFor("Encoding"));
Lei Zhangde579ab2018-05-25 21:49:49 +00002672 const CPDF_Array* descendant_array =
2673 font_dict->GetArrayFor("DescendantFonts");
Nicolas Penad03ca422017-03-06 13:54:33 -05002674 ASSERT_TRUE(descendant_array);
Lei Zhangf40380f2018-10-12 18:31:51 +00002675 EXPECT_EQ(1u, descendant_array->size());
Nicolas Penad03ca422017-03-06 13:54:33 -05002676
2677 // Check the CIDFontDict
Lei Zhangde579ab2018-05-25 21:49:49 +00002678 const CPDF_Dictionary* cidfont_dict = descendant_array->GetDictAt(0);
Lei Zhanga228ff32020-06-24 17:39:33 +00002679 EXPECT_EQ("Font", cidfont_dict->GetNameFor("Type"));
2680 EXPECT_EQ("CIDFontType2", cidfont_dict->GetNameFor("Subtype"));
2681 EXPECT_EQ("Arial-ItalicMT", cidfont_dict->GetNameFor("BaseFont"));
Lei Zhangb1ec2802018-05-25 21:55:24 +00002682 const CPDF_Dictionary* cidinfo_dict =
2683 cidfont_dict->GetDictFor("CIDSystemInfo");
Nicolas Penad03ca422017-03-06 13:54:33 -05002684 ASSERT_TRUE(cidinfo_dict);
2685 EXPECT_EQ("Adobe", cidinfo_dict->GetStringFor("Registry"));
2686 EXPECT_EQ("Identity", cidinfo_dict->GetStringFor("Ordering"));
2687 EXPECT_EQ(0, cidinfo_dict->GetNumberFor("Supplement"));
Tom Sepez20c41a52018-08-29 23:53:53 +00002688 CheckFontDescriptor(cidfont_dict, FPDF_FONT_TRUETYPE, false, true, span);
Nicolas Penad03ca422017-03-06 13:54:33 -05002689
2690 // Check widths
Lei Zhangde579ab2018-05-25 21:49:49 +00002691 const CPDF_Array* widths_array = cidfont_dict->GetArrayFor("W");
Nicolas Penad03ca422017-03-06 13:54:33 -05002692 ASSERT_TRUE(widths_array);
2693 CheckCompositeFontWidths(widths_array, typed_font);
Nicolas Penabe90aae2017-02-27 10:41:41 -05002694}
rbpotterce8e51e2017-04-28 12:42:47 -07002695
Lei Zhangab41f252018-12-23 03:10:50 +00002696TEST_F(FPDFEditEmbedderTest, NormalizeNegativeRotation) {
rbpotterce8e51e2017-04-28 12:42:47 -07002697 // Load document with a -90 degree rotation
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00002698 ASSERT_TRUE(OpenDocument("bug_713197.pdf"));
rbpotterce8e51e2017-04-28 12:42:47 -07002699 FPDF_PAGE page = LoadPage(0);
2700 EXPECT_NE(nullptr, page);
2701
2702 EXPECT_EQ(3, FPDFPage_GetRotation(page));
2703 UnloadPage(page);
2704}
Nicolas Penab3161852017-05-02 14:12:50 -04002705
Lei Zhang03e5e682019-09-16 19:45:55 +00002706// TODO(crbug.com/pdfium/11): Fix this test and enable.
2707#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
2708#define MAYBE_AddTrueTypeFontText DISABLED_AddTrueTypeFontText
2709#else
2710#define MAYBE_AddTrueTypeFontText AddTrueTypeFontText
2711#endif
2712TEST_F(FPDFEditEmbedderTest, MAYBE_AddTrueTypeFontText) {
Nicolas Penab3161852017-05-02 14:12:50 -04002713 // Start with a blank page
2714 FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792);
2715 {
Tom Sepezffff6c52019-07-30 21:56:19 +00002716 RetainPtr<CPDF_Font> stock_font =
2717 CPDF_Font::GetStockFont(cpdf_doc(), "Arial");
Tom Sepez20c41a52018-08-29 23:53:53 +00002718 pdfium::span<const uint8_t> span = stock_font->GetFont()->GetFontSpan();
2719 ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(),
2720 FPDF_FONT_TRUETYPE, 0));
Nicolas Penab3161852017-05-02 14:12:50 -04002721 ASSERT_TRUE(font.get());
2722
2723 // Add some text to the page
2724 FPDF_PAGEOBJECT text_object =
2725 FPDFPageObj_CreateTextObj(document(), font.get(), 12.0f);
2726 EXPECT_TRUE(text_object);
Hui Yingst3b6136a2020-06-09 00:39:33 +00002727 ScopedFPDFWideString text = GetFPDFWideString(kLoadedFontText);
Nicolas Penab3161852017-05-02 14:12:50 -04002728 EXPECT_TRUE(FPDFText_SetText(text_object, text.get()));
2729 FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 400, 400);
2730 FPDFPage_InsertObject(page, text_object);
Lei Zhang30ff2532019-01-31 21:37:55 +00002731 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingst3b6136a2020-06-09 00:39:33 +00002732 CompareBitmap(page_bitmap.get(), 612, 792, kLoadedFontTextChecksum);
Nicolas Penab3161852017-05-02 14:12:50 -04002733
2734 // Add some more text, same font
2735 FPDF_PAGEOBJECT text_object2 =
2736 FPDFPageObj_CreateTextObj(document(), font.get(), 15.0f);
Lei Zhangf0f67682019-04-08 17:03:21 +00002737 ScopedFPDFWideString text2 = GetFPDFWideString(L"Bigger font size");
Nicolas Penab3161852017-05-02 14:12:50 -04002738 EXPECT_TRUE(FPDFText_SetText(text_object2, text2.get()));
2739 FPDFPageObj_Transform(text_object2, 1, 0, 0, 1, 200, 200);
2740 FPDFPage_InsertObject(page, text_object2);
Nicolas Penab3161852017-05-02 14:12:50 -04002741 }
Lei Zhang30ff2532019-01-31 21:37:55 +00002742 ScopedFPDFBitmap page_bitmap2 = RenderPage(page);
Hui Yingst3b6136a2020-06-09 00:39:33 +00002743#if defined(OS_WIN)
2744 const char kInsertTrueTypeChecksum[] = "2199b579c49ab5f80c246a586a80ee90";
Lei Zhang0c03d632020-07-30 17:05:36 +00002745#elif defined(OS_APPLE)
Hui Yingst3b6136a2020-06-09 00:39:33 +00002746 const char kInsertTrueTypeChecksum[] = "9a1a7dfebe659513691aadd0d95b8d50";
Nicolas Penab3161852017-05-02 14:12:50 -04002747#else
Hui Yingst3b6136a2020-06-09 00:39:33 +00002748 const char kInsertTrueTypeChecksum[] = "c1d10cce1761c4a998a16b2562030568";
Lei Zhange4cdac52019-04-30 16:45:57 +00002749#endif
Hui Yingst3b6136a2020-06-09 00:39:33 +00002750 CompareBitmap(page_bitmap2.get(), 612, 792, kInsertTrueTypeChecksum);
Nicolas Penab3161852017-05-02 14:12:50 -04002751
Nicolas Pena207b7272017-05-26 17:37:06 -04002752 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Nicolas Penab3161852017-05-02 14:12:50 -04002753 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
2754 FPDF_ClosePage(page);
Dan Sinclair04e4dc82017-10-18 12:17:14 -04002755
Hui Yingst3b6136a2020-06-09 00:39:33 +00002756 VerifySavedDocument(612, 792, kInsertTrueTypeChecksum);
Nicolas Penab3161852017-05-02 14:12:50 -04002757}
Nicolas Penaf45ade32017-05-03 10:23:49 -04002758
Lei Zhangab41f252018-12-23 03:10:50 +00002759TEST_F(FPDFEditEmbedderTest, TransformAnnot) {
Jane Liueda65252017-06-07 11:31:27 -04002760 // Open a file with one annotation and load its first page.
2761 ASSERT_TRUE(OpenDocument("annotation_highlight_long_content.pdf"));
Lei Zhang75c81712018-02-08 17:22:39 +00002762 FPDF_PAGE page = LoadPage(0);
Jane Liueda65252017-06-07 11:31:27 -04002763 ASSERT_TRUE(page);
2764
Lei Zhanga21d5932018-02-05 18:28:38 +00002765 {
2766 // Add an underline annotation to the page without specifying its rectangle.
Tom Sepeze08d2b12018-04-25 18:49:32 +00002767 ScopedFPDFAnnotation annot(
Lei Zhanga21d5932018-02-05 18:28:38 +00002768 FPDFPage_CreateAnnot(page, FPDF_ANNOT_UNDERLINE));
2769 ASSERT_TRUE(annot);
Jane Liueda65252017-06-07 11:31:27 -04002770
Lei Zhanga21d5932018-02-05 18:28:38 +00002771 // FPDFPage_TransformAnnots() should run without errors when modifying
2772 // annotation rectangles.
2773 FPDFPage_TransformAnnots(page, 1, 2, 3, 4, 5, 6);
2774 }
Jane Liueda65252017-06-07 11:31:27 -04002775 UnloadPage(page);
2776}
2777
Nicolas Penaf45ade32017-05-03 10:23:49 -04002778// TODO(npm): Add tests using Japanese fonts in other OS.
Lei Zhangfe8e7582020-08-05 19:23:16 +00002779#if defined(OS_LINUX) || defined(OS_CHROMEOS)
Lei Zhang03e5e682019-09-16 19:45:55 +00002780// TODO(crbug.com/pdfium/11): Fix this test and enable.
2781#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
2782#define MAYBE_AddCIDFontText DISABLED_AddCIDFontText
2783#else
2784#define MAYBE_AddCIDFontText AddCIDFontText
2785#endif
2786TEST_F(FPDFEditEmbedderTest, MAYBE_AddCIDFontText) {
Nicolas Penaf45ade32017-05-03 10:23:49 -04002787 // Start with a blank page
2788 FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792);
2789 CFX_Font CIDfont;
2790 {
2791 // First, get the data from the font
2792 CIDfont.LoadSubst("IPAGothic", 1, 0, 400, 0, 932, 0);
2793 EXPECT_EQ("IPAGothic", CIDfont.GetFaceName());
Tom Sepez20c41a52018-08-29 23:53:53 +00002794 pdfium::span<const uint8_t> span = CIDfont.GetFontSpan();
Nicolas Penaf45ade32017-05-03 10:23:49 -04002795
2796 // Load the data into a FPDF_Font.
Tom Sepez20c41a52018-08-29 23:53:53 +00002797 ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(),
2798 FPDF_FONT_TRUETYPE, 1));
Nicolas Penaf45ade32017-05-03 10:23:49 -04002799 ASSERT_TRUE(font.get());
2800
2801 // Add some text to the page
2802 FPDF_PAGEOBJECT text_object =
2803 FPDFPageObj_CreateTextObj(document(), font.get(), 12.0f);
2804 ASSERT_TRUE(text_object);
2805 std::wstring wstr = L"ABCDEFGhijklmnop.";
Lei Zhangf0f67682019-04-08 17:03:21 +00002806 ScopedFPDFWideString text = GetFPDFWideString(wstr);
Nicolas Penaf45ade32017-05-03 10:23:49 -04002807 EXPECT_TRUE(FPDFText_SetText(text_object, text.get()));
2808 FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 200, 200);
2809 FPDFPage_InsertObject(page, text_object);
2810
2811 // And add some Japanese characters
2812 FPDF_PAGEOBJECT text_object2 =
2813 FPDFPageObj_CreateTextObj(document(), font.get(), 18.0f);
2814 ASSERT_TRUE(text_object2);
2815 std::wstring wstr2 =
2816 L"\u3053\u3093\u306B\u3061\u306f\u4e16\u754C\u3002\u3053\u3053\u306B1"
2817 L"\u756A";
Lei Zhangf0f67682019-04-08 17:03:21 +00002818 ScopedFPDFWideString text2 = GetFPDFWideString(wstr2);
Nicolas Penaf45ade32017-05-03 10:23:49 -04002819 EXPECT_TRUE(FPDFText_SetText(text_object2, text2.get()));
2820 FPDFPageObj_Transform(text_object2, 1, 0, 0, 1, 100, 500);
2821 FPDFPage_InsertObject(page, text_object2);
2822 }
2823
Nicolas Pena207b7272017-05-26 17:37:06 -04002824 // Check that the text renders properly.
Lei Zhanga2b70732019-06-25 08:34:22 +00002825 const char md5[] = "5159a72903fe57bf0cf645c894de8a74";
Lei Zhang107fa7b2018-02-09 21:48:15 +00002826 {
Lei Zhang30ff2532019-01-31 21:37:55 +00002827 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002828 CompareBitmap(page_bitmap.get(), 612, 792, md5);
2829 }
Nicolas Penaf45ade32017-05-03 10:23:49 -04002830
2831 // Save the document, close the page.
Nicolas Pena207b7272017-05-26 17:37:06 -04002832 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Nicolas Penaf45ade32017-05-03 10:23:49 -04002833 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
2834 FPDF_ClosePage(page);
Dan Sinclair04e4dc82017-10-18 12:17:14 -04002835
2836 VerifySavedDocument(612, 792, md5);
Nicolas Penaf45ade32017-05-03 10:23:49 -04002837}
Lei Zhangfe8e7582020-08-05 19:23:16 +00002838#endif // defined(OS_LINUX) || defined(OS_CHROMEOS)
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -04002839
Lei Zhang03e5e682019-09-16 19:45:55 +00002840// TODO(crbug.com/pdfium/11): Fix this test and enable.
2841#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
2842#define MAYBE_SaveAndRender DISABLED_SaveAndRender
2843#else
2844#define MAYBE_SaveAndRender SaveAndRender
2845#endif
2846TEST_F(FPDFEditEmbedderTest, MAYBE_SaveAndRender) {
Nicolas Penaa0b48aa2017-06-29 11:01:46 -04002847 const char md5[] = "3c20472b0552c0c22b88ab1ed8c6202b";
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -04002848 {
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00002849 ASSERT_TRUE(OpenDocument("bug_779.pdf"));
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -04002850 FPDF_PAGE page = LoadPage(0);
2851 ASSERT_NE(nullptr, page);
2852
2853 // Now add a more complex blue path.
2854 FPDF_PAGEOBJECT green_path = FPDFPageObj_CreateNewPath(20, 20);
Lei Zhang3475b482019-05-13 18:30:57 +00002855 EXPECT_TRUE(FPDFPageObj_SetFillColor(green_path, 0, 255, 0, 200));
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -04002856 // TODO(npm): stroking will cause the MD5s to differ.
2857 EXPECT_TRUE(FPDFPath_SetDrawMode(green_path, FPDF_FILLMODE_WINDING, 0));
2858 EXPECT_TRUE(FPDFPath_LineTo(green_path, 20, 63));
2859 EXPECT_TRUE(FPDFPath_BezierTo(green_path, 55, 55, 78, 78, 90, 90));
2860 EXPECT_TRUE(FPDFPath_LineTo(green_path, 133, 133));
2861 EXPECT_TRUE(FPDFPath_LineTo(green_path, 133, 33));
2862 EXPECT_TRUE(FPDFPath_BezierTo(green_path, 38, 33, 39, 36, 40, 40));
2863 EXPECT_TRUE(FPDFPath_Close(green_path));
2864 FPDFPage_InsertObject(page, green_path);
Tom Sepeze08d2b12018-04-25 18:49:32 +00002865 ScopedFPDFBitmap page_bitmap = RenderLoadedPage(page);
Lei Zhang107fa7b2018-02-09 21:48:15 +00002866 CompareBitmap(page_bitmap.get(), 612, 792, md5);
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -04002867
2868 // Now save the result, closing the page and document
2869 EXPECT_TRUE(FPDFPage_GenerateContent(page));
2870 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
2871 UnloadPage(page);
2872 }
Dan Sinclair04e4dc82017-10-18 12:17:14 -04002873
2874 VerifySavedDocument(612, 792, md5);
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -04002875}
Jane Liu28fb7ba2017-08-02 21:45:57 -04002876
Lei Zhangab41f252018-12-23 03:10:50 +00002877TEST_F(FPDFEditEmbedderTest, AddMark) {
Henrique Nakashimad8df8c32018-07-12 22:15:09 +00002878 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00002879 ASSERT_TRUE(OpenDocument("text_in_page_marked.pdf"));
Henrique Nakashimad8df8c32018-07-12 22:15:09 +00002880 FPDF_PAGE page = LoadPage(0);
2881 ASSERT_TRUE(page);
2882
Lei Zhang2697cb12019-07-03 18:14:29 +00002883 CheckMarkCounts(page, 1, 19, 8, 4, 9, 1);
Henrique Nakashimad8df8c32018-07-12 22:15:09 +00002884
2885 // Add to the first page object a "Bounds" mark with "Position": "First".
2886 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0);
2887 FPDF_PAGEOBJECTMARK mark = FPDFPageObj_AddMark(page_object, "Bounds");
2888 EXPECT_TRUE(mark);
Henrique Nakashimaa3406772018-07-13 19:10:53 +00002889 EXPECT_TRUE(FPDFPageObjMark_SetStringParam(document(), page_object, mark,
2890 "Position", "First"));
Henrique Nakashimad8df8c32018-07-12 22:15:09 +00002891
Lei Zhang2697cb12019-07-03 18:14:29 +00002892 CheckMarkCounts(page, 1, 19, 8, 4, 9, 2);
Henrique Nakashimad8df8c32018-07-12 22:15:09 +00002893
2894 // Save the file
2895 EXPECT_TRUE(FPDFPage_GenerateContent(page));
2896 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
2897 UnloadPage(page);
2898
2899 // Re-open the file and check the new mark is present.
Lei Zhang0b494052019-01-31 21:41:15 +00002900 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashimad8df8c32018-07-12 22:15:09 +00002901 FPDF_PAGE saved_page = LoadSavedPage(0);
2902
Lei Zhang2697cb12019-07-03 18:14:29 +00002903 CheckMarkCounts(saved_page, 1, 19, 8, 4, 9, 2);
2904
2905 CloseSavedPage(saved_page);
2906 CloseSavedDocument();
2907}
2908
Hui Yingst04440af2020-07-27 22:39:02 +00002909TEST_F(FPDFEditEmbedderTest, AddMarkCompressedStream) {
Lei Zhang2697cb12019-07-03 18:14:29 +00002910 // Load document with some text in a compressed stream.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00002911 ASSERT_TRUE(OpenDocument("hello_world_compressed_stream.pdf"));
Lei Zhang2697cb12019-07-03 18:14:29 +00002912 FPDF_PAGE page = LoadPage(0);
2913 ASSERT_TRUE(page);
2914
2915 // Render and check there are no marks.
2916 {
2917 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00002918 CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldChecksum);
Lei Zhang2697cb12019-07-03 18:14:29 +00002919 }
2920 CheckMarkCounts(page, 0, 2, 0, 0, 0, 0);
2921
2922 // Add to the first page object a "Bounds" mark with "Position": "First".
2923 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0);
2924 FPDF_PAGEOBJECTMARK mark = FPDFPageObj_AddMark(page_object, "Bounds");
2925 EXPECT_TRUE(mark);
2926 EXPECT_TRUE(FPDFPageObjMark_SetStringParam(document(), page_object, mark,
2927 "Position", "First"));
2928
2929 // Render and check there is 1 mark.
2930 {
2931 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00002932 CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldChecksum);
Lei Zhang2697cb12019-07-03 18:14:29 +00002933 }
2934 CheckMarkCounts(page, 0, 2, 0, 0, 0, 1);
2935
2936 // Save the file.
2937 EXPECT_TRUE(FPDFPage_GenerateContent(page));
2938 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
2939 UnloadPage(page);
2940
2941 // Re-open the file and check the new mark is present.
2942 ASSERT_TRUE(OpenSavedDocument());
2943 FPDF_PAGE saved_page = LoadSavedPage(0);
2944
2945 {
2946 ScopedFPDFBitmap page_bitmap = RenderPage(saved_page);
Hui Yingstb4baceb2020-04-28 23:46:10 +00002947 CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldChecksum);
Lei Zhang2697cb12019-07-03 18:14:29 +00002948 }
2949 CheckMarkCounts(saved_page, 0, 2, 0, 0, 0, 1);
Henrique Nakashimad8df8c32018-07-12 22:15:09 +00002950
2951 CloseSavedPage(saved_page);
2952 CloseSavedDocument();
2953}
2954
Lei Zhangab41f252018-12-23 03:10:50 +00002955TEST_F(FPDFEditEmbedderTest, SetMarkParam) {
Henrique Nakashimaa3406772018-07-13 19:10:53 +00002956 // Load document with some text.
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00002957 ASSERT_TRUE(OpenDocument("text_in_page_marked.pdf"));
Henrique Nakashimaa3406772018-07-13 19:10:53 +00002958 FPDF_PAGE page = LoadPage(0);
2959 ASSERT_TRUE(page);
2960
2961 constexpr int kExpectedObjectCount = 19;
2962 CheckMarkCounts(page, 1, kExpectedObjectCount, 8, 4, 9, 1);
2963
2964 // Check the "Bounds" mark's "Position" param is "Last".
2965 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 18);
2966 FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, 1);
2967 ASSERT_TRUE(mark);
2968 char buffer[256];
Henrique Nakashimac3099d12018-09-18 18:08:15 +00002969 unsigned long name_len = 999u;
2970 ASSERT_TRUE(FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len));
2971 EXPECT_EQ((6u + 1u) * 2u, name_len);
Henrique Nakashimaa3406772018-07-13 19:10:53 +00002972 ASSERT_EQ(L"Bounds",
2973 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)));
2974 unsigned long out_buffer_len;
2975 ASSERT_TRUE(FPDFPageObjMark_GetParamStringValue(
2976 mark, "Position", buffer, sizeof(buffer), &out_buffer_len));
2977 ASSERT_EQ(L"Last",
2978 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)));
2979
2980 // Set is to "End".
2981 EXPECT_TRUE(FPDFPageObjMark_SetStringParam(document(), page_object, mark,
2982 "Position", "End"));
2983
2984 // Verify the object passed must correspond to the mark passed.
2985 FPDF_PAGEOBJECT another_page_object = FPDFPage_GetObject(page, 17);
2986 EXPECT_FALSE(FPDFPageObjMark_SetStringParam(document(), another_page_object,
2987 mark, "Position", "End"));
2988
2989 // Verify nothing else changed.
2990 CheckMarkCounts(page, 1, kExpectedObjectCount, 8, 4, 9, 1);
2991
2992 // Verify "Position" now maps to "End".
2993 EXPECT_TRUE(FPDFPageObjMark_GetParamStringValue(
2994 mark, "Position", buffer, sizeof(buffer), &out_buffer_len));
2995 EXPECT_EQ(L"End",
2996 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)));
2997
2998 // Save the file
2999 EXPECT_TRUE(FPDFPage_GenerateContent(page));
3000 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
3001 UnloadPage(page);
3002
3003 // Re-open the file and cerify "Position" still maps to "End".
Lei Zhang0b494052019-01-31 21:41:15 +00003004 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashimaa3406772018-07-13 19:10:53 +00003005 FPDF_PAGE saved_page = LoadSavedPage(0);
3006
3007 CheckMarkCounts(saved_page, 1, kExpectedObjectCount, 8, 4, 9, 1);
3008 page_object = FPDFPage_GetObject(saved_page, 18);
3009 mark = FPDFPageObj_GetMark(page_object, 1);
3010 EXPECT_TRUE(FPDFPageObjMark_GetParamStringValue(
3011 mark, "Position", buffer, sizeof(buffer), &out_buffer_len));
3012 EXPECT_EQ(L"End",
3013 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)));
3014
3015 CloseSavedPage(saved_page);
3016 CloseSavedDocument();
3017}
3018
Lei Zhang03e5e682019-09-16 19:45:55 +00003019// TODO(crbug.com/pdfium/11): Fix this test and enable.
3020#if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_)
3021#define MAYBE_AddMarkedText DISABLED_AddMarkedText
3022#else
3023#define MAYBE_AddMarkedText AddMarkedText
3024#endif
3025TEST_F(FPDFEditEmbedderTest, MAYBE_AddMarkedText) {
Henrique Nakashima144107d2018-07-10 21:04:05 +00003026 // Start with a blank page.
3027 FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792);
3028
Tom Sepezffff6c52019-07-30 21:56:19 +00003029 RetainPtr<CPDF_Font> stock_font =
3030 CPDF_Font::GetStockFont(cpdf_doc(), "Arial");
Tom Sepez20c41a52018-08-29 23:53:53 +00003031 pdfium::span<const uint8_t> span = stock_font->GetFont()->GetFontSpan();
3032 ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(),
3033 FPDF_FONT_TRUETYPE, 0));
Henrique Nakashima144107d2018-07-10 21:04:05 +00003034 ASSERT_TRUE(font.get());
3035
3036 // Add some text to the page.
3037 FPDF_PAGEOBJECT text_object =
3038 FPDFPageObj_CreateTextObj(document(), font.get(), 12.0f);
3039
3040 EXPECT_TRUE(text_object);
Hui Yingst3b6136a2020-06-09 00:39:33 +00003041 ScopedFPDFWideString text1 = GetFPDFWideString(kLoadedFontText);
Henrique Nakashima144107d2018-07-10 21:04:05 +00003042 EXPECT_TRUE(FPDFText_SetText(text_object, text1.get()));
3043 FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 400, 400);
3044 FPDFPage_InsertObject(page, text_object);
3045
3046 // Add a mark with the tag "TestMarkName" to that text.
3047 EXPECT_EQ(0, FPDFPageObj_CountMarks(text_object));
Henrique Nakashima6fc8d872018-09-18 16:48:31 +00003048 FPDF_PAGEOBJECTMARK mark = FPDFPageObj_AddMark(text_object, "Test Mark Name");
Henrique Nakashima144107d2018-07-10 21:04:05 +00003049 EXPECT_TRUE(mark);
3050 EXPECT_EQ(1, FPDFPageObj_CountMarks(text_object));
3051 EXPECT_EQ(mark, FPDFPageObj_GetMark(text_object, 0));
3052 char buffer[256];
Henrique Nakashimac3099d12018-09-18 18:08:15 +00003053 unsigned long name_len = 999u;
3054 ASSERT_TRUE(FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len));
3055 EXPECT_EQ((14u + 1u) * 2, name_len);
Henrique Nakashima144107d2018-07-10 21:04:05 +00003056 std::wstring name =
3057 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer));
Henrique Nakashima6fc8d872018-09-18 16:48:31 +00003058 EXPECT_EQ(L"Test Mark Name", name);
Henrique Nakashima144107d2018-07-10 21:04:05 +00003059
3060 // Add parameters:
3061 // - int "IntKey" : 42
3062 // - string "StringKey": "StringValue"
Henrique Nakashima07520f62018-07-12 19:45:29 +00003063 // - blob "BlobKey": "\x01\x02\x03\0BlobValue1\0\0\0BlobValue2\0"
Lei Zhang590f4242019-05-15 20:57:26 +00003064 constexpr size_t kBlobLen = 28;
Lei Zhangd3b028b2018-11-30 22:22:00 +00003065 char block_value[kBlobLen];
3066 memcpy(block_value, "\x01\x02\x03\0BlobValue1\0\0\0BlobValue2\0", kBlobLen);
Henrique Nakashima144107d2018-07-10 21:04:05 +00003067 EXPECT_EQ(0, FPDFPageObjMark_CountParams(mark));
Henrique Nakashimaa3406772018-07-13 19:10:53 +00003068 EXPECT_TRUE(
3069 FPDFPageObjMark_SetIntParam(document(), text_object, mark, "IntKey", 42));
3070 EXPECT_TRUE(FPDFPageObjMark_SetStringParam(document(), text_object, mark,
3071 "StringKey", "StringValue"));
3072 EXPECT_TRUE(FPDFPageObjMark_SetBlobParam(document(), text_object, mark,
Lei Zhangd3b028b2018-11-30 22:22:00 +00003073 "BlobKey", block_value, kBlobLen));
Henrique Nakashima07520f62018-07-12 19:45:29 +00003074 EXPECT_EQ(3, FPDFPageObjMark_CountParams(mark));
Henrique Nakashima144107d2018-07-10 21:04:05 +00003075
3076 // Check the two parameters can be retrieved.
3077 EXPECT_EQ(FPDF_OBJECT_NUMBER,
Henrique Nakashima94230e52018-07-11 22:02:02 +00003078 FPDFPageObjMark_GetParamValueType(mark, "IntKey"));
Henrique Nakashima144107d2018-07-10 21:04:05 +00003079 int int_value;
Henrique Nakashima94230e52018-07-11 22:02:02 +00003080 EXPECT_TRUE(FPDFPageObjMark_GetParamIntValue(mark, "IntKey", &int_value));
Henrique Nakashima144107d2018-07-10 21:04:05 +00003081 EXPECT_EQ(42, int_value);
3082
3083 EXPECT_EQ(FPDF_OBJECT_STRING,
Henrique Nakashima94230e52018-07-11 22:02:02 +00003084 FPDFPageObjMark_GetParamValueType(mark, "StringKey"));
Henrique Nakashimac3099d12018-09-18 18:08:15 +00003085 unsigned long out_buffer_len = 999u;
Henrique Nakashimaa3406772018-07-13 19:10:53 +00003086 EXPECT_TRUE(FPDFPageObjMark_GetParamStringValue(
3087 mark, "StringKey", buffer, sizeof(buffer), &out_buffer_len));
Henrique Nakashima144107d2018-07-10 21:04:05 +00003088 EXPECT_GT(out_buffer_len, 0u);
Henrique Nakashimac3099d12018-09-18 18:08:15 +00003089 EXPECT_NE(999u, out_buffer_len);
Henrique Nakashima144107d2018-07-10 21:04:05 +00003090 name = GetPlatformWString(reinterpret_cast<unsigned short*>(buffer));
3091 EXPECT_EQ(L"StringValue", name);
3092
Henrique Nakashima07520f62018-07-12 19:45:29 +00003093 EXPECT_EQ(FPDF_OBJECT_STRING,
3094 FPDFPageObjMark_GetParamValueType(mark, "BlobKey"));
3095 out_buffer_len = 0;
Henrique Nakashimaa3406772018-07-13 19:10:53 +00003096 EXPECT_TRUE(FPDFPageObjMark_GetParamBlobValue(
3097 mark, "BlobKey", buffer, sizeof(buffer), &out_buffer_len));
Henrique Nakashima07520f62018-07-12 19:45:29 +00003098 EXPECT_EQ(kBlobLen, out_buffer_len);
Lei Zhangd3b028b2018-11-30 22:22:00 +00003099 EXPECT_EQ(0, memcmp(block_value, buffer, kBlobLen));
Henrique Nakashima07520f62018-07-12 19:45:29 +00003100
Lei Zhangf16376e2020-06-18 19:19:32 +00003101 // Render and check the bitmap is the expected one.
Henrique Nakashima144107d2018-07-10 21:04:05 +00003102 {
Lei Zhang30ff2532019-01-31 21:37:55 +00003103 ScopedFPDFBitmap page_bitmap = RenderPage(page);
Hui Yingst3b6136a2020-06-09 00:39:33 +00003104 CompareBitmap(page_bitmap.get(), 612, 792, kLoadedFontTextChecksum);
Henrique Nakashima144107d2018-07-10 21:04:05 +00003105 }
3106
Henrique Nakashimab4bcf692018-07-11 21:19:22 +00003107 // Now save the result.
3108 EXPECT_EQ(1, FPDFPage_CountObjects(page));
3109 EXPECT_TRUE(FPDFPage_GenerateContent(page));
3110 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
3111
Henrique Nakashima144107d2018-07-10 21:04:05 +00003112 FPDF_ClosePage(page);
3113
Henrique Nakashimab4bcf692018-07-11 21:19:22 +00003114 // Re-open the file and check the changes were kept in the saved .pdf.
Lei Zhang0b494052019-01-31 21:41:15 +00003115 ASSERT_TRUE(OpenSavedDocument());
Henrique Nakashimab4bcf692018-07-11 21:19:22 +00003116 FPDF_PAGE saved_page = LoadSavedPage(0);
3117 EXPECT_EQ(1, FPDFPage_CountObjects(saved_page));
3118
3119 text_object = FPDFPage_GetObject(saved_page, 0);
3120 EXPECT_TRUE(text_object);
3121 EXPECT_EQ(1, FPDFPageObj_CountMarks(text_object));
3122 mark = FPDFPageObj_GetMark(text_object, 0);
3123 EXPECT_TRUE(mark);
Henrique Nakashimac3099d12018-09-18 18:08:15 +00003124
3125 name_len = 999u;
3126 ASSERT_TRUE(FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len));
3127 EXPECT_EQ((14u + 1u) * 2, name_len);
Henrique Nakashimab4bcf692018-07-11 21:19:22 +00003128 name = GetPlatformWString(reinterpret_cast<unsigned short*>(buffer));
Henrique Nakashima6fc8d872018-09-18 16:48:31 +00003129 EXPECT_EQ(L"Test Mark Name", name);
Henrique Nakashimab4bcf692018-07-11 21:19:22 +00003130
3131 CloseSavedPage(saved_page);
3132 CloseSavedDocument();
Henrique Nakashima144107d2018-07-10 21:04:05 +00003133}
3134
Lei Zhangab41f252018-12-23 03:10:50 +00003135TEST_F(FPDFEditEmbedderTest, MarkGetName) {
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00003136 ASSERT_TRUE(OpenDocument("text_in_page_marked.pdf"));
Henrique Nakashimac3099d12018-09-18 18:08:15 +00003137 FPDF_PAGE page = LoadPage(0);
3138 ASSERT_TRUE(page);
3139 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 18);
3140 FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, 1);
3141 ASSERT_TRUE(mark);
3142
3143 char buffer[256];
3144 unsigned long out_len;
3145
3146 // Show the positive cases of FPDFPageObjMark_GetName.
3147 out_len = 999u;
3148 EXPECT_TRUE(FPDFPageObjMark_GetName(mark, nullptr, 0, &out_len));
3149 EXPECT_EQ((6u + 1u) * 2u, out_len);
3150
3151 out_len = 999u;
3152 EXPECT_TRUE(FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &out_len));
3153 EXPECT_EQ(L"Bounds",
3154 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)));
3155 EXPECT_EQ((6u + 1u) * 2u, out_len);
3156
3157 // Show the negative cases of FPDFPageObjMark_GetName.
3158 out_len = 999u;
3159 EXPECT_FALSE(
3160 FPDFPageObjMark_GetName(nullptr, buffer, sizeof(buffer), &out_len));
3161 EXPECT_EQ(999u, out_len);
3162
3163 EXPECT_FALSE(FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), nullptr));
3164
3165 UnloadPage(page);
3166}
3167
Lei Zhangab41f252018-12-23 03:10:50 +00003168TEST_F(FPDFEditEmbedderTest, MarkGetParamKey) {
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00003169 ASSERT_TRUE(OpenDocument("text_in_page_marked.pdf"));
Henrique Nakashimac3099d12018-09-18 18:08:15 +00003170 FPDF_PAGE page = LoadPage(0);
3171 ASSERT_TRUE(page);
3172 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 18);
3173 FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, 1);
3174 ASSERT_TRUE(mark);
3175
3176 char buffer[256];
3177 unsigned long out_len;
3178
3179 // Show the positive cases of FPDFPageObjMark_GetParamKey.
3180 out_len = 999u;
3181 EXPECT_TRUE(FPDFPageObjMark_GetParamKey(mark, 0, nullptr, 0, &out_len));
3182 EXPECT_EQ((8u + 1u) * 2u, out_len);
3183
3184 out_len = 999u;
3185 EXPECT_TRUE(
3186 FPDFPageObjMark_GetParamKey(mark, 0, buffer, sizeof(buffer), &out_len));
3187 EXPECT_EQ(L"Position",
3188 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)));
3189 EXPECT_EQ((8u + 1u) * 2u, out_len);
3190
3191 // Show the negative cases of FPDFPageObjMark_GetParamKey.
3192 out_len = 999u;
3193 EXPECT_FALSE(FPDFPageObjMark_GetParamKey(nullptr, 0, buffer, sizeof(buffer),
3194 &out_len));
3195 EXPECT_EQ(999u, out_len);
3196
3197 out_len = 999u;
3198 EXPECT_FALSE(
3199 FPDFPageObjMark_GetParamKey(mark, 1, buffer, sizeof(buffer), &out_len));
3200 EXPECT_EQ(999u, out_len);
3201
3202 EXPECT_FALSE(
3203 FPDFPageObjMark_GetParamKey(mark, 0, buffer, sizeof(buffer), nullptr));
3204
3205 UnloadPage(page);
3206}
3207
Lei Zhangab41f252018-12-23 03:10:50 +00003208TEST_F(FPDFEditEmbedderTest, MarkGetIntParam) {
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00003209 ASSERT_TRUE(OpenDocument("text_in_page_marked.pdf"));
Henrique Nakashimac3099d12018-09-18 18:08:15 +00003210 FPDF_PAGE page = LoadPage(0);
3211 ASSERT_TRUE(page);
3212 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 8);
3213 FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, 0);
3214 ASSERT_TRUE(mark);
3215
3216 int out_value;
3217
3218 // Show the positive cases of FPDFPageObjMark_GetParamIntValue.
3219 out_value = 999;
3220 EXPECT_TRUE(FPDFPageObjMark_GetParamIntValue(mark, "Factor", &out_value));
3221 EXPECT_EQ(3, out_value);
3222
3223 // Show the negative cases of FPDFPageObjMark_GetParamIntValue.
3224 out_value = 999;
3225 EXPECT_FALSE(FPDFPageObjMark_GetParamIntValue(nullptr, "Factor", &out_value));
3226 EXPECT_EQ(999, out_value);
3227
3228 out_value = 999;
3229 EXPECT_FALSE(FPDFPageObjMark_GetParamIntValue(mark, "ParamThatDoesNotExist",
3230 &out_value));
3231 EXPECT_EQ(999, out_value);
3232
3233 EXPECT_FALSE(FPDFPageObjMark_GetParamIntValue(mark, "Factor", nullptr));
3234
3235 page_object = FPDFPage_GetObject(page, 18);
3236 mark = FPDFPageObj_GetMark(page_object, 1);
3237 out_value = 999;
3238 EXPECT_FALSE(FPDFPageObjMark_GetParamIntValue(mark, "Position", &out_value));
3239 EXPECT_EQ(999, out_value);
3240
3241 UnloadPage(page);
3242}
3243
Lei Zhangab41f252018-12-23 03:10:50 +00003244TEST_F(FPDFEditEmbedderTest, MarkGetStringParam) {
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00003245 ASSERT_TRUE(OpenDocument("text_in_page_marked.pdf"));
Henrique Nakashimac3099d12018-09-18 18:08:15 +00003246 FPDF_PAGE page = LoadPage(0);
3247 ASSERT_TRUE(page);
3248 FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 18);
3249 FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, 1);
3250 ASSERT_TRUE(mark);
3251
3252 char buffer[256];
3253 unsigned long out_len;
3254
3255 // Show the positive cases of FPDFPageObjMark_GetParamStringValue.
3256 out_len = 999u;
3257 EXPECT_TRUE(FPDFPageObjMark_GetParamStringValue(mark, "Position", nullptr, 0,
3258 &out_len));
3259 EXPECT_EQ((4u + 1u) * 2u, out_len);
3260
3261 out_len = 999u;
3262 EXPECT_TRUE(FPDFPageObjMark_GetParamStringValue(mark, "Position", buffer,
3263 sizeof(buffer), &out_len));
3264 EXPECT_EQ(L"Last",
3265 GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)));
3266 EXPECT_EQ((4u + 1u) * 2u, out_len);
3267
3268 // Show the negative cases of FPDFPageObjMark_GetParamStringValue.
3269 out_len = 999u;
3270 EXPECT_FALSE(FPDFPageObjMark_GetParamStringValue(nullptr, "Position", buffer,
3271 sizeof(buffer), &out_len));
3272 EXPECT_EQ(999u, out_len);
3273
3274 out_len = 999u;
3275 EXPECT_FALSE(FPDFPageObjMark_GetParamStringValue(
3276 mark, "ParamThatDoesNotExist", buffer, sizeof(buffer), &out_len));
3277 EXPECT_EQ(999u, out_len);
3278
3279 EXPECT_FALSE(FPDFPageObjMark_GetParamStringValue(mark, "Position", buffer,
3280 sizeof(buffer), nullptr));
3281
3282 page_object = FPDFPage_GetObject(page, 8);
3283 mark = FPDFPageObj_GetMark(page_object, 0);
3284 out_len = 999u;
3285 EXPECT_FALSE(FPDFPageObjMark_GetParamStringValue(mark, "Factor", buffer,
3286 sizeof(buffer), &out_len));
3287 EXPECT_EQ(999u, out_len);
3288
3289 UnloadPage(page);
3290}
3291
Lei Zhang0ac7b4e2020-07-18 00:35:53 +00003292TEST_F(FPDFEditEmbedderTest, GetBitmap) {
Jane Liu28fb7ba2017-08-02 21:45:57 -04003293 ASSERT_TRUE(OpenDocument("embedded_images.pdf"));
3294 FPDF_PAGE page = LoadPage(0);
3295 ASSERT_TRUE(page);
Miklos Vajna92627612017-09-25 12:59:29 +02003296 ASSERT_EQ(39, FPDFPage_CountObjects(page));
Jane Liu28fb7ba2017-08-02 21:45:57 -04003297
3298 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 32);
3299 EXPECT_NE(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3300 EXPECT_FALSE(FPDFImageObj_GetBitmap(obj));
3301
Lei Zhang89ebf5c2020-07-14 20:59:05 +00003302 {
3303 obj = FPDFPage_GetObject(page, 33);
3304 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3305 ScopedFPDFBitmap bitmap(FPDFImageObj_GetBitmap(obj));
3306 EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap.get()));
Lei Zhang85166512020-07-14 23:28:56 +00003307 CompareBitmap(bitmap.get(), 109, 88, kEmbeddedImage33Checksum);
Lei Zhang89ebf5c2020-07-14 20:59:05 +00003308 }
Jane Liu28fb7ba2017-08-02 21:45:57 -04003309
Lei Zhang89ebf5c2020-07-14 20:59:05 +00003310 {
3311 obj = FPDFPage_GetObject(page, 34);
3312 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3313 ScopedFPDFBitmap bitmap(FPDFImageObj_GetBitmap(obj));
3314 EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap.get()));
3315 CompareBitmap(bitmap.get(), 103, 75, "c8d51fa6821ceb2a67f08446ff236c40");
3316 }
Jane Liu28fb7ba2017-08-02 21:45:57 -04003317
Lei Zhang89ebf5c2020-07-14 20:59:05 +00003318 {
3319 obj = FPDFPage_GetObject(page, 35);
3320 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3321 ScopedFPDFBitmap bitmap(FPDFImageObj_GetBitmap(obj));
3322 EXPECT_EQ(FPDFBitmap_Gray, FPDFBitmap_GetFormat(bitmap.get()));
3323 CompareBitmap(bitmap.get(), 92, 68, "9c6d76cb1e37ef8514f9455d759391f3");
3324 }
Jane Liu28fb7ba2017-08-02 21:45:57 -04003325
Lei Zhang89ebf5c2020-07-14 20:59:05 +00003326 {
3327 obj = FPDFPage_GetObject(page, 36);
3328 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3329 ScopedFPDFBitmap bitmap(FPDFImageObj_GetBitmap(obj));
3330 EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap.get()));
3331 CompareBitmap(bitmap.get(), 79, 60, "f4e72fb783a01c7b4614cdc25eaa98ac");
3332 }
Jane Liu28fb7ba2017-08-02 21:45:57 -04003333
Lei Zhang89ebf5c2020-07-14 20:59:05 +00003334 {
3335 obj = FPDFPage_GetObject(page, 37);
3336 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3337 ScopedFPDFBitmap bitmap(FPDFImageObj_GetBitmap(obj));
3338 EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap.get()));
3339 CompareBitmap(bitmap.get(), 126, 106, "2cf9e66414c72461f4ccbf9cdebdfa1b");
3340 }
Jane Liu28fb7ba2017-08-02 21:45:57 -04003341
Lei Zhang89ebf5c2020-07-14 20:59:05 +00003342 {
3343 obj = FPDFPage_GetObject(page, 38);
3344 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3345 ScopedFPDFBitmap bitmap(FPDFImageObj_GetBitmap(obj));
3346 EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap.get()));
3347 CompareBitmap(bitmap.get(), 194, 119, "a8f3a126cec274dab8242fd2ccdc1b8b");
3348 }
3349
Jane Liu28fb7ba2017-08-02 21:45:57 -04003350 UnloadPage(page);
3351}
Jane Liu548334e2017-08-03 16:33:40 -04003352
Lei Zhang0ac7b4e2020-07-18 00:35:53 +00003353TEST_F(FPDFEditEmbedderTest, GetBitmapIgnoresSetMatrix) {
Lei Zhang85166512020-07-14 23:28:56 +00003354 ASSERT_TRUE(OpenDocument("embedded_images.pdf"));
3355 FPDF_PAGE page = LoadPage(0);
3356 ASSERT_TRUE(page);
3357 ASSERT_EQ(39, FPDFPage_CountObjects(page));
3358
3359 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 33);
3360 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3361
3362 {
3363 // Render |obj| as is.
3364 ScopedFPDFBitmap bitmap(FPDFImageObj_GetBitmap(obj));
3365 EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap.get()));
3366 CompareBitmap(bitmap.get(), 109, 88, kEmbeddedImage33Checksum);
3367 }
3368
3369 // Check the matrix for |obj|.
3370 double a;
3371 double b;
3372 double c;
3373 double d;
3374 double e;
3375 double f;
3376 EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f));
3377 EXPECT_DOUBLE_EQ(53.0, a);
3378 EXPECT_DOUBLE_EQ(0.0, b);
3379 EXPECT_DOUBLE_EQ(0.0, c);
3380 EXPECT_DOUBLE_EQ(43.0, d);
3381 EXPECT_DOUBLE_EQ(72.0, e);
3382 EXPECT_DOUBLE_EQ(646.510009765625, f);
3383
3384 // Modify the matrix for |obj|.
3385 a = 120.0;
3386 EXPECT_TRUE(FPDFImageObj_SetMatrix(obj, a, b, c, d, e, f));
3387
3388 // Make sure the matrix modification took place.
3389 EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f));
3390 EXPECT_DOUBLE_EQ(120.0, a);
3391 EXPECT_DOUBLE_EQ(0.0, b);
3392 EXPECT_DOUBLE_EQ(0.0, c);
3393 EXPECT_DOUBLE_EQ(43.0, d);
3394 EXPECT_DOUBLE_EQ(72.0, e);
3395 EXPECT_DOUBLE_EQ(646.510009765625, f);
3396
3397 {
3398 // Render |obj| again. Note that the FPDFImageObj_SetMatrix() call has no
3399 // effect.
3400 ScopedFPDFBitmap bitmap(FPDFImageObj_GetBitmap(obj));
3401 EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap.get()));
3402 CompareBitmap(bitmap.get(), 109, 88, kEmbeddedImage33Checksum);
3403 }
3404
3405 UnloadPage(page);
3406}
3407
Lei Zhang0ac7b4e2020-07-18 00:35:53 +00003408TEST_F(FPDFEditEmbedderTest, GetBitmapForJBigImage) {
Lei Zhang53341dd2018-03-01 15:42:47 +00003409 ASSERT_TRUE(OpenDocument("bug_631912.pdf"));
3410 FPDF_PAGE page = LoadPage(0);
3411 ASSERT_TRUE(page);
3412 ASSERT_EQ(1, FPDFPage_CountObjects(page));
3413
3414 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 0);
3415 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3416 {
Tom Sepeze08d2b12018-04-25 18:49:32 +00003417 ScopedFPDFBitmap bitmap(FPDFImageObj_GetBitmap(obj));
Lei Zhang1330ebb2018-03-05 15:16:37 +00003418 ASSERT_TRUE(bitmap);
3419 EXPECT_EQ(FPDFBitmap_Gray, FPDFBitmap_GetFormat(bitmap.get()));
3420 CompareBitmap(bitmap.get(), 1152, 720, "3f6a48e2b3e91b799bf34567f55cb4de");
Lei Zhang53341dd2018-03-01 15:42:47 +00003421 }
3422
3423 UnloadPage(page);
3424}
3425
Lei Zhang0ac7b4e2020-07-18 00:35:53 +00003426TEST_F(FPDFEditEmbedderTest, GetBitmapIgnoresSMask) {
Lei Zhang2e1b4f22020-07-08 23:40:18 +00003427 ASSERT_TRUE(OpenDocument("matte.pdf"));
3428 FPDF_PAGE page = LoadPage(0);
3429 ASSERT_TRUE(page);
3430
3431 constexpr int kExpectedObjects = 4;
3432 ASSERT_EQ(kExpectedObjects, FPDFPage_CountObjects(page));
3433
3434 for (int i = 0; i < kExpectedObjects; ++i) {
3435 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, i);
3436 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3437 ScopedFPDFBitmap bitmap(FPDFImageObj_GetBitmap(obj));
3438 ASSERT_TRUE(bitmap);
3439 EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap.get()));
3440 CompareBitmap(bitmap.get(), 50, 50, "46c9a1dbe0b44765ce46017ad629a2fe");
3441 }
3442
3443 UnloadPage(page);
3444}
3445
Hui Yingst822fb8d2020-10-05 22:10:35 +00003446// TODO(crbug.com/pdfium/11): Fix this test and enable.
3447#if defined(_SKIA_SUPPORT_)
3448#define MAYBE_GetRenderedBitmapHandlesSetMatrix \
3449 DISABLED_GetRenderedBitmapHandlesSetMatrix
3450#else
3451#define MAYBE_GetRenderedBitmapHandlesSetMatrix \
3452 GetRenderedBitmapHandlesSetMatrix
3453#endif
3454TEST_F(FPDFEditEmbedderTest, MAYBE_GetRenderedBitmapHandlesSetMatrix) {
Lei Zhang0ac7b4e2020-07-18 00:35:53 +00003455 ASSERT_TRUE(OpenDocument("embedded_images.pdf"));
3456 FPDF_PAGE page = LoadPage(0);
3457 ASSERT_TRUE(page);
3458 ASSERT_EQ(39, FPDFPage_CountObjects(page));
3459
3460 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 33);
3461 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3462
3463 {
3464 // Render |obj| as is.
3465 ScopedFPDFBitmap bitmap(
3466 FPDFImageObj_GetRenderedBitmap(document(), page, obj));
3467 EXPECT_EQ(FPDFBitmap_BGRA, FPDFBitmap_GetFormat(bitmap.get()));
3468 CompareBitmap(bitmap.get(), 53, 43, "90fa16c2fb2bf8ad3654c2258417664c");
3469 }
3470
3471 // Check the matrix for |obj|.
3472 double a;
3473 double b;
3474 double c;
3475 double d;
3476 double e;
3477 double f;
3478 EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f));
3479 EXPECT_DOUBLE_EQ(53.0, a);
3480 EXPECT_DOUBLE_EQ(0.0, b);
3481 EXPECT_DOUBLE_EQ(0.0, c);
3482 EXPECT_DOUBLE_EQ(43.0, d);
3483 EXPECT_DOUBLE_EQ(72.0, e);
3484 EXPECT_DOUBLE_EQ(646.510009765625, f);
3485
3486 // Modify the matrix for |obj|.
3487 a = 120.0;
3488 EXPECT_TRUE(FPDFImageObj_SetMatrix(obj, a, b, c, d, e, f));
3489
3490 // Make sure the matrix modification took place.
3491 EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f));
3492 EXPECT_DOUBLE_EQ(120.0, a);
3493 EXPECT_DOUBLE_EQ(0.0, b);
3494 EXPECT_DOUBLE_EQ(0.0, c);
3495 EXPECT_DOUBLE_EQ(43.0, d);
3496 EXPECT_DOUBLE_EQ(72.0, e);
3497 EXPECT_DOUBLE_EQ(646.510009765625, f);
3498
3499 {
3500 // Render |obj| again. Note that the FPDFImageObj_SetMatrix() call has an
3501 // effect.
3502 ScopedFPDFBitmap bitmap(
3503 FPDFImageObj_GetRenderedBitmap(document(), page, obj));
3504 EXPECT_EQ(FPDFBitmap_BGRA, FPDFBitmap_GetFormat(bitmap.get()));
3505 CompareBitmap(bitmap.get(), 120, 43, "57ed8e15daa535490ff0c8b7640a36b4");
3506 }
3507
3508 UnloadPage(page);
3509}
3510
Hui Yingst822fb8d2020-10-05 22:10:35 +00003511// TODO(crbug.com/pdfium/11): Fix this test and enable.
3512#if defined(_SKIA_SUPPORT_)
3513#define MAYBE_GetRenderedBitmapHandlesSMask \
3514 DISABLED_GetRenderedBitmapHandlesSMask
3515#else
3516#define MAYBE_GetRenderedBitmapHandlesSMask GetRenderedBitmapHandlesSMask
3517#endif
3518TEST_F(FPDFEditEmbedderTest, MAYBE_GetRenderedBitmapHandlesSMask) {
Lei Zhang0ac7b4e2020-07-18 00:35:53 +00003519 ASSERT_TRUE(OpenDocument("matte.pdf"));
3520 FPDF_PAGE page = LoadPage(0);
3521 ASSERT_TRUE(page);
3522
3523 constexpr int kExpectedObjects = 4;
3524 ASSERT_EQ(kExpectedObjects, FPDFPage_CountObjects(page));
3525
3526 for (int i = 0; i < kExpectedObjects; ++i) {
3527 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, i);
3528 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3529 ScopedFPDFBitmap bitmap(
3530 FPDFImageObj_GetRenderedBitmap(document(), page, obj));
3531 ASSERT_TRUE(bitmap);
3532 EXPECT_EQ(FPDFBitmap_BGRA, FPDFBitmap_GetFormat(bitmap.get()));
3533 if (i == 0)
3534 CompareBitmap(bitmap.get(), 40, 60, "5a3ae4a660ce919e29c42ec2258142f1");
3535 else
3536 CompareBitmap(bitmap.get(), 40, 60, "67504e83f5d78214ea00efc19082c5c1");
3537 }
3538
3539 UnloadPage(page);
3540}
3541
3542TEST_F(FPDFEditEmbedderTest, GetRenderedBitmapBadParams) {
3543 ASSERT_TRUE(OpenDocument("embedded_images.pdf"));
3544 FPDF_PAGE page = LoadPage(0);
3545 ASSERT_TRUE(page);
3546
3547 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 33);
3548 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3549
3550 // Test various null parameters.
3551 EXPECT_FALSE(FPDFImageObj_GetRenderedBitmap(nullptr, nullptr, nullptr));
3552 EXPECT_FALSE(FPDFImageObj_GetRenderedBitmap(document(), nullptr, nullptr));
3553 EXPECT_FALSE(FPDFImageObj_GetRenderedBitmap(nullptr, page, nullptr));
3554 EXPECT_FALSE(FPDFImageObj_GetRenderedBitmap(nullptr, nullptr, obj));
3555 EXPECT_FALSE(FPDFImageObj_GetRenderedBitmap(document(), page, nullptr));
3556 EXPECT_FALSE(FPDFImageObj_GetRenderedBitmap(nullptr, page, obj));
3557
3558 // Test mismatch between document and page parameters.
3559 ScopedFPDFDocument new_document(FPDF_CreateNewDocument());
3560 EXPECT_FALSE(FPDFImageObj_GetRenderedBitmap(new_document.get(), page, obj));
3561
3562 UnloadPage(page);
3563}
3564
Lei Zhangab41f252018-12-23 03:10:50 +00003565TEST_F(FPDFEditEmbedderTest, GetImageData) {
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00003566 ASSERT_TRUE(OpenDocument("embedded_images.pdf"));
Jane Liu548334e2017-08-03 16:33:40 -04003567 FPDF_PAGE page = LoadPage(0);
3568 ASSERT_TRUE(page);
Miklos Vajna92627612017-09-25 12:59:29 +02003569 ASSERT_EQ(39, FPDFPage_CountObjects(page));
Jane Liu548334e2017-08-03 16:33:40 -04003570
3571 // Retrieve an image object with flate-encoded data stream.
3572 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 33);
3573 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3574
3575 // Check that the raw image data has the correct length and hash value.
3576 unsigned long len = FPDFImageObj_GetImageDataRaw(obj, nullptr, 0);
3577 std::vector<char> buf(len);
3578 EXPECT_EQ(4091u, FPDFImageObj_GetImageDataRaw(obj, buf.data(), len));
3579 EXPECT_EQ("f73802327d2e88e890f653961bcda81a",
3580 GenerateMD5Base16(reinterpret_cast<uint8_t*>(buf.data()), len));
3581
3582 // Check that the decoded image data has the correct length and hash value.
3583 len = FPDFImageObj_GetImageDataDecoded(obj, nullptr, 0);
3584 buf.clear();
3585 buf.resize(len);
3586 EXPECT_EQ(28776u, FPDFImageObj_GetImageDataDecoded(obj, buf.data(), len));
Lei Zhang85166512020-07-14 23:28:56 +00003587 EXPECT_EQ(kEmbeddedImage33Checksum,
Jane Liu548334e2017-08-03 16:33:40 -04003588 GenerateMD5Base16(reinterpret_cast<uint8_t*>(buf.data()), len));
3589
Lei Zhang351e8b02018-12-20 01:10:06 +00003590 // Retrieve an image object with DCTDecode-encoded data stream.
Jane Liu548334e2017-08-03 16:33:40 -04003591 obj = FPDFPage_GetObject(page, 37);
3592 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3593
3594 // Check that the raw image data has the correct length and hash value.
3595 len = FPDFImageObj_GetImageDataRaw(obj, nullptr, 0);
3596 buf.clear();
3597 buf.resize(len);
3598 EXPECT_EQ(4370u, FPDFImageObj_GetImageDataRaw(obj, buf.data(), len));
3599 EXPECT_EQ("6aae1f3710335023a9e12191be66b64b",
3600 GenerateMD5Base16(reinterpret_cast<uint8_t*>(buf.data()), len));
3601
3602 // Check that the decoded image data has the correct length and hash value,
3603 // which should be the same as those of the raw data, since this image is
3604 // encoded by a single DCTDecode filter and decoding is a noop.
3605 len = FPDFImageObj_GetImageDataDecoded(obj, nullptr, 0);
3606 buf.clear();
3607 buf.resize(len);
3608 EXPECT_EQ(4370u, FPDFImageObj_GetImageDataDecoded(obj, buf.data(), len));
3609 EXPECT_EQ("6aae1f3710335023a9e12191be66b64b",
3610 GenerateMD5Base16(reinterpret_cast<uint8_t*>(buf.data()), len));
3611
3612 UnloadPage(page);
3613}
Jane Liu2e5f0ae2017-08-08 15:23:27 -04003614
Lei Zhangab41f252018-12-23 03:10:50 +00003615TEST_F(FPDFEditEmbedderTest, GetImageMatrix) {
Lei Zhang866d6882018-10-24 17:31:01 +00003616 ASSERT_TRUE(OpenDocument("embedded_images.pdf"));
3617 FPDF_PAGE page = LoadPage(0);
3618 ASSERT_TRUE(page);
3619 ASSERT_EQ(39, FPDFPage_CountObjects(page));
3620
3621 FPDF_PAGEOBJECT obj;
3622 double a;
3623 double b;
3624 double c;
3625 double d;
3626 double e;
3627 double f;
3628
3629 obj = FPDFPage_GetObject(page, 33);
3630 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3631 EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f));
3632 EXPECT_DOUBLE_EQ(53.0, a);
3633 EXPECT_DOUBLE_EQ(0.0, b);
3634 EXPECT_DOUBLE_EQ(0.0, c);
3635 EXPECT_DOUBLE_EQ(43.0, d);
3636 EXPECT_DOUBLE_EQ(72.0, e);
3637 EXPECT_DOUBLE_EQ(646.510009765625, f);
3638
3639 obj = FPDFPage_GetObject(page, 34);
3640 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3641 EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f));
3642 EXPECT_DOUBLE_EQ(70.0, a);
3643 EXPECT_DOUBLE_EQ(0.0, b);
3644 EXPECT_DOUBLE_EQ(0.0, c);
3645 EXPECT_DOUBLE_EQ(51.0, d);
3646 EXPECT_DOUBLE_EQ(216.0, e);
3647 EXPECT_DOUBLE_EQ(646.510009765625, f);
3648
3649 obj = FPDFPage_GetObject(page, 35);
3650 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3651 EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f));
3652 EXPECT_DOUBLE_EQ(69.0, a);
3653 EXPECT_DOUBLE_EQ(0.0, b);
3654 EXPECT_DOUBLE_EQ(0.0, c);
3655 EXPECT_DOUBLE_EQ(51.0, d);
3656 EXPECT_DOUBLE_EQ(360.0, e);
3657 EXPECT_DOUBLE_EQ(646.510009765625, f);
3658
3659 obj = FPDFPage_GetObject(page, 36);
3660 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3661 EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f));
3662 EXPECT_DOUBLE_EQ(59.0, a);
3663 EXPECT_DOUBLE_EQ(0.0, b);
3664 EXPECT_DOUBLE_EQ(0.0, c);
3665 EXPECT_DOUBLE_EQ(45.0, d);
3666 EXPECT_DOUBLE_EQ(72.0, e);
3667 EXPECT_DOUBLE_EQ(553.510009765625, f);
3668
3669 obj = FPDFPage_GetObject(page, 37);
3670 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3671 EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f));
3672 EXPECT_DOUBLE_EQ(55.94000244140625, a);
3673 EXPECT_DOUBLE_EQ(0.0, b);
3674 EXPECT_DOUBLE_EQ(0.0, c);
3675 EXPECT_DOUBLE_EQ(46.950000762939453, d);
3676 EXPECT_DOUBLE_EQ(216.0, e);
3677 EXPECT_DOUBLE_EQ(552.510009765625, f);
3678
3679 obj = FPDFPage_GetObject(page, 38);
3680 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3681 EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f));
3682 EXPECT_DOUBLE_EQ(70.528999328613281, a);
3683 EXPECT_DOUBLE_EQ(0.0, b);
3684 EXPECT_DOUBLE_EQ(0.0, c);
3685 EXPECT_DOUBLE_EQ(43.149997711181641, d);
3686 EXPECT_DOUBLE_EQ(360.0, e);
3687 EXPECT_DOUBLE_EQ(553.3599853515625, f);
3688
3689 UnloadPage(page);
3690}
3691
Lei Zhangab41f252018-12-23 03:10:50 +00003692TEST_F(FPDFEditEmbedderTest, DestroyPageObject) {
Jane Liu2e5f0ae2017-08-08 15:23:27 -04003693 FPDF_PAGEOBJECT rect = FPDFPageObj_CreateNewRect(10, 10, 20, 20);
3694 ASSERT_TRUE(rect);
3695
3696 // There should be no memory leaks with a call to FPDFPageObj_Destroy().
3697 FPDFPageObj_Destroy(rect);
3698}
Jane Liube63ab92017-08-09 14:09:34 -04003699
Lei Zhangab41f252018-12-23 03:10:50 +00003700TEST_F(FPDFEditEmbedderTest, GetImageFilters) {
Daniel Hosseinian5af51b62020-07-18 00:53:43 +00003701 ASSERT_TRUE(OpenDocument("embedded_images.pdf"));
Jane Liube63ab92017-08-09 14:09:34 -04003702 FPDF_PAGE page = LoadPage(0);
3703 ASSERT_TRUE(page);
3704
3705 // Verify that retrieving the filter of a non-image object would fail.
3706 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 32);
3707 ASSERT_NE(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3708 ASSERT_EQ(0, FPDFImageObj_GetImageFilterCount(obj));
3709 EXPECT_EQ(0u, FPDFImageObj_GetImageFilter(obj, 0, nullptr, 0));
3710
3711 // Verify the returned filter string for an image object with a single filter.
3712 obj = FPDFPage_GetObject(page, 33);
3713 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3714 ASSERT_EQ(1, FPDFImageObj_GetImageFilterCount(obj));
3715 unsigned long len = FPDFImageObj_GetImageFilter(obj, 0, nullptr, 0);
3716 std::vector<char> buf(len);
Lei Zhang0733a1b2017-08-31 12:36:31 -07003717 static constexpr char kFlateDecode[] = "FlateDecode";
3718 EXPECT_EQ(sizeof(kFlateDecode),
3719 FPDFImageObj_GetImageFilter(obj, 0, buf.data(), len));
3720 EXPECT_STREQ(kFlateDecode, buf.data());
Jane Liube63ab92017-08-09 14:09:34 -04003721 EXPECT_EQ(0u, FPDFImageObj_GetImageFilter(obj, 1, nullptr, 0));
3722
3723 // Verify all the filters for an image object with a list of filters.
3724 obj = FPDFPage_GetObject(page, 38);
3725 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3726 ASSERT_EQ(2, FPDFImageObj_GetImageFilterCount(obj));
3727 len = FPDFImageObj_GetImageFilter(obj, 0, nullptr, 0);
3728 buf.clear();
3729 buf.resize(len);
Lei Zhang0733a1b2017-08-31 12:36:31 -07003730 static constexpr char kASCIIHexDecode[] = "ASCIIHexDecode";
3731 EXPECT_EQ(sizeof(kASCIIHexDecode),
3732 FPDFImageObj_GetImageFilter(obj, 0, buf.data(), len));
3733 EXPECT_STREQ(kASCIIHexDecode, buf.data());
Jane Liube63ab92017-08-09 14:09:34 -04003734
3735 len = FPDFImageObj_GetImageFilter(obj, 1, nullptr, 0);
3736 buf.clear();
3737 buf.resize(len);
Lei Zhang0733a1b2017-08-31 12:36:31 -07003738 static constexpr char kDCTDecode[] = "DCTDecode";
3739 EXPECT_EQ(sizeof(kDCTDecode),
3740 FPDFImageObj_GetImageFilter(obj, 1, buf.data(), len));
3741 EXPECT_STREQ(kDCTDecode, buf.data());
Jane Liube63ab92017-08-09 14:09:34 -04003742
3743 UnloadPage(page);
3744}
Jane Liuca898292017-08-16 11:25:35 -04003745
Lei Zhangab41f252018-12-23 03:10:50 +00003746TEST_F(FPDFEditEmbedderTest, GetImageMetadata) {
Jane Liuca898292017-08-16 11:25:35 -04003747 ASSERT_TRUE(OpenDocument("embedded_images.pdf"));
3748 FPDF_PAGE page = LoadPage(0);
3749 ASSERT_TRUE(page);
3750
3751 // Check that getting the metadata of a null object would fail.
3752 FPDF_IMAGEOBJ_METADATA metadata;
3753 EXPECT_FALSE(FPDFImageObj_GetImageMetadata(nullptr, page, &metadata));
3754
3755 // Check that receiving the metadata with a null metadata object would fail.
3756 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 35);
3757 EXPECT_FALSE(FPDFImageObj_GetImageMetadata(obj, page, nullptr));
3758
3759 // Check that when retrieving an image object's metadata without passing in
3760 // |page|, all values are correct, with the last two being default values.
3761 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3762 ASSERT_TRUE(FPDFImageObj_GetImageMetadata(obj, nullptr, &metadata));
Julian Lungerecd063e2017-12-27 10:18:50 -05003763 EXPECT_EQ(7, metadata.marked_content_id);
Jane Liuca898292017-08-16 11:25:35 -04003764 EXPECT_EQ(92u, metadata.width);
3765 EXPECT_EQ(68u, metadata.height);
Lei Zhang351e8b02018-12-20 01:10:06 +00003766 EXPECT_FLOAT_EQ(96.0f, metadata.horizontal_dpi);
3767 EXPECT_FLOAT_EQ(96.0f, metadata.vertical_dpi);
Jane Liuca898292017-08-16 11:25:35 -04003768 EXPECT_EQ(0u, metadata.bits_per_pixel);
3769 EXPECT_EQ(FPDF_COLORSPACE_UNKNOWN, metadata.colorspace);
3770
3771 // Verify the metadata of a bitmap image with indexed colorspace.
3772 ASSERT_TRUE(FPDFImageObj_GetImageMetadata(obj, page, &metadata));
Julian Lungerecd063e2017-12-27 10:18:50 -05003773 EXPECT_EQ(7, metadata.marked_content_id);
Jane Liuca898292017-08-16 11:25:35 -04003774 EXPECT_EQ(92u, metadata.width);
3775 EXPECT_EQ(68u, metadata.height);
Lei Zhang351e8b02018-12-20 01:10:06 +00003776 EXPECT_FLOAT_EQ(96.0f, metadata.horizontal_dpi);
3777 EXPECT_FLOAT_EQ(96.0f, metadata.vertical_dpi);
Jane Liuca898292017-08-16 11:25:35 -04003778 EXPECT_EQ(1u, metadata.bits_per_pixel);
3779 EXPECT_EQ(FPDF_COLORSPACE_INDEXED, metadata.colorspace);
3780
3781 // Verify the metadata of an image with RGB colorspace.
3782 obj = FPDFPage_GetObject(page, 37);
3783 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3784 ASSERT_TRUE(FPDFImageObj_GetImageMetadata(obj, page, &metadata));
Julian Lungerecd063e2017-12-27 10:18:50 -05003785 EXPECT_EQ(9, metadata.marked_content_id);
Jane Liuca898292017-08-16 11:25:35 -04003786 EXPECT_EQ(126u, metadata.width);
3787 EXPECT_EQ(106u, metadata.height);
Lei Zhang351e8b02018-12-20 01:10:06 +00003788 EXPECT_FLOAT_EQ(162.173752f, metadata.horizontal_dpi);
3789 EXPECT_FLOAT_EQ(162.555878f, metadata.vertical_dpi);
Jane Liuca898292017-08-16 11:25:35 -04003790 EXPECT_EQ(24u, metadata.bits_per_pixel);
3791 EXPECT_EQ(FPDF_COLORSPACE_DEVICERGB, metadata.colorspace);
3792
3793 UnloadPage(page);
3794}
Lei Zhangdc1b7392020-05-14 21:15:53 +00003795
3796TEST_F(FPDFEditEmbedderTest, GetImageMetadataJpxLzw) {
3797 ASSERT_TRUE(OpenDocument("jpx_lzw.pdf"));
3798 FPDF_PAGE page = LoadPage(0);
3799 ASSERT_TRUE(page);
3800
3801 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 0);
3802 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
3803
3804 FPDF_IMAGEOBJ_METADATA metadata;
3805 ASSERT_TRUE(FPDFImageObj_GetImageMetadata(obj, page, &metadata));
3806 EXPECT_EQ(-1, metadata.marked_content_id);
3807 EXPECT_EQ(612u, metadata.width);
3808 EXPECT_EQ(792u, metadata.height);
3809 EXPECT_FLOAT_EQ(72.0f, metadata.horizontal_dpi);
3810 EXPECT_FLOAT_EQ(72.0f, metadata.vertical_dpi);
3811 EXPECT_EQ(24u, metadata.bits_per_pixel);
3812 EXPECT_EQ(FPDF_COLORSPACE_UNKNOWN, metadata.colorspace);
3813
3814 UnloadPage(page);
3815}