summaryrefslogtreecommitdiffstats
path: root/src/core/crypto/ctr_encryption_layer.h
blob: a7bf810f4ce74beaac0433c1b951098361a23542 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
// Copyright 2018 yuzu emulator team
// Licensed under GPLv2 or any later version
// Refer to the license.txt file included.

#pragma once

#include <vector>
#include "core/crypto/aes_util.h"
#include "core/crypto/encryption_layer.h"
#include "core/crypto/key_manager.h"

namespace Core::Crypto {

// Sits on top of a VirtualFile and provides CTR-mode AES decription.
class CTREncryptionLayer : public EncryptionLayer {
public:
    CTREncryptionLayer(FileSys::VirtualFile base, Key128 key, std::size_t base_offset);

    std::size_t Read(u8* data, std::size_t length, std::size_t offset) const override;

    void SetIV(const std::vector<u8>& iv);

private:
    std::size_t base_offset;

    // Must be mutable as operations modify cipher contexts.
    mutable AESCipher<Key128> cipher;
    mutable std::vector<u8> iv;

    void UpdateIV(std::size_t offset) const;
};

} // namespace Core::Crypto