Frank Denis
7 years ago
17 changed files with 481 additions and 2 deletions
@ -0,0 +1,244 @@ |
|||||
|
/*
|
||||
|
* Argon2 source code package |
||||
|
* |
||||
|
* Written by Daniel Dinu and Dmitry Khovratovich, 2015 |
||||
|
* |
||||
|
* This work is licensed under a Creative Commons CC0 1.0 License/Waiver. |
||||
|
* |
||||
|
* You should have received a copy of the CC0 Public Domain Dedication along |
||||
|
* with |
||||
|
* this software. If not, see |
||||
|
* <http://creativecommons.org/publicdomain/zero/1.0/>.
|
||||
|
*/ |
||||
|
|
||||
|
#include <stdint.h> |
||||
|
#include <stdlib.h> |
||||
|
#include <string.h> |
||||
|
|
||||
|
#include "argon2-core.h" |
||||
|
#include "argon2.h" |
||||
|
#include "private/common.h" |
||||
|
#include "private/sse2_64_32.h" |
||||
|
|
||||
|
#if defined(HAVE_AVX512FINTRIN_H) && defined(HAVE_AVX2INTRIN_H) && \ |
||||
|
defined(HAVE_EMMINTRIN_H) && defined(HAVE_TMMINTRIN_H) && defined(HAVE_SMMINTRIN_H) |
||||
|
|
||||
|
# ifdef __GNUC__ |
||||
|
# pragma GCC target("sse2") |
||||
|
# pragma GCC target("ssse3") |
||||
|
# pragma GCC target("sse4.1") |
||||
|
# pragma GCC target("avx2") |
||||
|
# pragma GCC target("avx512f") |
||||
|
# endif |
||||
|
|
||||
|
# ifdef _MSC_VER |
||||
|
# include <intrin.h> /* for _mm_set_epi64x */ |
||||
|
# endif |
||||
|
#include <emmintrin.h> |
||||
|
#include <immintrin.h> |
||||
|
#include <smmintrin.h> |
||||
|
#include <tmmintrin.h> |
||||
|
|
||||
|
# include "blamka-round-avx512f.h" |
||||
|
|
||||
|
static void |
||||
|
fill_block(__m512i *state, const uint8_t *ref_block, uint8_t *next_block) |
||||
|
{ |
||||
|
__m512i block_XY[ARGON2_512BIT_WORDS_IN_BLOCK]; |
||||
|
uint32_t i; |
||||
|
|
||||
|
for (i = 0; i < ARGON2_512BIT_WORDS_IN_BLOCK; i++) { |
||||
|
block_XY[i] = state[i] = _mm512_xor_si512( |
||||
|
state[i], _mm512_loadu_si512((__m512i const *) (&ref_block[64 * i]))); |
||||
|
} |
||||
|
|
||||
|
for (i = 0; i < 2; ++i) { |
||||
|
BLAKE2_ROUND_1( |
||||
|
state[8 * i + 0], state[8 * i + 1], state[8 * i + 2], state[8 * i + 3], |
||||
|
state[8 * i + 4], state[8 * i + 5], state[8 * i + 6], state[8 * i + 7]); |
||||
|
} |
||||
|
|
||||
|
for (i = 0; i < 2; ++i) { |
||||
|
BLAKE2_ROUND_2( |
||||
|
state[2 * 0 + i], state[2 * 1 + i], state[2 * 2 + i], state[2 * 3 + i], |
||||
|
state[2 * 4 + i], state[2 * 5 + i], state[2 * 6 + i], state[2 * 7 + i]); |
||||
|
} |
||||
|
|
||||
|
for (i = 0; i < ARGON2_512BIT_WORDS_IN_BLOCK; i++) { |
||||
|
state[i] = _mm512_xor_si512(state[i], block_XY[i]); |
||||
|
_mm512_storeu_si512((__m512i *) (&next_block[64 * i]), state[i]); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
static void |
||||
|
fill_block_with_xor(__m512i *state, const uint8_t *ref_block, |
||||
|
uint8_t *next_block) |
||||
|
{ |
||||
|
__m512i block_XY[ARGON2_512BIT_WORDS_IN_BLOCK]; |
||||
|
uint32_t i; |
||||
|
|
||||
|
for (i = 0; i < ARGON2_512BIT_WORDS_IN_BLOCK; i++) { |
||||
|
state[i] = _mm512_xor_si512( |
||||
|
state[i], _mm512_loadu_si512((__m512i const *) (&ref_block[64 * i]))); |
||||
|
block_XY[i] = _mm512_xor_si512( |
||||
|
state[i], _mm512_loadu_si512((__m512i const *) (&next_block[64 * i]))); |
||||
|
} |
||||
|
|
||||
|
for (i = 0; i < 2; ++i) { |
||||
|
BLAKE2_ROUND_1( |
||||
|
state[8 * i + 0], state[8 * i + 1], state[8 * i + 2], state[8 * i + 3], |
||||
|
state[8 * i + 4], state[8 * i + 5], state[8 * i + 6], state[8 * i + 7]); |
||||
|
} |
||||
|
|
||||
|
for (i = 0; i < 2; ++i) { |
||||
|
BLAKE2_ROUND_2( |
||||
|
state[2 * 0 + i], state[2 * 1 + i], state[2 * 2 + i], state[2 * 3 + i], |
||||
|
state[2 * 4 + i], state[2 * 5 + i], state[2 * 6 + i], state[2 * 7 + i]); |
||||
|
} |
||||
|
|
||||
|
for (i = 0; i < ARGON2_512BIT_WORDS_IN_BLOCK; i++) { |
||||
|
state[i] = _mm512_xor_si512(state[i], block_XY[i]); |
||||
|
_mm512_storeu_si512((__m512i *) (&next_block[64 * i]), state[i]); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
static void |
||||
|
generate_addresses(const argon2_instance_t *instance, |
||||
|
const argon2_position_t *position, uint64_t *pseudo_rands) |
||||
|
{ |
||||
|
block address_block, input_block, tmp_block; |
||||
|
uint32_t i; |
||||
|
|
||||
|
init_block_value(&address_block, 0); |
||||
|
init_block_value(&input_block, 0); |
||||
|
|
||||
|
if (instance != NULL && position != NULL) { |
||||
|
input_block.v[0] = position->pass; |
||||
|
input_block.v[1] = position->lane; |
||||
|
input_block.v[2] = position->slice; |
||||
|
input_block.v[3] = instance->memory_blocks; |
||||
|
input_block.v[4] = instance->passes; |
||||
|
input_block.v[5] = instance->type; |
||||
|
|
||||
|
for (i = 0; i < instance->segment_length; ++i) { |
||||
|
if (i % ARGON2_ADDRESSES_IN_BLOCK == 0) { |
||||
|
/* Temporary zero-initialized blocks */ |
||||
|
__m512i zero_block[ARGON2_512BIT_WORDS_IN_BLOCK]; |
||||
|
__m512i zero2_block[ARGON2_512BIT_WORDS_IN_BLOCK]; |
||||
|
|
||||
|
memset(zero_block, 0, sizeof(zero_block)); |
||||
|
memset(zero2_block, 0, sizeof(zero2_block)); |
||||
|
init_block_value(&address_block, 0); |
||||
|
init_block_value(&tmp_block, 0); |
||||
|
/* Increasing index counter */ |
||||
|
input_block.v[6]++; |
||||
|
/* First iteration of G */ |
||||
|
fill_block_with_xor(zero_block, (uint8_t *) &input_block.v, |
||||
|
(uint8_t *) &tmp_block.v); |
||||
|
/* Second iteration of G */ |
||||
|
fill_block_with_xor(zero2_block, (uint8_t *) &tmp_block.v, |
||||
|
(uint8_t *) &address_block.v); |
||||
|
} |
||||
|
|
||||
|
pseudo_rands[i] = address_block.v[i % ARGON2_ADDRESSES_IN_BLOCK]; |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
void |
||||
|
fill_segment_avx512f(const argon2_instance_t *instance, |
||||
|
argon2_position_t position) |
||||
|
{ |
||||
|
block *ref_block = NULL, *curr_block = NULL; |
||||
|
uint64_t pseudo_rand, ref_index, ref_lane; |
||||
|
uint32_t prev_offset, curr_offset; |
||||
|
uint32_t starting_index, i; |
||||
|
__m512i state[ARGON2_512BIT_WORDS_IN_BLOCK]; |
||||
|
int data_independent_addressing = 1; |
||||
|
|
||||
|
/* Pseudo-random values that determine the reference block position */ |
||||
|
uint64_t *pseudo_rands = NULL; |
||||
|
|
||||
|
if (instance == NULL) { |
||||
|
return; |
||||
|
} |
||||
|
|
||||
|
if (instance->type == Argon2_id && |
||||
|
(position.pass != 0 || position.slice >= ARGON2_SYNC_POINTS / 2)) { |
||||
|
data_independent_addressing = 0; |
||||
|
} |
||||
|
|
||||
|
pseudo_rands = instance->pseudo_rands; |
||||
|
|
||||
|
if (data_independent_addressing) { |
||||
|
generate_addresses(instance, &position, pseudo_rands); |
||||
|
} |
||||
|
|
||||
|
starting_index = 0; |
||||
|
|
||||
|
if ((0 == position.pass) && (0 == position.slice)) { |
||||
|
starting_index = 2; /* we have already generated the first two blocks */ |
||||
|
} |
||||
|
|
||||
|
/* Offset of the current block */ |
||||
|
curr_offset = position.lane * instance->lane_length + |
||||
|
position.slice * instance->segment_length + starting_index; |
||||
|
|
||||
|
if (0 == curr_offset % instance->lane_length) { |
||||
|
/* Last block in this lane */ |
||||
|
prev_offset = curr_offset + instance->lane_length - 1; |
||||
|
} else { |
||||
|
/* Previous block */ |
||||
|
prev_offset = curr_offset - 1; |
||||
|
} |
||||
|
|
||||
|
memcpy(state, ((instance->region->memory + prev_offset)->v), |
||||
|
ARGON2_BLOCK_SIZE); |
||||
|
|
||||
|
for (i = starting_index; i < instance->segment_length; |
||||
|
++i, ++curr_offset, ++prev_offset) { |
||||
|
/*1.1 Rotating prev_offset if needed */ |
||||
|
if (curr_offset % instance->lane_length == 1) { |
||||
|
prev_offset = curr_offset - 1; |
||||
|
} |
||||
|
|
||||
|
/* 1.2 Computing the index of the reference block */ |
||||
|
/* 1.2.1 Taking pseudo-random value from the previous block */ |
||||
|
if (data_independent_addressing) { |
||||
|
#pragma warning(push) |
||||
|
#pragma warning(disable : 6385) |
||||
|
pseudo_rand = pseudo_rands[i]; |
||||
|
#pragma warning(pop) |
||||
|
} else { |
||||
|
pseudo_rand = instance->region->memory[prev_offset].v[0]; |
||||
|
} |
||||
|
|
||||
|
/* 1.2.2 Computing the lane of the reference block */ |
||||
|
ref_lane = ((pseudo_rand >> 32)) % instance->lanes; |
||||
|
|
||||
|
if ((position.pass == 0) && (position.slice == 0)) { |
||||
|
/* Can not reference other lanes yet */ |
||||
|
ref_lane = position.lane; |
||||
|
} |
||||
|
|
||||
|
/* 1.2.3 Computing the number of possible reference block within the
|
||||
|
* lane. |
||||
|
*/ |
||||
|
position.index = i; |
||||
|
ref_index = index_alpha(instance, &position, pseudo_rand & 0xFFFFFFFF, |
||||
|
ref_lane == position.lane); |
||||
|
|
||||
|
/* 2 Creating a new block */ |
||||
|
ref_block = instance->region->memory + |
||||
|
instance->lane_length * ref_lane + ref_index; |
||||
|
curr_block = instance->region->memory + curr_offset; |
||||
|
if (position.pass != 0) { |
||||
|
fill_block_with_xor(state, (uint8_t *) ref_block->v, |
||||
|
(uint8_t *) curr_block->v); |
||||
|
} else { |
||||
|
fill_block(state, (uint8_t *) ref_block->v, |
||||
|
(uint8_t *) curr_block->v); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
#endif |
@ -0,0 +1,145 @@ |
|||||
|
#ifndef blamka_round_avx512f_H |
||||
|
#define blamka_round_avx512f_H |
||||
|
|
||||
|
#include "private/common.h" |
||||
|
#include "private/sse2_64_32.h" |
||||
|
|
||||
|
#define ror64(x, n) _mm512_ror_epi64((x), (n)) |
||||
|
|
||||
|
static __m512i |
||||
|
muladd(__m512i x, __m512i y) |
||||
|
{ |
||||
|
__m512i z = _mm512_mul_epu32(x, y); |
||||
|
|
||||
|
return _mm512_add_epi64(_mm512_add_epi64(x, y), _mm512_add_epi64(z, z)); |
||||
|
} |
||||
|
|
||||
|
#define G1_AVX512F(A0, B0, C0, D0, A1, B1, C1, D1) \ |
||||
|
do { \ |
||||
|
A0 = muladd(A0, B0); \ |
||||
|
A1 = muladd(A1, B1); \ |
||||
|
\ |
||||
|
D0 = _mm512_xor_si512(D0, A0); \ |
||||
|
D1 = _mm512_xor_si512(D1, A1); \ |
||||
|
\ |
||||
|
D0 = ror64(D0, 32); \ |
||||
|
D1 = ror64(D1, 32); \ |
||||
|
\ |
||||
|
C0 = muladd(C0, D0); \ |
||||
|
C1 = muladd(C1, D1); \ |
||||
|
\ |
||||
|
B0 = _mm512_xor_si512(B0, C0); \ |
||||
|
B1 = _mm512_xor_si512(B1, C1); \ |
||||
|
\ |
||||
|
B0 = ror64(B0, 24); \ |
||||
|
B1 = ror64(B1, 24); \ |
||||
|
} while ((void)0, 0) |
||||
|
|
||||
|
#define G2_AVX512F(A0, B0, C0, D0, A1, B1, C1, D1) \ |
||||
|
do { \ |
||||
|
A0 = muladd(A0, B0); \ |
||||
|
A1 = muladd(A1, B1); \ |
||||
|
\ |
||||
|
D0 = _mm512_xor_si512(D0, A0); \ |
||||
|
D1 = _mm512_xor_si512(D1, A1); \ |
||||
|
\ |
||||
|
D0 = ror64(D0, 16); \ |
||||
|
D1 = ror64(D1, 16); \ |
||||
|
\ |
||||
|
C0 = muladd(C0, D0); \ |
||||
|
C1 = muladd(C1, D1); \ |
||||
|
\ |
||||
|
B0 = _mm512_xor_si512(B0, C0); \ |
||||
|
B1 = _mm512_xor_si512(B1, C1); \ |
||||
|
\ |
||||
|
B0 = ror64(B0, 63); \ |
||||
|
B1 = ror64(B1, 63); \ |
||||
|
} while ((void)0, 0) |
||||
|
|
||||
|
#define DIAGONALIZE(A0, B0, C0, D0, A1, B1, C1, D1) \ |
||||
|
do { \ |
||||
|
B0 = _mm512_permutex_epi64(B0, _MM_SHUFFLE(0, 3, 2, 1)); \ |
||||
|
B1 = _mm512_permutex_epi64(B1, _MM_SHUFFLE(0, 3, 2, 1)); \ |
||||
|
\ |
||||
|
C0 = _mm512_permutex_epi64(C0, _MM_SHUFFLE(1, 0, 3, 2)); \ |
||||
|
C1 = _mm512_permutex_epi64(C1, _MM_SHUFFLE(1, 0, 3, 2)); \ |
||||
|
\ |
||||
|
D0 = _mm512_permutex_epi64(D0, _MM_SHUFFLE(2, 1, 0, 3)); \ |
||||
|
D1 = _mm512_permutex_epi64(D1, _MM_SHUFFLE(2, 1, 0, 3)); \ |
||||
|
} while ((void)0, 0) |
||||
|
|
||||
|
#define UNDIAGONALIZE(A0, B0, C0, D0, A1, B1, C1, D1) \ |
||||
|
do { \ |
||||
|
B0 = _mm512_permutex_epi64(B0, _MM_SHUFFLE(2, 1, 0, 3)); \ |
||||
|
B1 = _mm512_permutex_epi64(B1, _MM_SHUFFLE(2, 1, 0, 3)); \ |
||||
|
\ |
||||
|
C0 = _mm512_permutex_epi64(C0, _MM_SHUFFLE(1, 0, 3, 2)); \ |
||||
|
C1 = _mm512_permutex_epi64(C1, _MM_SHUFFLE(1, 0, 3, 2)); \ |
||||
|
\ |
||||
|
D0 = _mm512_permutex_epi64(D0, _MM_SHUFFLE(0, 3, 2, 1)); \ |
||||
|
D1 = _mm512_permutex_epi64(D1, _MM_SHUFFLE(0, 3, 2, 1)); \ |
||||
|
} while ((void)0, 0) |
||||
|
|
||||
|
#define BLAKE2_ROUND(A0, B0, C0, D0, A1, B1, C1, D1) \ |
||||
|
do { \ |
||||
|
G1_AVX512F(A0, B0, C0, D0, A1, B1, C1, D1); \ |
||||
|
G2_AVX512F(A0, B0, C0, D0, A1, B1, C1, D1); \ |
||||
|
\ |
||||
|
DIAGONALIZE(A0, B0, C0, D0, A1, B1, C1, D1); \ |
||||
|
\ |
||||
|
G1_AVX512F(A0, B0, C0, D0, A1, B1, C1, D1); \ |
||||
|
G2_AVX512F(A0, B0, C0, D0, A1, B1, C1, D1); \ |
||||
|
\ |
||||
|
UNDIAGONALIZE(A0, B0, C0, D0, A1, B1, C1, D1); \ |
||||
|
} while ((void)0, 0) |
||||
|
|
||||
|
#define SWAP_HALVES(A0, A1) \ |
||||
|
do { \ |
||||
|
__m512i t0, t1; \ |
||||
|
t0 = _mm512_shuffle_i64x2(A0, A1, _MM_SHUFFLE(1, 0, 1, 0)); \ |
||||
|
t1 = _mm512_shuffle_i64x2(A0, A1, _MM_SHUFFLE(3, 2, 3, 2)); \ |
||||
|
A0 = t0; \ |
||||
|
A1 = t1; \ |
||||
|
} while((void)0, 0) |
||||
|
|
||||
|
#define SWAP_QUARTERS(A0, A1) \ |
||||
|
do { \ |
||||
|
SWAP_HALVES(A0, A1); \ |
||||
|
A0 = _mm512_permutexvar_epi64(_mm512_setr_epi64(0, 1, 4, 5, 2, 3, 6, 7), A0); \ |
||||
|
A1 = _mm512_permutexvar_epi64(_mm512_setr_epi64(0, 1, 4, 5, 2, 3, 6, 7), A1); \ |
||||
|
} while((void)0, 0) |
||||
|
|
||||
|
#define UNSWAP_QUARTERS(A0, A1) \ |
||||
|
do { \ |
||||
|
A0 = _mm512_permutexvar_epi64(_mm512_setr_epi64(0, 1, 4, 5, 2, 3, 6, 7), A0); \ |
||||
|
A1 = _mm512_permutexvar_epi64(_mm512_setr_epi64(0, 1, 4, 5, 2, 3, 6, 7), A1); \ |
||||
|
SWAP_HALVES(A0, A1); \ |
||||
|
} while((void)0, 0) |
||||
|
|
||||
|
#define BLAKE2_ROUND_1(A0, C0, B0, D0, A1, C1, B1, D1) \ |
||||
|
do { \ |
||||
|
SWAP_HALVES(A0, B0); \ |
||||
|
SWAP_HALVES(C0, D0); \ |
||||
|
SWAP_HALVES(A1, B1); \ |
||||
|
SWAP_HALVES(C1, D1); \ |
||||
|
BLAKE2_ROUND(A0, B0, C0, D0, A1, B1, C1, D1); \ |
||||
|
SWAP_HALVES(A0, B0); \ |
||||
|
SWAP_HALVES(C0, D0); \ |
||||
|
SWAP_HALVES(A1, B1); \ |
||||
|
SWAP_HALVES(C1, D1); \ |
||||
|
} while ((void)0, 0) |
||||
|
|
||||
|
#define BLAKE2_ROUND_2(A0, A1, B0, B1, C0, C1, D0, D1) \ |
||||
|
do { \ |
||||
|
SWAP_QUARTERS(A0, A1); \ |
||||
|
SWAP_QUARTERS(B0, B1); \ |
||||
|
SWAP_QUARTERS(C0, C1); \ |
||||
|
SWAP_QUARTERS(D0, D1); \ |
||||
|
BLAKE2_ROUND(A0, B0, C0, D0, A1, B1, C1, D1); \ |
||||
|
UNSWAP_QUARTERS(A0, A1); \ |
||||
|
UNSWAP_QUARTERS(B0, B1); \ |
||||
|
UNSWAP_QUARTERS(C0, C1); \ |
||||
|
UNSWAP_QUARTERS(D0, D1); \ |
||||
|
} while ((void)0, 0) |
||||
|
|
||||
|
#endif |
Loading…
Reference in new issue