blob: 9da342e5bb7aa50dca27fcb268b69cb70b6646a5 [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. 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,
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;
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,
bool bOwner);
void AES256_SetPerms(CPDF_Dictionary* pEncryptDict);
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;
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_