2018-07-27 20:55:23 -07:00
|
|
|
// Copyright 2018 yuzu emulator team
|
|
|
|
// Licensed under GPLv2 or any later version
|
|
|
|
// Refer to the license.txt file included.
|
|
|
|
|
|
|
|
#pragma once
|
|
|
|
|
2020-08-03 11:14:39 -07:00
|
|
|
#include <array>
|
|
|
|
|
2018-07-28 18:39:42 -07:00
|
|
|
#include "core/crypto/aes_util.h"
|
|
|
|
#include "core/crypto/encryption_layer.h"
|
|
|
|
#include "core/crypto/key_manager.h"
|
2018-07-27 20:55:23 -07:00
|
|
|
|
2018-07-28 18:39:42 -07:00
|
|
|
namespace Core::Crypto {
|
2018-07-27 20:55:23 -07:00
|
|
|
|
|
|
|
// Sits on top of a VirtualFile and provides CTR-mode AES decription.
|
2018-07-28 18:39:42 -07:00
|
|
|
class CTREncryptionLayer : public EncryptionLayer {
|
|
|
|
public:
|
2020-08-03 11:14:39 -07:00
|
|
|
using IVData = std::array<u8, 16>;
|
|
|
|
|
2021-05-15 22:46:30 -07:00
|
|
|
CTREncryptionLayer(FileSys::VirtualFile base_, Key128 key_, std::size_t base_offset_);
|
2018-07-27 20:55:23 -07:00
|
|
|
|
2018-09-15 06:21:06 -07:00
|
|
|
std::size_t Read(u8* data, std::size_t length, std::size_t offset) const override;
|
2018-07-27 20:55:23 -07:00
|
|
|
|
2020-08-03 11:14:39 -07:00
|
|
|
void SetIV(const IVData& iv);
|
2018-07-27 20:55:23 -07:00
|
|
|
|
|
|
|
private:
|
2018-09-15 06:21:06 -07:00
|
|
|
std::size_t base_offset;
|
2018-07-27 20:55:23 -07:00
|
|
|
|
|
|
|
// Must be mutable as operations modify cipher contexts.
|
|
|
|
mutable AESCipher<Key128> cipher;
|
2020-08-03 11:14:39 -07:00
|
|
|
mutable IVData iv{};
|
2018-07-27 20:55:23 -07:00
|
|
|
|
2018-09-15 06:21:06 -07:00
|
|
|
void UpdateIV(std::size_t offset) const;
|
2018-07-27 20:55:23 -07:00
|
|
|
};
|
|
|
|
|
2018-07-28 18:39:42 -07:00
|
|
|
} // namespace Core::Crypto
|