blob: ee6928330e5a91566e60e934189e00c14dd95a4f [file] [log] [blame]
// 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.
#include <memory>
#include "core/fxcrt/fx_string.h"
#include "core/fxcrt/fx_system.h"
#include "core/fxcrt/retain_ptr.h"
#define FXCIPHER_RC4 1
#define FXCIPHER_AES 2
#define FXCIPHER_AES2 3
class CPDF_Array;
class CPDF_CryptoHandler;
class CPDF_Dictionary;
class CPDF_Parser;
class CPDF_SecurityHandler : public Retainable {
template <typename T, typename... Args>
friend RetainPtr<T> pdfium::MakeRetain(Args&&... args);
bool OnInit(const CPDF_Dictionary* pEncryptDict,
const CPDF_Array* pIdArray,
const ByteString& password);
void OnCreate(CPDF_Dictionary* pEncryptDict,
const CPDF_Array* pIdArray,
const ByteString& user_password,
const ByteString& owner_password);
void OnCreate(CPDF_Dictionary* pEncryptDict,
const CPDF_Array* pIdArray,
const ByteString& user_password);
uint32_t GetPermissions() 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;
enum PasswordEncodingConversion {
~CPDF_SecurityHandler() override;
bool LoadDict(const CPDF_Dictionary* pEncryptDict);
bool LoadDict(const CPDF_Dictionary* pEncryptDict,
int* 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,
bool bOwner,
const uint8_t* key);
void AES256_SetPerms(CPDF_Dictionary* pEncryptDict,
uint32_t permission,
bool bEncryptMetadata,
const uint8_t* key);
void OnCreateInternal(CPDF_Dictionary* pEncryptDict,
const CPDF_Array* pIdArray,
const ByteString& user_password,
const ByteString& owner_password,
bool bDefault);
bool CheckSecurity(const ByteString& password);
void InitCryptoHandler();
bool m_bOwnerUnlocked = false;
int m_Version = 0;
int m_Revision = 0;
uint32_t m_Permissions = 0;
int m_Cipher = FXCIPHER_NONE;
size_t m_KeyLen = 0;
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];