在Qt下实现的AES-128 CBC模式的加密解密

#include "qaesencryption.h" #include <QDebug> #include <QVector> #ifdef USE_INTEL_AES_IF_AVAILABLE #include "aesni/aesni-key-exp.h" #include "aesni/aesni-enc-ecb.h" #include "aesni/aesni-enc-cbc.h" #endif /* * Static Functions * */ QByteArray QAESEncryption::Crypt(QAESEncryption::Aes level, QAESEncryption::Mode mode, const QByteArray &rawText, const QByteArray &key, const QByteArray &iv, QAESEncryption::Padding padding) { return QAESEncryption(level, mode, padding).encode(rawText, key, iv); } QByteArray QAESEncryption::Decrypt(QAESEncryption::Aes level, QAESEncryption::Mode mode, const QByteArray &rawText, const QByteArray &key, const QByteArray &iv, QAESEncryption::Padding padding) { return QAESEncryption(level, mode, padding).decode(rawText, key, iv); } QByteArray QAESEncryption::ExpandKey(QAESEncryption::Aes level, QAESEncryption::Mode mode, const QByteArray &key) { return QAESEncryption(level, mode).expandKey(key); } QByteArray QAESEncryption::RemovePadding(const QByteArray &rawText, QAESEncryption::Padding padding) { if (rawText.isEmpty()) return rawText; QByteArray ret(rawText); switch (padding) { case Padding::ZERO: //Works only if the last byte of the decoded array is not zero while (ret.at(ret.length()-1) == 0x00) ret.remove(ret.length()-1, 1); break; case Padding::PKCS7: #if QT_VERSION >= QT_VERSION_CHECK(5, 10, 0) ret.remove(ret.length() - ret.back(), ret.back()); #else ret.remove(ret.length() - ret.at(ret.length() - 1), ret.at(ret.length() - 1)); #endif break; case Padding::ISO: { // Find the last byte which is not zero int marker_index = ret.length() - 1; for (; marker_index >= 0; --marker_index) { if (ret.at(marker_index) != 0x00) { break; } } // And check if it's the byte for marking padding if (ret.at(marker_index) == '\x80') { ret.truncate(marker_index); } break; } default: //do nothing break; } return ret; } /* * End Static function declarations * */ /* * Local Functions * */ namespace { quint8 xTime(quint8 x) { return ((x<<1) ^ (((x>>7) & 1) * 0x1b)); } quint8 multiply(quint8 x, quint8 y) { return (((y & 1) * x) ^ ((y>>1 & 1) * xTime(x)) ^ ((y>>2 & 1) * xTime(xTime(x))) ^ ((y>>3 & 1) * xTime(xTime(xTime(x)))) ^ ((y>>4 & 1) * xTime(xTime(xTime(xTime(x)))))); } } /* * End Local functions * */ QAESEncryption::QAESEncryption(Aes level, Mode mode, Padding padding) : m_nb(4), m_blocklen(16), m_level(level), m_mode(mode), m_padding(padding) , m_aesNIAvailable(false), m_state(nullptr) { #ifdef USE_INTEL_AES_IF_AVAILABLE m_aesNIAvailable = check_aesni_support(); #endif switch (level) { case AES_128: { AES128 aes; m_nk = aes.nk; m_keyLen = aes.keylen; m_nr = aes.nr; m_expandedKey = aes.expandedKey; } break; case AES_192: { AES192 aes; m_nk = aes.nk; m_keyLen = aes.keylen; m_nr = aes.nr; m_expandedKey = aes.expandedKey; } break; case AES_256: { AES256 aes; m_nk = aes.nk; m_keyLen = aes.keylen; m_nr = aes.nr; m_expandedKey = aes.expandedKey; } break; default: { AES128 aes; m_nk = aes.nk; m_keyLen = aes.keylen; m_nr = aes.nr; m_expandedKey = aes.expandedKey; } break; } } QByteArray QAESEncryption::getPadding(int currSize, int alignment) { int size = (alignment - currSize % alignment) % alignment; switch(m_padding) { case Padding::ZERO: return QByteArray(size, 0x00); break; case Padding::PKCS7: if (size == 0) size = alignment; return QByteArray(size, size); break; case Padding::ISO: if (size > 0) return QByteArray (size - 1, 0x00).prepend('\x80'); break; default: return QByteArray(size, 0x00); break; } return QByteArray(); } QByteArray QAESEncryption::expandKey(const QByteArray &key) { #ifdef USE_INTEL_AES_IF_AVAILABLE if (true){ switch(m_level) { case AES_128: { AES128 aes128; quint8 ret[aes128.expandedKey]; memset(ret, 0x00, sizeof(ret)); quint8 uchar_key[key.size()]; memcpy(uchar_key, key.data(), key.size()); AES_128_Key_Expansion(uchar_key, ret); return QByteArray((char*) ret, aes128.expandedKey); } break; case AES_192: { AES192 aes192; quint8 ret[aes192.expandedKey]; memset(ret, 0x00, sizeof(ret)); quint8 uchar_key[key.size()]; memcpy(uchar_key, key.data(), key.size()); AES_192_Key_Expansion(uchar_key, ret); return QByteArray((char*) ret, aes192.expandedKey); } break; case AES_256: { AES256 aes256; quint8 ret[aes256.expandedKey]; memset(ret, 0x00, sizeof(ret)); quint8 uchar_key[key.size()]; memcpy(uchar_key, key.data(), key.size()); AES_256_Key_Expansion(uchar_key, ret); return QByteArray((char*) ret, aes256.expandedKey); } break; default: return QByteArray(); break; } } else #endif { int i, k; quint8 tempa[4]; // Used for the column/row operations QByteArray roundKey(key); // The first round key is the key itself. // All other round keys are found from the previous round keys. //i == Nk for(i = m_nk; i < m_nb * (m_nr + 1); i++) { tempa[0] = (quint8) roundKey.at((i-1) * 4 + 0); tempa[1] = (quint8) roundKey.at((i-1) * 4 + 1); tempa[2] = (quint8) roundKey.at((i-1) * 4 + 2); tempa[3] = (quint8) roundKey.at((i-1) * 4 + 3); if (i % m_nk == 0) { // This function shifts the 4 bytes in a word to the left once. // [a0,a1,a2,a3] becomes [a1,a2,a3,a0] // Function RotWord() k = tempa[0]; tempa[0] = tempa[1]; tempa[1] = tempa[2]; tempa[2] = tempa[3]; tempa[3] = k; // Function Subword() tempa[0] = getSBoxValue(tempa[0]); tempa[1] = getSBoxValue(tempa[1]); tempa[2] = getSBoxValue(tempa[2]); tempa[3] = getSBoxValue(tempa[3]); tempa[0] = tempa[0] ^ Rcon[i/m_nk]; } if (m_level == AES_256 && i % m_nk == 4) { // Function Subword() tempa[0] = getSBoxValue(tempa[0]); tempa[1] = getSBoxValue(tempa[1]); tempa[2] = getSBoxValue(tempa[2]); tempa[3] = getSBoxValue(tempa[3]); } roundKey.insert(i * 4 + 0, (quint8) roundKey.at((i - m_nk) * 4 + 0) ^ tempa[0]); roundKey.insert(i * 4 + 1, (quint8) roundKey.at((i - m_nk) * 4 + 1) ^ tempa[1]); roundKey.insert(i * 4 + 2, (quint8) roundKey.at((i - m_nk) * 4 + 2) ^ tempa[2]); roundKey.insert(i * 4 + 3, (quint8) roundKey.at((i - m_nk) * 4 + 3) ^ tempa[3]); } return roundKey; } } // This function adds the round key to state. // The round key is added to the state by an XOR function. void QAESEncryption::addRoundKey(const quint8 round, const QByteArray &expKey) { QByteArray::iterator it = m_state->begin(); for(int i=0; i < 16; ++i) it[i] = (quint8) it[i] ^ (quint8) expKey.at(round * m_nb * 4 + (i/4) * m_nb + (i%4)); } // The SubBytes Function Substitutes the values in the // state matrix with values in an S-box. void QAESEncryption::subBytes() { QByteArray::iterator it = m_state->begin();