diff --git a/library/ssl_tls.c b/library/ssl_tls.c index cde6795f9..fd5b5a33b 100644 --- a/library/ssl_tls.c +++ b/library/ssl_tls.c @@ -3239,6 +3239,10 @@ int ssl_set_hostname( ssl_context *ssl, const char *hostname ) return( POLARSSL_ERR_SSL_BAD_INPUT_DATA ); ssl->hostname_len = strlen( hostname ); + + if( ssl->hostname_len + 1 == 0 ) + return( POLARSSL_ERR_SSL_BAD_INPUT_DATA ); + ssl->hostname = (unsigned char *) malloc( ssl->hostname_len + 1 ); if( ssl->hostname == NULL ) @@ -3246,7 +3250,7 @@ int ssl_set_hostname( ssl_context *ssl, const char *hostname ) memcpy( ssl->hostname, (const unsigned char *) hostname, ssl->hostname_len ); - + ssl->hostname[ssl->hostname_len] = '\0'; return( 0 );