81e3560708
Batch #851 by ragnarok-clanker: 0 byte-identical, 6 confirmed by peer review. - fn_04417 AES::~AES: logic confirmed by peer review - fn_04418 AES::Decrypt: logic confirmed by peer review - fn_04419 AES::Initialize: logic confirmed by peer review - fn_04420 AES::Initialize: logic confirmed by peer review - fn_04421 AES::SetIV: logic confirmed by peer review - fn_04422 AES::SetMode: logic confirmed by peer review Reviewed-by: ragnarok-clanker Adds to the build: Source/Utility/AES.cpp
37 lines
967 B
C++
37 lines
967 B
C++
#ifndef RAGNAROK_SOURCE_UTILITY_AES_H
|
|
#define RAGNAROK_SOURCE_UTILITY_AES_H
|
|
|
|
#include <windows.h>
|
|
#include <bcrypt.h>
|
|
#include <vector>
|
|
|
|
class AES {
|
|
public:
|
|
enum Mode : int { ECB = 0, CBC = 1, CFB = 2 };
|
|
|
|
private:
|
|
/* 0000 */ BCRYPT_ALG_HANDLE m_algorithm;
|
|
/* 0004 */ BCRYPT_KEY_HANDLE m_key;
|
|
/* 0008 */ std::vector<UCHAR> m_keyObject;
|
|
/* 0014 */ std::vector<UCHAR> m_iv;
|
|
/* 0020 */ int m_mode;
|
|
|
|
public:
|
|
// INLINED
|
|
AES() : m_algorithm(nullptr), m_key(nullptr), m_mode(CBC) {
|
|
}
|
|
|
|
~AES();
|
|
ULONG Decrypt(PUCHAR input, ULONG inputSize, PUCHAR output, ULONG outputSize, UCHAR flags);
|
|
bool Initialize(PUCHAR secret, ULONG secretSize, const UCHAR* iv, ULONG ivSize, int mode);
|
|
bool Initialize(PUCHAR secret, ULONG secretSize, int mode);
|
|
bool SetIV(const UCHAR* iv, ULONG ivSize);
|
|
bool SetMode(int mode);
|
|
};
|
|
|
|
#if defined(_WIN32) && !defined(_WIN64)
|
|
static_assert(sizeof(std::vector<UCHAR>) == 12);
|
|
static_assert(sizeof(AES) == 36);
|
|
#endif
|
|
|
|
#endif
|