mirror of
https://github.com/Mbed-TLS/mbedtls.git
synced 2025-07-30 22:43:08 +03:00
Merge pull request #9872 from rojer/tls_hs_defrag_in
Defragment incoming TLS handshake messages
This commit is contained in:
@ -339,12 +339,13 @@ static void handle_buffer_resizing(mbedtls_ssl_context *ssl, int downsizing,
|
||||
size_t out_buf_new_len)
|
||||
{
|
||||
int modified = 0;
|
||||
size_t written_in = 0, iv_offset_in = 0, len_offset_in = 0;
|
||||
size_t written_in = 0, iv_offset_in = 0, len_offset_in = 0, hdr_in = 0;
|
||||
size_t written_out = 0, iv_offset_out = 0, len_offset_out = 0;
|
||||
if (ssl->in_buf != NULL) {
|
||||
written_in = ssl->in_msg - ssl->in_buf;
|
||||
iv_offset_in = ssl->in_iv - ssl->in_buf;
|
||||
len_offset_in = ssl->in_len - ssl->in_buf;
|
||||
hdr_in = ssl->in_hdr - ssl->in_buf;
|
||||
if (downsizing ?
|
||||
ssl->in_buf_len > in_buf_new_len && ssl->in_left < in_buf_new_len :
|
||||
ssl->in_buf_len < in_buf_new_len) {
|
||||
@ -376,7 +377,10 @@ static void handle_buffer_resizing(mbedtls_ssl_context *ssl, int downsizing,
|
||||
}
|
||||
if (modified) {
|
||||
/* Update pointers here to avoid doing it twice. */
|
||||
mbedtls_ssl_reset_in_out_pointers(ssl);
|
||||
ssl->in_hdr = ssl->in_buf + hdr_in;
|
||||
mbedtls_ssl_update_in_pointers(ssl);
|
||||
mbedtls_ssl_reset_out_pointers(ssl);
|
||||
|
||||
/* Fields below might not be properly updated with record
|
||||
* splitting or with CID, so they are manually updated here. */
|
||||
ssl->out_msg = ssl->out_buf + written_out;
|
||||
@ -1277,7 +1281,8 @@ int mbedtls_ssl_setup(mbedtls_ssl_context *ssl,
|
||||
goto error;
|
||||
}
|
||||
|
||||
mbedtls_ssl_reset_in_out_pointers(ssl);
|
||||
mbedtls_ssl_reset_in_pointers(ssl);
|
||||
mbedtls_ssl_reset_out_pointers(ssl);
|
||||
|
||||
#if defined(MBEDTLS_SSL_DTLS_SRTP)
|
||||
memset(&ssl->dtls_srtp_info, 0, sizeof(ssl->dtls_srtp_info));
|
||||
@ -1342,7 +1347,8 @@ void mbedtls_ssl_session_reset_msg_layer(mbedtls_ssl_context *ssl,
|
||||
/* Cancel any possibly running timer */
|
||||
mbedtls_ssl_set_timer(ssl, 0);
|
||||
|
||||
mbedtls_ssl_reset_in_out_pointers(ssl);
|
||||
mbedtls_ssl_reset_in_pointers(ssl);
|
||||
mbedtls_ssl_reset_out_pointers(ssl);
|
||||
|
||||
/* Reset incoming message parsing */
|
||||
ssl->in_offt = NULL;
|
||||
@ -1350,6 +1356,7 @@ void mbedtls_ssl_session_reset_msg_layer(mbedtls_ssl_context *ssl,
|
||||
ssl->in_msgtype = 0;
|
||||
ssl->in_msglen = 0;
|
||||
ssl->in_hslen = 0;
|
||||
ssl->in_hsfraglen = 0;
|
||||
ssl->keep_current_message = 0;
|
||||
ssl->transform_in = NULL;
|
||||
|
||||
|
Reference in New Issue
Block a user