| // 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 "build/build_config.h" |
| #include "constants/stream_dict_common.h" |
| #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" |
| #include "third_party/base/check.h" |
| #include "third_party/base/notreached.h" |
| #include "third_party/base/stl_util.h" |
| |
| namespace { |
| |
| #if defined(OS_APPLE) || defined(OS_WIN) |
| WideString ChangeSlashToPlatform(const wchar_t* str) { |
| WideString result; |
| while (*str) { |
| if (*str == '/') { |
| #if defined(OS_APPLE) |
| result += L':'; |
| #else |
| result += L'\\'; |
| #endif |
| } else { |
| result += *str; |
| } |
| str++; |
| } |
| return result; |
| } |
| |
| WideString ChangeSlashToPDF(const wchar_t* str) { |
| WideString result; |
| while (*str) { |
| if (*str == '\\' || *str == ':') |
| result += L'/'; |
| else |
| result += *str; |
| |
| str++; |
| } |
| return result; |
| } |
| #endif // defined(OS_APPLE) || defined(OS_WIN) |
| |
| } // namespace |
| |
| CPDF_FileSpec::CPDF_FileSpec(const CPDF_Object* pObj) : m_pObj(pObj) { |
| DCHECK(m_pObj); |
| } |
| |
| CPDF_FileSpec::CPDF_FileSpec(CPDF_Object* pObj) |
| : m_pObj(pObj), m_pWritableObj(pObj) { |
| DCHECK(m_pObj); |
| } |
| |
| CPDF_FileSpec::~CPDF_FileSpec() = default; |
| |
| WideString CPDF_FileSpec::DecodeFileName(const WideString& filepath) { |
| if (filepath.GetLength() <= 1) |
| return WideString(); |
| |
| #if defined(OS_APPLE) |
| if (filepath.First(sizeof("/Mac") - 1) == WideStringView(L"/Mac")) |
| return ChangeSlashToPlatform(filepath.c_str() + 1); |
| return ChangeSlashToPlatform(filepath.c_str()); |
| #elif defined(OS_WIN) |
| |
| if (filepath[0] != L'/') |
| return ChangeSlashToPlatform(filepath.c_str()); |
| if (filepath[1] == L'/') |
| return ChangeSlashToPlatform(filepath.c_str() + 1); |
| if (filepath[2] == L'/') { |
| WideString result; |
| result += filepath[1]; |
| result += L':'; |
| result += ChangeSlashToPlatform(filepath.c_str() + 2); |
| return result; |
| } |
| WideString result; |
| result += L'\\'; |
| result += ChangeSlashToPlatform(filepath.c_str()); |
| return result; |
| #else |
| return WideString(filepath); |
| #endif |
| } |
| |
| WideString CPDF_FileSpec::GetFileName() const { |
| WideString csFileName; |
| if (const CPDF_Dictionary* pDict = m_pObj->AsDictionary()) { |
| const CPDF_String* pUF = ToString(pDict->GetDirectObjectFor("UF")); |
| if (pUF) |
| csFileName = pUF->GetUnicodeText(); |
| if (csFileName.IsEmpty()) { |
| const CPDF_String* pK = |
| ToString(pDict->GetDirectObjectFor(pdfium::stream::kF)); |
| if (pK) |
| csFileName = WideString::FromDefANSI(pK->GetString().AsStringView()); |
| } |
| if (pDict->GetStringFor("FS") == "URL") |
| return csFileName; |
| |
| if (csFileName.IsEmpty()) { |
| for (const auto* key : {"DOS", "Mac", "Unix"}) { |
| const CPDF_String* pValue = ToString(pDict->GetDirectObjectFor(key)); |
| if (pValue) { |
| csFileName = |
| WideString::FromDefANSI(pValue->GetString().AsStringView()); |
| break; |
| } |
| } |
| } |
| } else if (const CPDF_String* pString = m_pObj->AsString()) { |
| csFileName = WideString::FromDefANSI(pString->GetString().AsStringView()); |
| } |
| return DecodeFileName(csFileName); |
| } |
| |
| const CPDF_Stream* CPDF_FileSpec::GetFileStream() const { |
| const CPDF_Dictionary* pDict = m_pObj->AsDictionary(); |
| if (!pDict) |
| return nullptr; |
| |
| // Get the embedded files dictionary. |
| const CPDF_Dictionary* pFiles = pDict->GetDictFor("EF"); |
| if (!pFiles) |
| return nullptr; |
| |
| // List of keys to check for the file specification string. |
| // Follows the same precedence order as GetFileName(). |
| static constexpr const char* kKeys[] = {"UF", "F", "DOS", "Mac", "Unix"}; |
| size_t end = pDict->GetStringFor("FS") == "URL" ? 2 : pdfium::size(kKeys); |
| for (size_t i = 0; i < end; ++i) { |
| ByteString key = kKeys[i]; |
| if (!pDict->GetUnicodeTextFor(key).IsEmpty()) { |
| const CPDF_Stream* pStream = pFiles->GetStreamFor(key); |
| if (pStream) |
| return pStream; |
| } |
| } |
| return nullptr; |
| } |
| |
| CPDF_Stream* CPDF_FileSpec::GetFileStream() { |
| return const_cast<CPDF_Stream*>( |
| static_cast<const CPDF_FileSpec*>(this)->GetFileStream()); |
| } |
| |
| const CPDF_Dictionary* CPDF_FileSpec::GetParamsDict() const { |
| const CPDF_Stream* pStream = GetFileStream(); |
| if (!pStream) |
| return nullptr; |
| |
| const CPDF_Dictionary* pDict = pStream->GetDict(); |
| return pDict ? pDict->GetDictFor("Params") : nullptr; |
| } |
| |
| CPDF_Dictionary* CPDF_FileSpec::GetParamsDict() { |
| return const_cast<CPDF_Dictionary*>( |
| static_cast<const CPDF_FileSpec*>(this)->GetParamsDict()); |
| } |
| |
| WideString CPDF_FileSpec::EncodeFileName(const WideString& filepath) { |
| if (filepath.GetLength() <= 1) |
| return WideString(); |
| |
| #if defined(OS_WIN) |
| if (filepath[1] == L':') { |
| WideString result(L'/'); |
| result += filepath[0]; |
| if (filepath[2] != L'\\') |
| result += L'/'; |
| |
| result += ChangeSlashToPDF(filepath.c_str() + 2); |
| return result; |
| } |
| if (filepath[0] == L'\\' && filepath[1] == L'\\') |
| return ChangeSlashToPDF(filepath.c_str() + 1); |
| |
| if (filepath[0] == L'\\') |
| return L'/' + ChangeSlashToPDF(filepath.c_str()); |
| return ChangeSlashToPDF(filepath.c_str()); |
| #elif defined(OS_APPLE) |
| if (filepath.First(sizeof("Mac") - 1).EqualsASCII("Mac")) |
| return L'/' + ChangeSlashToPDF(filepath.c_str()); |
| return ChangeSlashToPDF(filepath.c_str()); |
| #else |
| return WideString(filepath); |
| #endif |
| } |
| |
| void CPDF_FileSpec::SetFileName(const WideString& wsFileName) { |
| if (!m_pWritableObj) { |
| NOTREACHED(); |
| return; |
| } |
| |
| WideString wsStr = EncodeFileName(wsFileName); |
| if (m_pObj->IsString()) { |
| m_pWritableObj->SetString(wsStr.ToDefANSI()); |
| } else if (CPDF_Dictionary* pDict = m_pWritableObj->AsDictionary()) { |
| pDict->SetNewFor<CPDF_String>(pdfium::stream::kF, wsStr.ToDefANSI(), false); |
| pDict->SetNewFor<CPDF_String>("UF", wsStr); |
| } |
| } |