| // Copyright 2016 The PDFium Authors |
| // 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 |
| |
| #ifndef CORE_FPDFAPI_PARSER_CPDF_SECURITY_HANDLER_H_ |
| #define CORE_FPDFAPI_PARSER_CPDF_SECURITY_HANDLER_H_ |
| |
| #include <stddef.h> |
| #include <stdint.h> |
| |
| #include <memory> |
| |
| #include "core/fpdfapi/parser/cpdf_crypto_handler.h" |
| #include "core/fxcrt/bytestring.h" |
| #include "core/fxcrt/retain_ptr.h" |
| |
| class CPDF_Array; |
| class CPDF_Dictionary; |
| |
| class CPDF_SecurityHandler final : public Retainable { |
| public: |
| CONSTRUCT_VIA_MAKE_RETAIN; |
| |
| bool OnInit(const CPDF_Dictionary* pEncryptDict, |
| RetainPtr<const CPDF_Array> pIdArray, |
| const ByteString& password); |
| void OnCreate(CPDF_Dictionary* pEncryptDict, |
| const CPDF_Array* pIdArray, |
| const ByteString& password); |
| |
| // When `get_owner_perms` is true, returns full permissions if unlocked by |
| // owner. |
| uint32_t GetPermissions(bool get_owner_perms) const; |
| bool IsMetadataEncrypted() const; |
| |
| CPDF_CryptoHandler* GetCryptoHandler() const { |
| return m_pCryptoHandler.get(); |
| } |
| |
| // Take |password| and encode it, if necessary, based on the password encoding |
| // conversion. |
| ByteString GetEncodedPassword(ByteStringView password) const; |
| |
| private: |
| enum PasswordEncodingConversion { |
| kUnknown, |
| kNone, |
| kLatin1ToUtf8, |
| kUtf8toLatin1, |
| }; |
| |
| CPDF_SecurityHandler(); |
| ~CPDF_SecurityHandler() override; |
| |
| bool LoadDict(const CPDF_Dictionary* pEncryptDict); |
| bool LoadDict(const CPDF_Dictionary* pEncryptDict, |
| CPDF_CryptoHandler::Cipher* cipher, |
| size_t* key_len); |
| |
| ByteString GetUserPassword(const ByteString& owner_password) const; |
| bool CheckPassword(const ByteString& user_password, bool bOwner); |
| bool CheckPasswordImpl(const ByteString& password, bool bOwner); |
| bool CheckUserPassword(const ByteString& password, bool bIgnoreEncryptMeta); |
| bool CheckOwnerPassword(const ByteString& password); |
| bool AES256_CheckPassword(const ByteString& password, bool bOwner); |
| void AES256_SetPassword(CPDF_Dictionary* pEncryptDict, |
| const ByteString& password); |
| void AES256_SetPerms(CPDF_Dictionary* pEncryptDict); |
| bool CheckSecurity(const ByteString& password); |
| |
| void InitCryptoHandler(); |
| |
| bool m_bOwnerUnlocked = false; |
| int m_Version = 0; |
| int m_Revision = 0; |
| uint32_t m_Permissions = 0; |
| size_t m_KeyLen = 0; |
| CPDF_CryptoHandler::Cipher m_Cipher = CPDF_CryptoHandler::Cipher::kNone; |
| PasswordEncodingConversion m_PasswordEncodingConversion = kUnknown; |
| ByteString m_FileId; |
| RetainPtr<const CPDF_Dictionary> m_pEncryptDict; |
| std::unique_ptr<CPDF_CryptoHandler> m_pCryptoHandler; |
| uint8_t m_EncryptKey[32] = {}; |
| }; |
| |
| #endif // CORE_FPDFAPI_PARSER_CPDF_SECURITY_HANDLER_H_ |