fpdfsdk/fpdfview_embeddertest.cpp - platform/external/pdfium - Gitiles

 // Copyright 2015 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 <limits>
 #include <string>

 #include "core/fxcrt/fx_coordinates.h"
 #include "fpdfsdk/fpdfview_c_api_test.h"
 #include "public/fpdfview.h"
 #include "testing/embedder_test.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "testing/utils/path_service.h"

 namespace {

 class MockDownloadHints : public FX_DOWNLOADHINTS {
  public:
   static void SAddSegment(FX_DOWNLOADHINTS* pThis, size_t offset, size_t size) {
   }

   MockDownloadHints() {
     FX_DOWNLOADHINTS::version = 1;
     FX_DOWNLOADHINTS::AddSegment = SAddSegment;
   }

   ~MockDownloadHints() {}
 };

 }  // namespace

 TEST(fpdf, CApiTest) {
   EXPECT_TRUE(CheckPDFiumCApi());
 }

 class FPDFViewEmbeddertest : public EmbedderTest {};

 TEST_F(FPDFViewEmbeddertest, Document) {
   EXPECT_TRUE(OpenDocument("about_blank.pdf"));
   EXPECT_EQ(1, GetPageCount());
   EXPECT_EQ(0, GetFirstPageNum());

   int version;
   EXPECT_TRUE(FPDF_GetFileVersion(document(), &version));
   EXPECT_EQ(14, version);

   EXPECT_EQ(0xFFFFFFFF, FPDF_GetDocPermissions(document()));
   EXPECT_EQ(-1, FPDF_GetSecurityHandlerRevision(document()));
 }

 TEST_F(FPDFViewEmbeddertest, LoadNonexistentDocument) {
   FPDF_DOCUMENT doc = FPDF_LoadDocument("nonexistent_document.pdf", "");
   ASSERT_FALSE(doc);
   EXPECT_EQ(static_cast<int>(FPDF_GetLastError()), FPDF_ERR_FILE);
 }

 // See bug 465.
 TEST_F(FPDFViewEmbeddertest, EmptyDocument) {
   EXPECT_TRUE(CreateEmptyDocument());

   {
     int version = 42;
     EXPECT_FALSE(FPDF_GetFileVersion(document(), &version));
     EXPECT_EQ(0, version);
   }

   {
 #ifndef PDF_ENABLE_XFA
     const unsigned long kExpected = 0;
 #else
     const unsigned long kExpected = static_cast<uint32_t>(-1);
 #endif
     EXPECT_EQ(kExpected, FPDF_GetDocPermissions(document()));
   }

   EXPECT_EQ(-1, FPDF_GetSecurityHandlerRevision(document()));

   EXPECT_EQ(0, FPDF_GetPageCount(document()));

   EXPECT_TRUE(FPDF_VIEWERREF_GetPrintScaling(document()));
   EXPECT_EQ(1, FPDF_VIEWERREF_GetNumCopies(document()));
   EXPECT_EQ(DuplexUndefined, FPDF_VIEWERREF_GetDuplex(document()));

   char buf[100];
   EXPECT_EQ(0U, FPDF_VIEWERREF_GetName(document(), "foo", nullptr, 0));
   EXPECT_EQ(0U, FPDF_VIEWERREF_GetName(document(), "foo", buf, sizeof(buf)));

   EXPECT_EQ(0u, FPDF_CountNamedDests(document()));
 }

 TEST_F(FPDFViewEmbeddertest, LinearizedDocument) {
   EXPECT_TRUE(OpenDocument("feature_linearized_loading.pdf", nullptr, true));
   int version;
   EXPECT_TRUE(FPDF_GetFileVersion(document(), &version));
   EXPECT_EQ(16, version);
 }

 TEST_F(FPDFViewEmbeddertest, Page) {
   EXPECT_TRUE(OpenDocument("about_blank.pdf"));
   FPDF_PAGE page = LoadPage(0);
   EXPECT_NE(nullptr, page);
   EXPECT_EQ(612.0, FPDF_GetPageWidth(page));
   EXPECT_EQ(792.0, FPDF_GetPageHeight(page));
   UnloadPage(page);
   EXPECT_EQ(nullptr, LoadPage(1));
 }

 TEST_F(FPDFViewEmbeddertest, ViewerRefDummy) {
   EXPECT_TRUE(OpenDocument("about_blank.pdf"));
   EXPECT_TRUE(FPDF_VIEWERREF_GetPrintScaling(document()));
   EXPECT_EQ(1, FPDF_VIEWERREF_GetNumCopies(document()));
   EXPECT_EQ(DuplexUndefined, FPDF_VIEWERREF_GetDuplex(document()));

   char buf[100];
   EXPECT_EQ(0U, FPDF_VIEWERREF_GetName(document(), "foo", nullptr, 0));
   EXPECT_EQ(0U, FPDF_VIEWERREF_GetName(document(), "foo", buf, sizeof(buf)));
 }

 TEST_F(FPDFViewEmbeddertest, ViewerRef) {
   EXPECT_TRUE(OpenDocument("viewer_ref.pdf"));
   EXPECT_TRUE(FPDF_VIEWERREF_GetPrintScaling(document()));
   EXPECT_EQ(5, FPDF_VIEWERREF_GetNumCopies(document()));
   EXPECT_EQ(DuplexUndefined, FPDF_VIEWERREF_GetDuplex(document()));

   // Test some corner cases.
   char buf[100];
   EXPECT_EQ(0U, FPDF_VIEWERREF_GetName(document(), "", buf, sizeof(buf)));
   EXPECT_EQ(0U, FPDF_VIEWERREF_GetName(document(), "foo", nullptr, 0));
   EXPECT_EQ(0U, FPDF_VIEWERREF_GetName(document(), "foo", buf, sizeof(buf)));

   // Make sure |buf| does not get written into when it appears to be too small.
   // NOLINTNEXTLINE(runtime/printf)
   strcpy(buf, "ABCD");
   EXPECT_EQ(4U, FPDF_VIEWERREF_GetName(document(), "Foo", buf, 1));
   EXPECT_STREQ("ABCD", buf);

   // Note "Foo" is a different key from "foo".
   EXPECT_EQ(4U,
             FPDF_VIEWERREF_GetName(document(), "Foo", nullptr, sizeof(buf)));
   ASSERT_EQ(4U, FPDF_VIEWERREF_GetName(document(), "Foo", buf, sizeof(buf)));
   EXPECT_STREQ("foo", buf);

   // Try to retrieve a boolean and an integer.
   EXPECT_EQ(
       0U, FPDF_VIEWERREF_GetName(document(), "HideToolbar", buf, sizeof(buf)));
   EXPECT_EQ(0U,
             FPDF_VIEWERREF_GetName(document(), "NumCopies", buf, sizeof(buf)));

   // Try more valid cases.
   ASSERT_EQ(4U,
             FPDF_VIEWERREF_GetName(document(), "Direction", buf, sizeof(buf)));
   EXPECT_STREQ("R2L", buf);
   ASSERT_EQ(8U,
             FPDF_VIEWERREF_GetName(document(), "ViewArea", buf, sizeof(buf)));
   EXPECT_STREQ("CropBox", buf);
 }

 TEST_F(FPDFViewEmbeddertest, NamedDests) {
   EXPECT_TRUE(OpenDocument("named_dests.pdf"));
   long buffer_size;
   char fixed_buffer[512];
   FPDF_DEST dest;

   // Query the size of the first item.
   buffer_size = 2000000;  // Absurdly large, check not used for this case.
   dest = FPDF_GetNamedDest(document(), 0, nullptr, &buffer_size);
   EXPECT_NE(nullptr, dest);
   EXPECT_EQ(12, buffer_size);

   // Try to retrieve the first item with too small a buffer.
   buffer_size = 10;
   dest = FPDF_GetNamedDest(document(), 0, fixed_buffer, &buffer_size);
   EXPECT_NE(nullptr, dest);
   EXPECT_EQ(-1, buffer_size);

   // Try to retrieve the first item with correctly sized buffer. Item is
   // taken from Dests NameTree in named_dests.pdf.
   buffer_size = 12;
   dest = FPDF_GetNamedDest(document(), 0, fixed_buffer, &buffer_size);
   EXPECT_NE(nullptr, dest);
   EXPECT_EQ(12, buffer_size);
   EXPECT_EQ(std::string("F\0i\0r\0s\0t\0\0\0", 12),
             std::string(fixed_buffer, buffer_size));

   // Try to retrieve the second item with ample buffer. Item is taken
   // from Dests NameTree but has a sub-dictionary in named_dests.pdf.
   buffer_size = sizeof(fixed_buffer);
   dest = FPDF_GetNamedDest(document(), 1, fixed_buffer, &buffer_size);
   EXPECT_NE(nullptr, dest);
   EXPECT_EQ(10, buffer_size);
   EXPECT_EQ(std::string("N\0e\0x\0t\0\0\0", 10),
             std::string(fixed_buffer, buffer_size));

   // Try to retrieve third item with ample buffer. Item is taken
   // from Dests NameTree but has a bad sub-dictionary in named_dests.pdf.
   // in named_dests.pdf).
   buffer_size = sizeof(fixed_buffer);
   dest = FPDF_GetNamedDest(document(), 2, fixed_buffer, &buffer_size);
   EXPECT_EQ(nullptr, dest);
   EXPECT_EQ(sizeof(fixed_buffer),
             static_cast<size_t>(buffer_size));  // unmodified.

   // Try to retrieve the forth item with ample buffer. Item is taken
   // from Dests NameTree but has a vale of the wrong type in named_dests.pdf.
   buffer_size = sizeof(fixed_buffer);
   dest = FPDF_GetNamedDest(document(), 3, fixed_buffer, &buffer_size);
   EXPECT_EQ(nullptr, dest);
   EXPECT_EQ(sizeof(fixed_buffer),
             static_cast<size_t>(buffer_size));  // unmodified.

   // Try to retrieve fifth item with ample buffer. Item taken from the
   // old-style Dests dictionary object in named_dests.pdf.
   buffer_size = sizeof(fixed_buffer);
   dest = FPDF_GetNamedDest(document(), 4, fixed_buffer, &buffer_size);
   EXPECT_NE(nullptr, dest);
   EXPECT_EQ(30, buffer_size);
   EXPECT_EQ(std::string("F\0i\0r\0s\0t\0A\0l\0t\0e\0r\0n\0a\0t\0e\0\0\0", 30),
             std::string(fixed_buffer, buffer_size));

   // Try to retrieve sixth item with ample buffer. Item istaken from the
   // old-style Dests dictionary object but has a sub-dictionary in
   // named_dests.pdf.
   buffer_size = sizeof(fixed_buffer);
   dest = FPDF_GetNamedDest(document(), 5, fixed_buffer, &buffer_size);
   EXPECT_NE(nullptr, dest);
   EXPECT_EQ(28, buffer_size);
   EXPECT_EQ(std::string("L\0a\0s\0t\0A\0l\0t\0e\0r\0n\0a\0t\0e\0\0\0", 28),
             std::string(fixed_buffer, buffer_size));

   // Try to retrieve non-existent item with ample buffer.
   buffer_size = sizeof(fixed_buffer);
   dest = FPDF_GetNamedDest(document(), 6, fixed_buffer, &buffer_size);
   EXPECT_EQ(nullptr, dest);
   EXPECT_EQ(sizeof(fixed_buffer),
             static_cast<size_t>(buffer_size));  // unmodified.

   // Try to underflow/overflow the integer index.
   buffer_size = sizeof(fixed_buffer);
   dest = FPDF_GetNamedDest(document(), std::numeric_limits<int>::max(),
                            fixed_buffer, &buffer_size);
   EXPECT_EQ(nullptr, dest);
   EXPECT_EQ(sizeof(fixed_buffer),
             static_cast<size_t>(buffer_size));  // unmodified.

   buffer_size = sizeof(fixed_buffer);
   dest = FPDF_GetNamedDest(document(), std::numeric_limits<int>::min(),
                            fixed_buffer, &buffer_size);
   EXPECT_EQ(nullptr, dest);
   EXPECT_EQ(sizeof(fixed_buffer),
             static_cast<size_t>(buffer_size));  // unmodified.

   buffer_size = sizeof(fixed_buffer);
   dest = FPDF_GetNamedDest(document(), -1, fixed_buffer, &buffer_size);
   EXPECT_EQ(nullptr, dest);
   EXPECT_EQ(sizeof(fixed_buffer),
             static_cast<size_t>(buffer_size));  // unmodified.
 }

 TEST_F(FPDFViewEmbeddertest, NamedDestsByName) {
   EXPECT_TRUE(OpenDocument("named_dests.pdf"));

   // Null pointer returns nullptr.
   FPDF_DEST dest = FPDF_GetNamedDestByName(document(), nullptr);
   EXPECT_EQ(nullptr, dest);

   // Empty string returns nullptr.
   dest = FPDF_GetNamedDestByName(document(), "");
   EXPECT_EQ(nullptr, dest);

   // Item from Dests NameTree.
   dest = FPDF_GetNamedDestByName(document(), "First");
   EXPECT_NE(nullptr, dest);

   long ignore_len = 0;
   FPDF_DEST dest_by_index =
       FPDF_GetNamedDest(document(), 0, nullptr, &ignore_len);
   EXPECT_EQ(dest_by_index, dest);

   // Item from Dests dictionary.
   dest = FPDF_GetNamedDestByName(document(), "FirstAlternate");
   EXPECT_NE(nullptr, dest);

   ignore_len = 0;
   dest_by_index = FPDF_GetNamedDest(document(), 4, nullptr, &ignore_len);
   EXPECT_EQ(dest_by_index, dest);

   // Bad value type for item from Dests NameTree array.
   dest = FPDF_GetNamedDestByName(document(), "WrongType");
   EXPECT_EQ(nullptr, dest);

   // No such destination in either Dest NameTree or dictionary.
   dest = FPDF_GetNamedDestByName(document(), "Bogus");
   EXPECT_EQ(nullptr, dest);
 }

 // The following tests pass if the document opens without crashing.
 TEST_F(FPDFViewEmbeddertest, Crasher_113) {
   EXPECT_TRUE(OpenDocument("bug_113.pdf"));
 }

 TEST_F(FPDFViewEmbeddertest, Crasher_451830) {
   // Document is damaged and can't be opened.
   EXPECT_FALSE(OpenDocument("bug_451830.pdf"));
 }

 TEST_F(FPDFViewEmbeddertest, Crasher_452455) {
   EXPECT_TRUE(OpenDocument("bug_452455.pdf"));
   FPDF_PAGE page = LoadPage(0);
   EXPECT_NE(nullptr, page);
   UnloadPage(page);
 }

 TEST_F(FPDFViewEmbeddertest, Crasher_454695) {
   // Document is damaged and can't be opened.
   EXPECT_FALSE(OpenDocument("bug_454695.pdf"));
 }

 TEST_F(FPDFViewEmbeddertest, Crasher_572871) {
   EXPECT_TRUE(OpenDocument("bug_572871.pdf"));
 }

 // It tests that document can still be loaded even the trailer has no 'Size'
 // field if other information is right.
 TEST_F(FPDFViewEmbeddertest, Failed_213) {
   EXPECT_TRUE(OpenDocument("bug_213.pdf"));
 }

 // The following tests pass if the document opens without infinite looping.
 TEST_F(FPDFViewEmbeddertest, Hang_298) {
   EXPECT_FALSE(OpenDocument("bug_298.pdf"));
 }

 // Test if the document opens without infinite looping.
 // Previously this test will hang in a loop inside LoadAllCrossRefV4. After
 // the fix, LoadAllCrossRefV4 will return false after detecting a cross
 // reference loop. Cross references will be rebuilt successfully.
 TEST_F(FPDFViewEmbeddertest, CrossRefV4Loop) {
   EXPECT_TRUE(OpenDocument("bug_xrefv4_loop.pdf"));
   MockDownloadHints hints;

   // Make sure calling FPDFAvail_IsDocAvail() on this file does not infinite
   // loop either. See bug 875.
   int ret = PDF_DATA_NOTAVAIL;
   while (ret == PDF_DATA_NOTAVAIL)
     ret = FPDFAvail_IsDocAvail(avail_, &hints);
   EXPECT_EQ(PDF_DATA_AVAIL, ret);
 }

 // The test should pass when circular references to ParseIndirectObject will not
 // cause infinite loop.
 TEST_F(FPDFViewEmbeddertest, Hang_343) {
   EXPECT_FALSE(OpenDocument("bug_343.pdf"));
 }

 // The test should pass when the absence of 'Contents' field in a signature
 // dictionary will not cause an infinite loop in CPDF_SyntaxParser::GetObject().
 TEST_F(FPDFViewEmbeddertest, Hang_344) {
   EXPECT_FALSE(OpenDocument("bug_344.pdf"));
 }

 // The test should pass when there is no infinite recursion in
 // CPDF_SyntaxParser::GetString().
 TEST_F(FPDFViewEmbeddertest, Hang_355) {
   EXPECT_FALSE(OpenDocument("bug_355.pdf"));
 }
 // The test should pass even when the file has circular references to pages.
 TEST_F(FPDFViewEmbeddertest, Hang_360) {
   EXPECT_FALSE(OpenDocument("bug_360.pdf"));
 }

 TEST_F(FPDFViewEmbeddertest, FPDF_RenderPageBitmapWithMatrix) {
   const char* const kRotatedMD5[4] = {
       "0a90de37f52127619c3dfb642b5fa2fe", "d599429574ff0dcad3bc898ea8b874ca",
       "0113386bb0bd45125bacc6dee78bfe78", "051fcfa4c1f9de28765705633a8ef3a9"};
   const char kTopLeftQuarterMD5[] = "4982be08db3f6d2e6409186ebbced9eb";
   const char kHoriStretchedMD5[] = "004bf38f3c5c76a644e6fca204747f21";
   const char kRotateandStretchMD5[] = "0ea95cacc716d003cf063a2c5ed6c8d7";

   EXPECT_TRUE(OpenDocument("rectangles.pdf"));
   FPDF_PAGE page = LoadPage(0);
   EXPECT_NE(nullptr, page);
   const int initial_width = static_cast<int>(FPDF_GetPageWidth(page));
   const int initial_height = static_cast<int>(FPDF_GetPageHeight(page));
   EXPECT_EQ(200, initial_width);
   EXPECT_EQ(300, initial_height);

   FPDF_BITMAP bitmap = RenderPage(page);
   CompareBitmap(bitmap, initial_width, initial_height, kRotatedMD5[0]);
   FPDFBitmap_Destroy(bitmap);

   int width;
   int height;
   FS_RECTF rect;
   rect.left = 0;
   rect.top = 0;
   FS_MATRIX matrix;

   // Try the easy rotations: 0, 90, 180, 270 clockwise. The output should be the
   // same as FPDF_RenderPageBitmap with the appropriate rotation flag. Per PDF
   // spec section 4.2.2, a t degree rotation is represented by [cos(t) sin(t)
   // -sin(t) cos(t) 0 0] (matrix goes on the right in the multiplication).
   rect.right = initial_width;
   rect.bottom = initial_height;
   CFX_Matrix rot_matrices[4] = {
       CFX_Matrix(1, 0, 0, 1, 0, 0), CFX_Matrix(0, -1, 1, 0, 0, 0),
       CFX_Matrix(-1, 0, 0, -1, 0, 0), CFX_Matrix(0, 1, -1, 0, 0, 0)};
   for (int rot = 0; rot < 4; ++rot) {
     matrix.a = rot_matrices[rot].a;
     matrix.b = rot_matrices[rot].b;
     matrix.c = rot_matrices[rot].c;
     matrix.d = rot_matrices[rot].d;
     matrix.e = rot_matrices[rot].e;
     matrix.f = rot_matrices[rot].f;
     if (rot % 2 == 0) {
       width = initial_width;
       height = initial_height;
     } else {
       width = initial_height;
       height = initial_width;
     }
     rect.right = width;
     rect.bottom = height;

     bitmap = FPDFBitmap_Create(width, height, 0);
     FPDFBitmap_FillRect(bitmap, 0, 0, width, height, 0xFFFFFFFF);
     FPDF_RenderPageBitmap(bitmap, page, 0, 0, width, height, rot, 0);
     CompareBitmap(bitmap, width, height, kRotatedMD5[rot]);
     FPDFBitmap_Destroy(bitmap);

     bitmap = FPDFBitmap_Create(width, height, 0);
     FPDFBitmap_FillRect(bitmap, 0, 0, width, height, 0xFFFFFFFF);
     FPDF_RenderPageBitmapWithMatrix(bitmap, page, &matrix, &rect, 0);
     CompareBitmap(bitmap, width, height, kRotatedMD5[rot]);
     FPDFBitmap_Destroy(bitmap);
   }
   // TODO(npm): what to do with transformations that do not align the page with
   // the axis, like a 45 degree rotation (currently, part of the page gets cut
   // out). pdfium:849

   // Now render again with the image scaled smaller.
   width = initial_width / 2;
   height = initial_height / 2;
   matrix.a = 0.5;
   matrix.b = 0;
   matrix.c = 0;
   matrix.d = 0.5;

   rect.right = width;
   rect.bottom = height;

   bitmap = FPDFBitmap_Create(width, height, 0);
   FPDFBitmap_FillRect(bitmap, 0, 0, width, height, 0xFFFFFFFF);
   FPDF_RenderPageBitmapWithMatrix(bitmap, page, &matrix, &rect, 0);
   CompareBitmap(bitmap, width, height, kTopLeftQuarterMD5);
   FPDFBitmap_Destroy(bitmap);

   // Now render again with the image scaled larger horizontally.
   width = initial_width * 2;
   height = initial_height;
   matrix.a = 2;
   matrix.d = 1;
   rect.right = width;
   rect.bottom = height;
   bitmap = FPDFBitmap_Create(width, height, 0);
   FPDFBitmap_FillRect(bitmap, 0, 0, width, height, 0xFFFFFFFF);
   FPDF_RenderPageBitmapWithMatrix(bitmap, page, &matrix, &rect, 0);
   CompareBitmap(bitmap, width, height, kHoriStretchedMD5);
   FPDFBitmap_Destroy(bitmap);

   // Test a rotation followed by a stretch.
   width = initial_height * 2;
   height = initial_width;
   matrix.a = 0;
   matrix.b = -1;
   matrix.c = 2;
   matrix.d = 0;
   matrix.e = 0;
   matrix.f = 0;
   rect.right = width;
   rect.bottom = height;
   bitmap = FPDFBitmap_Create(width, height, 0);
   FPDFBitmap_FillRect(bitmap, 0, 0, width, height, 0xFFFFFFFF);
   FPDF_RenderPageBitmapWithMatrix(bitmap, page, &matrix, &rect, 0);
   CompareBitmap(bitmap, width, height, kRotateandStretchMD5);
   FPDFBitmap_Destroy(bitmap);

   UnloadPage(page);
 }

 class UnSupRecordDelegate : public EmbedderTest::Delegate {
  public:
   UnSupRecordDelegate() : type_(-1) {}
   ~UnSupRecordDelegate() override {}

   void UnsupportedHandler(int type) override { type_ = type; }

   int type_;
 };

 TEST_F(FPDFViewEmbeddertest, UnSupportedOperations_NotFound) {
   UnSupRecordDelegate delegate;
   SetDelegate(&delegate);
   ASSERT_TRUE(OpenDocument("hello_world.pdf"));
   EXPECT_EQ(delegate.type_, -1);
   SetDelegate(nullptr);
 }

 TEST_F(FPDFViewEmbeddertest, UnSupportedOperations_LoadCustomDocument) {
   UnSupRecordDelegate delegate;
   SetDelegate(&delegate);
   ASSERT_TRUE(OpenDocument("unsupported_feature.pdf"));
   EXPECT_EQ(FPDF_UNSP_DOC_PORTABLECOLLECTION, delegate.type_);
   SetDelegate(nullptr);
 }

 TEST_F(FPDFViewEmbeddertest, UnSupportedOperations_LoadDocument) {
   std::string file_path;
   ASSERT_TRUE(
       PathService::GetTestFilePath("unsupported_feature.pdf", &file_path));

   UnSupRecordDelegate delegate;
   SetDelegate(&delegate);
   FPDF_DOCUMENT doc = FPDF_LoadDocument(file_path.c_str(), "");
   EXPECT_TRUE(doc != nullptr);
   EXPECT_EQ(FPDF_UNSP_DOC_PORTABLECOLLECTION, delegate.type_);
   FPDF_CloseDocument(doc);
   SetDelegate(nullptr);
 }
	// Copyright 2015 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 <limits>
	#include <string>

	#include "core/fxcrt/fx_coordinates.h"
	#include "fpdfsdk/fpdfview_c_api_test.h"
	#include "public/fpdfview.h"
	#include "testing/embedder_test.h"
	#include "testing/gtest/include/gtest/gtest.h"
	#include "testing/utils/path_service.h"

	namespace {

	class MockDownloadHints : public FX_DOWNLOADHINTS {
	public:
	static void SAddSegment(FX_DOWNLOADHINTS* pThis, size_t offset, size_t size) {
	}

	MockDownloadHints() {
	FX_DOWNLOADHINTS::version = 1;
	FX_DOWNLOADHINTS::AddSegment = SAddSegment;
	}

	~MockDownloadHints() {}
	};

	} // namespace

	TEST(fpdf, CApiTest) {
	EXPECT_TRUE(CheckPDFiumCApi());
	}

	class FPDFViewEmbeddertest : public EmbedderTest {};

	TEST_F(FPDFViewEmbeddertest, Document) {
	EXPECT_TRUE(OpenDocument("about_blank.pdf"));
	EXPECT_EQ(1, GetPageCount());
	EXPECT_EQ(0, GetFirstPageNum());

	int version;
	EXPECT_TRUE(FPDF_GetFileVersion(document(), &version));
	EXPECT_EQ(14, version);

	EXPECT_EQ(0xFFFFFFFF, FPDF_GetDocPermissions(document()));
	EXPECT_EQ(-1, FPDF_GetSecurityHandlerRevision(document()));
	}

	TEST_F(FPDFViewEmbeddertest, LoadNonexistentDocument) {
	FPDF_DOCUMENT doc = FPDF_LoadDocument("nonexistent_document.pdf", "");
	ASSERT_FALSE(doc);
	EXPECT_EQ(static_cast<int>(FPDF_GetLastError()), FPDF_ERR_FILE);
	}

	// See bug 465.
	TEST_F(FPDFViewEmbeddertest, EmptyDocument) {
	EXPECT_TRUE(CreateEmptyDocument());

	{
	int version = 42;
	EXPECT_FALSE(FPDF_GetFileVersion(document(), &version));
	EXPECT_EQ(0, version);
	}

	{
	#ifndef PDF_ENABLE_XFA
	const unsigned long kExpected = 0;
	#else
	const unsigned long kExpected = static_cast<uint32_t>(-1);
	#endif
	EXPECT_EQ(kExpected, FPDF_GetDocPermissions(document()));
	}

	EXPECT_EQ(-1, FPDF_GetSecurityHandlerRevision(document()));

	EXPECT_EQ(0, FPDF_GetPageCount(document()));

	EXPECT_TRUE(FPDF_VIEWERREF_GetPrintScaling(document()));
	EXPECT_EQ(1, FPDF_VIEWERREF_GetNumCopies(document()));
	EXPECT_EQ(DuplexUndefined, FPDF_VIEWERREF_GetDuplex(document()));

	char buf[100];
	EXPECT_EQ(0U, FPDF_VIEWERREF_GetName(document(), "foo", nullptr, 0));
	EXPECT_EQ(0U, FPDF_VIEWERREF_GetName(document(), "foo", buf, sizeof(buf)));

	EXPECT_EQ(0u, FPDF_CountNamedDests(document()));
	}

	TEST_F(FPDFViewEmbeddertest, LinearizedDocument) {
	EXPECT_TRUE(OpenDocument("feature_linearized_loading.pdf", nullptr, true));
	int version;
	EXPECT_TRUE(FPDF_GetFileVersion(document(), &version));
	EXPECT_EQ(16, version);
	}

	TEST_F(FPDFViewEmbeddertest, Page) {
	EXPECT_TRUE(OpenDocument("about_blank.pdf"));
	FPDF_PAGE page = LoadPage(0);
	EXPECT_NE(nullptr, page);
	EXPECT_EQ(612.0, FPDF_GetPageWidth(page));
	EXPECT_EQ(792.0, FPDF_GetPageHeight(page));
	UnloadPage(page);
	EXPECT_EQ(nullptr, LoadPage(1));
	}

	TEST_F(FPDFViewEmbeddertest, ViewerRefDummy) {
	EXPECT_TRUE(OpenDocument("about_blank.pdf"));
	EXPECT_TRUE(FPDF_VIEWERREF_GetPrintScaling(document()));
	EXPECT_EQ(1, FPDF_VIEWERREF_GetNumCopies(document()));
	EXPECT_EQ(DuplexUndefined, FPDF_VIEWERREF_GetDuplex(document()));

	char buf[100];
	EXPECT_EQ(0U, FPDF_VIEWERREF_GetName(document(), "foo", nullptr, 0));
	EXPECT_EQ(0U, FPDF_VIEWERREF_GetName(document(), "foo", buf, sizeof(buf)));
	}

	TEST_F(FPDFViewEmbeddertest, ViewerRef) {
	EXPECT_TRUE(OpenDocument("viewer_ref.pdf"));
	EXPECT_TRUE(FPDF_VIEWERREF_GetPrintScaling(document()));
	EXPECT_EQ(5, FPDF_VIEWERREF_GetNumCopies(document()));
	EXPECT_EQ(DuplexUndefined, FPDF_VIEWERREF_GetDuplex(document()));

	// Test some corner cases.
	char buf[100];
	EXPECT_EQ(0U, FPDF_VIEWERREF_GetName(document(), "", buf, sizeof(buf)));
	EXPECT_EQ(0U, FPDF_VIEWERREF_GetName(document(), "foo", nullptr, 0));
	EXPECT_EQ(0U, FPDF_VIEWERREF_GetName(document(), "foo", buf, sizeof(buf)));

	// Make sure \|buf\| does not get written into when it appears to be too small.
	// NOLINTNEXTLINE(runtime/printf)
	strcpy(buf, "ABCD");
	EXPECT_EQ(4U, FPDF_VIEWERREF_GetName(document(), "Foo", buf, 1));
	EXPECT_STREQ("ABCD", buf);

	// Note "Foo" is a different key from "foo".
	EXPECT_EQ(4U,
	FPDF_VIEWERREF_GetName(document(), "Foo", nullptr, sizeof(buf)));
	ASSERT_EQ(4U, FPDF_VIEWERREF_GetName(document(), "Foo", buf, sizeof(buf)));
	EXPECT_STREQ("foo", buf);

	// Try to retrieve a boolean and an integer.
	EXPECT_EQ(
	0U, FPDF_VIEWERREF_GetName(document(), "HideToolbar", buf, sizeof(buf)));
	EXPECT_EQ(0U,
	FPDF_VIEWERREF_GetName(document(), "NumCopies", buf, sizeof(buf)));

	// Try more valid cases.
	ASSERT_EQ(4U,
	FPDF_VIEWERREF_GetName(document(), "Direction", buf, sizeof(buf)));
	EXPECT_STREQ("R2L", buf);
	ASSERT_EQ(8U,
	FPDF_VIEWERREF_GetName(document(), "ViewArea", buf, sizeof(buf)));
	EXPECT_STREQ("CropBox", buf);
	}

	TEST_F(FPDFViewEmbeddertest, NamedDests) {
	EXPECT_TRUE(OpenDocument("named_dests.pdf"));
	long buffer_size;
	char fixed_buffer[512];
	FPDF_DEST dest;

	// Query the size of the first item.
	buffer_size = 2000000; // Absurdly large, check not used for this case.
	dest = FPDF_GetNamedDest(document(), 0, nullptr, &buffer_size);
	EXPECT_NE(nullptr, dest);
	EXPECT_EQ(12, buffer_size);

	// Try to retrieve the first item with too small a buffer.
	buffer_size = 10;
	dest = FPDF_GetNamedDest(document(), 0, fixed_buffer, &buffer_size);
	EXPECT_NE(nullptr, dest);
	EXPECT_EQ(-1, buffer_size);

	// Try to retrieve the first item with correctly sized buffer. Item is
	// taken from Dests NameTree in named_dests.pdf.
	buffer_size = 12;
	dest = FPDF_GetNamedDest(document(), 0, fixed_buffer, &buffer_size);
	EXPECT_NE(nullptr, dest);
	EXPECT_EQ(12, buffer_size);
	EXPECT_EQ(std::string("F\0i\0r\0s\0t\0\0\0", 12),
	std::string(fixed_buffer, buffer_size));

	// Try to retrieve the second item with ample buffer. Item is taken
	// from Dests NameTree but has a sub-dictionary in named_dests.pdf.
	buffer_size = sizeof(fixed_buffer);
	dest = FPDF_GetNamedDest(document(), 1, fixed_buffer, &buffer_size);
	EXPECT_NE(nullptr, dest);
	EXPECT_EQ(10, buffer_size);
	EXPECT_EQ(std::string("N\0e\0x\0t\0\0\0", 10),
	std::string(fixed_buffer, buffer_size));

	// Try to retrieve third item with ample buffer. Item is taken
	// from Dests NameTree but has a bad sub-dictionary in named_dests.pdf.
	// in named_dests.pdf).
	buffer_size = sizeof(fixed_buffer);
	dest = FPDF_GetNamedDest(document(), 2, fixed_buffer, &buffer_size);
	EXPECT_EQ(nullptr, dest);
	EXPECT_EQ(sizeof(fixed_buffer),
	static_cast<size_t>(buffer_size)); // unmodified.

	// Try to retrieve the forth item with ample buffer. Item is taken
	// from Dests NameTree but has a vale of the wrong type in named_dests.pdf.
	buffer_size = sizeof(fixed_buffer);
	dest = FPDF_GetNamedDest(document(), 3, fixed_buffer, &buffer_size);
	EXPECT_EQ(nullptr, dest);
	EXPECT_EQ(sizeof(fixed_buffer),
	static_cast<size_t>(buffer_size)); // unmodified.

	// Try to retrieve fifth item with ample buffer. Item taken from the
	// old-style Dests dictionary object in named_dests.pdf.
	buffer_size = sizeof(fixed_buffer);
	dest = FPDF_GetNamedDest(document(), 4, fixed_buffer, &buffer_size);
	EXPECT_NE(nullptr, dest);
	EXPECT_EQ(30, buffer_size);
	EXPECT_EQ(std::string("F\0i\0r\0s\0t\0A\0l\0t\0e\0r\0n\0a\0t\0e\0\0\0", 30),
	std::string(fixed_buffer, buffer_size));

	// Try to retrieve sixth item with ample buffer. Item istaken from the
	// old-style Dests dictionary object but has a sub-dictionary in
	// named_dests.pdf.
	buffer_size = sizeof(fixed_buffer);
	dest = FPDF_GetNamedDest(document(), 5, fixed_buffer, &buffer_size);
	EXPECT_NE(nullptr, dest);
	EXPECT_EQ(28, buffer_size);
	EXPECT_EQ(std::string("L\0a\0s\0t\0A\0l\0t\0e\0r\0n\0a\0t\0e\0\0\0", 28),
	std::string(fixed_buffer, buffer_size));

	// Try to retrieve non-existent item with ample buffer.
	buffer_size = sizeof(fixed_buffer);
	dest = FPDF_GetNamedDest(document(), 6, fixed_buffer, &buffer_size);
	EXPECT_EQ(nullptr, dest);
	EXPECT_EQ(sizeof(fixed_buffer),
	static_cast<size_t>(buffer_size)); // unmodified.

	// Try to underflow/overflow the integer index.
	buffer_size = sizeof(fixed_buffer);
	dest = FPDF_GetNamedDest(document(), std::numeric_limits<int>::max(),
	fixed_buffer, &buffer_size);
	EXPECT_EQ(nullptr, dest);
	EXPECT_EQ(sizeof(fixed_buffer),
	static_cast<size_t>(buffer_size)); // unmodified.

	buffer_size = sizeof(fixed_buffer);
	dest = FPDF_GetNamedDest(document(), std::numeric_limits<int>::min(),
	fixed_buffer, &buffer_size);
	EXPECT_EQ(nullptr, dest);
	EXPECT_EQ(sizeof(fixed_buffer),
	static_cast<size_t>(buffer_size)); // unmodified.

	buffer_size = sizeof(fixed_buffer);
	dest = FPDF_GetNamedDest(document(), -1, fixed_buffer, &buffer_size);
	EXPECT_EQ(nullptr, dest);
	EXPECT_EQ(sizeof(fixed_buffer),
	static_cast<size_t>(buffer_size)); // unmodified.
	}

	TEST_F(FPDFViewEmbeddertest, NamedDestsByName) {
	EXPECT_TRUE(OpenDocument("named_dests.pdf"));

	// Null pointer returns nullptr.
	FPDF_DEST dest = FPDF_GetNamedDestByName(document(), nullptr);
	EXPECT_EQ(nullptr, dest);

	// Empty string returns nullptr.
	dest = FPDF_GetNamedDestByName(document(), "");
	EXPECT_EQ(nullptr, dest);

	// Item from Dests NameTree.
	dest = FPDF_GetNamedDestByName(document(), "First");
	EXPECT_NE(nullptr, dest);

	long ignore_len = 0;
	FPDF_DEST dest_by_index =
	FPDF_GetNamedDest(document(), 0, nullptr, &ignore_len);
	EXPECT_EQ(dest_by_index, dest);

	// Item from Dests dictionary.
	dest = FPDF_GetNamedDestByName(document(), "FirstAlternate");
	EXPECT_NE(nullptr, dest);

	ignore_len = 0;
	dest_by_index = FPDF_GetNamedDest(document(), 4, nullptr, &ignore_len);
	EXPECT_EQ(dest_by_index, dest);

	// Bad value type for item from Dests NameTree array.
	dest = FPDF_GetNamedDestByName(document(), "WrongType");
	EXPECT_EQ(nullptr, dest);

	// No such destination in either Dest NameTree or dictionary.
	dest = FPDF_GetNamedDestByName(document(), "Bogus");
	EXPECT_EQ(nullptr, dest);
	}

	// The following tests pass if the document opens without crashing.
	TEST_F(FPDFViewEmbeddertest, Crasher_113) {
	EXPECT_TRUE(OpenDocument("bug_113.pdf"));
	}

	TEST_F(FPDFViewEmbeddertest, Crasher_451830) {
	// Document is damaged and can't be opened.
	EXPECT_FALSE(OpenDocument("bug_451830.pdf"));
	}

	TEST_F(FPDFViewEmbeddertest, Crasher_452455) {
	EXPECT_TRUE(OpenDocument("bug_452455.pdf"));
	FPDF_PAGE page = LoadPage(0);
	EXPECT_NE(nullptr, page);
	UnloadPage(page);
	}

	TEST_F(FPDFViewEmbeddertest, Crasher_454695) {
	// Document is damaged and can't be opened.
	EXPECT_FALSE(OpenDocument("bug_454695.pdf"));
	}

	TEST_F(FPDFViewEmbeddertest, Crasher_572871) {
	EXPECT_TRUE(OpenDocument("bug_572871.pdf"));
	}

	// It tests that document can still be loaded even the trailer has no 'Size'
	// field if other information is right.
	TEST_F(FPDFViewEmbeddertest, Failed_213) {
	EXPECT_TRUE(OpenDocument("bug_213.pdf"));
	}

	// The following tests pass if the document opens without infinite looping.
	TEST_F(FPDFViewEmbeddertest, Hang_298) {
	EXPECT_FALSE(OpenDocument("bug_298.pdf"));
	}

	// Test if the document opens without infinite looping.
	// Previously this test will hang in a loop inside LoadAllCrossRefV4. After
	// the fix, LoadAllCrossRefV4 will return false after detecting a cross
	// reference loop. Cross references will be rebuilt successfully.
	TEST_F(FPDFViewEmbeddertest, CrossRefV4Loop) {
	EXPECT_TRUE(OpenDocument("bug_xrefv4_loop.pdf"));
	MockDownloadHints hints;

	// Make sure calling FPDFAvail_IsDocAvail() on this file does not infinite
	// loop either. See bug 875.
	int ret = PDF_DATA_NOTAVAIL;
	while (ret == PDF_DATA_NOTAVAIL)
	ret = FPDFAvail_IsDocAvail(avail_, &hints);
	EXPECT_EQ(PDF_DATA_AVAIL, ret);
	}

	// The test should pass when circular references to ParseIndirectObject will not
	// cause infinite loop.
	TEST_F(FPDFViewEmbeddertest, Hang_343) {
	EXPECT_FALSE(OpenDocument("bug_343.pdf"));
	}

	// The test should pass when the absence of 'Contents' field in a signature
	// dictionary will not cause an infinite loop in CPDF_SyntaxParser::GetObject().
	TEST_F(FPDFViewEmbeddertest, Hang_344) {
	EXPECT_FALSE(OpenDocument("bug_344.pdf"));
	}

	// The test should pass when there is no infinite recursion in
	// CPDF_SyntaxParser::GetString().
	TEST_F(FPDFViewEmbeddertest, Hang_355) {
	EXPECT_FALSE(OpenDocument("bug_355.pdf"));
	}
	// The test should pass even when the file has circular references to pages.
	TEST_F(FPDFViewEmbeddertest, Hang_360) {
	EXPECT_FALSE(OpenDocument("bug_360.pdf"));
	}

	TEST_F(FPDFViewEmbeddertest, FPDF_RenderPageBitmapWithMatrix) {
	const char* const kRotatedMD5[4] = {
	"0a90de37f52127619c3dfb642b5fa2fe", "d599429574ff0dcad3bc898ea8b874ca",
	"0113386bb0bd45125bacc6dee78bfe78", "051fcfa4c1f9de28765705633a8ef3a9"};
	const char kTopLeftQuarterMD5[] = "4982be08db3f6d2e6409186ebbced9eb";
	const char kHoriStretchedMD5[] = "004bf38f3c5c76a644e6fca204747f21";
	const char kRotateandStretchMD5[] = "0ea95cacc716d003cf063a2c5ed6c8d7";

	EXPECT_TRUE(OpenDocument("rectangles.pdf"));
	FPDF_PAGE page = LoadPage(0);
	EXPECT_NE(nullptr, page);
	const int initial_width = static_cast<int>(FPDF_GetPageWidth(page));
	const int initial_height = static_cast<int>(FPDF_GetPageHeight(page));
	EXPECT_EQ(200, initial_width);
	EXPECT_EQ(300, initial_height);

	FPDF_BITMAP bitmap = RenderPage(page);
	CompareBitmap(bitmap, initial_width, initial_height, kRotatedMD5[0]);
	FPDFBitmap_Destroy(bitmap);

	int width;
	int height;
	FS_RECTF rect;
	rect.left = 0;
	rect.top = 0;
	FS_MATRIX matrix;

	// Try the easy rotations: 0, 90, 180, 270 clockwise. The output should be the
	// same as FPDF_RenderPageBitmap with the appropriate rotation flag. Per PDF
	// spec section 4.2.2, a t degree rotation is represented by [cos(t) sin(t)
	// -sin(t) cos(t) 0 0] (matrix goes on the right in the multiplication).
	rect.right = initial_width;
	rect.bottom = initial_height;
	CFX_Matrix rot_matrices[4] = {
	CFX_Matrix(1, 0, 0, 1, 0, 0), CFX_Matrix(0, -1, 1, 0, 0, 0),
	CFX_Matrix(-1, 0, 0, -1, 0, 0), CFX_Matrix(0, 1, -1, 0, 0, 0)};
	for (int rot = 0; rot < 4; ++rot) {
	matrix.a = rot_matrices[rot].a;
	matrix.b = rot_matrices[rot].b;
	matrix.c = rot_matrices[rot].c;
	matrix.d = rot_matrices[rot].d;
	matrix.e = rot_matrices[rot].e;
	matrix.f = rot_matrices[rot].f;
	if (rot % 2 == 0) {
	width = initial_width;
	height = initial_height;
	} else {
	width = initial_height;
	height = initial_width;
	}
	rect.right = width;
	rect.bottom = height;

	bitmap = FPDFBitmap_Create(width, height, 0);
	FPDFBitmap_FillRect(bitmap, 0, 0, width, height, 0xFFFFFFFF);
	FPDF_RenderPageBitmap(bitmap, page, 0, 0, width, height, rot, 0);
	CompareBitmap(bitmap, width, height, kRotatedMD5[rot]);
	FPDFBitmap_Destroy(bitmap);

	bitmap = FPDFBitmap_Create(width, height, 0);
	FPDFBitmap_FillRect(bitmap, 0, 0, width, height, 0xFFFFFFFF);
	FPDF_RenderPageBitmapWithMatrix(bitmap, page, &matrix, &rect, 0);
	CompareBitmap(bitmap, width, height, kRotatedMD5[rot]);
	FPDFBitmap_Destroy(bitmap);
	}
	// TODO(npm): what to do with transformations that do not align the page with
	// the axis, like a 45 degree rotation (currently, part of the page gets cut
	// out). pdfium:849

	// Now render again with the image scaled smaller.
	width = initial_width / 2;
	height = initial_height / 2;
	matrix.a = 0.5;
	matrix.b = 0;
	matrix.c = 0;
	matrix.d = 0.5;

	rect.right = width;
	rect.bottom = height;

	bitmap = FPDFBitmap_Create(width, height, 0);
	FPDFBitmap_FillRect(bitmap, 0, 0, width, height, 0xFFFFFFFF);
	FPDF_RenderPageBitmapWithMatrix(bitmap, page, &matrix, &rect, 0);
	CompareBitmap(bitmap, width, height, kTopLeftQuarterMD5);
	FPDFBitmap_Destroy(bitmap);

	// Now render again with the image scaled larger horizontally.
	width = initial_width * 2;
	height = initial_height;
	matrix.a = 2;
	matrix.d = 1;
	rect.right = width;
	rect.bottom = height;
	bitmap = FPDFBitmap_Create(width, height, 0);
	FPDFBitmap_FillRect(bitmap, 0, 0, width, height, 0xFFFFFFFF);
	FPDF_RenderPageBitmapWithMatrix(bitmap, page, &matrix, &rect, 0);
	CompareBitmap(bitmap, width, height, kHoriStretchedMD5);
	FPDFBitmap_Destroy(bitmap);

	// Test a rotation followed by a stretch.
	width = initial_height * 2;
	height = initial_width;
	matrix.a = 0;
	matrix.b = -1;
	matrix.c = 2;
	matrix.d = 0;
	matrix.e = 0;
	matrix.f = 0;
	rect.right = width;
	rect.bottom = height;
	bitmap = FPDFBitmap_Create(width, height, 0);
	FPDFBitmap_FillRect(bitmap, 0, 0, width, height, 0xFFFFFFFF);
	FPDF_RenderPageBitmapWithMatrix(bitmap, page, &matrix, &rect, 0);
	CompareBitmap(bitmap, width, height, kRotateandStretchMD5);
	FPDFBitmap_Destroy(bitmap);

	UnloadPage(page);
	}

	class UnSupRecordDelegate : public EmbedderTest::Delegate {
	public:
	UnSupRecordDelegate() : type_(-1) {}
	~UnSupRecordDelegate() override {}

	void UnsupportedHandler(int type) override { type_ = type; }

	int type_;
	};

	TEST_F(FPDFViewEmbeddertest, UnSupportedOperations_NotFound) {
	UnSupRecordDelegate delegate;
	SetDelegate(&delegate);
	ASSERT_TRUE(OpenDocument("hello_world.pdf"));
	EXPECT_EQ(delegate.type_, -1);
	SetDelegate(nullptr);
	}

	TEST_F(FPDFViewEmbeddertest, UnSupportedOperations_LoadCustomDocument) {
	UnSupRecordDelegate delegate;
	SetDelegate(&delegate);
	ASSERT_TRUE(OpenDocument("unsupported_feature.pdf"));
	EXPECT_EQ(FPDF_UNSP_DOC_PORTABLECOLLECTION, delegate.type_);
	SetDelegate(nullptr);
	}

	TEST_F(FPDFViewEmbeddertest, UnSupportedOperations_LoadDocument) {
	std::string file_path;
	ASSERT_TRUE(
	PathService::GetTestFilePath("unsupported_feature.pdf", &file_path));

	UnSupRecordDelegate delegate;
	SetDelegate(&delegate);
	FPDF_DOCUMENT doc = FPDF_LoadDocument(file_path.c_str(), "");
	EXPECT_TRUE(doc != nullptr);
	EXPECT_EQ(FPDF_UNSP_DOC_PORTABLECOLLECTION, delegate.type_);
	FPDF_CloseDocument(doc);
	SetDelegate(nullptr);
	}