blob: 74e85f6795ac30af5e0d6406cb3ae3731abfd85e [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
Nicolas Penabe90aae2017-02-27 10:41:41 -050010#include "core/fpdfapi/font/cpdf_font.h"
Nicolas Penaa4ad01f2017-02-15 16:26:48 -050011#include "core/fpdfapi/page/cpdf_page.h"
Nicolas Penabe90aae2017-02-27 10:41:41 -050012#include "core/fpdfapi/parser/cpdf_array.h"
Nicolas Penaa4ad01f2017-02-15 16:26:48 -050013#include "core/fpdfapi/parser/cpdf_dictionary.h"
Nicolas Penad03ca422017-03-06 13:54:33 -050014#include "core/fpdfapi/parser/cpdf_number.h"
Nicolas Penabe90aae2017-02-27 10:41:41 -050015#include "core/fpdfapi/parser/cpdf_stream.h"
Nicolas Pena0fc185e2017-02-08 12:13:20 -050016#include "core/fxcrt/fx_system.h"
Nicolas Penaa4ad01f2017-02-15 16:26:48 -050017#include "fpdfsdk/fsdk_define.h"
Nicolas Penab3161852017-05-02 14:12:50 -040018#include "public/cpp/fpdf_deleters.h"
Jane Liueda65252017-06-07 11:31:27 -040019#include "public/fpdf_annot.h"
Tom Sepezd483eb42016-01-06 10:03:59 -080020#include "public/fpdf_edit.h"
21#include "public/fpdfview.h"
22#include "testing/embedder_test.h"
Tom Sepez0aec19b2016-01-07 12:22:44 -080023#include "testing/gmock/include/gmock/gmock-matchers.h"
Tom Sepezd483eb42016-01-06 10:03:59 -080024#include "testing/gtest/include/gtest/gtest.h"
Tom Sepez0aec19b2016-01-07 12:22:44 -080025#include "testing/test_support.h"
Tom Sepezd483eb42016-01-06 10:03:59 -080026
Nicolas Pena3ff54002017-07-05 11:55:35 -040027class FPDFEditEmbeddertest : public EmbedderTest {
Nicolas Penad03ca422017-03-06 13:54:33 -050028 protected:
29 FPDF_DOCUMENT CreateNewDocument() {
30 document_ = FPDF_CreateNewDocument();
Tom Sepez41066c12017-05-18 09:28:49 -070031 cpdf_doc_ = CPDFDocumentFromFPDFDocument(document_);
Nicolas Penad03ca422017-03-06 13:54:33 -050032 return document_;
33 }
34
35 void CheckFontDescriptor(CPDF_Dictionary* font_dict,
36 int font_type,
37 bool bold,
38 bool italic,
39 uint32_t size,
40 const uint8_t* data) {
41 CPDF_Dictionary* font_desc = font_dict->GetDictFor("FontDescriptor");
42 ASSERT_TRUE(font_desc);
43 EXPECT_EQ("FontDescriptor", font_desc->GetStringFor("Type"));
44 EXPECT_EQ(font_dict->GetStringFor("BaseFont"),
45 font_desc->GetStringFor("FontName"));
46
47 // Check that the font descriptor has the required keys according to spec
48 // 1.7 Table 5.19
49 ASSERT_TRUE(font_desc->KeyExist("Flags"));
Dan Sinclair10e1f052017-09-28 15:59:42 -040050
Nicolas Penad03ca422017-03-06 13:54:33 -050051 int font_flags = font_desc->GetIntegerFor("Flags");
Dan Sinclair10e1f052017-09-28 15:59:42 -040052 EXPECT_EQ(bold, FontStyleIsBold(font_flags));
53 EXPECT_EQ(italic, FontStyleIsItalic(font_flags));
54 EXPECT_TRUE(FontStyleIsNonSymbolic(font_flags));
Nicolas Penad03ca422017-03-06 13:54:33 -050055 ASSERT_TRUE(font_desc->KeyExist("FontBBox"));
Nicolás Peña5f95f362017-09-28 13:00:45 +090056
57 CPDF_Array* fontBBox = font_desc->GetArrayFor("FontBBox");
58 ASSERT_TRUE(fontBBox);
59 EXPECT_EQ(4U, fontBBox->GetCount());
60 // Check that the coordinates are in the preferred order according to spec
61 // 1.7 Section 3.8.4
62 EXPECT_TRUE(fontBBox->GetIntegerAt(0) < fontBBox->GetIntegerAt(2));
63 EXPECT_TRUE(fontBBox->GetIntegerAt(1) < fontBBox->GetIntegerAt(3));
64
Nicolas Penad03ca422017-03-06 13:54:33 -050065 EXPECT_TRUE(font_desc->KeyExist("ItalicAngle"));
66 EXPECT_TRUE(font_desc->KeyExist("Ascent"));
67 EXPECT_TRUE(font_desc->KeyExist("Descent"));
68 EXPECT_TRUE(font_desc->KeyExist("CapHeight"));
69 EXPECT_TRUE(font_desc->KeyExist("StemV"));
Ryan Harrison275e2602017-09-18 14:23:18 -040070 ByteString present("FontFile");
71 ByteString absent("FontFile2");
Nicolas Penad03ca422017-03-06 13:54:33 -050072 if (font_type == FPDF_FONT_TRUETYPE)
73 std::swap(present, absent);
74 EXPECT_TRUE(font_desc->KeyExist(present));
75 EXPECT_FALSE(font_desc->KeyExist(absent));
76
77 // Check that the font stream is the one that was provided
78 CPDF_Stream* font_stream = font_desc->GetStreamFor(present);
79 ASSERT_EQ(size, font_stream->GetRawSize());
Nicolás Peña79eab232017-09-28 13:29:05 +090080 if (font_type == FPDF_FONT_TRUETYPE) {
81 ASSERT_EQ(static_cast<int>(size),
82 font_stream->GetDict()->GetIntegerFor("Length1"));
83 }
Nicolas Penad03ca422017-03-06 13:54:33 -050084 uint8_t* stream_data = font_stream->GetRawData();
85 for (size_t j = 0; j < size; j++)
86 EXPECT_EQ(data[j], stream_data[j]) << " at byte " << j;
87 }
88
89 void CheckCompositeFontWidths(CPDF_Array* widths_array,
90 CPDF_Font* typed_font) {
91 // Check that W array is in a format that conforms to PDF spec 1.7 section
92 // "Glyph Metrics in CIDFonts" (these checks are not
93 // implementation-specific).
94 EXPECT_GT(widths_array->GetCount(), 1U);
95 int num_cids_checked = 0;
96 int cur_cid = 0;
97 for (size_t idx = 0; idx < widths_array->GetCount(); idx++) {
98 int cid = widths_array->GetNumberAt(idx);
99 EXPECT_GE(cid, cur_cid);
100 ASSERT_FALSE(++idx == widths_array->GetCount());
101 CPDF_Object* next = widths_array->GetObjectAt(idx);
102 if (next->IsArray()) {
103 // We are in the c [w1 w2 ...] case
104 CPDF_Array* arr = next->AsArray();
105 int cnt = static_cast<int>(arr->GetCount());
106 size_t inner_idx = 0;
107 for (cur_cid = cid; cur_cid < cid + cnt; cur_cid++) {
108 int width = arr->GetNumberAt(inner_idx++);
109 EXPECT_EQ(width, typed_font->GetCharWidthF(cur_cid)) << " at cid "
110 << cur_cid;
111 }
112 num_cids_checked += cnt;
113 continue;
114 }
115 // Otherwise, are in the c_first c_last w case.
116 ASSERT_TRUE(next->IsNumber());
117 int last_cid = next->AsNumber()->GetInteger();
118 ASSERT_FALSE(++idx == widths_array->GetCount());
119 int width = widths_array->GetNumberAt(idx);
120 for (cur_cid = cid; cur_cid <= last_cid; cur_cid++) {
121 EXPECT_EQ(width, typed_font->GetCharWidthF(cur_cid)) << " at cid "
122 << cur_cid;
123 }
124 num_cids_checked += last_cid - cid + 1;
125 }
126 // Make sure we have a good amount of cids described
127 EXPECT_GT(num_cids_checked, 900);
128 }
129 CPDF_Document* cpdf_doc() { return cpdf_doc_; }
130
131 private:
132 CPDF_Document* cpdf_doc_;
133};
Tom Sepezd483eb42016-01-06 10:03:59 -0800134
etienneb7712c262016-04-26 08:13:45 -0700135namespace {
thestigdc7ec032016-11-21 15:32:52 -0800136
etienneb7712c262016-04-26 08:13:45 -0700137const char kExpectedPDF[] =
138 "%PDF-1.7\r\n"
139 "%\xA1\xB3\xC5\xD7\r\n"
140 "1 0 obj\r\n"
141 "<</Pages 2 0 R /Type/Catalog>>\r\n"
142 "endobj\r\n"
143 "2 0 obj\r\n"
144 "<</Count 1/Kids\\[ 4 0 R \\]/Type/Pages>>\r\n"
145 "endobj\r\n"
146 "3 0 obj\r\n"
147 "<</CreationDate\\(D:.*\\)/Creator\\(PDFium\\)>>\r\n"
148 "endobj\r\n"
149 "4 0 obj\r\n"
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400150 "<</MediaBox\\[ 0 0 640 480\\]/Parent 2 0 R "
151 "/Resources<</ExtGState<</FXE1 5 0 R >>>>"
Nicolas Penad9d6c292017-06-06 16:12:10 -0400152 "/Rotate 0/Type/Page"
etienneb7712c262016-04-26 08:13:45 -0700153 ">>\r\n"
154 "endobj\r\n"
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400155 "5 0 obj\r\n"
156 "<</BM/Normal/CA 1/ca 1>>\r\n"
157 "endobj\r\n"
etienneb7712c262016-04-26 08:13:45 -0700158 "xref\r\n"
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400159 "0 6\r\n"
etienneb7712c262016-04-26 08:13:45 -0700160 "0000000000 65535 f\r\n"
161 "0000000017 00000 n\r\n"
162 "0000000066 00000 n\r\n"
163 "0000000122 00000 n\r\n"
164 "0000000192 00000 n\r\n"
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400165 "0000000311 00000 n\r\n"
etienneb7712c262016-04-26 08:13:45 -0700166 "trailer\r\n"
167 "<<\r\n"
168 "/Root 1 0 R\r\n"
169 "/Info 3 0 R\r\n"
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400170 "/Size 6/ID\\[<.*><.*>\\]>>\r\n"
etienneb7712c262016-04-26 08:13:45 -0700171 "startxref\r\n"
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400172 "354\r\n"
etienneb7712c262016-04-26 08:13:45 -0700173 "%%EOF\r\n";
thestigdc7ec032016-11-21 15:32:52 -0800174
etienneb7712c262016-04-26 08:13:45 -0700175} // namespace
176
Tom Sepezd483eb42016-01-06 10:03:59 -0800177TEST_F(FPDFEditEmbeddertest, EmptyCreation) {
178 EXPECT_TRUE(CreateEmptyDocument());
weili9b777de2016-08-19 16:19:46 -0700179 FPDF_PAGE page = FPDFPage_New(document(), 0, 640.0, 480.0);
Tom Sepezd483eb42016-01-06 10:03:59 -0800180 EXPECT_NE(nullptr, page);
Nicolas Penad9d6c292017-06-06 16:12:10 -0400181 // The FPDFPage_GenerateContent call should do nothing.
Tom Sepezd483eb42016-01-06 10:03:59 -0800182 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Tom Sepez0aec19b2016-01-07 12:22:44 -0800183 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
etienneb7712c262016-04-26 08:13:45 -0700184
Nicolas Penad9d6c292017-06-06 16:12:10 -0400185 EXPECT_THAT(GetString(), testing::MatchesRegex(std::string(
186 kExpectedPDF, sizeof(kExpectedPDF))));
weili9b777de2016-08-19 16:19:46 -0700187 FPDF_ClosePage(page);
Tom Sepezd483eb42016-01-06 10:03:59 -0800188}
thestigdc7ec032016-11-21 15:32:52 -0800189
190// Regression test for https://crbug.com/667012
191TEST_F(FPDFEditEmbeddertest, RasterizePDF) {
192 const char kAllBlackMd5sum[] = "5708fc5c4a8bd0abde99c8e8f0390615";
193
194 // Get the bitmap for the original document/
195 FPDF_BITMAP orig_bitmap;
196 {
197 EXPECT_TRUE(OpenDocument("black.pdf"));
198 FPDF_PAGE orig_page = LoadPage(0);
199 EXPECT_NE(nullptr, orig_page);
200 orig_bitmap = RenderPage(orig_page);
201 CompareBitmap(orig_bitmap, 612, 792, kAllBlackMd5sum);
202 UnloadPage(orig_page);
203 }
204
205 // Create a new document from |orig_bitmap| and save it.
206 {
207 FPDF_DOCUMENT temp_doc = FPDF_CreateNewDocument();
208 FPDF_PAGE temp_page = FPDFPage_New(temp_doc, 0, 612, 792);
209
210 // Add the bitmap to an image object and add the image object to the output
211 // page.
Lei Zhangcbd89572017-03-15 17:35:47 -0700212 FPDF_PAGEOBJECT temp_img = FPDFPageObj_NewImageObj(temp_doc);
thestigdc7ec032016-11-21 15:32:52 -0800213 EXPECT_TRUE(FPDFImageObj_SetBitmap(&temp_page, 1, temp_img, orig_bitmap));
214 EXPECT_TRUE(FPDFImageObj_SetMatrix(temp_img, 612, 0, 0, 792, 0, 0));
215 FPDFPage_InsertObject(temp_page, temp_img);
216 EXPECT_TRUE(FPDFPage_GenerateContent(temp_page));
217 EXPECT_TRUE(FPDF_SaveAsCopy(temp_doc, this, 0));
218 FPDF_ClosePage(temp_page);
219 FPDF_CloseDocument(temp_doc);
220 }
221 FPDFBitmap_Destroy(orig_bitmap);
222
223 // Get the generated content. Make sure it is at least as big as the original
224 // PDF.
Nicolas Penaa0b48aa2017-06-29 11:01:46 -0400225 EXPECT_GT(GetString().size(), 923U);
Nicolas Pena3ff54002017-07-05 11:55:35 -0400226 TestAndCloseSaved(612, 792, kAllBlackMd5sum);
thestigdc7ec032016-11-21 15:32:52 -0800227}
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500228
229TEST_F(FPDFEditEmbeddertest, AddPaths) {
230 // Start with a blank page
Nicolas Penad03ca422017-03-06 13:54:33 -0500231 FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792);
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500232
233 // We will first add a red rectangle
234 FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(10, 10, 20, 20);
235 ASSERT_NE(nullptr, red_rect);
236 // Expect false when trying to set colors out of range
237 EXPECT_FALSE(FPDFPath_SetStrokeColor(red_rect, 100, 100, 100, 300));
238 EXPECT_FALSE(FPDFPath_SetFillColor(red_rect, 200, 256, 200, 0));
239
240 // Fill rectangle with red and insert to the page
241 EXPECT_TRUE(FPDFPath_SetFillColor(red_rect, 255, 0, 0, 255));
242 EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0));
243 FPDFPage_InsertObject(page, red_rect);
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500244 FPDF_BITMAP page_bitmap = RenderPage(page);
245 CompareBitmap(page_bitmap, 612, 792, "66d02eaa6181e2c069ce2ea99beda497");
246 FPDFBitmap_Destroy(page_bitmap);
247
248 // Now add to that a green rectangle with some medium alpha
249 FPDF_PAGEOBJECT green_rect = FPDFPageObj_CreateNewRect(100, 100, 40, 40);
250 EXPECT_TRUE(FPDFPath_SetFillColor(green_rect, 0, 255, 0, 128));
Miklos Vajnaed4705b2017-04-05 09:24:50 +0200251
Miklos Vajna1ef04c92017-05-08 18:14:19 +0200252 // Make sure the type of the rectangle is a path.
253 EXPECT_EQ(FPDF_PAGEOBJ_PATH, FPDFPageObj_GetType(green_rect));
254
Miklos Vajnaed4705b2017-04-05 09:24:50 +0200255 // Make sure we get back the same color we set previously.
256 unsigned int R;
257 unsigned int G;
258 unsigned int B;
259 unsigned int A;
260 EXPECT_TRUE(FPDFPath_GetFillColor(green_rect, &R, &G, &B, &A));
261 EXPECT_EQ(0U, R);
262 EXPECT_EQ(255U, G);
263 EXPECT_EQ(0U, B);
264 EXPECT_EQ(128U, A);
265
Miklos Vajna12abfd02017-09-15 07:49:03 +0200266 // Make sure the path has 5 points (1 FXPT_TYPE::MoveTo and 4
267 // FXPT_TYPE::LineTo).
Miklos Vajna0150a542017-09-21 21:46:56 +0200268 ASSERT_EQ(5, FPDFPath_CountSegments(green_rect));
Miklos Vajna36eed872017-09-20 22:52:43 +0200269 // Verify actual coordinates.
270 FPDF_PATHSEGMENT segment = FPDFPath_GetPathSegment(green_rect, 0);
271 float x;
272 float y;
273 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
274 EXPECT_EQ(100, x);
275 EXPECT_EQ(100, y);
276 EXPECT_EQ(FPDF_SEGMENT_MOVETO, FPDFPathSegment_GetType(segment));
277 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
278 segment = FPDFPath_GetPathSegment(green_rect, 1);
279 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
280 EXPECT_EQ(100, x);
281 EXPECT_EQ(140, y);
282 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
283 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
284 segment = FPDFPath_GetPathSegment(green_rect, 2);
285 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
286 EXPECT_EQ(140, x);
287 EXPECT_EQ(140, y);
288 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
289 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
290 segment = FPDFPath_GetPathSegment(green_rect, 3);
291 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
292 EXPECT_EQ(140, x);
293 EXPECT_EQ(100, y);
294 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
295 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
296 segment = FPDFPath_GetPathSegment(green_rect, 4);
297 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
298 EXPECT_EQ(100, x);
299 EXPECT_EQ(100, y);
300 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
301 EXPECT_TRUE(FPDFPathSegment_GetClose(segment));
Miklos Vajna12abfd02017-09-15 07:49:03 +0200302
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500303 EXPECT_TRUE(FPDFPath_SetDrawMode(green_rect, FPDF_FILLMODE_WINDING, 0));
304 FPDFPage_InsertObject(page, green_rect);
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500305 page_bitmap = RenderPage(page);
306 CompareBitmap(page_bitmap, 612, 792, "7b0b87604594e773add528fae567a558");
307 FPDFBitmap_Destroy(page_bitmap);
308
309 // Add a black triangle.
310 FPDF_PAGEOBJECT black_path = FPDFPageObj_CreateNewPath(400, 100);
311 EXPECT_TRUE(FPDFPath_SetFillColor(black_path, 0, 0, 0, 200));
312 EXPECT_TRUE(FPDFPath_SetDrawMode(black_path, FPDF_FILLMODE_ALTERNATE, 0));
313 EXPECT_TRUE(FPDFPath_LineTo(black_path, 400, 200));
314 EXPECT_TRUE(FPDFPath_LineTo(black_path, 300, 100));
315 EXPECT_TRUE(FPDFPath_Close(black_path));
Miklos Vajna12abfd02017-09-15 07:49:03 +0200316
317 // Make sure the path has 3 points (1 FXPT_TYPE::MoveTo and 2
318 // FXPT_TYPE::LineTo).
Miklos Vajna0150a542017-09-21 21:46:56 +0200319 ASSERT_EQ(3, FPDFPath_CountSegments(black_path));
Miklos Vajna36eed872017-09-20 22:52:43 +0200320 // Verify actual coordinates.
321 segment = FPDFPath_GetPathSegment(black_path, 0);
322 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
323 EXPECT_EQ(400, x);
324 EXPECT_EQ(100, y);
325 EXPECT_EQ(FPDF_SEGMENT_MOVETO, FPDFPathSegment_GetType(segment));
326 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
327 segment = FPDFPath_GetPathSegment(black_path, 1);
328 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
329 EXPECT_EQ(400, x);
330 EXPECT_EQ(200, y);
331 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
332 EXPECT_FALSE(FPDFPathSegment_GetClose(segment));
333 segment = FPDFPath_GetPathSegment(black_path, 2);
334 EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y));
335 EXPECT_EQ(300, x);
336 EXPECT_EQ(100, y);
337 EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment));
338 EXPECT_TRUE(FPDFPathSegment_GetClose(segment));
339 // Make sure out of bounds index access fails properly.
340 EXPECT_EQ(nullptr, FPDFPath_GetPathSegment(black_path, 3));
Miklos Vajna12abfd02017-09-15 07:49:03 +0200341
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500342 FPDFPage_InsertObject(page, black_path);
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500343 page_bitmap = RenderPage(page);
344 CompareBitmap(page_bitmap, 612, 792, "eadc8020a14dfcf091da2688733d8806");
345 FPDFBitmap_Destroy(page_bitmap);
346
347 // Now add a more complex blue path.
348 FPDF_PAGEOBJECT blue_path = FPDFPageObj_CreateNewPath(200, 200);
349 EXPECT_TRUE(FPDFPath_SetFillColor(blue_path, 0, 0, 255, 255));
350 EXPECT_TRUE(FPDFPath_SetDrawMode(blue_path, FPDF_FILLMODE_WINDING, 0));
351 EXPECT_TRUE(FPDFPath_LineTo(blue_path, 230, 230));
352 EXPECT_TRUE(FPDFPath_BezierTo(blue_path, 250, 250, 280, 280, 300, 300));
353 EXPECT_TRUE(FPDFPath_LineTo(blue_path, 325, 325));
354 EXPECT_TRUE(FPDFPath_LineTo(blue_path, 350, 325));
355 EXPECT_TRUE(FPDFPath_BezierTo(blue_path, 375, 330, 390, 360, 400, 400));
356 EXPECT_TRUE(FPDFPath_Close(blue_path));
357 FPDFPage_InsertObject(page, blue_path);
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500358 page_bitmap = RenderPage(page);
Nicolas Penaa0b48aa2017-06-29 11:01:46 -0400359 const char last_md5[] = "9823e1a21bd9b72b6a442ba4f12af946";
360 CompareBitmap(page_bitmap, 612, 792, last_md5);
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500361 FPDFBitmap_Destroy(page_bitmap);
362
363 // Now save the result, closing the page and document
Nicolas Pena207b7272017-05-26 17:37:06 -0400364 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Nicolas Penad03ca422017-03-06 13:54:33 -0500365 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500366 FPDF_ClosePage(page);
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500367
368 // Render the saved result
Nicolas Pena3ff54002017-07-05 11:55:35 -0400369 TestAndCloseSaved(612, 792, last_md5);
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500370}
371
Miklos Vajna12abfd02017-09-15 07:49:03 +0200372TEST_F(FPDFEditEmbeddertest, PathsPoints) {
373 CreateNewDocument();
374 FPDF_PAGEOBJECT img = FPDFPageObj_NewImageObj(document_);
375 // This should fail gracefully, even if img is not a path.
Miklos Vajna0150a542017-09-21 21:46:56 +0200376 ASSERT_EQ(-1, FPDFPath_CountSegments(img));
Miklos Vajna12abfd02017-09-15 07:49:03 +0200377
378 // This should fail gracefully, even if path is NULL.
Miklos Vajna0150a542017-09-21 21:46:56 +0200379 ASSERT_EQ(-1, FPDFPath_CountSegments(nullptr));
Miklos Vajna12abfd02017-09-15 07:49:03 +0200380
Miklos Vajna36eed872017-09-20 22:52:43 +0200381 // FPDFPath_GetPathSegment() with a non-path.
382 ASSERT_EQ(nullptr, FPDFPath_GetPathSegment(img, 0));
383 // FPDFPath_GetPathSegment() with a NULL path.
384 ASSERT_EQ(nullptr, FPDFPath_GetPathSegment(nullptr, 0));
385 float x;
386 float y;
387 // FPDFPathSegment_GetPoint() with a NULL segment.
388 EXPECT_FALSE(FPDFPathSegment_GetPoint(nullptr, &x, &y));
389
390 // FPDFPathSegment_GetType() with a NULL segment.
391 ASSERT_EQ(FPDF_SEGMENT_UNKNOWN, FPDFPathSegment_GetType(nullptr));
392
393 // FPDFPathSegment_GetClose() with a NULL segment.
394 EXPECT_FALSE(FPDFPathSegment_GetClose(nullptr));
395
Miklos Vajna12abfd02017-09-15 07:49:03 +0200396 FPDFPageObj_Destroy(img);
397}
398
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500399TEST_F(FPDFEditEmbeddertest, PathOnTopOfText) {
400 // Load document with some text
401 EXPECT_TRUE(OpenDocument("hello_world.pdf"));
402 FPDF_PAGE page = LoadPage(0);
403 EXPECT_NE(nullptr, page);
404
405 // Add an opaque rectangle on top of some of the text.
406 FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(20, 100, 50, 50);
407 EXPECT_TRUE(FPDFPath_SetFillColor(red_rect, 255, 0, 0, 255));
408 EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0));
409 FPDFPage_InsertObject(page, red_rect);
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500410
411 // Add a transparent triangle on top of other part of the text.
412 FPDF_PAGEOBJECT black_path = FPDFPageObj_CreateNewPath(20, 50);
413 EXPECT_TRUE(FPDFPath_SetFillColor(black_path, 0, 0, 0, 100));
414 EXPECT_TRUE(FPDFPath_SetDrawMode(black_path, FPDF_FILLMODE_ALTERNATE, 0));
415 EXPECT_TRUE(FPDFPath_LineTo(black_path, 30, 80));
416 EXPECT_TRUE(FPDFPath_LineTo(black_path, 40, 10));
417 EXPECT_TRUE(FPDFPath_Close(black_path));
418 FPDFPage_InsertObject(page, black_path);
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500419
420 // Render and check the result. Text is slightly different on Mac.
421 FPDF_BITMAP bitmap = RenderPage(page);
Dan Sinclair698aed72017-09-26 16:24:49 -0400422#if _FX_PLATFORM_ == _FX_PLATFORM_APPLE_
Lei Zhang0d6d1782017-03-24 15:52:00 -0700423 const char md5[] = "f9e6fa74230f234286bfcada9f7606d8";
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500424#else
Nicolas Pena00c3cfd2017-07-10 17:29:54 -0400425 const char md5[] = "bc6e6eb50dda4695ba0fb4d04ed82ada";
Nicolas Pena5bcd9a32017-03-22 11:04:35 -0400426#endif
Lei Zhang0d6d1782017-03-24 15:52:00 -0700427 CompareBitmap(bitmap, 200, 200, md5);
Nicolas Pena0fc185e2017-02-08 12:13:20 -0500428 FPDFBitmap_Destroy(bitmap);
429 UnloadPage(page);
430}
Nicolas Pena2eb1a702017-02-09 18:17:33 -0500431
wileyryae858aa42017-05-31 14:49:05 -0500432TEST_F(FPDFEditEmbeddertest, EditOverExistingContent) {
433 // Load document with existing content
434 EXPECT_TRUE(OpenDocument("bug_717.pdf"));
435 FPDF_PAGE page = LoadPage(0);
436 EXPECT_NE(nullptr, page);
437
438 // Add a transparent rectangle on top of the existing content
439 FPDF_PAGEOBJECT red_rect2 = FPDFPageObj_CreateNewRect(90, 700, 25, 50);
440 EXPECT_TRUE(FPDFPath_SetFillColor(red_rect2, 255, 0, 0, 100));
441 EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect2, FPDF_FILLMODE_ALTERNATE, 0));
442 FPDFPage_InsertObject(page, red_rect2);
443
444 // Add an opaque rectangle on top of the existing content
445 FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(115, 700, 25, 50);
446 EXPECT_TRUE(FPDFPath_SetFillColor(red_rect, 255, 0, 0, 255));
447 EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0));
448 FPDFPage_InsertObject(page, red_rect);
449
450 FPDF_BITMAP bitmap = RenderPage(page);
451 CompareBitmap(bitmap, 612, 792, "ad04e5bd0f471a9a564fb034bd0fb073");
452 FPDFBitmap_Destroy(bitmap);
453 EXPECT_TRUE(FPDFPage_GenerateContent(page));
454
455 // Now save the result, closing the page and document
456 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
Nicolas Pena3ff54002017-07-05 11:55:35 -0400457 UnloadPage(page);
wileyryae858aa42017-05-31 14:49:05 -0500458
Nicolas Pena3ff54002017-07-05 11:55:35 -0400459 // Render the saved result without closing the page and document
460 TestSaved(612, 792, "ad04e5bd0f471a9a564fb034bd0fb073");
wileyryae858aa42017-05-31 14:49:05 -0500461
462 ClearString();
463 // Add another opaque rectangle on top of the existing content
464 FPDF_PAGEOBJECT green_rect = FPDFPageObj_CreateNewRect(150, 700, 25, 50);
465 EXPECT_TRUE(FPDFPath_SetFillColor(green_rect, 0, 255, 0, 255));
466 EXPECT_TRUE(FPDFPath_SetDrawMode(green_rect, FPDF_FILLMODE_ALTERNATE, 0));
Nicolas Pena3ff54002017-07-05 11:55:35 -0400467 FPDFPage_InsertObject(m_SavedPage, green_rect);
wileyryae858aa42017-05-31 14:49:05 -0500468
469 // Add another transparent rectangle on top of existing content
470 FPDF_PAGEOBJECT green_rect2 = FPDFPageObj_CreateNewRect(175, 700, 25, 50);
471 EXPECT_TRUE(FPDFPath_SetFillColor(green_rect2, 0, 255, 0, 100));
472 EXPECT_TRUE(FPDFPath_SetDrawMode(green_rect2, FPDF_FILLMODE_ALTERNATE, 0));
Nicolas Pena3ff54002017-07-05 11:55:35 -0400473 FPDFPage_InsertObject(m_SavedPage, green_rect2);
474 FPDF_BITMAP new_bitmap = RenderPageWithFlags(m_SavedPage, m_SavedForm, 0);
Nicolas Penaa0b48aa2017-06-29 11:01:46 -0400475 const char last_md5[] = "4b5b00f824620f8c9b8801ebb98e1cdd";
476 CompareBitmap(new_bitmap, 612, 792, last_md5);
wileyryae858aa42017-05-31 14:49:05 -0500477 FPDFBitmap_Destroy(new_bitmap);
Nicolas Pena3ff54002017-07-05 11:55:35 -0400478 EXPECT_TRUE(FPDFPage_GenerateContent(m_SavedPage));
wileyryae858aa42017-05-31 14:49:05 -0500479
480 // Now save the result, closing the page and document
Nicolas Pena3ff54002017-07-05 11:55:35 -0400481 EXPECT_TRUE(FPDF_SaveAsCopy(m_SavedDocument, this, 0));
482 CloseSaved();
wileyryae858aa42017-05-31 14:49:05 -0500483
484 // Render the saved result
Nicolas Pena3ff54002017-07-05 11:55:35 -0400485 TestAndCloseSaved(612, 792, last_md5);
wileyryae858aa42017-05-31 14:49:05 -0500486}
487
Nicolas Pena2eb1a702017-02-09 18:17:33 -0500488TEST_F(FPDFEditEmbeddertest, AddStrokedPaths) {
489 // Start with a blank page
Nicolas Penad03ca422017-03-06 13:54:33 -0500490 FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792);
Nicolas Pena2eb1a702017-02-09 18:17:33 -0500491
492 // Add a large stroked rectangle (fill color should not affect it).
493 FPDF_PAGEOBJECT rect = FPDFPageObj_CreateNewRect(20, 20, 200, 400);
494 EXPECT_TRUE(FPDFPath_SetFillColor(rect, 255, 0, 0, 255));
495 EXPECT_TRUE(FPDFPath_SetStrokeColor(rect, 0, 255, 0, 255));
496 EXPECT_TRUE(FPDFPath_SetStrokeWidth(rect, 15.0f));
497 EXPECT_TRUE(FPDFPath_SetDrawMode(rect, 0, 1));
498 FPDFPage_InsertObject(page, rect);
Nicolas Pena2eb1a702017-02-09 18:17:33 -0500499 FPDF_BITMAP page_bitmap = RenderPage(page);
500 CompareBitmap(page_bitmap, 612, 792, "64bd31f862a89e0a9e505a5af6efd506");
501 FPDFBitmap_Destroy(page_bitmap);
502
503 // Add crossed-checkmark
504 FPDF_PAGEOBJECT check = FPDFPageObj_CreateNewPath(300, 500);
505 EXPECT_TRUE(FPDFPath_LineTo(check, 400, 400));
506 EXPECT_TRUE(FPDFPath_LineTo(check, 600, 600));
507 EXPECT_TRUE(FPDFPath_MoveTo(check, 400, 600));
508 EXPECT_TRUE(FPDFPath_LineTo(check, 600, 400));
509 EXPECT_TRUE(FPDFPath_SetStrokeColor(check, 128, 128, 128, 180));
510 EXPECT_TRUE(FPDFPath_SetStrokeWidth(check, 8.35f));
511 EXPECT_TRUE(FPDFPath_SetDrawMode(check, 0, 1));
512 FPDFPage_InsertObject(page, check);
Nicolas Pena2eb1a702017-02-09 18:17:33 -0500513 page_bitmap = RenderPage(page);
514 CompareBitmap(page_bitmap, 612, 792, "4b6f3b9d25c4e194821217d5016c3724");
515 FPDFBitmap_Destroy(page_bitmap);
516
517 // Add stroked and filled oval-ish path.
518 FPDF_PAGEOBJECT path = FPDFPageObj_CreateNewPath(250, 100);
519 EXPECT_TRUE(FPDFPath_BezierTo(path, 180, 166, 180, 233, 250, 300));
520 EXPECT_TRUE(FPDFPath_LineTo(path, 255, 305));
521 EXPECT_TRUE(FPDFPath_BezierTo(path, 325, 233, 325, 166, 255, 105));
522 EXPECT_TRUE(FPDFPath_Close(path));
523 EXPECT_TRUE(FPDFPath_SetFillColor(path, 200, 128, 128, 100));
524 EXPECT_TRUE(FPDFPath_SetStrokeColor(path, 128, 200, 128, 150));
525 EXPECT_TRUE(FPDFPath_SetStrokeWidth(path, 10.5f));
526 EXPECT_TRUE(FPDFPath_SetDrawMode(path, FPDF_FILLMODE_ALTERNATE, 1));
527 FPDFPage_InsertObject(page, path);
Nicolas Pena2eb1a702017-02-09 18:17:33 -0500528 page_bitmap = RenderPage(page);
529 CompareBitmap(page_bitmap, 612, 792, "ff3e6a22326754944cc6e56609acd73b");
530 FPDFBitmap_Destroy(page_bitmap);
531 FPDF_ClosePage(page);
Nicolas Pena2eb1a702017-02-09 18:17:33 -0500532}
Nicolas Pena49058402017-02-14 18:26:20 -0500533
534TEST_F(FPDFEditEmbeddertest, AddStandardFontText) {
535 // Start with a blank page
Nicolas Penad03ca422017-03-06 13:54:33 -0500536 FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792);
Nicolas Pena49058402017-02-14 18:26:20 -0500537
538 // Add some text to the page
Nicolas Penab3161852017-05-02 14:12:50 -0400539 FPDF_PAGEOBJECT text_object1 =
540 FPDFPageObj_NewTextObj(document(), "Arial", 12.0f);
541 EXPECT_TRUE(text_object1);
542 std::unique_ptr<unsigned short, pdfium::FreeDeleter> text1 =
543 GetFPDFWideString(L"I'm at the bottom of the page");
544 EXPECT_TRUE(FPDFText_SetText(text_object1, text1.get()));
545 FPDFPageObj_Transform(text_object1, 1, 0, 0, 1, 20, 20);
546 FPDFPage_InsertObject(page, text_object1);
Nicolas Pena49058402017-02-14 18:26:20 -0500547 FPDF_BITMAP page_bitmap = RenderPage(page);
Dan Sinclair698aed72017-09-26 16:24:49 -0400548#if _FX_PLATFORM_ == _FX_PLATFORM_APPLE_
Lei Zhang0d6d1782017-03-24 15:52:00 -0700549 const char md5[] = "a4dddc1a3930fa694bbff9789dab4161";
Nicolas Pena49058402017-02-14 18:26:20 -0500550#else
Nicolas Pena00c3cfd2017-07-10 17:29:54 -0400551 const char md5[] = "7a35771853a1cbba38f6775807878625";
Nicolas Pena5bcd9a32017-03-22 11:04:35 -0400552#endif
Lei Zhang0d6d1782017-03-24 15:52:00 -0700553 CompareBitmap(page_bitmap, 612, 792, md5);
Nicolas Pena49058402017-02-14 18:26:20 -0500554 FPDFBitmap_Destroy(page_bitmap);
555
556 // Try another font
Nicolas Penab3161852017-05-02 14:12:50 -0400557 FPDF_PAGEOBJECT text_object2 =
Nicolas Penad03ca422017-03-06 13:54:33 -0500558 FPDFPageObj_NewTextObj(document(), "TimesNewRomanBold", 15.0f);
Nicolas Penab3161852017-05-02 14:12:50 -0400559 EXPECT_TRUE(text_object2);
560 std::unique_ptr<unsigned short, pdfium::FreeDeleter> text2 =
561 GetFPDFWideString(L"Hi, I'm Bold. Times New Roman Bold.");
562 EXPECT_TRUE(FPDFText_SetText(text_object2, text2.get()));
563 FPDFPageObj_Transform(text_object2, 1, 0, 0, 1, 100, 600);
564 FPDFPage_InsertObject(page, text_object2);
Nicolas Pena49058402017-02-14 18:26:20 -0500565 page_bitmap = RenderPage(page);
Dan Sinclair698aed72017-09-26 16:24:49 -0400566#if _FX_PLATFORM_ == _FX_PLATFORM_APPLE_
Lei Zhang0d6d1782017-03-24 15:52:00 -0700567 const char md5_2[] = "a5c4ace4c6f27644094813fe1441a21c";
Dan Sinclair698aed72017-09-26 16:24:49 -0400568#elif _FX_PLATFORM_ == _FX_PLATFORM_WINDOWS_
Nicolas Pena00c3cfd2017-07-10 17:29:54 -0400569 const char md5_2[] = "b231b329a4b566fb9b42bfc15fe59bb7";
Nicolas Pena49058402017-02-14 18:26:20 -0500570#else
Nicolas Pena00c3cfd2017-07-10 17:29:54 -0400571 const char md5_2[] = "f85fae151851436072b7b3c6703e506a";
Nicolas Pena5bcd9a32017-03-22 11:04:35 -0400572#endif
Lei Zhang0d6d1782017-03-24 15:52:00 -0700573 CompareBitmap(page_bitmap, 612, 792, md5_2);
Nicolas Pena49058402017-02-14 18:26:20 -0500574 FPDFBitmap_Destroy(page_bitmap);
575
576 // And some randomly transformed text
Nicolas Penab3161852017-05-02 14:12:50 -0400577 FPDF_PAGEOBJECT text_object3 =
Nicolas Penad03ca422017-03-06 13:54:33 -0500578 FPDFPageObj_NewTextObj(document(), "Courier-Bold", 20.0f);
Nicolas Penab3161852017-05-02 14:12:50 -0400579 EXPECT_TRUE(text_object3);
580 std::unique_ptr<unsigned short, pdfium::FreeDeleter> text3 =
581 GetFPDFWideString(L"Can you read me? <:)>");
582 EXPECT_TRUE(FPDFText_SetText(text_object3, text3.get()));
583 FPDFPageObj_Transform(text_object3, 1, 1.5, 2, 0.5, 200, 200);
584 FPDFPage_InsertObject(page, text_object3);
Nicolas Pena49058402017-02-14 18:26:20 -0500585 page_bitmap = RenderPage(page);
Dan Sinclair698aed72017-09-26 16:24:49 -0400586#if _FX_PLATFORM_ == _FX_PLATFORM_APPLE_
Lei Zhang0d6d1782017-03-24 15:52:00 -0700587 const char md5_3[] = "40b3ef04f915ff4c4208948001763544";
Dan Sinclair698aed72017-09-26 16:24:49 -0400588#elif _FX_PLATFORM_ == _FX_PLATFORM_WINDOWS_
Nicolas Pena00c3cfd2017-07-10 17:29:54 -0400589 const char md5_3[] = "ba874b3b137f984510c4e287ed4ba7ae";
Nicolas Pena49058402017-02-14 18:26:20 -0500590#else
Nicolas Pena00c3cfd2017-07-10 17:29:54 -0400591 const char md5_3[] = "c5aed6a8ef05558c8c47d58c87cbcb46";
Nicolas Pena5bcd9a32017-03-22 11:04:35 -0400592#endif
Lei Zhang0d6d1782017-03-24 15:52:00 -0700593 CompareBitmap(page_bitmap, 612, 792, md5_3);
Nicolas Pena49058402017-02-14 18:26:20 -0500594 FPDFBitmap_Destroy(page_bitmap);
595
596 // TODO(npm): Why are there issues with text rotated by 90 degrees?
597 // TODO(npm): FPDF_SaveAsCopy not giving the desired result after this.
598 FPDF_ClosePage(page);
Nicolas Pena49058402017-02-14 18:26:20 -0500599}
Nicolas Penaa4ad01f2017-02-15 16:26:48 -0500600
Nicolas Pena603a31d2017-06-14 11:41:18 -0400601TEST_F(FPDFEditEmbeddertest, GraphicsData) {
602 // New page
603 std::unique_ptr<void, FPDFPageDeleter> page(
604 FPDFPage_New(CreateNewDocument(), 0, 612, 792));
605
606 // Create a rect with nontrivial graphics
607 FPDF_PAGEOBJECT rect1 = FPDFPageObj_CreateNewRect(10, 10, 100, 100);
608 FPDFPageObj_SetBlendMode(rect1, "Color");
609 FPDFPage_InsertObject(page.get(), rect1);
610 EXPECT_TRUE(FPDFPage_GenerateContent(page.get()));
611
612 // Check that the ExtGState was created
613 CPDF_Page* the_page = CPDFPageFromFPDFPage(page.get());
614 CPDF_Dictionary* graphics_dict =
615 the_page->m_pResources->GetDictFor("ExtGState");
616 ASSERT_TRUE(graphics_dict);
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400617 EXPECT_EQ(2, static_cast<int>(graphics_dict->GetCount()));
Nicolas Pena603a31d2017-06-14 11:41:18 -0400618
619 // Add a text object causing no change to the graphics dictionary
620 FPDF_PAGEOBJECT text1 = FPDFPageObj_NewTextObj(document(), "Arial", 12.0f);
621 // Only alpha, the last component, matters for the graphics dictionary. And
622 // the default value is 255.
623 EXPECT_TRUE(FPDFText_SetFillColor(text1, 100, 100, 100, 255));
624 FPDFPage_InsertObject(page.get(), text1);
625 EXPECT_TRUE(FPDFPage_GenerateContent(page.get()));
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400626 EXPECT_EQ(2, static_cast<int>(graphics_dict->GetCount()));
Nicolas Pena603a31d2017-06-14 11:41:18 -0400627
628 // Add a text object increasing the size of the graphics dictionary
629 FPDF_PAGEOBJECT text2 =
630 FPDFPageObj_NewTextObj(document(), "Times-Roman", 12.0f);
631 FPDFPage_InsertObject(page.get(), text2);
632 FPDFPageObj_SetBlendMode(text2, "Darken");
633 EXPECT_TRUE(FPDFText_SetFillColor(text2, 0, 0, 255, 150));
634 EXPECT_TRUE(FPDFPage_GenerateContent(page.get()));
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400635 EXPECT_EQ(3, static_cast<int>(graphics_dict->GetCount()));
Nicolas Pena603a31d2017-06-14 11:41:18 -0400636
637 // Add a path that should reuse graphics
Nicolas Penace67be42017-06-14 14:52:49 -0400638 FPDF_PAGEOBJECT path = FPDFPageObj_CreateNewPath(400, 100);
Nicolas Pena603a31d2017-06-14 11:41:18 -0400639 FPDFPageObj_SetBlendMode(path, "Darken");
640 EXPECT_TRUE(FPDFPath_SetFillColor(path, 200, 200, 100, 150));
641 FPDFPage_InsertObject(page.get(), path);
642 EXPECT_TRUE(FPDFPage_GenerateContent(page.get()));
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400643 EXPECT_EQ(3, static_cast<int>(graphics_dict->GetCount()));
Nicolas Pena603a31d2017-06-14 11:41:18 -0400644
645 // Add a rect increasing the size of the graphics dictionary
646 FPDF_PAGEOBJECT rect2 = FPDFPageObj_CreateNewRect(10, 10, 100, 100);
647 FPDFPageObj_SetBlendMode(rect2, "Darken");
648 EXPECT_TRUE(FPDFPath_SetFillColor(rect2, 0, 0, 255, 150));
649 EXPECT_TRUE(FPDFPath_SetStrokeColor(rect2, 0, 0, 0, 200));
650 FPDFPage_InsertObject(page.get(), rect2);
651 EXPECT_TRUE(FPDFPage_GenerateContent(page.get()));
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400652 EXPECT_EQ(4, static_cast<int>(graphics_dict->GetCount()));
Nicolas Pena603a31d2017-06-14 11:41:18 -0400653}
654
Nicolas Penaa4ad01f2017-02-15 16:26:48 -0500655TEST_F(FPDFEditEmbeddertest, DoubleGenerating) {
656 // Start with a blank page
Nicolas Penad03ca422017-03-06 13:54:33 -0500657 FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792);
Nicolas Penaa4ad01f2017-02-15 16:26:48 -0500658
659 // Add a red rectangle with some non-default alpha
660 FPDF_PAGEOBJECT rect = FPDFPageObj_CreateNewRect(10, 10, 100, 100);
661 EXPECT_TRUE(FPDFPath_SetFillColor(rect, 255, 0, 0, 128));
662 EXPECT_TRUE(FPDFPath_SetDrawMode(rect, FPDF_FILLMODE_WINDING, 0));
663 FPDFPage_InsertObject(page, rect);
664 EXPECT_TRUE(FPDFPage_GenerateContent(page));
665
666 // Check the ExtGState
667 CPDF_Page* the_page = CPDFPageFromFPDFPage(page);
668 CPDF_Dictionary* graphics_dict =
669 the_page->m_pResources->GetDictFor("ExtGState");
670 ASSERT_TRUE(graphics_dict);
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400671 EXPECT_EQ(2, static_cast<int>(graphics_dict->GetCount()));
Nicolas Penaa4ad01f2017-02-15 16:26:48 -0500672
673 // Check the bitmap
674 FPDF_BITMAP page_bitmap = RenderPage(page);
675 CompareBitmap(page_bitmap, 612, 792, "5384da3406d62360ffb5cac4476fff1c");
676 FPDFBitmap_Destroy(page_bitmap);
677
678 // Never mind, my new favorite color is blue, increase alpha
679 EXPECT_TRUE(FPDFPath_SetFillColor(rect, 0, 0, 255, 180));
680 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400681 EXPECT_EQ(3, static_cast<int>(graphics_dict->GetCount()));
Nicolas Penaa4ad01f2017-02-15 16:26:48 -0500682
683 // Check that bitmap displays changed content
684 page_bitmap = RenderPage(page);
685 CompareBitmap(page_bitmap, 612, 792, "2e51656f5073b0bee611d9cd086aa09c");
686 FPDFBitmap_Destroy(page_bitmap);
687
688 // And now generate, without changes
689 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400690 EXPECT_EQ(3, static_cast<int>(graphics_dict->GetCount()));
Nicolas Penaa4ad01f2017-02-15 16:26:48 -0500691 page_bitmap = RenderPage(page);
692 CompareBitmap(page_bitmap, 612, 792, "2e51656f5073b0bee611d9cd086aa09c");
693 FPDFBitmap_Destroy(page_bitmap);
694
695 // Add some text to the page
Nicolas Penab3161852017-05-02 14:12:50 -0400696 FPDF_PAGEOBJECT text_object =
697 FPDFPageObj_NewTextObj(document(), "Arial", 12.0f);
698 std::unique_ptr<unsigned short, pdfium::FreeDeleter> text =
699 GetFPDFWideString(L"Something something #text# something");
700 EXPECT_TRUE(FPDFText_SetText(text_object, text.get()));
701 FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 300, 300);
702 FPDFPage_InsertObject(page, text_object);
Nicolas Penaa4ad01f2017-02-15 16:26:48 -0500703 EXPECT_TRUE(FPDFPage_GenerateContent(page));
704 CPDF_Dictionary* font_dict = the_page->m_pResources->GetDictFor("Font");
705 ASSERT_TRUE(font_dict);
706 EXPECT_EQ(1, static_cast<int>(font_dict->GetCount()));
707
708 // Generate yet again, check dicts are reasonably sized
709 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400710 EXPECT_EQ(3, static_cast<int>(graphics_dict->GetCount()));
Nicolas Penaa4ad01f2017-02-15 16:26:48 -0500711 EXPECT_EQ(1, static_cast<int>(font_dict->GetCount()));
712 FPDF_ClosePage(page);
Nicolas Penaa4ad01f2017-02-15 16:26:48 -0500713}
Nicolas Penabe90aae2017-02-27 10:41:41 -0500714
Nicolas Penad03ca422017-03-06 13:54:33 -0500715TEST_F(FPDFEditEmbeddertest, LoadSimpleType1Font) {
716 CreateNewDocument();
717 // TODO(npm): use other fonts after disallowing loading any font as any type
718 const CPDF_Font* stock_font =
719 CPDF_Font::GetStockFont(cpdf_doc(), "Times-Bold");
Lei Zhangd74da7b2017-05-04 13:30:29 -0700720 const uint8_t* data = stock_font->GetFont()->GetFontData();
721 const uint32_t size = stock_font->GetFont()->GetSize();
Nicolas Penab3161852017-05-02 14:12:50 -0400722 std::unique_ptr<void, FPDFFontDeleter> font(
723 FPDFText_LoadFont(document(), data, size, FPDF_FONT_TYPE1, false));
724 ASSERT_TRUE(font.get());
Nicolas Pena46abb662017-05-17 17:23:22 -0400725 CPDF_Font* typed_font = static_cast<CPDF_Font*>(font.get());
Nicolas Penad03ca422017-03-06 13:54:33 -0500726 EXPECT_TRUE(typed_font->IsType1Font());
Nicolas Penabe90aae2017-02-27 10:41:41 -0500727
Nicolas Penad03ca422017-03-06 13:54:33 -0500728 CPDF_Dictionary* font_dict = typed_font->GetFontDict();
Nicolas Penabe90aae2017-02-27 10:41:41 -0500729 EXPECT_EQ("Font", font_dict->GetStringFor("Type"));
730 EXPECT_EQ("Type1", font_dict->GetStringFor("Subtype"));
731 EXPECT_EQ("Times New Roman Bold", font_dict->GetStringFor("BaseFont"));
732 ASSERT_TRUE(font_dict->KeyExist("FirstChar"));
733 ASSERT_TRUE(font_dict->KeyExist("LastChar"));
734 EXPECT_EQ(32, font_dict->GetIntegerFor("FirstChar"));
Nicolas Penad03ca422017-03-06 13:54:33 -0500735 EXPECT_EQ(255, font_dict->GetIntegerFor("LastChar"));
736
Nicolas Penabe90aae2017-02-27 10:41:41 -0500737 CPDF_Array* widths_array = font_dict->GetArrayFor("Widths");
Nicolas Penad03ca422017-03-06 13:54:33 -0500738 ASSERT_TRUE(widths_array);
739 ASSERT_EQ(224U, widths_array->GetCount());
Nicolas Penabe90aae2017-02-27 10:41:41 -0500740 EXPECT_EQ(250, widths_array->GetNumberAt(0));
Nicolas Penad03ca422017-03-06 13:54:33 -0500741 EXPECT_EQ(569, widths_array->GetNumberAt(11));
742 EXPECT_EQ(500, widths_array->GetNumberAt(223));
743 CheckFontDescriptor(font_dict, FPDF_FONT_TYPE1, true, false, size, data);
744}
Nicolas Penabe90aae2017-02-27 10:41:41 -0500745
Nicolas Penad03ca422017-03-06 13:54:33 -0500746TEST_F(FPDFEditEmbeddertest, LoadSimpleTrueTypeFont) {
747 CreateNewDocument();
748 const CPDF_Font* stock_font = CPDF_Font::GetStockFont(cpdf_doc(), "Courier");
Lei Zhangd74da7b2017-05-04 13:30:29 -0700749 const uint8_t* data = stock_font->GetFont()->GetFontData();
750 const uint32_t size = stock_font->GetFont()->GetSize();
Nicolas Penab3161852017-05-02 14:12:50 -0400751 std::unique_ptr<void, FPDFFontDeleter> font(
752 FPDFText_LoadFont(document(), data, size, FPDF_FONT_TRUETYPE, false));
753 ASSERT_TRUE(font.get());
Nicolas Pena46abb662017-05-17 17:23:22 -0400754 CPDF_Font* typed_font = static_cast<CPDF_Font*>(font.get());
Nicolas Penad03ca422017-03-06 13:54:33 -0500755 EXPECT_TRUE(typed_font->IsTrueTypeFont());
Nicolas Penabe90aae2017-02-27 10:41:41 -0500756
Nicolas Penad03ca422017-03-06 13:54:33 -0500757 CPDF_Dictionary* font_dict = typed_font->GetFontDict();
758 EXPECT_EQ("Font", font_dict->GetStringFor("Type"));
759 EXPECT_EQ("TrueType", font_dict->GetStringFor("Subtype"));
760 EXPECT_EQ("Courier New", font_dict->GetStringFor("BaseFont"));
761 ASSERT_TRUE(font_dict->KeyExist("FirstChar"));
762 ASSERT_TRUE(font_dict->KeyExist("LastChar"));
763 EXPECT_EQ(32, font_dict->GetIntegerFor("FirstChar"));
764 EXPECT_EQ(255, font_dict->GetIntegerFor("LastChar"));
Nicolas Penabe90aae2017-02-27 10:41:41 -0500765
Nicolas Penad03ca422017-03-06 13:54:33 -0500766 CPDF_Array* widths_array = font_dict->GetArrayFor("Widths");
767 ASSERT_TRUE(widths_array);
768 ASSERT_EQ(224U, widths_array->GetCount());
769 EXPECT_EQ(600, widths_array->GetNumberAt(33));
770 EXPECT_EQ(600, widths_array->GetNumberAt(74));
771 EXPECT_EQ(600, widths_array->GetNumberAt(223));
772 CheckFontDescriptor(font_dict, FPDF_FONT_TRUETYPE, false, false, size, data);
773}
774
775TEST_F(FPDFEditEmbeddertest, LoadCIDType0Font) {
776 CreateNewDocument();
777 const CPDF_Font* stock_font =
778 CPDF_Font::GetStockFont(cpdf_doc(), "Times-Roman");
Lei Zhangd74da7b2017-05-04 13:30:29 -0700779 const uint8_t* data = stock_font->GetFont()->GetFontData();
780 const uint32_t size = stock_font->GetFont()->GetSize();
Nicolas Penab3161852017-05-02 14:12:50 -0400781 std::unique_ptr<void, FPDFFontDeleter> font(
782 FPDFText_LoadFont(document(), data, size, FPDF_FONT_TYPE1, 1));
783 ASSERT_TRUE(font.get());
Nicolas Pena46abb662017-05-17 17:23:22 -0400784 CPDF_Font* typed_font = static_cast<CPDF_Font*>(font.get());
Nicolas Penad03ca422017-03-06 13:54:33 -0500785 EXPECT_TRUE(typed_font->IsCIDFont());
786
787 // Check font dictionary entries
788 CPDF_Dictionary* font_dict = typed_font->GetFontDict();
789 EXPECT_EQ("Font", font_dict->GetStringFor("Type"));
790 EXPECT_EQ("Type0", font_dict->GetStringFor("Subtype"));
791 EXPECT_EQ("Times New Roman-Identity-H", font_dict->GetStringFor("BaseFont"));
792 EXPECT_EQ("Identity-H", font_dict->GetStringFor("Encoding"));
793 CPDF_Array* descendant_array = font_dict->GetArrayFor("DescendantFonts");
794 ASSERT_TRUE(descendant_array);
795 EXPECT_EQ(1U, descendant_array->GetCount());
796
797 // Check the CIDFontDict
798 CPDF_Dictionary* cidfont_dict = descendant_array->GetDictAt(0);
799 EXPECT_EQ("Font", cidfont_dict->GetStringFor("Type"));
800 EXPECT_EQ("CIDFontType0", cidfont_dict->GetStringFor("Subtype"));
801 EXPECT_EQ("Times New Roman", cidfont_dict->GetStringFor("BaseFont"));
802 CPDF_Dictionary* cidinfo_dict = cidfont_dict->GetDictFor("CIDSystemInfo");
803 ASSERT_TRUE(cidinfo_dict);
804 EXPECT_EQ("Adobe", cidinfo_dict->GetStringFor("Registry"));
805 EXPECT_EQ("Identity", cidinfo_dict->GetStringFor("Ordering"));
806 EXPECT_EQ(0, cidinfo_dict->GetNumberFor("Supplement"));
807 CheckFontDescriptor(cidfont_dict, FPDF_FONT_TYPE1, false, false, size, data);
808
809 // Check widths
810 CPDF_Array* widths_array = cidfont_dict->GetArrayFor("W");
811 ASSERT_TRUE(widths_array);
Nicolas Penaf45ade32017-05-03 10:23:49 -0400812 EXPECT_GT(widths_array->GetCount(), 1U);
Nicolas Penad03ca422017-03-06 13:54:33 -0500813 CheckCompositeFontWidths(widths_array, typed_font);
814}
815
816TEST_F(FPDFEditEmbeddertest, LoadCIDType2Font) {
817 CreateNewDocument();
818 const CPDF_Font* stock_font =
819 CPDF_Font::GetStockFont(cpdf_doc(), "Helvetica-Oblique");
Lei Zhangd74da7b2017-05-04 13:30:29 -0700820 const uint8_t* data = stock_font->GetFont()->GetFontData();
821 const uint32_t size = stock_font->GetFont()->GetSize();
Nicolas Penad03ca422017-03-06 13:54:33 -0500822
Nicolas Penab3161852017-05-02 14:12:50 -0400823 std::unique_ptr<void, FPDFFontDeleter> font(
824 FPDFText_LoadFont(document(), data, size, FPDF_FONT_TRUETYPE, 1));
825 ASSERT_TRUE(font.get());
Nicolas Pena46abb662017-05-17 17:23:22 -0400826 CPDF_Font* typed_font = static_cast<CPDF_Font*>(font.get());
Nicolas Penad03ca422017-03-06 13:54:33 -0500827 EXPECT_TRUE(typed_font->IsCIDFont());
828
829 // Check font dictionary entries
830 CPDF_Dictionary* font_dict = typed_font->GetFontDict();
831 EXPECT_EQ("Font", font_dict->GetStringFor("Type"));
832 EXPECT_EQ("Type0", font_dict->GetStringFor("Subtype"));
833 EXPECT_EQ("Arial Italic", font_dict->GetStringFor("BaseFont"));
834 EXPECT_EQ("Identity-H", font_dict->GetStringFor("Encoding"));
835 CPDF_Array* descendant_array = font_dict->GetArrayFor("DescendantFonts");
836 ASSERT_TRUE(descendant_array);
837 EXPECT_EQ(1U, descendant_array->GetCount());
838
839 // Check the CIDFontDict
840 CPDF_Dictionary* cidfont_dict = descendant_array->GetDictAt(0);
841 EXPECT_EQ("Font", cidfont_dict->GetStringFor("Type"));
842 EXPECT_EQ("CIDFontType2", cidfont_dict->GetStringFor("Subtype"));
843 EXPECT_EQ("Arial Italic", cidfont_dict->GetStringFor("BaseFont"));
844 CPDF_Dictionary* cidinfo_dict = cidfont_dict->GetDictFor("CIDSystemInfo");
845 ASSERT_TRUE(cidinfo_dict);
846 EXPECT_EQ("Adobe", cidinfo_dict->GetStringFor("Registry"));
847 EXPECT_EQ("Identity", cidinfo_dict->GetStringFor("Ordering"));
848 EXPECT_EQ(0, cidinfo_dict->GetNumberFor("Supplement"));
849 CheckFontDescriptor(cidfont_dict, FPDF_FONT_TRUETYPE, false, true, size,
850 data);
851
852 // Check widths
853 CPDF_Array* widths_array = cidfont_dict->GetArrayFor("W");
854 ASSERT_TRUE(widths_array);
855 CheckCompositeFontWidths(widths_array, typed_font);
Nicolas Penabe90aae2017-02-27 10:41:41 -0500856}
rbpotterce8e51e2017-04-28 12:42:47 -0700857
858TEST_F(FPDFEditEmbeddertest, NormalizeNegativeRotation) {
859 // Load document with a -90 degree rotation
860 EXPECT_TRUE(OpenDocument("bug_713197.pdf"));
861 FPDF_PAGE page = LoadPage(0);
862 EXPECT_NE(nullptr, page);
863
864 EXPECT_EQ(3, FPDFPage_GetRotation(page));
865 UnloadPage(page);
866}
Nicolas Penab3161852017-05-02 14:12:50 -0400867
868TEST_F(FPDFEditEmbeddertest, AddTrueTypeFontText) {
869 // Start with a blank page
870 FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792);
871 {
872 const CPDF_Font* stock_font = CPDF_Font::GetStockFont(cpdf_doc(), "Arial");
Lei Zhangd74da7b2017-05-04 13:30:29 -0700873 const uint8_t* data = stock_font->GetFont()->GetFontData();
874 const uint32_t size = stock_font->GetFont()->GetSize();
Nicolas Penab3161852017-05-02 14:12:50 -0400875 std::unique_ptr<void, FPDFFontDeleter> font(
876 FPDFText_LoadFont(document(), data, size, FPDF_FONT_TRUETYPE, 0));
877 ASSERT_TRUE(font.get());
878
879 // Add some text to the page
880 FPDF_PAGEOBJECT text_object =
881 FPDFPageObj_CreateTextObj(document(), font.get(), 12.0f);
882 EXPECT_TRUE(text_object);
883 std::unique_ptr<unsigned short, pdfium::FreeDeleter> text =
884 GetFPDFWideString(L"I am testing my loaded font, WEE.");
885 EXPECT_TRUE(FPDFText_SetText(text_object, text.get()));
886 FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 400, 400);
887 FPDFPage_InsertObject(page, text_object);
Nicolas Penab3161852017-05-02 14:12:50 -0400888 FPDF_BITMAP page_bitmap = RenderPage(page);
Dan Sinclair698aed72017-09-26 16:24:49 -0400889#if _FX_PLATFORM_ == _FX_PLATFORM_APPLE_
Nicolas Penab3161852017-05-02 14:12:50 -0400890 const char md5[] = "17d2b6cd574cf66170b09c8927529a94";
891#else
Nicolas Pena00c3cfd2017-07-10 17:29:54 -0400892 const char md5[] = "1722c6a9deed953d730de9cd13dcbd55";
Dan Sinclair698aed72017-09-26 16:24:49 -0400893#endif // _FX_PLATFORM_ == _FX_PLATFORM_APPLE_
Nicolas Penab3161852017-05-02 14:12:50 -0400894 CompareBitmap(page_bitmap, 612, 792, md5);
895 FPDFBitmap_Destroy(page_bitmap);
896
897 // Add some more text, same font
898 FPDF_PAGEOBJECT text_object2 =
899 FPDFPageObj_CreateTextObj(document(), font.get(), 15.0f);
900 std::unique_ptr<unsigned short, pdfium::FreeDeleter> text2 =
901 GetFPDFWideString(L"Bigger font size");
902 EXPECT_TRUE(FPDFText_SetText(text_object2, text2.get()));
903 FPDFPageObj_Transform(text_object2, 1, 0, 0, 1, 200, 200);
904 FPDFPage_InsertObject(page, text_object2);
Nicolas Penab3161852017-05-02 14:12:50 -0400905 }
906 FPDF_BITMAP page_bitmap2 = RenderPage(page);
Dan Sinclair698aed72017-09-26 16:24:49 -0400907#if _FX_PLATFORM_ == _FX_PLATFORM_APPLE_
Nicolas Penab3161852017-05-02 14:12:50 -0400908 const char md5_2[] = "8eded4193ff1f0f77b8b600a825e97ea";
909#else
Nicolas Pena00c3cfd2017-07-10 17:29:54 -0400910 const char md5_2[] = "9d7885072058f6c3e68ecaf32e917f30";
Dan Sinclair698aed72017-09-26 16:24:49 -0400911#endif // _FX_PLATFORM_ == _FX_PLATFORM_APPLE_
Nicolas Penab3161852017-05-02 14:12:50 -0400912 CompareBitmap(page_bitmap2, 612, 792, md5_2);
913 FPDFBitmap_Destroy(page_bitmap2);
914
Nicolas Pena207b7272017-05-26 17:37:06 -0400915 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Nicolas Penab3161852017-05-02 14:12:50 -0400916 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
917 FPDF_ClosePage(page);
Nicolas Pena3ff54002017-07-05 11:55:35 -0400918 TestAndCloseSaved(612, 792, md5_2);
Nicolas Penab3161852017-05-02 14:12:50 -0400919}
Nicolas Penaf45ade32017-05-03 10:23:49 -0400920
Jane Liueda65252017-06-07 11:31:27 -0400921TEST_F(FPDFEditEmbeddertest, TransformAnnot) {
922 // Open a file with one annotation and load its first page.
923 ASSERT_TRUE(OpenDocument("annotation_highlight_long_content.pdf"));
924 FPDF_PAGE page = FPDF_LoadPage(document(), 0);
925 ASSERT_TRUE(page);
926
927 // Add an underline annotation to the page without specifying its rectangle.
Jane Liud60e9ad2017-06-26 11:28:36 -0400928 FPDF_ANNOTATION annot = FPDFPage_CreateAnnot(page, FPDF_ANNOT_UNDERLINE);
929 ASSERT_TRUE(annot);
Jane Liueda65252017-06-07 11:31:27 -0400930
931 // FPDFPage_TransformAnnots() should run without errors when modifying
932 // annotation rectangles.
933 FPDFPage_TransformAnnots(page, 1, 2, 3, 4, 5, 6);
934
Jane Liud60e9ad2017-06-26 11:28:36 -0400935 FPDFPage_CloseAnnot(annot);
Jane Liueda65252017-06-07 11:31:27 -0400936 UnloadPage(page);
937}
938
Nicolas Penaf45ade32017-05-03 10:23:49 -0400939// TODO(npm): Add tests using Japanese fonts in other OS.
Dan Sinclair698aed72017-09-26 16:24:49 -0400940#if _FX_PLATFORM_ == _FX_PLATFORM_LINUX_
Nicolas Penaf45ade32017-05-03 10:23:49 -0400941TEST_F(FPDFEditEmbeddertest, AddCIDFontText) {
942 // Start with a blank page
943 FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792);
944 CFX_Font CIDfont;
945 {
946 // First, get the data from the font
947 CIDfont.LoadSubst("IPAGothic", 1, 0, 400, 0, 932, 0);
948 EXPECT_EQ("IPAGothic", CIDfont.GetFaceName());
949 const uint8_t* data = CIDfont.GetFontData();
950 const uint32_t size = CIDfont.GetSize();
951
952 // Load the data into a FPDF_Font.
953 std::unique_ptr<void, FPDFFontDeleter> font(
954 FPDFText_LoadFont(document(), data, size, FPDF_FONT_TRUETYPE, 1));
955 ASSERT_TRUE(font.get());
956
957 // Add some text to the page
958 FPDF_PAGEOBJECT text_object =
959 FPDFPageObj_CreateTextObj(document(), font.get(), 12.0f);
960 ASSERT_TRUE(text_object);
961 std::wstring wstr = L"ABCDEFGhijklmnop.";
962 std::unique_ptr<unsigned short, pdfium::FreeDeleter> text =
963 GetFPDFWideString(wstr);
964 EXPECT_TRUE(FPDFText_SetText(text_object, text.get()));
965 FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 200, 200);
966 FPDFPage_InsertObject(page, text_object);
967
968 // And add some Japanese characters
969 FPDF_PAGEOBJECT text_object2 =
970 FPDFPageObj_CreateTextObj(document(), font.get(), 18.0f);
971 ASSERT_TRUE(text_object2);
972 std::wstring wstr2 =
973 L"\u3053\u3093\u306B\u3061\u306f\u4e16\u754C\u3002\u3053\u3053\u306B1"
974 L"\u756A";
975 std::unique_ptr<unsigned short, pdfium::FreeDeleter> text2 =
976 GetFPDFWideString(wstr2);
977 EXPECT_TRUE(FPDFText_SetText(text_object2, text2.get()));
978 FPDFPageObj_Transform(text_object2, 1, 0, 0, 1, 100, 500);
979 FPDFPage_InsertObject(page, text_object2);
980 }
981
Nicolas Pena207b7272017-05-26 17:37:06 -0400982 // Check that the text renders properly.
Nicolas Penaf45ade32017-05-03 10:23:49 -0400983 FPDF_BITMAP page_bitmap = RenderPage(page);
984 const char md5[] = "2bc6c1aaa2252e73246a75775ccf38c2";
985 CompareBitmap(page_bitmap, 612, 792, md5);
986 FPDFBitmap_Destroy(page_bitmap);
987
988 // Save the document, close the page.
Nicolas Pena207b7272017-05-26 17:37:06 -0400989 EXPECT_TRUE(FPDFPage_GenerateContent(page));
Nicolas Penaf45ade32017-05-03 10:23:49 -0400990 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
991 FPDF_ClosePage(page);
Nicolas Pena3ff54002017-07-05 11:55:35 -0400992 TestAndCloseSaved(612, 792, md5);
Nicolas Penaf45ade32017-05-03 10:23:49 -0400993}
Dan Sinclair698aed72017-09-26 16:24:49 -0400994#endif // _FX_PLATFORM_ == _FX_PLATFORM_LINUX_
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400995
996TEST_F(FPDFEditEmbeddertest, SaveAndRender) {
Nicolas Penaa0b48aa2017-06-29 11:01:46 -0400997 const char md5[] = "3c20472b0552c0c22b88ab1ed8c6202b";
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -0400998 {
999 EXPECT_TRUE(OpenDocument("bug_779.pdf"));
1000 FPDF_PAGE page = LoadPage(0);
1001 ASSERT_NE(nullptr, page);
1002
1003 // Now add a more complex blue path.
1004 FPDF_PAGEOBJECT green_path = FPDFPageObj_CreateNewPath(20, 20);
1005 EXPECT_TRUE(FPDFPath_SetFillColor(green_path, 0, 255, 0, 200));
1006 // TODO(npm): stroking will cause the MD5s to differ.
1007 EXPECT_TRUE(FPDFPath_SetDrawMode(green_path, FPDF_FILLMODE_WINDING, 0));
1008 EXPECT_TRUE(FPDFPath_LineTo(green_path, 20, 63));
1009 EXPECT_TRUE(FPDFPath_BezierTo(green_path, 55, 55, 78, 78, 90, 90));
1010 EXPECT_TRUE(FPDFPath_LineTo(green_path, 133, 133));
1011 EXPECT_TRUE(FPDFPath_LineTo(green_path, 133, 33));
1012 EXPECT_TRUE(FPDFPath_BezierTo(green_path, 38, 33, 39, 36, 40, 40));
1013 EXPECT_TRUE(FPDFPath_Close(green_path));
1014 FPDFPage_InsertObject(page, green_path);
1015 FPDF_BITMAP page_bitmap = RenderPage(page);
Nicolas Penaa0b48aa2017-06-29 11:01:46 -04001016 CompareBitmap(page_bitmap, 612, 792, md5);
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -04001017 FPDFBitmap_Destroy(page_bitmap);
1018
1019 // Now save the result, closing the page and document
1020 EXPECT_TRUE(FPDFPage_GenerateContent(page));
1021 EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0));
1022 UnloadPage(page);
1023 }
Nicolas Pena3ff54002017-07-05 11:55:35 -04001024 TestAndCloseSaved(612, 792, md5);
Nicolas Pena9ba8fbc2017-06-28 15:31:56 -04001025}
Jane Liu28fb7ba2017-08-02 21:45:57 -04001026
1027TEST_F(FPDFEditEmbeddertest, ExtractImageBitmap) {
1028 ASSERT_TRUE(OpenDocument("embedded_images.pdf"));
1029 FPDF_PAGE page = LoadPage(0);
1030 ASSERT_TRUE(page);
Miklos Vajna92627612017-09-25 12:59:29 +02001031 ASSERT_EQ(39, FPDFPage_CountObjects(page));
Jane Liu28fb7ba2017-08-02 21:45:57 -04001032
1033 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 32);
1034 EXPECT_NE(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
1035 EXPECT_FALSE(FPDFImageObj_GetBitmap(obj));
1036
1037 obj = FPDFPage_GetObject(page, 33);
1038 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
1039 FPDF_BITMAP bitmap = FPDFImageObj_GetBitmap(obj);
1040 EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap));
1041 CompareBitmap(bitmap, 109, 88, "d65e98d968d196abf13f78aec655ffae");
1042 FPDFBitmap_Destroy(bitmap);
1043
1044 obj = FPDFPage_GetObject(page, 34);
1045 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
1046 bitmap = FPDFImageObj_GetBitmap(obj);
1047 EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap));
1048 CompareBitmap(bitmap, 103, 75, "1287711c84dbef767c435d11697661d6");
1049 FPDFBitmap_Destroy(bitmap);
1050
1051 obj = FPDFPage_GetObject(page, 35);
1052 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
1053 bitmap = FPDFImageObj_GetBitmap(obj);
1054 EXPECT_EQ(FPDFBitmap_Gray, FPDFBitmap_GetFormat(bitmap));
1055 CompareBitmap(bitmap, 92, 68, "9c6d76cb1e37ef8514f9455d759391f3");
1056 FPDFBitmap_Destroy(bitmap);
1057
1058 obj = FPDFPage_GetObject(page, 36);
1059 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
1060 bitmap = FPDFImageObj_GetBitmap(obj);
1061 EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap));
1062 CompareBitmap(bitmap, 79, 60, "15cb6a49a2e354ed0e9f45dd34e3da1a");
1063 FPDFBitmap_Destroy(bitmap);
1064
1065 obj = FPDFPage_GetObject(page, 37);
1066 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
1067 bitmap = FPDFImageObj_GetBitmap(obj);
1068 EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap));
1069 CompareBitmap(bitmap, 126, 106, "be5a64ba7890d2657522af6524118534");
1070 FPDFBitmap_Destroy(bitmap);
1071
1072 obj = FPDFPage_GetObject(page, 38);
1073 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
1074 bitmap = FPDFImageObj_GetBitmap(obj);
1075 EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap));
1076 CompareBitmap(bitmap, 194, 119, "f9e24207ee1bc0db6c543d33a5f12ec5");
1077 FPDFBitmap_Destroy(bitmap);
1078 UnloadPage(page);
1079}
Jane Liu548334e2017-08-03 16:33:40 -04001080
1081TEST_F(FPDFEditEmbeddertest, GetImageData) {
1082 EXPECT_TRUE(OpenDocument("embedded_images.pdf"));
1083 FPDF_PAGE page = LoadPage(0);
1084 ASSERT_TRUE(page);
Miklos Vajna92627612017-09-25 12:59:29 +02001085 ASSERT_EQ(39, FPDFPage_CountObjects(page));
Jane Liu548334e2017-08-03 16:33:40 -04001086
1087 // Retrieve an image object with flate-encoded data stream.
1088 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 33);
1089 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
1090
1091 // Check that the raw image data has the correct length and hash value.
1092 unsigned long len = FPDFImageObj_GetImageDataRaw(obj, nullptr, 0);
1093 std::vector<char> buf(len);
1094 EXPECT_EQ(4091u, FPDFImageObj_GetImageDataRaw(obj, buf.data(), len));
1095 EXPECT_EQ("f73802327d2e88e890f653961bcda81a",
1096 GenerateMD5Base16(reinterpret_cast<uint8_t*>(buf.data()), len));
1097
1098 // Check that the decoded image data has the correct length and hash value.
1099 len = FPDFImageObj_GetImageDataDecoded(obj, nullptr, 0);
1100 buf.clear();
1101 buf.resize(len);
1102 EXPECT_EQ(28776u, FPDFImageObj_GetImageDataDecoded(obj, buf.data(), len));
1103 EXPECT_EQ("cb3637934bb3b95a6e4ae1ea9eb9e56e",
1104 GenerateMD5Base16(reinterpret_cast<uint8_t*>(buf.data()), len));
1105
1106 // Retrieve an image obejct with DCTDecode-encoded data stream.
1107 obj = FPDFPage_GetObject(page, 37);
1108 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
1109
1110 // Check that the raw image data has the correct length and hash value.
1111 len = FPDFImageObj_GetImageDataRaw(obj, nullptr, 0);
1112 buf.clear();
1113 buf.resize(len);
1114 EXPECT_EQ(4370u, FPDFImageObj_GetImageDataRaw(obj, buf.data(), len));
1115 EXPECT_EQ("6aae1f3710335023a9e12191be66b64b",
1116 GenerateMD5Base16(reinterpret_cast<uint8_t*>(buf.data()), len));
1117
1118 // Check that the decoded image data has the correct length and hash value,
1119 // which should be the same as those of the raw data, since this image is
1120 // encoded by a single DCTDecode filter and decoding is a noop.
1121 len = FPDFImageObj_GetImageDataDecoded(obj, nullptr, 0);
1122 buf.clear();
1123 buf.resize(len);
1124 EXPECT_EQ(4370u, FPDFImageObj_GetImageDataDecoded(obj, buf.data(), len));
1125 EXPECT_EQ("6aae1f3710335023a9e12191be66b64b",
1126 GenerateMD5Base16(reinterpret_cast<uint8_t*>(buf.data()), len));
1127
1128 UnloadPage(page);
1129}
Jane Liu2e5f0ae2017-08-08 15:23:27 -04001130
1131TEST_F(FPDFEditEmbeddertest, DestroyPageObject) {
1132 FPDF_PAGEOBJECT rect = FPDFPageObj_CreateNewRect(10, 10, 20, 20);
1133 ASSERT_TRUE(rect);
1134
1135 // There should be no memory leaks with a call to FPDFPageObj_Destroy().
1136 FPDFPageObj_Destroy(rect);
1137}
Jane Liube63ab92017-08-09 14:09:34 -04001138
1139TEST_F(FPDFEditEmbeddertest, GetImageFilters) {
1140 EXPECT_TRUE(OpenDocument("embedded_images.pdf"));
1141 FPDF_PAGE page = LoadPage(0);
1142 ASSERT_TRUE(page);
1143
1144 // Verify that retrieving the filter of a non-image object would fail.
1145 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 32);
1146 ASSERT_NE(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
1147 ASSERT_EQ(0, FPDFImageObj_GetImageFilterCount(obj));
1148 EXPECT_EQ(0u, FPDFImageObj_GetImageFilter(obj, 0, nullptr, 0));
1149
1150 // Verify the returned filter string for an image object with a single filter.
1151 obj = FPDFPage_GetObject(page, 33);
1152 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
1153 ASSERT_EQ(1, FPDFImageObj_GetImageFilterCount(obj));
1154 unsigned long len = FPDFImageObj_GetImageFilter(obj, 0, nullptr, 0);
1155 std::vector<char> buf(len);
Lei Zhang0733a1b2017-08-31 12:36:31 -07001156 static constexpr char kFlateDecode[] = "FlateDecode";
1157 EXPECT_EQ(sizeof(kFlateDecode),
1158 FPDFImageObj_GetImageFilter(obj, 0, buf.data(), len));
1159 EXPECT_STREQ(kFlateDecode, buf.data());
Jane Liube63ab92017-08-09 14:09:34 -04001160 EXPECT_EQ(0u, FPDFImageObj_GetImageFilter(obj, 1, nullptr, 0));
1161
1162 // Verify all the filters for an image object with a list of filters.
1163 obj = FPDFPage_GetObject(page, 38);
1164 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
1165 ASSERT_EQ(2, FPDFImageObj_GetImageFilterCount(obj));
1166 len = FPDFImageObj_GetImageFilter(obj, 0, nullptr, 0);
1167 buf.clear();
1168 buf.resize(len);
Lei Zhang0733a1b2017-08-31 12:36:31 -07001169 static constexpr char kASCIIHexDecode[] = "ASCIIHexDecode";
1170 EXPECT_EQ(sizeof(kASCIIHexDecode),
1171 FPDFImageObj_GetImageFilter(obj, 0, buf.data(), len));
1172 EXPECT_STREQ(kASCIIHexDecode, buf.data());
Jane Liube63ab92017-08-09 14:09:34 -04001173
1174 len = FPDFImageObj_GetImageFilter(obj, 1, nullptr, 0);
1175 buf.clear();
1176 buf.resize(len);
Lei Zhang0733a1b2017-08-31 12:36:31 -07001177 static constexpr char kDCTDecode[] = "DCTDecode";
1178 EXPECT_EQ(sizeof(kDCTDecode),
1179 FPDFImageObj_GetImageFilter(obj, 1, buf.data(), len));
1180 EXPECT_STREQ(kDCTDecode, buf.data());
Jane Liube63ab92017-08-09 14:09:34 -04001181
1182 UnloadPage(page);
1183}
Jane Liuca898292017-08-16 11:25:35 -04001184
1185TEST_F(FPDFEditEmbeddertest, GetImageMetadata) {
1186 ASSERT_TRUE(OpenDocument("embedded_images.pdf"));
1187 FPDF_PAGE page = LoadPage(0);
1188 ASSERT_TRUE(page);
1189
1190 // Check that getting the metadata of a null object would fail.
1191 FPDF_IMAGEOBJ_METADATA metadata;
1192 EXPECT_FALSE(FPDFImageObj_GetImageMetadata(nullptr, page, &metadata));
1193
1194 // Check that receiving the metadata with a null metadata object would fail.
1195 FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 35);
1196 EXPECT_FALSE(FPDFImageObj_GetImageMetadata(obj, page, nullptr));
1197
1198 // Check that when retrieving an image object's metadata without passing in
1199 // |page|, all values are correct, with the last two being default values.
1200 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
1201 ASSERT_TRUE(FPDFImageObj_GetImageMetadata(obj, nullptr, &metadata));
1202 EXPECT_EQ(92u, metadata.width);
1203 EXPECT_EQ(68u, metadata.height);
1204 EXPECT_NEAR(96.000000, metadata.horizontal_dpi, 0.001);
1205 EXPECT_NEAR(96.000000, metadata.vertical_dpi, 0.001);
1206 EXPECT_EQ(0u, metadata.bits_per_pixel);
1207 EXPECT_EQ(FPDF_COLORSPACE_UNKNOWN, metadata.colorspace);
1208
1209 // Verify the metadata of a bitmap image with indexed colorspace.
1210 ASSERT_TRUE(FPDFImageObj_GetImageMetadata(obj, page, &metadata));
1211 EXPECT_EQ(92u, metadata.width);
1212 EXPECT_EQ(68u, metadata.height);
1213 EXPECT_NEAR(96.000000, metadata.horizontal_dpi, 0.001);
1214 EXPECT_NEAR(96.000000, metadata.vertical_dpi, 0.001);
1215 EXPECT_EQ(1u, metadata.bits_per_pixel);
1216 EXPECT_EQ(FPDF_COLORSPACE_INDEXED, metadata.colorspace);
1217
1218 // Verify the metadata of an image with RGB colorspace.
1219 obj = FPDFPage_GetObject(page, 37);
1220 ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj));
1221 ASSERT_TRUE(FPDFImageObj_GetImageMetadata(obj, page, &metadata));
1222 EXPECT_EQ(126u, metadata.width);
1223 EXPECT_EQ(106u, metadata.height);
1224 EXPECT_NEAR(162.173752, metadata.horizontal_dpi, 0.001);
1225 EXPECT_NEAR(162.555878, metadata.vertical_dpi, 0.001);
1226 EXPECT_EQ(24u, metadata.bits_per_pixel);
1227 EXPECT_EQ(FPDF_COLORSPACE_DEVICERGB, metadata.colorspace);
1228
1229 UnloadPage(page);
1230}