diff options
author | bunnei <bunneidev@gmail.com> | 2018-09-06 00:06:11 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2018-09-06 00:06:11 +0200 |
commit | a6ae7654105fe6ec46ff0bcabb714b8447b83899 (patch) | |
tree | 0ff4d2396cb0730ec5952181e4e67947b64832ec /src/core/crypto/aes_util.cpp | |
parent | Merge pull request #1245 from degasus/optimizations (diff) | |
parent | bktr: Fix bucket overlap error (diff) | |
download | yuzu-a6ae7654105fe6ec46ff0bcabb714b8447b83899.tar yuzu-a6ae7654105fe6ec46ff0bcabb714b8447b83899.tar.gz yuzu-a6ae7654105fe6ec46ff0bcabb714b8447b83899.tar.bz2 yuzu-a6ae7654105fe6ec46ff0bcabb714b8447b83899.tar.lz yuzu-a6ae7654105fe6ec46ff0bcabb714b8447b83899.tar.xz yuzu-a6ae7654105fe6ec46ff0bcabb714b8447b83899.tar.zst yuzu-a6ae7654105fe6ec46ff0bcabb714b8447b83899.zip |
Diffstat (limited to 'src/core/crypto/aes_util.cpp')
-rw-r--r-- | src/core/crypto/aes_util.cpp | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/src/core/crypto/aes_util.cpp b/src/core/crypto/aes_util.cpp index 72e4bed67..89ade5000 100644 --- a/src/core/crypto/aes_util.cpp +++ b/src/core/crypto/aes_util.cpp @@ -82,11 +82,25 @@ void AESCipher<Key, KeySize>::Transcode(const u8* src, size_t size, u8* dest, Op } } else { const auto block_size = mbedtls_cipher_get_block_size(context); + if (size < block_size) { + std::vector<u8> block(block_size); + std::memcpy(block.data(), src, size); + Transcode(block.data(), block.size(), block.data(), op); + std::memcpy(dest, block.data(), size); + return; + } for (size_t offset = 0; offset < size; offset += block_size) { auto length = std::min<size_t>(block_size, size - offset); mbedtls_cipher_update(context, src + offset, length, dest + offset, &written); if (written != length) { + if (length < block_size) { + std::vector<u8> block(block_size); + std::memcpy(block.data(), src + offset, length); + Transcode(block.data(), block.size(), block.data(), op); + std::memcpy(dest + offset, block.data(), length); + return; + } LOG_WARNING(Crypto, "Not all data was decrypted requested={:016X}, actual={:016X}.", length, written); } |