blob: 9cab3d7275cf9cfd6a34da27c03cd3f960655ae4 [file] [log] [blame]
// 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_