// 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 "core/fpdfdoc/include/fpdf_doc.h" #include <memory> #include <vector> #include "core/fpdfapi/fpdf_parser/include/cpdf_name.h" #include "core/fpdfapi/fpdf_parser/include/cpdf_string.h" #include "testing/gtest/include/gtest/gtest.h" #include "testing/test_support.h" namespace { using ScopedObj = std::unique_ptr<CPDF_Object, ReleaseDeleter<CPDF_Object>>; using ScopedDict = std::unique_ptr<CPDF_Dictionary, ReleaseDeleter<CPDF_Dictionary>>; } TEST(doc_basic_filespec, EncodeDecodeFileName) { std::vector<pdfium::NullTermWstrFuncTestData> test_data = { // Empty src string. {L"", L""}, // only file name. {L"test.pdf", L"test.pdf"}, #if _FXM_PLATFORM_ == _FXM_PLATFORM_WINDOWS_ // With drive identifier. {L"r:\\pdfdocs\\spec.pdf", L"/r/pdfdocs/spec.pdf"}, // Relative path. {L"My Document\\test.pdf", L"My Document/test.pdf"}, // Absolute path without drive identifier. {L"\\pdfdocs\\spec.pdf", L"//pdfdocs/spec.pdf"}, // Absolute path with double backslashes. {L"\\\\pdfdocs\\spec.pdf", L"/pdfdocs/spec.pdf"}, // Network resource name. It is not supported yet. // {L"pclib/eng:\\pdfdocs\\spec.pdf", L"/pclib/eng/pdfdocs/spec.pdf"}, #elif _FXM_PLATFORM_ == _FXM_PLATFORM_APPLE_ // Absolute path with colon separator. {L"Mac HD:PDFDocs:spec.pdf", L"/Mac HD/PDFDocs/spec.pdf"}, // Relative path with colon separator. {L"PDFDocs:spec.pdf", L"PDFDocs/spec.pdf"}, #else // Relative path. {L"./docs/test.pdf", L"./docs/test.pdf"}, // Relative path with parent dir. {L"../test_docs/test.pdf", L"../test_docs/test.pdf"}, // Absolute path. {L"/usr/local/home/test.pdf", L"/usr/local/home/test.pdf"}, #endif }; for (const auto& data : test_data) { CFX_WideString encoded_str = CPDF_FileSpec::EncodeFileName(data.input); EXPECT_TRUE(encoded_str == data.expected); // DecodeFileName is the reverse procedure of EncodeFileName. CFX_WideString decoded_str = CPDF_FileSpec::DecodeFileName(data.expected); EXPECT_TRUE(decoded_str == data.input); } } TEST(doc_basic_filespec, GetFileName) { { // String object. pdfium::NullTermWstrFuncTestData test_data = { #if _FXM_PLATFORM_ == _FXM_PLATFORM_WINDOWS_ L"/C/docs/test.pdf", L"C:\\docs\\test.pdf" #elif _FXM_PLATFORM_ == _FXM_PLATFORM_APPLE_ L"/Mac HD/docs/test.pdf", L"Mac HD:docs:test.pdf" #else L"/docs/test.pdf", L"/docs/test.pdf" #endif }; ScopedObj str_obj(new CPDF_String(test_data.input)); CPDF_FileSpec file_spec(str_obj.get()); CFX_WideString file_name; EXPECT_TRUE(file_spec.GetFileName(&file_name)); EXPECT_TRUE(file_name == test_data.expected); } { // Dictionary object. pdfium::NullTermWstrFuncTestData test_data[5] = { #if _FXM_PLATFORM_ == _FXM_PLATFORM_WINDOWS_ {L"/C/docs/test.pdf", L"C:\\docs\\test.pdf"}, {L"/D/docs/test.pdf", L"D:\\docs\\test.pdf"}, {L"/E/docs/test.pdf", L"E:\\docs\\test.pdf"}, {L"/F/docs/test.pdf", L"F:\\docs\\test.pdf"}, {L"/G/docs/test.pdf", L"G:\\docs\\test.pdf"}, #elif _FXM_PLATFORM_ == _FXM_PLATFORM_APPLE_ {L"/Mac HD/docs1/test.pdf", L"Mac HD:docs1:test.pdf"}, {L"/Mac HD/docs2/test.pdf", L"Mac HD:docs2:test.pdf"}, {L"/Mac HD/docs3/test.pdf", L"Mac HD:docs3:test.pdf"}, {L"/Mac HD/docs4/test.pdf", L"Mac HD:docs4:test.pdf"}, {L"/Mac HD/docs5/test.pdf", L"Mac HD:docs5:test.pdf"}, #else {L"/docs/a/test.pdf", L"/docs/a/test.pdf"}, {L"/docs/b/test.pdf", L"/docs/b/test.pdf"}, {L"/docs/c/test.pdf", L"/docs/c/test.pdf"}, {L"/docs/d/test.pdf", L"/docs/d/test.pdf"}, {L"/docs/e/test.pdf", L"/docs/e/test.pdf"}, #endif }; // Keyword fields in reverse order of precedence to retrieve the file name. const char* const keywords[5] = {"Unix", "Mac", "DOS", "F", "UF"}; ScopedDict dict_obj(new CPDF_Dictionary); CPDF_FileSpec file_spec(dict_obj.get()); CFX_WideString file_name; for (int i = 0; i < 5; ++i) { dict_obj->SetAt(keywords[i], new CPDF_String(test_data[i].input)); EXPECT_TRUE(file_spec.GetFileName(&file_name)); EXPECT_TRUE(file_name == test_data[i].expected); } // With all the former fields and 'FS' field suggests 'URL' type. dict_obj->SetAtString("FS", "URL"); EXPECT_TRUE(file_spec.GetFileName(&file_name)); // Url string is not decoded. EXPECT_TRUE(file_name == test_data[4].input); } { // Invalid object. ScopedObj name_obj(new CPDF_Name("test.pdf")); CPDF_FileSpec file_spec(name_obj.get()); CFX_WideString file_name; EXPECT_FALSE(file_spec.GetFileName(&file_name)); } } TEST(doc_basic_filespec, SetFileName) { pdfium::NullTermWstrFuncTestData test_data = { #if _FXM_PLATFORM_ == _FXM_PLATFORM_WINDOWS_ L"C:\\docs\\test.pdf", L"/C/docs/test.pdf" #elif _FXM_PLATFORM_ == _FXM_PLATFORM_APPLE_ L"Mac HD:docs:test.pdf", L"/Mac HD/docs/test.pdf" #else L"/docs/test.pdf", L"/docs/test.pdf" #endif }; // String object. ScopedObj str_obj(new CPDF_String(L"babababa")); CPDF_FileSpec file_spec1(str_obj.get()); file_spec1.SetFileName(test_data.input); // Check internal object value. CFX_ByteString str = CFX_ByteString::FromUnicode(test_data.expected); EXPECT_TRUE(str == str_obj->GetString()); // Check we can get the file name back. CFX_WideString file_name; EXPECT_TRUE(file_spec1.GetFileName(&file_name)); EXPECT_TRUE(file_name == test_data.input); // Dictionary object. ScopedDict dict_obj(new CPDF_Dictionary); CPDF_FileSpec file_spec2(dict_obj.get()); file_spec2.SetFileName(test_data.input); // Check internal object value. file_name = dict_obj->GetUnicodeTextBy("F"); EXPECT_TRUE(file_name == test_data.expected); file_name = dict_obj->GetUnicodeTextBy("UF"); EXPECT_TRUE(file_name == test_data.expected); // Check we can get the file name back. EXPECT_TRUE(file_spec2.GetFileName(&file_name)); EXPECT_TRUE(file_name == test_data.input); }