// Copyright 2016 PDFium Authors. All rights reserved. // Use of this source code is governed by a BSD-style license that can be // found in the LICENSE file. #include #include #include #include #include "core/fpdfapi/font/cpdf_font.h" #include "core/fpdfapi/page/cpdf_page.h" #include "core/fpdfapi/parser/cpdf_array.h" #include "core/fpdfapi/parser/cpdf_dictionary.h" #include "core/fpdfapi/parser/cpdf_number.h" #include "core/fpdfapi/parser/cpdf_stream.h" #include "core/fxcrt/fx_system.h" #include "fpdfsdk/fsdk_define.h" #include "public/cpp/fpdf_deleters.h" #include "public/fpdf_annot.h" #include "public/fpdf_edit.h" #include "public/fpdfview.h" #include "testing/embedder_test.h" #include "testing/gmock/include/gmock/gmock-matchers.h" #include "testing/gtest/include/gtest/gtest.h" #include "testing/test_support.h" class FPDFEditEmbeddertest : public EmbedderTest { protected: FPDF_DOCUMENT CreateNewDocument() { document_ = FPDF_CreateNewDocument(); cpdf_doc_ = CPDFDocumentFromFPDFDocument(document_); return document_; } void CheckFontDescriptor(CPDF_Dictionary* font_dict, int font_type, bool bold, bool italic, uint32_t size, const uint8_t* data) { CPDF_Dictionary* font_desc = font_dict->GetDictFor("FontDescriptor"); ASSERT_TRUE(font_desc); EXPECT_EQ("FontDescriptor", font_desc->GetStringFor("Type")); EXPECT_EQ(font_dict->GetStringFor("BaseFont"), font_desc->GetStringFor("FontName")); // Check that the font descriptor has the required keys according to spec // 1.7 Table 5.19 ASSERT_TRUE(font_desc->KeyExist("Flags")); int font_flags = font_desc->GetIntegerFor("Flags"); if (bold) EXPECT_TRUE(font_flags & FXFONT_BOLD); else EXPECT_FALSE(font_flags & FXFONT_BOLD); if (italic) EXPECT_TRUE(font_flags & FXFONT_ITALIC); else EXPECT_FALSE(font_flags & FXFONT_ITALIC); EXPECT_TRUE(font_flags & FXFONT_NONSYMBOLIC); ASSERT_TRUE(font_desc->KeyExist("FontBBox")); EXPECT_EQ(4U, font_desc->GetArrayFor("FontBBox")->GetCount()); EXPECT_TRUE(font_desc->KeyExist("ItalicAngle")); EXPECT_TRUE(font_desc->KeyExist("Ascent")); EXPECT_TRUE(font_desc->KeyExist("Descent")); EXPECT_TRUE(font_desc->KeyExist("CapHeight")); EXPECT_TRUE(font_desc->KeyExist("StemV")); ByteString present("FontFile"); ByteString absent("FontFile2"); if (font_type == FPDF_FONT_TRUETYPE) std::swap(present, absent); EXPECT_TRUE(font_desc->KeyExist(present)); EXPECT_FALSE(font_desc->KeyExist(absent)); // Check that the font stream is the one that was provided CPDF_Stream* font_stream = font_desc->GetStreamFor(present); ASSERT_EQ(size, font_stream->GetRawSize()); uint8_t* stream_data = font_stream->GetRawData(); for (size_t j = 0; j < size; j++) EXPECT_EQ(data[j], stream_data[j]) << " at byte " << j; } void CheckCompositeFontWidths(CPDF_Array* widths_array, CPDF_Font* typed_font) { // Check that W array is in a format that conforms to PDF spec 1.7 section // "Glyph Metrics in CIDFonts" (these checks are not // implementation-specific). EXPECT_GT(widths_array->GetCount(), 1U); int num_cids_checked = 0; int cur_cid = 0; for (size_t idx = 0; idx < widths_array->GetCount(); idx++) { int cid = widths_array->GetNumberAt(idx); EXPECT_GE(cid, cur_cid); ASSERT_FALSE(++idx == widths_array->GetCount()); CPDF_Object* next = widths_array->GetObjectAt(idx); if (next->IsArray()) { // We are in the c [w1 w2 ...] case CPDF_Array* arr = next->AsArray(); int cnt = static_cast(arr->GetCount()); size_t inner_idx = 0; for (cur_cid = cid; cur_cid < cid + cnt; cur_cid++) { int width = arr->GetNumberAt(inner_idx++); EXPECT_EQ(width, typed_font->GetCharWidthF(cur_cid)) << " at cid " << cur_cid; } num_cids_checked += cnt; continue; } // Otherwise, are in the c_first c_last w case. ASSERT_TRUE(next->IsNumber()); int last_cid = next->AsNumber()->GetInteger(); ASSERT_FALSE(++idx == widths_array->GetCount()); int width = widths_array->GetNumberAt(idx); for (cur_cid = cid; cur_cid <= last_cid; cur_cid++) { EXPECT_EQ(width, typed_font->GetCharWidthF(cur_cid)) << " at cid " << cur_cid; } num_cids_checked += last_cid - cid + 1; } // Make sure we have a good amount of cids described EXPECT_GT(num_cids_checked, 900); } CPDF_Document* cpdf_doc() { return cpdf_doc_; } private: CPDF_Document* cpdf_doc_; }; namespace { const char kExpectedPDF[] = "%PDF-1.7\r\n" "%\xA1\xB3\xC5\xD7\r\n" "1 0 obj\r\n" "<>\r\n" "endobj\r\n" "2 0 obj\r\n" "<>\r\n" "endobj\r\n" "3 0 obj\r\n" "<>\r\n" "endobj\r\n" "4 0 obj\r\n" "<>>>" "/Rotate 0/Type/Page" ">>\r\n" "endobj\r\n" "5 0 obj\r\n" "<>\r\n" "endobj\r\n" "xref\r\n" "0 6\r\n" "0000000000 65535 f\r\n" "0000000017 00000 n\r\n" "0000000066 00000 n\r\n" "0000000122 00000 n\r\n" "0000000192 00000 n\r\n" "0000000311 00000 n\r\n" "trailer\r\n" "<<\r\n" "/Root 1 0 R\r\n" "/Info 3 0 R\r\n" "/Size 6/ID\\[<.*><.*>\\]>>\r\n" "startxref\r\n" "354\r\n" "%%EOF\r\n"; } // namespace TEST_F(FPDFEditEmbeddertest, EmptyCreation) { EXPECT_TRUE(CreateEmptyDocument()); FPDF_PAGE page = FPDFPage_New(document(), 0, 640.0, 480.0); EXPECT_NE(nullptr, page); // The FPDFPage_GenerateContent call should do nothing. EXPECT_TRUE(FPDFPage_GenerateContent(page)); EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); EXPECT_THAT(GetString(), testing::MatchesRegex(std::string( kExpectedPDF, sizeof(kExpectedPDF)))); FPDF_ClosePage(page); } // Regression test for https://crbug.com/667012 TEST_F(FPDFEditEmbeddertest, RasterizePDF) { const char kAllBlackMd5sum[] = "5708fc5c4a8bd0abde99c8e8f0390615"; // Get the bitmap for the original document/ FPDF_BITMAP orig_bitmap; { EXPECT_TRUE(OpenDocument("black.pdf")); FPDF_PAGE orig_page = LoadPage(0); EXPECT_NE(nullptr, orig_page); orig_bitmap = RenderPage(orig_page); CompareBitmap(orig_bitmap, 612, 792, kAllBlackMd5sum); UnloadPage(orig_page); } // Create a new document from |orig_bitmap| and save it. { FPDF_DOCUMENT temp_doc = FPDF_CreateNewDocument(); FPDF_PAGE temp_page = FPDFPage_New(temp_doc, 0, 612, 792); // Add the bitmap to an image object and add the image object to the output // page. FPDF_PAGEOBJECT temp_img = FPDFPageObj_NewImageObj(temp_doc); EXPECT_TRUE(FPDFImageObj_SetBitmap(&temp_page, 1, temp_img, orig_bitmap)); EXPECT_TRUE(FPDFImageObj_SetMatrix(temp_img, 612, 0, 0, 792, 0, 0)); FPDFPage_InsertObject(temp_page, temp_img); EXPECT_TRUE(FPDFPage_GenerateContent(temp_page)); EXPECT_TRUE(FPDF_SaveAsCopy(temp_doc, this, 0)); FPDF_ClosePage(temp_page); FPDF_CloseDocument(temp_doc); } FPDFBitmap_Destroy(orig_bitmap); // Get the generated content. Make sure it is at least as big as the original // PDF. EXPECT_GT(GetString().size(), 923U); TestAndCloseSaved(612, 792, kAllBlackMd5sum); } TEST_F(FPDFEditEmbeddertest, AddPaths) { // Start with a blank page FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792); // We will first add a red rectangle FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(10, 10, 20, 20); ASSERT_NE(nullptr, red_rect); // Expect false when trying to set colors out of range EXPECT_FALSE(FPDFPath_SetStrokeColor(red_rect, 100, 100, 100, 300)); EXPECT_FALSE(FPDFPath_SetFillColor(red_rect, 200, 256, 200, 0)); // Fill rectangle with red and insert to the page EXPECT_TRUE(FPDFPath_SetFillColor(red_rect, 255, 0, 0, 255)); EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0)); FPDFPage_InsertObject(page, red_rect); FPDF_BITMAP page_bitmap = RenderPage(page); CompareBitmap(page_bitmap, 612, 792, "66d02eaa6181e2c069ce2ea99beda497"); FPDFBitmap_Destroy(page_bitmap); // Now add to that a green rectangle with some medium alpha FPDF_PAGEOBJECT green_rect = FPDFPageObj_CreateNewRect(100, 100, 40, 40); EXPECT_TRUE(FPDFPath_SetFillColor(green_rect, 0, 255, 0, 128)); // Make sure the type of the rectangle is a path. EXPECT_EQ(FPDF_PAGEOBJ_PATH, FPDFPageObj_GetType(green_rect)); // Make sure we get back the same color we set previously. unsigned int R; unsigned int G; unsigned int B; unsigned int A; EXPECT_TRUE(FPDFPath_GetFillColor(green_rect, &R, &G, &B, &A)); EXPECT_EQ(0U, R); EXPECT_EQ(255U, G); EXPECT_EQ(0U, B); EXPECT_EQ(128U, A); // Make sure the path has 5 points (1 FXPT_TYPE::MoveTo and 4 // FXPT_TYPE::LineTo). ASSERT_EQ(5, FPDFPath_CountPoint(green_rect)); EXPECT_TRUE(FPDFPath_SetDrawMode(green_rect, FPDF_FILLMODE_WINDING, 0)); FPDFPage_InsertObject(page, green_rect); page_bitmap = RenderPage(page); CompareBitmap(page_bitmap, 612, 792, "7b0b87604594e773add528fae567a558"); FPDFBitmap_Destroy(page_bitmap); // Add a black triangle. FPDF_PAGEOBJECT black_path = FPDFPageObj_CreateNewPath(400, 100); EXPECT_TRUE(FPDFPath_SetFillColor(black_path, 0, 0, 0, 200)); EXPECT_TRUE(FPDFPath_SetDrawMode(black_path, FPDF_FILLMODE_ALTERNATE, 0)); EXPECT_TRUE(FPDFPath_LineTo(black_path, 400, 200)); EXPECT_TRUE(FPDFPath_LineTo(black_path, 300, 100)); EXPECT_TRUE(FPDFPath_Close(black_path)); // Make sure the path has 3 points (1 FXPT_TYPE::MoveTo and 2 // FXPT_TYPE::LineTo). ASSERT_EQ(3, FPDFPath_CountPoint(black_path)); FPDFPage_InsertObject(page, black_path); page_bitmap = RenderPage(page); CompareBitmap(page_bitmap, 612, 792, "eadc8020a14dfcf091da2688733d8806"); FPDFBitmap_Destroy(page_bitmap); // Now add a more complex blue path. FPDF_PAGEOBJECT blue_path = FPDFPageObj_CreateNewPath(200, 200); EXPECT_TRUE(FPDFPath_SetFillColor(blue_path, 0, 0, 255, 255)); EXPECT_TRUE(FPDFPath_SetDrawMode(blue_path, FPDF_FILLMODE_WINDING, 0)); EXPECT_TRUE(FPDFPath_LineTo(blue_path, 230, 230)); EXPECT_TRUE(FPDFPath_BezierTo(blue_path, 250, 250, 280, 280, 300, 300)); EXPECT_TRUE(FPDFPath_LineTo(blue_path, 325, 325)); EXPECT_TRUE(FPDFPath_LineTo(blue_path, 350, 325)); EXPECT_TRUE(FPDFPath_BezierTo(blue_path, 375, 330, 390, 360, 400, 400)); EXPECT_TRUE(FPDFPath_Close(blue_path)); FPDFPage_InsertObject(page, blue_path); page_bitmap = RenderPage(page); const char last_md5[] = "9823e1a21bd9b72b6a442ba4f12af946"; CompareBitmap(page_bitmap, 612, 792, last_md5); FPDFBitmap_Destroy(page_bitmap); // Now save the result, closing the page and document EXPECT_TRUE(FPDFPage_GenerateContent(page)); EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); FPDF_ClosePage(page); // Render the saved result TestAndCloseSaved(612, 792, last_md5); } TEST_F(FPDFEditEmbeddertest, PathsPoints) { CreateNewDocument(); FPDF_PAGEOBJECT img = FPDFPageObj_NewImageObj(document_); // This should fail gracefully, even if img is not a path. ASSERT_EQ(-1, FPDFPath_CountPoint(img)); // This should fail gracefully, even if path is NULL. ASSERT_EQ(-1, FPDFPath_CountPoint(nullptr)); FPDFPageObj_Destroy(img); } TEST_F(FPDFEditEmbeddertest, PathOnTopOfText) { // Load document with some text EXPECT_TRUE(OpenDocument("hello_world.pdf")); FPDF_PAGE page = LoadPage(0); EXPECT_NE(nullptr, page); // Add an opaque rectangle on top of some of the text. FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(20, 100, 50, 50); EXPECT_TRUE(FPDFPath_SetFillColor(red_rect, 255, 0, 0, 255)); EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0)); FPDFPage_InsertObject(page, red_rect); // Add a transparent triangle on top of other part of the text. FPDF_PAGEOBJECT black_path = FPDFPageObj_CreateNewPath(20, 50); EXPECT_TRUE(FPDFPath_SetFillColor(black_path, 0, 0, 0, 100)); EXPECT_TRUE(FPDFPath_SetDrawMode(black_path, FPDF_FILLMODE_ALTERNATE, 0)); EXPECT_TRUE(FPDFPath_LineTo(black_path, 30, 80)); EXPECT_TRUE(FPDFPath_LineTo(black_path, 40, 10)); EXPECT_TRUE(FPDFPath_Close(black_path)); FPDFPage_InsertObject(page, black_path); // Render and check the result. Text is slightly different on Mac. FPDF_BITMAP bitmap = RenderPage(page); #if _FXM_PLATFORM_ == _FXM_PLATFORM_APPLE_ const char md5[] = "f9e6fa74230f234286bfcada9f7606d8"; #else const char md5[] = "bc6e6eb50dda4695ba0fb4d04ed82ada"; #endif CompareBitmap(bitmap, 200, 200, md5); FPDFBitmap_Destroy(bitmap); UnloadPage(page); } TEST_F(FPDFEditEmbeddertest, EditOverExistingContent) { // Load document with existing content EXPECT_TRUE(OpenDocument("bug_717.pdf")); FPDF_PAGE page = LoadPage(0); EXPECT_NE(nullptr, page); // Add a transparent rectangle on top of the existing content FPDF_PAGEOBJECT red_rect2 = FPDFPageObj_CreateNewRect(90, 700, 25, 50); EXPECT_TRUE(FPDFPath_SetFillColor(red_rect2, 255, 0, 0, 100)); EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect2, FPDF_FILLMODE_ALTERNATE, 0)); FPDFPage_InsertObject(page, red_rect2); // Add an opaque rectangle on top of the existing content FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(115, 700, 25, 50); EXPECT_TRUE(FPDFPath_SetFillColor(red_rect, 255, 0, 0, 255)); EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0)); FPDFPage_InsertObject(page, red_rect); FPDF_BITMAP bitmap = RenderPage(page); CompareBitmap(bitmap, 612, 792, "ad04e5bd0f471a9a564fb034bd0fb073"); FPDFBitmap_Destroy(bitmap); EXPECT_TRUE(FPDFPage_GenerateContent(page)); // Now save the result, closing the page and document EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); UnloadPage(page); // Render the saved result without closing the page and document TestSaved(612, 792, "ad04e5bd0f471a9a564fb034bd0fb073"); ClearString(); // Add another opaque rectangle on top of the existing content FPDF_PAGEOBJECT green_rect = FPDFPageObj_CreateNewRect(150, 700, 25, 50); EXPECT_TRUE(FPDFPath_SetFillColor(green_rect, 0, 255, 0, 255)); EXPECT_TRUE(FPDFPath_SetDrawMode(green_rect, FPDF_FILLMODE_ALTERNATE, 0)); FPDFPage_InsertObject(m_SavedPage, green_rect); // Add another transparent rectangle on top of existing content FPDF_PAGEOBJECT green_rect2 = FPDFPageObj_CreateNewRect(175, 700, 25, 50); EXPECT_TRUE(FPDFPath_SetFillColor(green_rect2, 0, 255, 0, 100)); EXPECT_TRUE(FPDFPath_SetDrawMode(green_rect2, FPDF_FILLMODE_ALTERNATE, 0)); FPDFPage_InsertObject(m_SavedPage, green_rect2); FPDF_BITMAP new_bitmap = RenderPageWithFlags(m_SavedPage, m_SavedForm, 0); const char last_md5[] = "4b5b00f824620f8c9b8801ebb98e1cdd"; CompareBitmap(new_bitmap, 612, 792, last_md5); FPDFBitmap_Destroy(new_bitmap); EXPECT_TRUE(FPDFPage_GenerateContent(m_SavedPage)); // Now save the result, closing the page and document EXPECT_TRUE(FPDF_SaveAsCopy(m_SavedDocument, this, 0)); CloseSaved(); // Render the saved result TestAndCloseSaved(612, 792, last_md5); } TEST_F(FPDFEditEmbeddertest, AddStrokedPaths) { // Start with a blank page FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792); // Add a large stroked rectangle (fill color should not affect it). FPDF_PAGEOBJECT rect = FPDFPageObj_CreateNewRect(20, 20, 200, 400); EXPECT_TRUE(FPDFPath_SetFillColor(rect, 255, 0, 0, 255)); EXPECT_TRUE(FPDFPath_SetStrokeColor(rect, 0, 255, 0, 255)); EXPECT_TRUE(FPDFPath_SetStrokeWidth(rect, 15.0f)); EXPECT_TRUE(FPDFPath_SetDrawMode(rect, 0, 1)); FPDFPage_InsertObject(page, rect); FPDF_BITMAP page_bitmap = RenderPage(page); CompareBitmap(page_bitmap, 612, 792, "64bd31f862a89e0a9e505a5af6efd506"); FPDFBitmap_Destroy(page_bitmap); // Add crossed-checkmark FPDF_PAGEOBJECT check = FPDFPageObj_CreateNewPath(300, 500); EXPECT_TRUE(FPDFPath_LineTo(check, 400, 400)); EXPECT_TRUE(FPDFPath_LineTo(check, 600, 600)); EXPECT_TRUE(FPDFPath_MoveTo(check, 400, 600)); EXPECT_TRUE(FPDFPath_LineTo(check, 600, 400)); EXPECT_TRUE(FPDFPath_SetStrokeColor(check, 128, 128, 128, 180)); EXPECT_TRUE(FPDFPath_SetStrokeWidth(check, 8.35f)); EXPECT_TRUE(FPDFPath_SetDrawMode(check, 0, 1)); FPDFPage_InsertObject(page, check); page_bitmap = RenderPage(page); CompareBitmap(page_bitmap, 612, 792, "4b6f3b9d25c4e194821217d5016c3724"); FPDFBitmap_Destroy(page_bitmap); // Add stroked and filled oval-ish path. FPDF_PAGEOBJECT path = FPDFPageObj_CreateNewPath(250, 100); EXPECT_TRUE(FPDFPath_BezierTo(path, 180, 166, 180, 233, 250, 300)); EXPECT_TRUE(FPDFPath_LineTo(path, 255, 305)); EXPECT_TRUE(FPDFPath_BezierTo(path, 325, 233, 325, 166, 255, 105)); EXPECT_TRUE(FPDFPath_Close(path)); EXPECT_TRUE(FPDFPath_SetFillColor(path, 200, 128, 128, 100)); EXPECT_TRUE(FPDFPath_SetStrokeColor(path, 128, 200, 128, 150)); EXPECT_TRUE(FPDFPath_SetStrokeWidth(path, 10.5f)); EXPECT_TRUE(FPDFPath_SetDrawMode(path, FPDF_FILLMODE_ALTERNATE, 1)); FPDFPage_InsertObject(page, path); page_bitmap = RenderPage(page); CompareBitmap(page_bitmap, 612, 792, "ff3e6a22326754944cc6e56609acd73b"); FPDFBitmap_Destroy(page_bitmap); FPDF_ClosePage(page); } TEST_F(FPDFEditEmbeddertest, AddStandardFontText) { // Start with a blank page FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792); // Add some text to the page FPDF_PAGEOBJECT text_object1 = FPDFPageObj_NewTextObj(document(), "Arial", 12.0f); EXPECT_TRUE(text_object1); std::unique_ptr text1 = GetFPDFWideString(L"I'm at the bottom of the page"); EXPECT_TRUE(FPDFText_SetText(text_object1, text1.get())); FPDFPageObj_Transform(text_object1, 1, 0, 0, 1, 20, 20); FPDFPage_InsertObject(page, text_object1); FPDF_BITMAP page_bitmap = RenderPage(page); #if _FXM_PLATFORM_ == _FXM_PLATFORM_APPLE_ const char md5[] = "a4dddc1a3930fa694bbff9789dab4161"; #else const char md5[] = "7a35771853a1cbba38f6775807878625"; #endif CompareBitmap(page_bitmap, 612, 792, md5); FPDFBitmap_Destroy(page_bitmap); // Try another font FPDF_PAGEOBJECT text_object2 = FPDFPageObj_NewTextObj(document(), "TimesNewRomanBold", 15.0f); EXPECT_TRUE(text_object2); std::unique_ptr text2 = GetFPDFWideString(L"Hi, I'm Bold. Times New Roman Bold."); EXPECT_TRUE(FPDFText_SetText(text_object2, text2.get())); FPDFPageObj_Transform(text_object2, 1, 0, 0, 1, 100, 600); FPDFPage_InsertObject(page, text_object2); page_bitmap = RenderPage(page); #if _FXM_PLATFORM_ == _FXM_PLATFORM_APPLE_ const char md5_2[] = "a5c4ace4c6f27644094813fe1441a21c"; #elif _FXM_PLATFORM_ == _FXM_PLATFORM_WINDOWS_ const char md5_2[] = "b231b329a4b566fb9b42bfc15fe59bb7"; #else const char md5_2[] = "f85fae151851436072b7b3c6703e506a"; #endif CompareBitmap(page_bitmap, 612, 792, md5_2); FPDFBitmap_Destroy(page_bitmap); // And some randomly transformed text FPDF_PAGEOBJECT text_object3 = FPDFPageObj_NewTextObj(document(), "Courier-Bold", 20.0f); EXPECT_TRUE(text_object3); std::unique_ptr text3 = GetFPDFWideString(L"Can you read me? <:)>"); EXPECT_TRUE(FPDFText_SetText(text_object3, text3.get())); FPDFPageObj_Transform(text_object3, 1, 1.5, 2, 0.5, 200, 200); FPDFPage_InsertObject(page, text_object3); page_bitmap = RenderPage(page); #if _FXM_PLATFORM_ == _FXM_PLATFORM_APPLE_ const char md5_3[] = "40b3ef04f915ff4c4208948001763544"; #elif _FXM_PLATFORM_ == _FXM_PLATFORM_WINDOWS_ const char md5_3[] = "ba874b3b137f984510c4e287ed4ba7ae"; #else const char md5_3[] = "c5aed6a8ef05558c8c47d58c87cbcb46"; #endif CompareBitmap(page_bitmap, 612, 792, md5_3); FPDFBitmap_Destroy(page_bitmap); // TODO(npm): Why are there issues with text rotated by 90 degrees? // TODO(npm): FPDF_SaveAsCopy not giving the desired result after this. FPDF_ClosePage(page); } TEST_F(FPDFEditEmbeddertest, GraphicsData) { // New page std::unique_ptr page( FPDFPage_New(CreateNewDocument(), 0, 612, 792)); // Create a rect with nontrivial graphics FPDF_PAGEOBJECT rect1 = FPDFPageObj_CreateNewRect(10, 10, 100, 100); FPDFPageObj_SetBlendMode(rect1, "Color"); FPDFPage_InsertObject(page.get(), rect1); EXPECT_TRUE(FPDFPage_GenerateContent(page.get())); // Check that the ExtGState was created CPDF_Page* the_page = CPDFPageFromFPDFPage(page.get()); CPDF_Dictionary* graphics_dict = the_page->m_pResources->GetDictFor("ExtGState"); ASSERT_TRUE(graphics_dict); EXPECT_EQ(2, static_cast(graphics_dict->GetCount())); // Add a text object causing no change to the graphics dictionary FPDF_PAGEOBJECT text1 = FPDFPageObj_NewTextObj(document(), "Arial", 12.0f); // Only alpha, the last component, matters for the graphics dictionary. And // the default value is 255. EXPECT_TRUE(FPDFText_SetFillColor(text1, 100, 100, 100, 255)); FPDFPage_InsertObject(page.get(), text1); EXPECT_TRUE(FPDFPage_GenerateContent(page.get())); EXPECT_EQ(2, static_cast(graphics_dict->GetCount())); // Add a text object increasing the size of the graphics dictionary FPDF_PAGEOBJECT text2 = FPDFPageObj_NewTextObj(document(), "Times-Roman", 12.0f); FPDFPage_InsertObject(page.get(), text2); FPDFPageObj_SetBlendMode(text2, "Darken"); EXPECT_TRUE(FPDFText_SetFillColor(text2, 0, 0, 255, 150)); EXPECT_TRUE(FPDFPage_GenerateContent(page.get())); EXPECT_EQ(3, static_cast(graphics_dict->GetCount())); // Add a path that should reuse graphics FPDF_PAGEOBJECT path = FPDFPageObj_CreateNewPath(400, 100); FPDFPageObj_SetBlendMode(path, "Darken"); EXPECT_TRUE(FPDFPath_SetFillColor(path, 200, 200, 100, 150)); FPDFPage_InsertObject(page.get(), path); EXPECT_TRUE(FPDFPage_GenerateContent(page.get())); EXPECT_EQ(3, static_cast(graphics_dict->GetCount())); // Add a rect increasing the size of the graphics dictionary FPDF_PAGEOBJECT rect2 = FPDFPageObj_CreateNewRect(10, 10, 100, 100); FPDFPageObj_SetBlendMode(rect2, "Darken"); EXPECT_TRUE(FPDFPath_SetFillColor(rect2, 0, 0, 255, 150)); EXPECT_TRUE(FPDFPath_SetStrokeColor(rect2, 0, 0, 0, 200)); FPDFPage_InsertObject(page.get(), rect2); EXPECT_TRUE(FPDFPage_GenerateContent(page.get())); EXPECT_EQ(4, static_cast(graphics_dict->GetCount())); } TEST_F(FPDFEditEmbeddertest, DoubleGenerating) { // Start with a blank page FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792); // Add a red rectangle with some non-default alpha FPDF_PAGEOBJECT rect = FPDFPageObj_CreateNewRect(10, 10, 100, 100); EXPECT_TRUE(FPDFPath_SetFillColor(rect, 255, 0, 0, 128)); EXPECT_TRUE(FPDFPath_SetDrawMode(rect, FPDF_FILLMODE_WINDING, 0)); FPDFPage_InsertObject(page, rect); EXPECT_TRUE(FPDFPage_GenerateContent(page)); // Check the ExtGState CPDF_Page* the_page = CPDFPageFromFPDFPage(page); CPDF_Dictionary* graphics_dict = the_page->m_pResources->GetDictFor("ExtGState"); ASSERT_TRUE(graphics_dict); EXPECT_EQ(2, static_cast(graphics_dict->GetCount())); // Check the bitmap FPDF_BITMAP page_bitmap = RenderPage(page); CompareBitmap(page_bitmap, 612, 792, "5384da3406d62360ffb5cac4476fff1c"); FPDFBitmap_Destroy(page_bitmap); // Never mind, my new favorite color is blue, increase alpha EXPECT_TRUE(FPDFPath_SetFillColor(rect, 0, 0, 255, 180)); EXPECT_TRUE(FPDFPage_GenerateContent(page)); EXPECT_EQ(3, static_cast(graphics_dict->GetCount())); // Check that bitmap displays changed content page_bitmap = RenderPage(page); CompareBitmap(page_bitmap, 612, 792, "2e51656f5073b0bee611d9cd086aa09c"); FPDFBitmap_Destroy(page_bitmap); // And now generate, without changes EXPECT_TRUE(FPDFPage_GenerateContent(page)); EXPECT_EQ(3, static_cast(graphics_dict->GetCount())); page_bitmap = RenderPage(page); CompareBitmap(page_bitmap, 612, 792, "2e51656f5073b0bee611d9cd086aa09c"); FPDFBitmap_Destroy(page_bitmap); // Add some text to the page FPDF_PAGEOBJECT text_object = FPDFPageObj_NewTextObj(document(), "Arial", 12.0f); std::unique_ptr text = GetFPDFWideString(L"Something something #text# something"); EXPECT_TRUE(FPDFText_SetText(text_object, text.get())); FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 300, 300); FPDFPage_InsertObject(page, text_object); EXPECT_TRUE(FPDFPage_GenerateContent(page)); CPDF_Dictionary* font_dict = the_page->m_pResources->GetDictFor("Font"); ASSERT_TRUE(font_dict); EXPECT_EQ(1, static_cast(font_dict->GetCount())); // Generate yet again, check dicts are reasonably sized EXPECT_TRUE(FPDFPage_GenerateContent(page)); EXPECT_EQ(3, static_cast(graphics_dict->GetCount())); EXPECT_EQ(1, static_cast(font_dict->GetCount())); FPDF_ClosePage(page); } TEST_F(FPDFEditEmbeddertest, LoadSimpleType1Font) { CreateNewDocument(); // TODO(npm): use other fonts after disallowing loading any font as any type const CPDF_Font* stock_font = CPDF_Font::GetStockFont(cpdf_doc(), "Times-Bold"); const uint8_t* data = stock_font->GetFont()->GetFontData(); const uint32_t size = stock_font->GetFont()->GetSize(); std::unique_ptr font( FPDFText_LoadFont(document(), data, size, FPDF_FONT_TYPE1, false)); ASSERT_TRUE(font.get()); CPDF_Font* typed_font = static_cast(font.get()); EXPECT_TRUE(typed_font->IsType1Font()); CPDF_Dictionary* font_dict = typed_font->GetFontDict(); EXPECT_EQ("Font", font_dict->GetStringFor("Type")); EXPECT_EQ("Type1", font_dict->GetStringFor("Subtype")); EXPECT_EQ("Times New Roman Bold", font_dict->GetStringFor("BaseFont")); ASSERT_TRUE(font_dict->KeyExist("FirstChar")); ASSERT_TRUE(font_dict->KeyExist("LastChar")); EXPECT_EQ(32, font_dict->GetIntegerFor("FirstChar")); EXPECT_EQ(255, font_dict->GetIntegerFor("LastChar")); CPDF_Array* widths_array = font_dict->GetArrayFor("Widths"); ASSERT_TRUE(widths_array); ASSERT_EQ(224U, widths_array->GetCount()); EXPECT_EQ(250, widths_array->GetNumberAt(0)); EXPECT_EQ(569, widths_array->GetNumberAt(11)); EXPECT_EQ(500, widths_array->GetNumberAt(223)); CheckFontDescriptor(font_dict, FPDF_FONT_TYPE1, true, false, size, data); } TEST_F(FPDFEditEmbeddertest, LoadSimpleTrueTypeFont) { CreateNewDocument(); const CPDF_Font* stock_font = CPDF_Font::GetStockFont(cpdf_doc(), "Courier"); const uint8_t* data = stock_font->GetFont()->GetFontData(); const uint32_t size = stock_font->GetFont()->GetSize(); std::unique_ptr font( FPDFText_LoadFont(document(), data, size, FPDF_FONT_TRUETYPE, false)); ASSERT_TRUE(font.get()); CPDF_Font* typed_font = static_cast(font.get()); EXPECT_TRUE(typed_font->IsTrueTypeFont()); CPDF_Dictionary* font_dict = typed_font->GetFontDict(); EXPECT_EQ("Font", font_dict->GetStringFor("Type")); EXPECT_EQ("TrueType", font_dict->GetStringFor("Subtype")); EXPECT_EQ("Courier New", font_dict->GetStringFor("BaseFont")); ASSERT_TRUE(font_dict->KeyExist("FirstChar")); ASSERT_TRUE(font_dict->KeyExist("LastChar")); EXPECT_EQ(32, font_dict->GetIntegerFor("FirstChar")); EXPECT_EQ(255, font_dict->GetIntegerFor("LastChar")); CPDF_Array* widths_array = font_dict->GetArrayFor("Widths"); ASSERT_TRUE(widths_array); ASSERT_EQ(224U, widths_array->GetCount()); EXPECT_EQ(600, widths_array->GetNumberAt(33)); EXPECT_EQ(600, widths_array->GetNumberAt(74)); EXPECT_EQ(600, widths_array->GetNumberAt(223)); CheckFontDescriptor(font_dict, FPDF_FONT_TRUETYPE, false, false, size, data); } TEST_F(FPDFEditEmbeddertest, LoadCIDType0Font) { CreateNewDocument(); const CPDF_Font* stock_font = CPDF_Font::GetStockFont(cpdf_doc(), "Times-Roman"); const uint8_t* data = stock_font->GetFont()->GetFontData(); const uint32_t size = stock_font->GetFont()->GetSize(); std::unique_ptr font( FPDFText_LoadFont(document(), data, size, FPDF_FONT_TYPE1, 1)); ASSERT_TRUE(font.get()); CPDF_Font* typed_font = static_cast(font.get()); EXPECT_TRUE(typed_font->IsCIDFont()); // Check font dictionary entries CPDF_Dictionary* font_dict = typed_font->GetFontDict(); EXPECT_EQ("Font", font_dict->GetStringFor("Type")); EXPECT_EQ("Type0", font_dict->GetStringFor("Subtype")); EXPECT_EQ("Times New Roman-Identity-H", font_dict->GetStringFor("BaseFont")); EXPECT_EQ("Identity-H", font_dict->GetStringFor("Encoding")); CPDF_Array* descendant_array = font_dict->GetArrayFor("DescendantFonts"); ASSERT_TRUE(descendant_array); EXPECT_EQ(1U, descendant_array->GetCount()); // Check the CIDFontDict CPDF_Dictionary* cidfont_dict = descendant_array->GetDictAt(0); EXPECT_EQ("Font", cidfont_dict->GetStringFor("Type")); EXPECT_EQ("CIDFontType0", cidfont_dict->GetStringFor("Subtype")); EXPECT_EQ("Times New Roman", cidfont_dict->GetStringFor("BaseFont")); CPDF_Dictionary* cidinfo_dict = cidfont_dict->GetDictFor("CIDSystemInfo"); ASSERT_TRUE(cidinfo_dict); EXPECT_EQ("Adobe", cidinfo_dict->GetStringFor("Registry")); EXPECT_EQ("Identity", cidinfo_dict->GetStringFor("Ordering")); EXPECT_EQ(0, cidinfo_dict->GetNumberFor("Supplement")); CheckFontDescriptor(cidfont_dict, FPDF_FONT_TYPE1, false, false, size, data); // Check widths CPDF_Array* widths_array = cidfont_dict->GetArrayFor("W"); ASSERT_TRUE(widths_array); EXPECT_GT(widths_array->GetCount(), 1U); CheckCompositeFontWidths(widths_array, typed_font); } TEST_F(FPDFEditEmbeddertest, LoadCIDType2Font) { CreateNewDocument(); const CPDF_Font* stock_font = CPDF_Font::GetStockFont(cpdf_doc(), "Helvetica-Oblique"); const uint8_t* data = stock_font->GetFont()->GetFontData(); const uint32_t size = stock_font->GetFont()->GetSize(); std::unique_ptr font( FPDFText_LoadFont(document(), data, size, FPDF_FONT_TRUETYPE, 1)); ASSERT_TRUE(font.get()); CPDF_Font* typed_font = static_cast(font.get()); EXPECT_TRUE(typed_font->IsCIDFont()); // Check font dictionary entries CPDF_Dictionary* font_dict = typed_font->GetFontDict(); EXPECT_EQ("Font", font_dict->GetStringFor("Type")); EXPECT_EQ("Type0", font_dict->GetStringFor("Subtype")); EXPECT_EQ("Arial Italic", font_dict->GetStringFor("BaseFont")); EXPECT_EQ("Identity-H", font_dict->GetStringFor("Encoding")); CPDF_Array* descendant_array = font_dict->GetArrayFor("DescendantFonts"); ASSERT_TRUE(descendant_array); EXPECT_EQ(1U, descendant_array->GetCount()); // Check the CIDFontDict CPDF_Dictionary* cidfont_dict = descendant_array->GetDictAt(0); EXPECT_EQ("Font", cidfont_dict->GetStringFor("Type")); EXPECT_EQ("CIDFontType2", cidfont_dict->GetStringFor("Subtype")); EXPECT_EQ("Arial Italic", cidfont_dict->GetStringFor("BaseFont")); CPDF_Dictionary* cidinfo_dict = cidfont_dict->GetDictFor("CIDSystemInfo"); ASSERT_TRUE(cidinfo_dict); EXPECT_EQ("Adobe", cidinfo_dict->GetStringFor("Registry")); EXPECT_EQ("Identity", cidinfo_dict->GetStringFor("Ordering")); EXPECT_EQ(0, cidinfo_dict->GetNumberFor("Supplement")); CheckFontDescriptor(cidfont_dict, FPDF_FONT_TRUETYPE, false, true, size, data); // Check widths CPDF_Array* widths_array = cidfont_dict->GetArrayFor("W"); ASSERT_TRUE(widths_array); CheckCompositeFontWidths(widths_array, typed_font); } TEST_F(FPDFEditEmbeddertest, NormalizeNegativeRotation) { // Load document with a -90 degree rotation EXPECT_TRUE(OpenDocument("bug_713197.pdf")); FPDF_PAGE page = LoadPage(0); EXPECT_NE(nullptr, page); EXPECT_EQ(3, FPDFPage_GetRotation(page)); UnloadPage(page); } TEST_F(FPDFEditEmbeddertest, AddTrueTypeFontText) { // Start with a blank page FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792); { const CPDF_Font* stock_font = CPDF_Font::GetStockFont(cpdf_doc(), "Arial"); const uint8_t* data = stock_font->GetFont()->GetFontData(); const uint32_t size = stock_font->GetFont()->GetSize(); std::unique_ptr font( FPDFText_LoadFont(document(), data, size, FPDF_FONT_TRUETYPE, 0)); ASSERT_TRUE(font.get()); // Add some text to the page FPDF_PAGEOBJECT text_object = FPDFPageObj_CreateTextObj(document(), font.get(), 12.0f); EXPECT_TRUE(text_object); std::unique_ptr text = GetFPDFWideString(L"I am testing my loaded font, WEE."); EXPECT_TRUE(FPDFText_SetText(text_object, text.get())); FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 400, 400); FPDFPage_InsertObject(page, text_object); FPDF_BITMAP page_bitmap = RenderPage(page); #if _FXM_PLATFORM_ == _FXM_PLATFORM_APPLE_ const char md5[] = "17d2b6cd574cf66170b09c8927529a94"; #else const char md5[] = "1722c6a9deed953d730de9cd13dcbd55"; #endif // _FXM_PLATFORM_ == _FXM_PLATFORM_APPLE_ CompareBitmap(page_bitmap, 612, 792, md5); FPDFBitmap_Destroy(page_bitmap); // Add some more text, same font FPDF_PAGEOBJECT text_object2 = FPDFPageObj_CreateTextObj(document(), font.get(), 15.0f); std::unique_ptr text2 = GetFPDFWideString(L"Bigger font size"); EXPECT_TRUE(FPDFText_SetText(text_object2, text2.get())); FPDFPageObj_Transform(text_object2, 1, 0, 0, 1, 200, 200); FPDFPage_InsertObject(page, text_object2); } FPDF_BITMAP page_bitmap2 = RenderPage(page); #if _FXM_PLATFORM_ == _FXM_PLATFORM_APPLE_ const char md5_2[] = "8eded4193ff1f0f77b8b600a825e97ea"; #else const char md5_2[] = "9d7885072058f6c3e68ecaf32e917f30"; #endif // _FXM_PLATFORM_ == _FXM_PLATFORM_APPLE_ CompareBitmap(page_bitmap2, 612, 792, md5_2); FPDFBitmap_Destroy(page_bitmap2); EXPECT_TRUE(FPDFPage_GenerateContent(page)); EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); FPDF_ClosePage(page); TestAndCloseSaved(612, 792, md5_2); } TEST_F(FPDFEditEmbeddertest, TransformAnnot) { // Open a file with one annotation and load its first page. ASSERT_TRUE(OpenDocument("annotation_highlight_long_content.pdf")); FPDF_PAGE page = FPDF_LoadPage(document(), 0); ASSERT_TRUE(page); // Add an underline annotation to the page without specifying its rectangle. FPDF_ANNOTATION annot = FPDFPage_CreateAnnot(page, FPDF_ANNOT_UNDERLINE); ASSERT_TRUE(annot); // FPDFPage_TransformAnnots() should run without errors when modifying // annotation rectangles. FPDFPage_TransformAnnots(page, 1, 2, 3, 4, 5, 6); FPDFPage_CloseAnnot(annot); UnloadPage(page); } // TODO(npm): Add tests using Japanese fonts in other OS. #if _FXM_PLATFORM_ == _FXM_PLATFORM_LINUX_ TEST_F(FPDFEditEmbeddertest, AddCIDFontText) { // Start with a blank page FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792); CFX_Font CIDfont; { // First, get the data from the font CIDfont.LoadSubst("IPAGothic", 1, 0, 400, 0, 932, 0); EXPECT_EQ("IPAGothic", CIDfont.GetFaceName()); const uint8_t* data = CIDfont.GetFontData(); const uint32_t size = CIDfont.GetSize(); // Load the data into a FPDF_Font. std::unique_ptr font( FPDFText_LoadFont(document(), data, size, FPDF_FONT_TRUETYPE, 1)); ASSERT_TRUE(font.get()); // Add some text to the page FPDF_PAGEOBJECT text_object = FPDFPageObj_CreateTextObj(document(), font.get(), 12.0f); ASSERT_TRUE(text_object); std::wstring wstr = L"ABCDEFGhijklmnop."; std::unique_ptr text = GetFPDFWideString(wstr); EXPECT_TRUE(FPDFText_SetText(text_object, text.get())); FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 200, 200); FPDFPage_InsertObject(page, text_object); // And add some Japanese characters FPDF_PAGEOBJECT text_object2 = FPDFPageObj_CreateTextObj(document(), font.get(), 18.0f); ASSERT_TRUE(text_object2); std::wstring wstr2 = L"\u3053\u3093\u306B\u3061\u306f\u4e16\u754C\u3002\u3053\u3053\u306B1" L"\u756A"; std::unique_ptr text2 = GetFPDFWideString(wstr2); EXPECT_TRUE(FPDFText_SetText(text_object2, text2.get())); FPDFPageObj_Transform(text_object2, 1, 0, 0, 1, 100, 500); FPDFPage_InsertObject(page, text_object2); } // Check that the text renders properly. FPDF_BITMAP page_bitmap = RenderPage(page); const char md5[] = "2bc6c1aaa2252e73246a75775ccf38c2"; CompareBitmap(page_bitmap, 612, 792, md5); FPDFBitmap_Destroy(page_bitmap); // Save the document, close the page. EXPECT_TRUE(FPDFPage_GenerateContent(page)); EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); FPDF_ClosePage(page); TestAndCloseSaved(612, 792, md5); } #endif // _FXM_PLATFORM_ == _FXM_PLATFORM_LINUX_ TEST_F(FPDFEditEmbeddertest, SaveAndRender) { const char md5[] = "3c20472b0552c0c22b88ab1ed8c6202b"; { EXPECT_TRUE(OpenDocument("bug_779.pdf")); FPDF_PAGE page = LoadPage(0); ASSERT_NE(nullptr, page); // Now add a more complex blue path. FPDF_PAGEOBJECT green_path = FPDFPageObj_CreateNewPath(20, 20); EXPECT_TRUE(FPDFPath_SetFillColor(green_path, 0, 255, 0, 200)); // TODO(npm): stroking will cause the MD5s to differ. EXPECT_TRUE(FPDFPath_SetDrawMode(green_path, FPDF_FILLMODE_WINDING, 0)); EXPECT_TRUE(FPDFPath_LineTo(green_path, 20, 63)); EXPECT_TRUE(FPDFPath_BezierTo(green_path, 55, 55, 78, 78, 90, 90)); EXPECT_TRUE(FPDFPath_LineTo(green_path, 133, 133)); EXPECT_TRUE(FPDFPath_LineTo(green_path, 133, 33)); EXPECT_TRUE(FPDFPath_BezierTo(green_path, 38, 33, 39, 36, 40, 40)); EXPECT_TRUE(FPDFPath_Close(green_path)); FPDFPage_InsertObject(page, green_path); FPDF_BITMAP page_bitmap = RenderPage(page); CompareBitmap(page_bitmap, 612, 792, md5); FPDFBitmap_Destroy(page_bitmap); // Now save the result, closing the page and document EXPECT_TRUE(FPDFPage_GenerateContent(page)); EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); UnloadPage(page); } TestAndCloseSaved(612, 792, md5); } TEST_F(FPDFEditEmbeddertest, ExtractImageBitmap) { ASSERT_TRUE(OpenDocument("embedded_images.pdf")); FPDF_PAGE page = LoadPage(0); ASSERT_TRUE(page); ASSERT_EQ(39, FPDFPage_CountObject(page)); FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 32); EXPECT_NE(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); EXPECT_FALSE(FPDFImageObj_GetBitmap(obj)); obj = FPDFPage_GetObject(page, 33); ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); FPDF_BITMAP bitmap = FPDFImageObj_GetBitmap(obj); EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap)); CompareBitmap(bitmap, 109, 88, "d65e98d968d196abf13f78aec655ffae"); FPDFBitmap_Destroy(bitmap); obj = FPDFPage_GetObject(page, 34); ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); bitmap = FPDFImageObj_GetBitmap(obj); EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap)); CompareBitmap(bitmap, 103, 75, "1287711c84dbef767c435d11697661d6"); FPDFBitmap_Destroy(bitmap); obj = FPDFPage_GetObject(page, 35); ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); bitmap = FPDFImageObj_GetBitmap(obj); EXPECT_EQ(FPDFBitmap_Gray, FPDFBitmap_GetFormat(bitmap)); CompareBitmap(bitmap, 92, 68, "9c6d76cb1e37ef8514f9455d759391f3"); FPDFBitmap_Destroy(bitmap); obj = FPDFPage_GetObject(page, 36); ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); bitmap = FPDFImageObj_GetBitmap(obj); EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap)); CompareBitmap(bitmap, 79, 60, "15cb6a49a2e354ed0e9f45dd34e3da1a"); FPDFBitmap_Destroy(bitmap); obj = FPDFPage_GetObject(page, 37); ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); bitmap = FPDFImageObj_GetBitmap(obj); EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap)); CompareBitmap(bitmap, 126, 106, "be5a64ba7890d2657522af6524118534"); FPDFBitmap_Destroy(bitmap); obj = FPDFPage_GetObject(page, 38); ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); bitmap = FPDFImageObj_GetBitmap(obj); EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap)); CompareBitmap(bitmap, 194, 119, "f9e24207ee1bc0db6c543d33a5f12ec5"); FPDFBitmap_Destroy(bitmap); UnloadPage(page); } TEST_F(FPDFEditEmbeddertest, GetImageData) { EXPECT_TRUE(OpenDocument("embedded_images.pdf")); FPDF_PAGE page = LoadPage(0); ASSERT_TRUE(page); ASSERT_EQ(39, FPDFPage_CountObject(page)); // Retrieve an image object with flate-encoded data stream. FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 33); ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); // Check that the raw image data has the correct length and hash value. unsigned long len = FPDFImageObj_GetImageDataRaw(obj, nullptr, 0); std::vector buf(len); EXPECT_EQ(4091u, FPDFImageObj_GetImageDataRaw(obj, buf.data(), len)); EXPECT_EQ("f73802327d2e88e890f653961bcda81a", GenerateMD5Base16(reinterpret_cast(buf.data()), len)); // Check that the decoded image data has the correct length and hash value. len = FPDFImageObj_GetImageDataDecoded(obj, nullptr, 0); buf.clear(); buf.resize(len); EXPECT_EQ(28776u, FPDFImageObj_GetImageDataDecoded(obj, buf.data(), len)); EXPECT_EQ("cb3637934bb3b95a6e4ae1ea9eb9e56e", GenerateMD5Base16(reinterpret_cast(buf.data()), len)); // Retrieve an image obejct with DCTDecode-encoded data stream. obj = FPDFPage_GetObject(page, 37); ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); // Check that the raw image data has the correct length and hash value. len = FPDFImageObj_GetImageDataRaw(obj, nullptr, 0); buf.clear(); buf.resize(len); EXPECT_EQ(4370u, FPDFImageObj_GetImageDataRaw(obj, buf.data(), len)); EXPECT_EQ("6aae1f3710335023a9e12191be66b64b", GenerateMD5Base16(reinterpret_cast(buf.data()), len)); // Check that the decoded image data has the correct length and hash value, // which should be the same as those of the raw data, since this image is // encoded by a single DCTDecode filter and decoding is a noop. len = FPDFImageObj_GetImageDataDecoded(obj, nullptr, 0); buf.clear(); buf.resize(len); EXPECT_EQ(4370u, FPDFImageObj_GetImageDataDecoded(obj, buf.data(), len)); EXPECT_EQ("6aae1f3710335023a9e12191be66b64b", GenerateMD5Base16(reinterpret_cast(buf.data()), len)); UnloadPage(page); } TEST_F(FPDFEditEmbeddertest, DestroyPageObject) { FPDF_PAGEOBJECT rect = FPDFPageObj_CreateNewRect(10, 10, 20, 20); ASSERT_TRUE(rect); // There should be no memory leaks with a call to FPDFPageObj_Destroy(). FPDFPageObj_Destroy(rect); } TEST_F(FPDFEditEmbeddertest, GetImageFilters) { EXPECT_TRUE(OpenDocument("embedded_images.pdf")); FPDF_PAGE page = LoadPage(0); ASSERT_TRUE(page); // Verify that retrieving the filter of a non-image object would fail. FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 32); ASSERT_NE(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); ASSERT_EQ(0, FPDFImageObj_GetImageFilterCount(obj)); EXPECT_EQ(0u, FPDFImageObj_GetImageFilter(obj, 0, nullptr, 0)); // Verify the returned filter string for an image object with a single filter. obj = FPDFPage_GetObject(page, 33); ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); ASSERT_EQ(1, FPDFImageObj_GetImageFilterCount(obj)); unsigned long len = FPDFImageObj_GetImageFilter(obj, 0, nullptr, 0); std::vector buf(len); static constexpr char kFlateDecode[] = "FlateDecode"; EXPECT_EQ(sizeof(kFlateDecode), FPDFImageObj_GetImageFilter(obj, 0, buf.data(), len)); EXPECT_STREQ(kFlateDecode, buf.data()); EXPECT_EQ(0u, FPDFImageObj_GetImageFilter(obj, 1, nullptr, 0)); // Verify all the filters for an image object with a list of filters. obj = FPDFPage_GetObject(page, 38); ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); ASSERT_EQ(2, FPDFImageObj_GetImageFilterCount(obj)); len = FPDFImageObj_GetImageFilter(obj, 0, nullptr, 0); buf.clear(); buf.resize(len); static constexpr char kASCIIHexDecode[] = "ASCIIHexDecode"; EXPECT_EQ(sizeof(kASCIIHexDecode), FPDFImageObj_GetImageFilter(obj, 0, buf.data(), len)); EXPECT_STREQ(kASCIIHexDecode, buf.data()); len = FPDFImageObj_GetImageFilter(obj, 1, nullptr, 0); buf.clear(); buf.resize(len); static constexpr char kDCTDecode[] = "DCTDecode"; EXPECT_EQ(sizeof(kDCTDecode), FPDFImageObj_GetImageFilter(obj, 1, buf.data(), len)); EXPECT_STREQ(kDCTDecode, buf.data()); UnloadPage(page); } TEST_F(FPDFEditEmbeddertest, GetImageMetadata) { ASSERT_TRUE(OpenDocument("embedded_images.pdf")); FPDF_PAGE page = LoadPage(0); ASSERT_TRUE(page); // Check that getting the metadata of a null object would fail. FPDF_IMAGEOBJ_METADATA metadata; EXPECT_FALSE(FPDFImageObj_GetImageMetadata(nullptr, page, &metadata)); // Check that receiving the metadata with a null metadata object would fail. FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 35); EXPECT_FALSE(FPDFImageObj_GetImageMetadata(obj, page, nullptr)); // Check that when retrieving an image object's metadata without passing in // |page|, all values are correct, with the last two being default values. ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); ASSERT_TRUE(FPDFImageObj_GetImageMetadata(obj, nullptr, &metadata)); EXPECT_EQ(92u, metadata.width); EXPECT_EQ(68u, metadata.height); EXPECT_NEAR(96.000000, metadata.horizontal_dpi, 0.001); EXPECT_NEAR(96.000000, metadata.vertical_dpi, 0.001); EXPECT_EQ(0u, metadata.bits_per_pixel); EXPECT_EQ(FPDF_COLORSPACE_UNKNOWN, metadata.colorspace); // Verify the metadata of a bitmap image with indexed colorspace. ASSERT_TRUE(FPDFImageObj_GetImageMetadata(obj, page, &metadata)); EXPECT_EQ(92u, metadata.width); EXPECT_EQ(68u, metadata.height); EXPECT_NEAR(96.000000, metadata.horizontal_dpi, 0.001); EXPECT_NEAR(96.000000, metadata.vertical_dpi, 0.001); EXPECT_EQ(1u, metadata.bits_per_pixel); EXPECT_EQ(FPDF_COLORSPACE_INDEXED, metadata.colorspace); // Verify the metadata of an image with RGB colorspace. obj = FPDFPage_GetObject(page, 37); ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); ASSERT_TRUE(FPDFImageObj_GetImageMetadata(obj, page, &metadata)); EXPECT_EQ(126u, metadata.width); EXPECT_EQ(106u, metadata.height); EXPECT_NEAR(162.173752, metadata.horizontal_dpi, 0.001); EXPECT_NEAR(162.555878, metadata.vertical_dpi, 0.001); EXPECT_EQ(24u, metadata.bits_per_pixel); EXPECT_EQ(FPDF_COLORSPACE_DEVICERGB, metadata.colorspace); UnloadPage(page); }