Правильный способ шифрования AES NI в C ++ - PullRequest
Новая
       50

Правильный способ шифрования AES NI в C ++

1 голос
/ 07 ноября 2019

Я пытаюсь включить функцию шифрования AES 256 бит (режим CBC) с помощью специального набора инструкций (AES-NI) от Intel.

На данный момент это мой код: 

int Nb = 4;
int Nk = 8;
int Nr = 14;

unsigned int BLOCK_SIZE = 16;

void block_encryption_special(unsigned char input[], unsigned char result[], __m256i *key)
{

    __m256i block = _mm256_loadu_si256((__m256i *) input);

    block = _mm256_xor_si256(block, key[0]);

    for (int i = 1; i < Nr; i++)
    {
        block = _mm256_aesenc_epi128(block, key[i]);
    }

    block = _mm256_aesenclast_epi128(block, key[Nr]);

    _mm256_storeu_si256((__m256i *) result, block);

}

unsigned char* encrypt(unsigned char input[], unsigned int input_length, unsigned  char key[], unsigned char *iv, bool special)
{
    unsigned int i = 0;
    unsigned int total_lenght;
    unsigned int length_padded = (input_length / BLOCK_SIZE);

    if(input_length % BLOCK_SIZE)
    {
        length_padded++;
    }

    length_padded *=  BLOCK_SIZE;

    total_lenght = length_padded;

    unsigned char *align  = null_padding(input, input_length, total_lenght);
    unsigned char *result = new unsigned char[total_lenght];
    unsigned char *block = new unsigned char[BLOCK_SIZE];

    memcpy(block, iv, BLOCK_SIZE);

    for (i = 0; i < total_lenght; i += BLOCK_SIZE)
    {
        block_encryption_special(block, result + i, (__m256i *)key);
        memcpy(block, result + i, BLOCK_SIZE);
    }

    delete[] block;
    delete[] align;

    return result;
}

Я думаю, что я делаю что-то не так, потому что я получаю пустой вывод из этого. Я что-то пропустил, пожалуйста?

1 Ответ

1 голос
/ 07 ноября 2019

Пожалуйста, прочтите документацию к memcpy. Первый аргумент должен быть буфером назначения. В своем коде вы копируете данные в блок и затем удаляете их. Также использование memcpy в цикле for неправильно. Вы должны увеличить указатель буфера назначения.

...