1
1
mirror of https://github.com/BLAKE2/BLAKE2 synced 2024-11-29 10:45:04 +01:00

style, consistency, wip

This commit is contained in:
Samuel Neves 2016-10-11 15:04:29 +01:00
parent c038f88aec
commit 1dd6358997
3 changed files with 134 additions and 100 deletions

@ -64,6 +64,19 @@ static BLAKE2_INLINE uint64_t load64( const void *src )
#endif #endif
} }
static BLAKE2_INLINE uint16_t load16( const void *src )
{
#if defined(NATIVE_LITTLE_ENDIAN)
uint16_t w;
memcpy(&w, src, sizeof w);
return w;
#else
const uint8_t *p = ( const uint8_t * )src;
return (( uint16_t )( p[0] ) << 0) |
(( uint16_t )( p[1] ) << 8) ;
#endif
}
static BLAKE2_INLINE void store16( void *dst, uint16_t w ) static BLAKE2_INLINE void store16( void *dst, uint16_t w )
{ {
#if defined(NATIVE_LITTLE_ENDIAN) #if defined(NATIVE_LITTLE_ENDIAN)

@ -5,141 +5,162 @@
#include "blake2.h" #include "blake2.h"
#include "blake2-impl.h" #include "blake2-impl.h"
typedef struct blake2xs_state__ {
int blake2xs_init( blake2s_state *S, const uint16_t outlen, const void *key, const uint8_t keylen) blake2s_state S[1];
{
blake2s_param P[1]; blake2s_param P[1];
} blake2xs_state;
if ( !outlen ) return -1;
P->digest_length = BLAKE2S_OUTBYTES; int blake2xs_init( blake2xs_state *S, const size_t outlen, const void *key, size_t keylen )
P->key_length = keylen; {
P->fanout = 1; if ( outlen == 0 || outlen > 0xFFFFUL ) {
P->depth = 1; return -1;
store32( &P->leaf_length, 0 ); }
store32( &P->node_offset, 0 );
store16( &P->xof_length, outlen );
P->node_depth = 0;
P->inner_length = 0;
memset( P->salt, 0, sizeof( P->salt ) );
memset( P->personal, 0, sizeof( P->personal ) );
if( blake2s_init_param( S, P ) < 0 ) return -1; if (key == NULL || keylen > BLAKE2S_KEYBYTES) {
return -1;
}
if (keylen) /* Initialize parameter block */
{ S->P->digest_length = BLAKE2S_OUTBYTES;
if ( !key || keylen > BLAKE2S_KEYBYTES ) return -1; S->P->key_length = keylen;
uint8_t block[BLAKE2S_BLOCKBYTES]; S->P->fanout = 1;
memset( block, 0, BLAKE2S_BLOCKBYTES ); S->P->depth = 1;
memcpy( block, key, keylen ); store32( &S->P->leaf_length, 0 );
blake2s_update( S, block, BLAKE2S_BLOCKBYTES ); store32( &S->P->node_offset, 0 );
secure_zero_memory( block, BLAKE2S_BLOCKBYTES ); store16( &S->P->xof_length, outlen );
} S->P->node_depth = 0;
S->P->inner_length = 0;
memset( S->P->salt, 0, sizeof( S->P->salt ) );
memset( S->P->personal, 0, sizeof( S->P->personal ) );
if( blake2s_init_param( S->S, S->P ) < 0 ) {
return -1;
}
if (keylen > 0) {
uint8_t block[BLAKE2S_BLOCKBYTES];
memset(block, 0, BLAKE2S_BLOCKBYTES);
memcpy(block, key, keylen);
blake2s_update(S->S, block, BLAKE2S_BLOCKBYTES);
secure_zero_memory(block, BLAKE2S_BLOCKBYTES);
}
return 0; return 0;
} }
int blake2xs_update( blake2xs_state *S, const void *in, size_t inlen ) {
int blake2xs_update( blake2s_state *S, const uint8_t *in, uint64_t inlen ) return blake2s_update( S->S, in, inlen );
{
if ( !in ) return -1;
return blake2s_update( S, in, inlen );
} }
int blake2xs_final( blake2s_state *S, uint8_t *out, const uint16_t outlen ) { int blake2xs_final(blake2xs_state *S, void *out, size_t outlen)
{
blake2s_state C[1];
blake2s_param P[1];
uint16_t xof_length = load16(&S->P->xof_length);
uint8_t root[BLAKE2S_BLOCKBYTES];
size_t i;
blake2s_state C[1]; if (out == NULL) {
blake2s_param P[1]; return -1;
int i; }
uint16_t last_index = outlen/BLAKE2S_OUTBYTES;
uint8_t root[BLAKE2S_BLOCKBYTES];
if ( !out ) return 1; /* outlen must match the output size defined in xof_length, */
/* unless it was -1, in which case anything goes except 0. */
/* Finalize the root hash */ if(xof_length == 0xFFFFUL) {
if (blake2s_final( S, root, BLAKE2S_OUTBYTES ) < 0 ) if(outlen == 0) {
return -1; return -1;
/* Set common block structure values */
P->digest_length = BLAKE2S_OUTBYTES;
P->key_length = 0;
P->fanout = 0;
P->depth = 0;
store32( &P->leaf_length, BLAKE2S_OUTBYTES );
store16( &P->xof_length, outlen );
P->inner_length = 0;
P->node_depth = 0;
memset( P->salt, 0, sizeof( P->salt ) );
memset( P->personal, 0, sizeof( P->personal ) );
for( i = 0; i < last_index; ++ i ) {
/* Initialize state */
store32( &P->node_offset, i );
if( blake2s_init_param( C, P ) < 0 ) return -1;
/* Process key if needed */
blake2s_update( C, root, BLAKE2S_OUTBYTES);
blake2s_final( C, out + i*BLAKE2S_OUTBYTES, BLAKE2S_OUTBYTES );
} }
/* Last instance */ } else {
store32( &P->node_offset, last_index); if(outlen != xof_length) {
P->digest_length = outlen % BLAKE2S_OUTBYTES; return -1;
if( blake2s_init_param( C, P ) < 0 ) return -1; }
blake2s_update( C, root, BLAKE2S_OUTBYTES); }
blake2s_final( C, out + last_index*BLAKE2S_OUTBYTES, outlen % BLAKE2S_OUTBYTES );
return 0; /* Finalize the root hash */
if (blake2s_final(S->S, root, BLAKE2S_OUTBYTES) < 0) {
return -1;
}
/* Set common block structure values */
/* Copy values from parent instance, and only change the ones below */
memcpy(P, S->P, sizeof(blake2s_param));
P->fanout = 0;
P->depth = 0;
store32(&P->leaf_length, BLAKE2S_OUTBYTES);
P->inner_length = BLAKE2S_OUTBYTES;
P->node_depth = 0;
for (i = 0; outlen > 0; ++i) {
const size_t block_size = (outlen < BLAKE2S_OUTBYTES) ? outlen : BLAKE2S_OUTBYTES;
/* Initialize state */
store32(&P->node_offset, i);
blake2s_init_param(C, P);
/* Process key if needed */
blake2s_update(C, root, BLAKE2S_OUTBYTES);
blake2s_final(C, (uint8_t *)out + i * BLAKE2S_OUTBYTES, BLAKE2S_OUTBYTES);
outlen -= block_size;
}
secure_zero_memory(root, sizeof(root));
secure_zero_memory(P, sizeof(P));
secure_zero_memory(C, sizeof(C));
/* Put blake2xs in an invalid state? cf. blake2s_is_lastblock */
return 0;
} }
int blake2xs( uint8_t *out, const void *in, const void *key, const int blake2xs(void *out, size_t outlen, const void *in, size_t inlen, const void *key, size_t keylen)
uint16_t outlen, const uint64_t inlen, const uint8_t keylen )
{ {
blake2s_state S[1]; blake2xs_state S[1];
/* Verify parameters */ /* Verify parameters */
if ( NULL == in && inlen > 0 ) return -1; if (NULL == in && inlen > 0)
return -1;
if ( NULL == out ) return -1; if (NULL == out)
return -1;
if ( NULL == key && keylen > 0) return -1; if (NULL == key && keylen > 0)
return -1;
if( keylen > BLAKE2S_KEYBYTES ) return -1; if (keylen > BLAKE2S_KEYBYTES)
return -1;
if( !outlen ) return -1; if (outlen == 0)
return -1;
/* Initialize the root block structure */ /* Initialize the root block structure */
if ( blake2xs_init( S, outlen, key, keylen ) < 0 ) { if (blake2xs_init(S, outlen, key, keylen) < 0) {
return -1; return -1;
} }
/* Compute the root of the tree */ /* Compute the root of the tree */
if ( blake2xs_update( S, ( const uint8_t * )in, inlen ) < 0 ) if (blake2xs_update(S, in, inlen) < 0) {
return -1; return -1;
}
/* Compute the final hash using the counter construction */ /* Compute the final hash using the counter construction */
blake2xs_final( S, out, outlen ); return blake2xs_final(S, out, outlen);
return 0;
} }
#if defined(BLAKE2XS_SELFTEST) #if defined(BLAKE2XS_SELFTEST)
#include <string.h> #include <string.h>
#include "blake2-kat.h" #include "blake2-kat.h"
int main( int argc, char **argv ) int main( void )
{ {
uint8_t key[BLAKE2S_KEYBYTES]; uint8_t key[BLAKE2S_KEYBYTES];
uint8_t buf[BLAKE2_KAT_LENGTH]; uint8_t buf[BLAKE2_KAT_LENGTH];
for( size_t i = 0; i < BLAKE2S_KEYBYTES; ++i ) for( size_t i = 0; i < BLAKE2S_KEYBYTES; ++i ) {
key[i] = ( uint8_t )i; key[i] = ( uint8_t )i;
}
for( size_t i = 0; i < BLAKE2_KAT_LENGTH; ++i ) for( size_t i = 0; i < BLAKE2_KAT_LENGTH; ++i ) {
buf[i] = ( uint8_t )i; buf[i] = ( uint8_t )i;
}
for( size_t i = 0; i < BLAKE2_KAT_LENGTH; ++i ) for( size_t i = 1; i < BLAKE2_KAT_LENGTH; ++i )
{ {
uint8_t hash[BLAKE2_KAT_LENGTH]; uint8_t hash[BLAKE2_KAT_LENGTH] = {0};
blake2xs( hash, buf, key, i, BLAKE2_KAT_LENGTH, BLAKE2S_KEYBYTES ); blake2xs( hash, i, buf, BLAKE2_KAT_LENGTH, key, BLAKE2S_KEYBYTES );
for( size_t j = 0; j < i; ++j ) { for( size_t j = 0; j < i; ++j ) {
printf("%02x", hash[j]); printf("%02x", hash[j]);