yuzu/mbedtls
1
0
Fork 0
mirror of https://github.com/yuzu-emu/mbedtls.git synced 2024-10-18 16:21:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix cookie context usage

Browse source
This commit is contained in:
Manuel Pégourié-Gonnard 2014-07-23 17:26:48 +02:00 committed by Paul Bakker
parent 232edd46be
commit e4de06145a
1 changed files with 10 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
library/ssl_cookie.c
View file

@ -106,20 +106,23 @@ int ssl_cookie_setup( ssl_cookie_ctx *ctx,
/* /*
* Generate cookie for DTLS ClientHello verification * Generate cookie for DTLS ClientHello verification
*/ */
int ssl_cookie_write( void *ctx, int ssl_cookie_write( void *p_ctx,
unsigned char **p, unsigned char *end, unsigned char **p, unsigned char *end,
const unsigned char *cli_id, size_t cli_id_len ) const unsigned char *cli_id, size_t cli_id_len )
{ {
int ret; int ret;
unsigned char hmac_out[HVR_MD_LEN]; unsigned char hmac_out[HVR_MD_LEN];
md_context_t *hmac_ctx = (md_context_t *) ctx; ssl_cookie_ctx *ctx = (ssl_cookie_ctx *) p_ctx;
if( ctx == NULL )
return( POLARSSL_ERR_SSL_BAD_INPUT_DATA );
if( (size_t)( end - *p ) < HVR_MD_USE ) if( (size_t)( end - *p ) < HVR_MD_USE )
return( POLARSSL_ERR_SSL_BAD_INPUT_DATA ); return( POLARSSL_ERR_SSL_BAD_INPUT_DATA );
if( ( ret = md_hmac_reset( hmac_ctx ) ) != 0 || if( ( ret = md_hmac_reset( &ctx->hmac_ctx ) ) != 0 ||
( ret = md_hmac_update( hmac_ctx, cli_id, cli_id_len ) ) != 0 || ( ret = md_hmac_update( &ctx->hmac_ctx, cli_id, cli_id_len ) ) != 0 ||
( ret = md_hmac_finish( hmac_ctx, hmac_out ) ) != 0 ) ( ret = md_hmac_finish( &ctx->hmac_ctx, hmac_out ) ) != 0 )
{ {
return( POLARSSL_ERR_SSL_INTERNAL_ERROR ); return( POLARSSL_ERR_SSL_INTERNAL_ERROR );
} }
@ -133,18 +136,17 @@ int ssl_cookie_write( void *ctx,
/* /*
* Check a cookie * Check a cookie
*/ */
int ssl_cookie_check( void *ctx, int ssl_cookie_check( void *p_ctx,
const unsigned char *cookie, size_t cookie_len, const unsigned char *cookie, size_t cookie_len,
const unsigned char *cli_id, size_t cli_id_len ) const unsigned char *cli_id, size_t cli_id_len )
{ {
unsigned char ref_cookie[HVR_MD_USE]; unsigned char ref_cookie[HVR_MD_USE];
unsigned char *p = ref_cookie; unsigned char *p = ref_cookie;
md_context_t *hmac_ctx = (md_context_t *) ctx;
if( cookie_len != HVR_MD_USE ) if( cookie_len != HVR_MD_USE )
return( -1 ); return( -1 );
if( ssl_cookie_write( hmac_ctx, if( ssl_cookie_write( p_ctx,
&p, p + sizeof( ref_cookie ), &p, p + sizeof( ref_cookie ),
cli_id, cli_id_len ) != 0 ) cli_id, cli_id_len ) != 0 )
return( -1 ); return( -1 );