101 lines
4.2 KiB
C
101 lines
4.2 KiB
C
#ifndef __BLAKE2_DIGEST_H__
|
|
#define __BLAKE2_DIGEST_H__
|
|
|
|
#include "blake2.h"
|
|
#include <cpuid.h>
|
|
|
|
#if defined(__cplusplus)
|
|
extern "C" {
|
|
#endif
|
|
|
|
typedef int (*blake2b_init_funcptr)( blake2b_state *S, const uint8_t outlen );
|
|
typedef int (*blake2b_init_key_funcptr)( blake2b_state *S, const uint8_t outlen, const void *key, const uint8_t keylen );
|
|
typedef int (*blake2b_init_param_funcptr)( blake2b_state *S, const blake2b_param *P );
|
|
typedef int (*blake2b_update_funcptr)( blake2b_state *S, const uint8_t *in, uint64_t inlen );
|
|
typedef int (*blake2b_final_funcptr)( blake2b_state *S, uint8_t *out, uint8_t outlen );
|
|
typedef int (*blake2bp_init_funcptr)( blake2bp_state *S, const uint8_t outlen );
|
|
typedef int (*blake2bp_init_key_funcptr)( blake2bp_state *S, const uint8_t outlen, const void *key, const uint8_t keylen );
|
|
typedef int (*blake2bp_update_funcptr)( blake2bp_state *S, const uint8_t *in, uint64_t inlen );
|
|
typedef int (*blake2bp_final_funcptr)( blake2bp_state *S, uint8_t *out, uint8_t outlen );
|
|
|
|
typedef int (*blake2b_funcptr)( uint8_t *out, const void *in, const void *key, const uint8_t outlen, const uint64_t inlen, uint8_t keylen );
|
|
typedef int (*blake2bp_funcptr)( uint8_t *out, const void *in, const void *key, const uint8_t outlen, const uint64_t inlen, uint8_t keylen );
|
|
|
|
/*
|
|
* BLAKE2 function pointers. These are set to the optimized routines
|
|
* based on CPU capabilities.
|
|
*/
|
|
struct blake2_dispatch {
|
|
blake2b_init_funcptr blake2b_init;
|
|
blake2b_init_key_funcptr blake2b_init_key;
|
|
blake2b_init_param_funcptr blake2b_init_param;
|
|
blake2b_update_funcptr blake2b_update;
|
|
blake2b_final_funcptr blake2b_final;
|
|
blake2bp_init_funcptr blake2bp_init;
|
|
blake2bp_init_key_funcptr blake2bp_init_key;
|
|
blake2bp_update_funcptr blake2bp_update;
|
|
blake2bp_final_funcptr blake2bp_final;
|
|
blake2b_funcptr blake2b;
|
|
blake2bp_funcptr blake2bp;
|
|
};
|
|
|
|
static void blake2_module_init(struct blake2_dispatch *dsp, processor_info_t *pc)
|
|
{
|
|
dsp->blake2b_init = blake2b_init_sse2;
|
|
dsp->blake2b_init_key = blake2b_init_key_sse2;
|
|
dsp->blake2b_init_param = blake2b_init_param_sse2;
|
|
dsp->blake2b_update = blake2b_update_sse2;
|
|
dsp->blake2b_final = blake2b_final_sse2;
|
|
dsp->blake2bp_init = blake2bp_init_sse2;
|
|
dsp->blake2bp_init_key = blake2bp_init_key_sse2;
|
|
dsp->blake2bp_update = blake2bp_update_sse2;
|
|
dsp->blake2bp_final = blake2bp_final_sse2;
|
|
dsp->blake2b = blake2b_sse2;
|
|
dsp->blake2bp = blake2bp_sse2;
|
|
|
|
if (pc->sse_level == 3 && pc->sse_sub_level == 1) {
|
|
dsp->blake2b_init = blake2b_init_ssse3;
|
|
dsp->blake2b_init_key = blake2b_init_key_ssse3;
|
|
dsp->blake2b_init_param = blake2b_init_param_ssse3;
|
|
dsp->blake2b_update = blake2b_update_ssse3;
|
|
dsp->blake2b_final = blake2b_final_ssse3;
|
|
dsp->blake2bp_init = blake2bp_init_ssse3;
|
|
dsp->blake2bp_init_key = blake2bp_init_key_ssse3;
|
|
dsp->blake2bp_update = blake2bp_update_ssse3;
|
|
dsp->blake2bp_final = blake2bp_final_ssse3;
|
|
dsp->blake2b = blake2b_ssse3;
|
|
dsp->blake2bp = blake2bp_ssse3;
|
|
|
|
} else if (pc->sse_level == 4 && pc->sse_sub_level >= 1) {
|
|
dsp->blake2b_init = blake2b_init_sse41;
|
|
dsp->blake2b_init_key = blake2b_init_key_sse41;
|
|
dsp->blake2b_init_param = blake2b_init_param_sse41;
|
|
dsp->blake2b_update = blake2b_update_sse41;
|
|
dsp->blake2b_final = blake2b_final_sse41;
|
|
dsp->blake2bp_init = blake2bp_init_sse41;
|
|
dsp->blake2bp_init_key = blake2bp_init_key_sse41;
|
|
dsp->blake2bp_update = blake2bp_update_sse41;
|
|
dsp->blake2bp_final = blake2bp_final_sse41;
|
|
dsp->blake2b = blake2b_sse41;
|
|
dsp->blake2bp = blake2bp_sse41;
|
|
}
|
|
if (pc->avx_level >= 1) {
|
|
dsp->blake2b_init = blake2b_init_avx;
|
|
dsp->blake2b_init_key = blake2b_init_key_avx;
|
|
dsp->blake2b_init_param = blake2b_init_param_avx;
|
|
dsp->blake2b_update = blake2b_update_avx;
|
|
dsp->blake2b_final = blake2b_final_avx;
|
|
dsp->blake2bp_init = blake2bp_init_avx;
|
|
dsp->blake2bp_init_key = blake2bp_init_key_avx;
|
|
dsp->blake2bp_update = blake2bp_update_avx;
|
|
dsp->blake2bp_final = blake2bp_final_avx;
|
|
dsp->blake2b = blake2b_avx;
|
|
dsp->blake2bp = blake2bp_avx;
|
|
}
|
|
}
|
|
|
|
#if defined(__cplusplus)
|
|
}
|
|
#endif
|
|
|
|
#endif
|