| // 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 <memory> | |
| #include <string> | |
| #include "build/build_config.h" | |
| #include "core/fpdfapi/parser/cpdf_dictionary.h" | |
| #include "core/fpdfapi/parser/cpdf_document.h" | |
| #include "core/fpdfapi/parser/cpdf_parser.h" | |
| #include "core/fxcrt/fx_system.h" | |
| #include "public/cpp/fpdf_scopers.h" | |
| #include "public/fpdf_edit.h" | |
| #include "public/fpdf_save.h" | |
| #include "public/fpdfview.h" | |
| #include "testing/embedder_test.h" | |
| #include "testing/embedder_test_constants.h" | |
| #include "testing/gtest/include/gtest/gtest.h" | |
| namespace { | |
| const char kAgeUTF8[] = | |
| "\xc3\xa2" | |
| "ge"; | |
| const char kAgeLatin1[] = | |
| "\xe2" | |
| "ge"; | |
| const char kHotelUTF8[] = | |
| "h" | |
| "\xc3\xb4" | |
| "tel"; | |
| const char kHotelLatin1[] = | |
| "h" | |
| "\xf4" | |
| "tel"; | |
| } // namespace | |
| class CPDFSecurityHandlerEmbedderTest : public EmbedderTest { | |
| protected: | |
| void OpenAndVerifyHelloWorldDocumentWithPassword(const char* filename, | |
| const char* password) { | |
| ASSERT_TRUE(OpenDocumentWithPassword(filename, password)); | |
| FPDF_PAGE page = LoadPage(0); | |
| VerifyHelloWorldPage(page); | |
| UnloadPage(page); | |
| } | |
| void VerifySavedHelloWorldDocumentWithPassword(const char* password) { | |
| ASSERT_TRUE(OpenSavedDocumentWithPassword(password)); | |
| FPDF_PAGE page = LoadSavedPage(0); | |
| VerifyHelloWorldPage(page); | |
| CloseSavedPage(page); | |
| CloseSavedDocument(); | |
| } | |
| void VerifySavedModifiedHelloWorldDocumentWithPassword(const char* password) { | |
| ASSERT_TRUE(OpenSavedDocumentWithPassword(password)); | |
| FPDF_PAGE page = LoadSavedPage(0); | |
| VerifyModifiedHelloWorldPage(page); | |
| CloseSavedPage(page); | |
| CloseSavedDocument(); | |
| } | |
| void RemoveTrailerIdFromDocument() { | |
| // This is cheating slightly to avoid a layering violation, since this file | |
| // cannot include fpdfsdk/cpdfsdk_helpers.h to get access to | |
| // CPDFDocumentFromFPDFDocument(). | |
| CPDF_Document* doc = reinterpret_cast<CPDF_Document*>((document())); | |
| ASSERT_TRUE(doc); | |
| CPDF_Parser* parser = doc->GetParser(); | |
| ASSERT_TRUE(parser); | |
| CPDF_Dictionary* trailer = parser->GetMutableTrailerForTesting(); | |
| ASSERT_TRUE(trailer); | |
| ASSERT_TRUE(trailer->RemoveFor("ID")); | |
| } | |
| void RemoveGoodbyeObject() { | |
| FPDF_PAGE page = LoadPage(0); | |
| { | |
| ScopedFPDFPageObject goodbye_object(FPDFPage_GetObject(page, 1)); | |
| ASSERT_TRUE(goodbye_object); | |
| ASSERT_TRUE(FPDFPage_RemoveObject(page, goodbye_object.get())); | |
| } | |
| ASSERT_TRUE(FPDFPage_GenerateContent(page)); | |
| VerifyModifiedHelloWorldPage(page); | |
| UnloadPage(page); | |
| } | |
| private: | |
| void VerifyHelloWorldPage(FPDF_PAGE page) { | |
| ASSERT_TRUE(page); | |
| ScopedFPDFBitmap page_bitmap = RenderPage(page); | |
| CompareBitmap(page_bitmap.get(), 200, 200, pdfium::kHelloWorldChecksum); | |
| } | |
| void VerifyModifiedHelloWorldPage(FPDF_PAGE page) { | |
| ASSERT_TRUE(page); | |
| ScopedFPDFBitmap page_bitmap = RenderPage(page); | |
| CompareBitmap(page_bitmap.get(), 200, 200, | |
| pdfium::kHelloWorldRemovedChecksum); | |
| } | |
| }; | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, Unencrypted) { | |
| ASSERT_TRUE(OpenDocument("about_blank.pdf")); | |
| EXPECT_EQ(0xFFFFFFFF, FPDF_GetDocPermissions(document())); | |
| } | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, UnencryptedWithPassword) { | |
| ASSERT_TRUE(OpenDocumentWithPassword("about_blank.pdf", "foobar")); | |
| EXPECT_EQ(0xFFFFFFFF, FPDF_GetDocPermissions(document())); | |
| } | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, NoPassword) { | |
| EXPECT_FALSE(OpenDocument("encrypted.pdf")); | |
| } | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, BadPassword) { | |
| EXPECT_FALSE(OpenDocumentWithPassword("encrypted.pdf", "tiger")); | |
| } | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, UserPassword) { | |
| ASSERT_TRUE(OpenDocumentWithPassword("encrypted.pdf", "1234")); | |
| EXPECT_EQ(0xFFFFF2C0, FPDF_GetDocPermissions(document())); | |
| } | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, OwnerPassword) { | |
| ASSERT_TRUE(OpenDocumentWithPassword("encrypted.pdf", "5678")); | |
| EXPECT_EQ(0xFFFFFFFC, FPDF_GetDocPermissions(document())); | |
| } | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, PasswordAfterGenerateSave) { | |
| #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) | |
| #if defined(OS_WIN) | |
| const char kChecksum[] = "06fe5a97341b3e0f0a22ccc242fd9040"; | |
| #else | |
| const char kChecksum[] = "169c8e3acea8fba5a40f695bbbc96273"; | |
| #endif // defined(OS_WIN) | |
| #else | |
| #if defined(OS_WIN) | |
| const char kChecksum[] = "041c2fb541c8907cc22ce101b686c79e"; | |
| #elif defined(OS_APPLE) | |
| const char kChecksum[] = "1ace03eb7c466c132aacf319cb9d69d3"; | |
| #else | |
| const char kChecksum[] = "7048dca58e2ed8f93339008b91e4eb4e"; | |
| #endif | |
| #endif // defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) | |
| { | |
| ASSERT_TRUE(OpenDocumentWithOptions("encrypted.pdf", "5678", | |
| LinearizeOption::kMustLinearize, | |
| JavaScriptOption::kEnableJavaScript)); | |
| FPDF_PAGE page = LoadPage(0); | |
| ASSERT_TRUE(page); | |
| FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(10, 10, 20, 20); | |
| ASSERT_TRUE(red_rect); | |
| EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 0, 0, 255)); | |
| EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0)); | |
| FPDFPage_InsertObject(page, red_rect); | |
| ScopedFPDFBitmap bitmap = RenderLoadedPage(page); | |
| CompareBitmap(bitmap.get(), 612, 792, kChecksum); | |
| EXPECT_TRUE(FPDFPage_GenerateContent(page)); | |
| SetWholeFileAvailable(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| UnloadPage(page); | |
| } | |
| std::string new_file = GetString(); | |
| FPDF_FILEACCESS file_access; | |
| memset(&file_access, 0, sizeof(file_access)); | |
| file_access.m_FileLen = new_file.size(); | |
| file_access.m_GetBlock = GetBlockFromString; | |
| file_access.m_Param = &new_file; | |
| EXPECT_FALSE(FPDF_LoadCustomDocument(&file_access, nullptr)); | |
| struct { | |
| const char* password; | |
| const unsigned long permissions; | |
| } tests[] = {{"1234", 0xFFFFF2C0}, {"5678", 0xFFFFFFFC}}; | |
| for (const auto& test : tests) { | |
| ASSERT_TRUE(OpenSavedDocumentWithPassword(test.password)); | |
| FPDF_PAGE page = LoadSavedPage(0); | |
| VerifySavedRendering(page, 612, 792, kChecksum); | |
| EXPECT_EQ(test.permissions, FPDF_GetDocPermissions(saved_document_)); | |
| CloseSavedPage(page); | |
| CloseSavedDocument(); | |
| } | |
| } | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, NoPasswordVersion5) { | |
| ASSERT_FALSE(OpenDocument("bug_644.pdf")); | |
| } | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, BadPasswordVersion5) { | |
| ASSERT_FALSE(OpenDocumentWithPassword("bug_644.pdf", "tiger")); | |
| } | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, OwnerPasswordVersion5) { | |
| ASSERT_TRUE(OpenDocumentWithPassword("bug_644.pdf", "a")); | |
| EXPECT_EQ(0xFFFFFFFC, FPDF_GetDocPermissions(document())); | |
| } | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, UserPasswordVersion5) { | |
| ASSERT_TRUE(OpenDocumentWithPassword("bug_644.pdf", "b")); | |
| EXPECT_EQ(0xFFFFFFFC, FPDF_GetDocPermissions(document())); | |
| } | |
| // Should not crash. https://crbug.com/pdfium/1436 | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, BadOkeyVersion2) { | |
| EXPECT_FALSE( | |
| OpenDocumentWithPassword("encrypted_hello_world_r2_bad_okey.pdf", "a")); | |
| } | |
| // Should not crash. https://crbug.com/pdfium/1436 | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, BadOkeyVersion3) { | |
| EXPECT_FALSE( | |
| OpenDocumentWithPassword("encrypted_hello_world_r3_bad_okey.pdf", "a")); | |
| } | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, OwnerPasswordVersion2UTF8) { | |
| // The password is "age", where the 'a' has a circumflex. Encoding the | |
| // password as UTF-8 works. | |
| OpenAndVerifyHelloWorldDocumentWithPassword("encrypted_hello_world_r2.pdf", | |
| kAgeUTF8); | |
| EXPECT_EQ(2, FPDF_GetSecurityHandlerRevision(document())); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveTrailerIdFromDocument(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| // With revision 2 and 3, the owner password is not tied to the document ID in | |
| // the trailer, so the owner password entry remains in the copy and is still | |
| // valid, even though the document ID has changed. | |
| // The user password is tied to the document ID, so without an existing ID, | |
| // the user password entry has to be regenerated with the owner password. | |
| // Since the user password was not used to decrypt the document, it cannot be | |
| // recovered. Thus only verify the owner password, which is now also the user | |
| // password. | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| ClearString(); | |
| RemoveGoodbyeObject(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| } | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, OwnerPasswordVersion2Latin1) { | |
| // The same password encoded as Latin-1 also works at revision 2. | |
| OpenAndVerifyHelloWorldDocumentWithPassword("encrypted_hello_world_r2.pdf", | |
| kAgeLatin1); | |
| EXPECT_EQ(2, FPDF_GetSecurityHandlerRevision(document())); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveTrailerIdFromDocument(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| ClearString(); | |
| RemoveGoodbyeObject(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| } | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, OwnerPasswordVersion3UTF8) { | |
| // Same as OwnerPasswordVersion2UTF8 test above. | |
| OpenAndVerifyHelloWorldDocumentWithPassword("encrypted_hello_world_r3.pdf", | |
| kAgeUTF8); | |
| EXPECT_EQ(3, FPDF_GetSecurityHandlerRevision(document())); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveTrailerIdFromDocument(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| ClearString(); | |
| RemoveGoodbyeObject(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| } | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, OwnerPasswordVersion3Latin1) { | |
| // Same as OwnerPasswordVersion2Latin1 test above. | |
| OpenAndVerifyHelloWorldDocumentWithPassword("encrypted_hello_world_r3.pdf", | |
| kAgeLatin1); | |
| EXPECT_EQ(3, FPDF_GetSecurityHandlerRevision(document())); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveTrailerIdFromDocument(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| ClearString(); | |
| RemoveGoodbyeObject(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| } | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, OwnerPasswordVersion5UTF8) { | |
| // Same as OwnerPasswordVersion2UTF8 test above. | |
| OpenAndVerifyHelloWorldDocumentWithPassword("encrypted_hello_world_r5.pdf", | |
| kAgeUTF8); | |
| EXPECT_EQ(5, FPDF_GetSecurityHandlerRevision(document())); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveTrailerIdFromDocument(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveGoodbyeObject(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| } | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, OwnerPasswordVersion5Latin1) { | |
| // Same as OwnerPasswordVersion2Latin1 test above. | |
| OpenAndVerifyHelloWorldDocumentWithPassword("encrypted_hello_world_r5.pdf", | |
| kAgeLatin1); | |
| EXPECT_EQ(5, FPDF_GetSecurityHandlerRevision(document())); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveTrailerIdFromDocument(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveGoodbyeObject(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| } | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, OwnerPasswordVersion6UTF8) { | |
| // Same as OwnerPasswordVersion2UTF8 test above. | |
| OpenAndVerifyHelloWorldDocumentWithPassword("encrypted_hello_world_r6.pdf", | |
| kAgeUTF8); | |
| EXPECT_EQ(6, FPDF_GetSecurityHandlerRevision(document())); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveTrailerIdFromDocument(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveGoodbyeObject(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| } | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, OwnerPasswordVersion6Latin1) { | |
| // Same as OwnerPasswordVersion2Latin1 test above. | |
| OpenAndVerifyHelloWorldDocumentWithPassword("encrypted_hello_world_r6.pdf", | |
| kAgeLatin1); | |
| EXPECT_EQ(6, FPDF_GetSecurityHandlerRevision(document())); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveTrailerIdFromDocument(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveGoodbyeObject(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| } | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, UserPasswordVersion2UTF8) { | |
| // The password is "hotel", where the 'o' has a circumflex. Encoding the | |
| // password as UTF-8 works. | |
| OpenAndVerifyHelloWorldDocumentWithPassword("encrypted_hello_world_r2.pdf", | |
| kHotelUTF8); | |
| EXPECT_EQ(2, FPDF_GetSecurityHandlerRevision(document())); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveTrailerIdFromDocument(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| // Unlike the OwnerPasswordVersion2UTF8 test case, the user password was used | |
| // to decrypt the document, so it is available to regenerated the user | |
| // password entry. Thus it is possible to verify with both the unmodified | |
| // owner password, and the updated user password. | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveGoodbyeObject(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| } | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, UserPasswordVersion2Latin1) { | |
| // The same password encoded as Latin-1 also works at revision 2. | |
| OpenAndVerifyHelloWorldDocumentWithPassword("encrypted_hello_world_r2.pdf", | |
| kHotelLatin1); | |
| EXPECT_EQ(2, FPDF_GetSecurityHandlerRevision(document())); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveTrailerIdFromDocument(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveGoodbyeObject(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| } | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, UserPasswordVersion3UTF8) { | |
| // Same as UserPasswordVersion2UTF8 test above. | |
| OpenAndVerifyHelloWorldDocumentWithPassword("encrypted_hello_world_r3.pdf", | |
| kHotelUTF8); | |
| EXPECT_EQ(3, FPDF_GetSecurityHandlerRevision(document())); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveTrailerIdFromDocument(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveGoodbyeObject(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| } | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, UserPasswordVersion3Latin1) { | |
| // Same as UserPasswordVersion2Latin1 test above. | |
| OpenAndVerifyHelloWorldDocumentWithPassword("encrypted_hello_world_r3.pdf", | |
| kHotelLatin1); | |
| EXPECT_EQ(3, FPDF_GetSecurityHandlerRevision(document())); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveTrailerIdFromDocument(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveGoodbyeObject(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| } | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, UserPasswordVersion5UTF8) { | |
| // Same as UserPasswordVersion2UTF8 test above. | |
| OpenAndVerifyHelloWorldDocumentWithPassword("encrypted_hello_world_r5.pdf", | |
| kHotelUTF8); | |
| EXPECT_EQ(5, FPDF_GetSecurityHandlerRevision(document())); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveTrailerIdFromDocument(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveGoodbyeObject(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| } | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, UserPasswordVersion5Latin1) { | |
| // Same as UserPasswordVersion2Latin1 test above. | |
| OpenAndVerifyHelloWorldDocumentWithPassword("encrypted_hello_world_r5.pdf", | |
| kHotelLatin1); | |
| EXPECT_EQ(5, FPDF_GetSecurityHandlerRevision(document())); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveTrailerIdFromDocument(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveGoodbyeObject(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| } | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, UserPasswordVersion6UTF8) { | |
| // Same as UserPasswordVersion2UTF8 test above. | |
| OpenAndVerifyHelloWorldDocumentWithPassword("encrypted_hello_world_r6.pdf", | |
| kHotelUTF8); | |
| EXPECT_EQ(6, FPDF_GetSecurityHandlerRevision(document())); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveTrailerIdFromDocument(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveGoodbyeObject(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| } | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, UserPasswordVersion6Latin1) { | |
| // Same as UserPasswordVersion2Latin1 test above. | |
| OpenAndVerifyHelloWorldDocumentWithPassword("encrypted_hello_world_r6.pdf", | |
| kHotelLatin1); | |
| EXPECT_EQ(6, FPDF_GetSecurityHandlerRevision(document())); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveTrailerIdFromDocument(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| ClearString(); | |
| RemoveGoodbyeObject(); | |
| EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kAgeUTF8); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kHotelLatin1); | |
| VerifySavedModifiedHelloWorldDocumentWithPassword(kHotelUTF8); | |
| } | |
| TEST_F(CPDFSecurityHandlerEmbedderTest, BUG_1124998) { | |
| OpenAndVerifyHelloWorldDocumentWithPassword("bug_1124998.pdf", "test"); | |
| } |