// 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. // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com #include "core/fpdfdoc/cpdf_filespec.h" #include <vector> #include "core/fpdfapi/parser/cpdf_dictionary.h" #include "core/fpdfapi/parser/cpdf_name.h" #include "core/fpdfapi/parser/cpdf_object.h" #include "core/fpdfapi/parser/cpdf_stream.h" #include "core/fpdfapi/parser/cpdf_string.h" #include "core/fpdfapi/parser/fpdf_parser_decode.h" #include "core/fxcrt/fx_system.h" namespace { #if _FXM_PLATFORM_ == _FXM_PLATFORM_APPLE_ || \ _FXM_PLATFORM_ == _FXM_PLATFORM_WINDOWS_ CFX_WideString ChangeSlashToPlatform(const wchar_t* str) { CFX_WideString result; while (*str) { if (*str == '/') { #if _FXM_PLATFORM_ == _FXM_PLATFORM_APPLE_ result += L':'; #else result += L'\\'; #endif } else { result += *str; } str++; } return result; } CFX_WideString ChangeSlashToPDF(const wchar_t* str) { CFX_WideString result; while (*str) { if (*str == '\\' || *str == ':') result += L'/'; else result += *str; str++; } return result; } #endif // _FXM_PLATFORM_APPLE_ || _FXM_PLATFORM_WINDOWS_ } // namespace CPDF_FileSpec::CPDF_FileSpec(CPDF_Object* pObj) : m_pObj(pObj) { ASSERT(m_pObj); } CPDF_FileSpec::~CPDF_FileSpec() {} CFX_WideString CPDF_FileSpec::DecodeFileName(const CFX_WideString& filepath) { if (filepath.GetLength() <= 1) return CFX_WideString(); #if _FXM_PLATFORM_ == _FXM_PLATFORM_APPLE_ if (filepath.Left(sizeof("/Mac") - 1) == CFX_WideStringC(L"/Mac")) return ChangeSlashToPlatform(filepath.c_str() + 1); return ChangeSlashToPlatform(filepath.c_str()); #elif _FXM_PLATFORM_ == _FXM_PLATFORM_WINDOWS_ if (filepath.GetAt(0) != L'/') return ChangeSlashToPlatform(filepath.c_str()); if (filepath.GetAt(1) == L'/') return ChangeSlashToPlatform(filepath.c_str() + 1); if (filepath.GetAt(2) == L'/') { CFX_WideString result; result += filepath.GetAt(1); result += L':'; result += ChangeSlashToPlatform(filepath.c_str() + 2); return result; } CFX_WideString result; result += L'\\'; result += ChangeSlashToPlatform(filepath.c_str()); return result; #else return CFX_WideString(filepath); #endif } CFX_WideString CPDF_FileSpec::GetFileName() const { CFX_WideString csFileName; if (CPDF_Dictionary* pDict = m_pObj->AsDictionary()) { csFileName = pDict->GetUnicodeTextFor("UF"); if (csFileName.IsEmpty()) { csFileName = CFX_WideString::FromLocal(pDict->GetStringFor("F").AsStringC()); } if (pDict->GetStringFor("FS") == "URL") return csFileName; if (csFileName.IsEmpty()) { constexpr const char* keys[] = {"DOS", "Mac", "Unix"}; for (const auto* key : keys) { if (pDict->KeyExist(key)) { csFileName = CFX_WideString::FromLocal(pDict->GetStringFor(key).AsStringC()); break; } } } } else if (m_pObj->IsString()) { csFileName = CFX_WideString::FromLocal(m_pObj->GetString().AsStringC()); } return DecodeFileName(csFileName); } CPDF_Stream* CPDF_FileSpec::GetFileStream() const { CPDF_Dictionary* pDict = m_pObj->AsDictionary(); if (!pDict) return nullptr; // Get the embedded files dictionary. CPDF_Dictionary* pFiles = pDict->GetDictFor("EF"); if (!pFiles) return nullptr; // Get the file stream of the highest precedence with its file specification // string available. Follows the same precedence order as GetFileName(). constexpr const char* keys[] = {"UF", "F", "DOS", "Mac", "Unix"}; size_t end = pDict->GetStringFor("FS") == "URL" ? 2 : FX_ArraySize(keys); for (size_t i = 0; i < end; ++i) { const CFX_ByteString& key = keys[i]; if (!pDict->GetUnicodeTextFor(key).IsEmpty()) { CPDF_Stream* pStream = pFiles->GetStreamFor(key); if (pStream) return pStream; } } return nullptr; } CPDF_Dictionary* CPDF_FileSpec::GetParamsDict() const { CPDF_Stream* pStream = GetFileStream(); if (!pStream) return nullptr; CPDF_Dictionary* pDict = pStream->GetDict(); if (!pDict) return nullptr; return pDict->GetDictFor("Params"); } CFX_WideString CPDF_FileSpec::EncodeFileName(const CFX_WideString& filepath) { if (filepath.GetLength() <= 1) return CFX_WideString(); #if _FXM_PLATFORM_ == _FXM_PLATFORM_WINDOWS_ if (filepath.GetAt(1) == L':') { CFX_WideString result(L'/'); result += filepath.GetAt(0); if (filepath.GetAt(2) != L'\\') result += L'/'; result += ChangeSlashToPDF(filepath.c_str() + 2); return result; } if (filepath.GetAt(0) == L'\\' && filepath.GetAt(1) == L'\\') return ChangeSlashToPDF(filepath.c_str() + 1); if (filepath.GetAt(0) == L'\\') return L'/' + ChangeSlashToPDF(filepath.c_str()); return ChangeSlashToPDF(filepath.c_str()); #elif _FXM_PLATFORM_ == _FXM_PLATFORM_APPLE_ if (filepath.Left(sizeof("Mac") - 1) == L"Mac") return L'/' + ChangeSlashToPDF(filepath.c_str()); return ChangeSlashToPDF(filepath.c_str()); #else return CFX_WideString(filepath); #endif } void CPDF_FileSpec::SetFileName(const CFX_WideString& wsFileName) { CFX_WideString wsStr = EncodeFileName(wsFileName); if (m_pObj->IsString()) { m_pObj->SetString(CFX_ByteString::FromUnicode(wsStr)); } else if (CPDF_Dictionary* pDict = m_pObj->AsDictionary()) { pDict->SetNewFor<CPDF_String>("F", CFX_ByteString::FromUnicode(wsStr), false); pDict->SetNewFor<CPDF_String>("UF", PDF_EncodeText(wsStr), false); } }