mirror of
https://github.com/QuasarApp/Qt-AES.git
synced 2025-04-26 21:54:32 +00:00
* Pass expKey parameter of addRoundKey() by const reference to avoid unnecessary copy. * Use C++11 nullptr instead of NULL, make it clear that default value of iv parameter in encode() and decode() is empty QByteArray instead of implicit conversion from null pointer via QByteArray(const char *, int = -1) constructor. * Change parameter names in declarations of cipher(), invCipher() and byteXor() to match definitions. * Convert AES-NI-related files to headers, place functions with internal linkage to anonymous namespace to avoid exporting them, don't use inline specifier (inline keyword have different meaning in C++ rather than in C). * Use char literals instead of implementation-defined int-to-signed-char conversions where possible. * Set default value for padding argument in static RemovePadding() to match sample in README.
62 lines
2.0 KiB
C
62 lines
2.0 KiB
C
#ifndef AESNIENCECB_H
|
||
#define AESNIENCECB_H
|
||
|
||
#include <wmmintrin.h>
|
||
|
||
namespace {
|
||
|
||
/* Note – the length of the output buffer is assumed to be a multiple of 16 bytes */
|
||
void AES_ECB_encrypt(const unsigned char *in, //pointer to the PLAINTEXT
|
||
unsigned char *out, //pointer to the CIPHERTEXT buffer
|
||
unsigned long length, //text length in bytes
|
||
const char *key, //pointer to the expanded key schedule
|
||
int number_of_rounds) //number of AES rounds 10,12 or 14
|
||
{
|
||
__m128i tmp;
|
||
unsigned long i;
|
||
int j;
|
||
if(length%16)
|
||
length = length/16+1;
|
||
else
|
||
length = length/16;
|
||
for(i = 0; i < length; i++) {
|
||
tmp = _mm_loadu_si128 (&((__m128i*)in)[i]);
|
||
tmp = _mm_xor_si128 (tmp,((__m128i*)key)[0]);
|
||
for(j=1; j <number_of_rounds; j++) {
|
||
tmp = _mm_aesenc_si128 (tmp, ((__m128i*)key)[j]);
|
||
}
|
||
tmp = _mm_aesenclast_si128 (tmp,((__m128i*)key)[j]);
|
||
_mm_storeu_si128 (&((__m128i*)out)[i],tmp);
|
||
}
|
||
}
|
||
|
||
#if 0
|
||
void AES_ECB_decrypt(const unsigned char *in, //pointer to the CIPHERTEXT
|
||
unsigned char *out, //pointer to the DECRYPTED TEXT buffer
|
||
unsigned long length, //text length in bytes
|
||
const char *key, //pointer to the expanded key schedule
|
||
int number_of_rounds) //number of AES rounds 10,12 or 14
|
||
{
|
||
__m128i tmp;
|
||
unsigned long i;
|
||
int j;
|
||
if(length%16)
|
||
length = length/16+1;
|
||
else
|
||
length = length/16;
|
||
for(i = 0; i < length; i++) {
|
||
tmp = _mm_loadu_si128 (&((__m128i*)in)[i]);
|
||
tmp = _mm_xor_si128 (tmp,((__m128i*)key)[0]);
|
||
for(j = 1; j < number_of_rounds; j++) {
|
||
tmp = _mm_aesdec_si128 (tmp,((__m128i*)key)[j]);
|
||
}
|
||
tmp = _mm_aesdeclast_si128 (tmp,((__m128i*)key)[j]);
|
||
_mm_storeu_si128 (&((__m128i*)out)[i],tmp);
|
||
}
|
||
}
|
||
#endif
|
||
|
||
}
|
||
|
||
#endif // AESNIENCECB_H
|