75 lines
1.5 KiB
C++
75 lines
1.5 KiB
C++
#ifndef ENCRYPTION_H
|
|
#define ENCRYPTION_H
|
|
|
|
#include <QByteArray>
|
|
#include <openssl/blowfish.h>
|
|
#include <openssl/sha.h>
|
|
#include "twofish.h"
|
|
|
|
class BlowfishCipher
|
|
{
|
|
protected:
|
|
QByteArray mKey;
|
|
QByteArray mIV;
|
|
BF_KEY mContext;
|
|
|
|
public:
|
|
BlowfishCipher();
|
|
~BlowfishCipher();
|
|
void setKey(const QByteArray& ba);
|
|
const QByteArray& key() const;
|
|
void setIV(const QByteArray& ba);
|
|
QByteArray& IV() const;
|
|
|
|
void encrypt(const QByteArray& plain, QByteArray& encrypted);
|
|
void decrypt(const QByteArray& encrypted, QByteArray& plain);
|
|
};
|
|
|
|
class TwofishCipher
|
|
{
|
|
protected:
|
|
QByteArray mKey;
|
|
QByteArray mIV;
|
|
TwofishKey mKeyStruct;
|
|
Twofish* mContext;
|
|
|
|
public:
|
|
TwofishCipher();
|
|
~TwofishCipher();
|
|
|
|
int blocksize() const { return 16; }
|
|
void setKey(const QByteArray& ba);
|
|
const QByteArray& key() const;
|
|
|
|
// IV vector must be 16 bytes
|
|
void setIV(const QByteArray& ba);
|
|
const QByteArray& IV() const;
|
|
|
|
// Plain and encrypted must be padded to 16 bytes boundary before call
|
|
void encrypt(const QByteArray& plain, int plainOffset, QByteArray& encrypted, int encryptedOffset);
|
|
void decrypt(const QByteArray& encrypted, int encryptedOffset, QByteArray& plain, int plainOffset);
|
|
};
|
|
|
|
|
|
class SHA256
|
|
{
|
|
protected:
|
|
SHA256_CTX mContext;
|
|
QByteArray mDigest;
|
|
public:
|
|
SHA256();
|
|
~SHA256();
|
|
|
|
QByteArray& digest();
|
|
void update(const void* data, int length);
|
|
void final();
|
|
};
|
|
|
|
class IV
|
|
{
|
|
public:
|
|
static void Generate(QByteArray& buffer);
|
|
};
|
|
|
|
#endif // ENCRYPTION_H
|