79 lines
3.1 KiB
C++
79 lines
3.1 KiB
C++
#pragma once
|
|
#ifndef MESSMER_CPPUTILS_CRYPTO_SYMMETRIC_CFBCIPHER_H_
|
|
#define MESSMER_CPPUTILS_CRYPTO_SYMMETRIC_CFBCIPHER_H_
|
|
|
|
#include "../../data/FixedSizeData.h"
|
|
#include "../../data/Data.h"
|
|
#include "../../random/Random.h"
|
|
#include <boost/optional.hpp>
|
|
#include <vendor_cryptopp/modes.h>
|
|
#include "Cipher.h"
|
|
#include "EncryptionKey.h"
|
|
|
|
namespace cpputils {
|
|
|
|
template<typename BlockCipher, unsigned int KeySize>
|
|
class CFB_Cipher {
|
|
public:
|
|
using EncryptionKey = cpputils::EncryptionKey;
|
|
|
|
static constexpr unsigned int KEYSIZE = KeySize;
|
|
static constexpr unsigned int STRING_KEYSIZE = 2 * KEYSIZE;
|
|
|
|
static constexpr unsigned int ciphertextSize(unsigned int plaintextBlockSize) {
|
|
return plaintextBlockSize + IV_SIZE;
|
|
}
|
|
|
|
static constexpr unsigned int plaintextSize(unsigned int ciphertextBlockSize) {
|
|
return ciphertextBlockSize - IV_SIZE;
|
|
}
|
|
|
|
static Data encrypt(const CryptoPP::byte *plaintext, unsigned int plaintextSize, const EncryptionKey &encKey);
|
|
static boost::optional<Data> decrypt(const CryptoPP::byte *ciphertext, unsigned int ciphertextSize, const EncryptionKey &encKey);
|
|
|
|
private:
|
|
static constexpr unsigned int IV_SIZE = BlockCipher::BLOCKSIZE;
|
|
};
|
|
|
|
template<class BlockCipher, unsigned int KeySize>
|
|
constexpr unsigned int CFB_Cipher<BlockCipher, KeySize>::KEYSIZE;
|
|
template<class BlockCipher, unsigned int KeySize>
|
|
constexpr unsigned int CFB_Cipher<BlockCipher, KeySize>::STRING_KEYSIZE;
|
|
|
|
template<typename BlockCipher, unsigned int KeySize>
|
|
Data CFB_Cipher<BlockCipher, KeySize>::encrypt(const CryptoPP::byte *plaintext, unsigned int plaintextSize, const EncryptionKey &encKey) {
|
|
ASSERT(encKey.binaryLength() == KeySize, "Wrong key size");
|
|
|
|
FixedSizeData<IV_SIZE> iv = Random::PseudoRandom().getFixedSize<IV_SIZE>();
|
|
auto encryption = typename CryptoPP::CFB_Mode<BlockCipher>::Encryption(static_cast<const CryptoPP::byte*>(encKey.data()), encKey.binaryLength(), iv.data());
|
|
Data ciphertext(ciphertextSize(plaintextSize));
|
|
iv.ToBinary(ciphertext.data());
|
|
if (plaintextSize > 0) {
|
|
encryption.ProcessData(static_cast<CryptoPP::byte*>(ciphertext.data()) + IV_SIZE, plaintext, plaintextSize);
|
|
}
|
|
return ciphertext;
|
|
}
|
|
|
|
template<typename BlockCipher, unsigned int KeySize>
|
|
boost::optional<Data> CFB_Cipher<BlockCipher, KeySize>::decrypt(const CryptoPP::byte *ciphertext, unsigned int ciphertextSize, const EncryptionKey &encKey) {
|
|
ASSERT(encKey.binaryLength() == KeySize, "Wrong key size");
|
|
|
|
if (ciphertextSize < IV_SIZE) {
|
|
return boost::none;
|
|
}
|
|
|
|
const CryptoPP::byte *ciphertextIV = ciphertext;
|
|
const CryptoPP::byte *ciphertextData = ciphertext + IV_SIZE;
|
|
auto decryption = typename CryptoPP::CFB_Mode<BlockCipher>::Decryption(static_cast<const CryptoPP::byte*>(encKey.data()), encKey.binaryLength(), ciphertextIV);
|
|
Data plaintext(plaintextSize(ciphertextSize));
|
|
if (plaintext.size() > 0) {
|
|
// TODO Shouldn't we pass in ciphertextSize instead of plaintext.size() here as last argument (and also in the if above)?
|
|
decryption.ProcessData(static_cast<CryptoPP::byte*>(plaintext.data()), ciphertextData, plaintext.size());
|
|
}
|
|
return plaintext;
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|