cuberite/polarssl
1
0
Fork 0
mirror of https://github.com/cuberite/polarssl.git synced 2025-09-30 17:09:41 -04:00
Code Issues Packages Projects Releases Wiki Activity

tests: write early data: Improve tls13_cli_max_early_data_size

Browse Source 
Signed-off-by: Ronald Cron <ronald.cron@arm.com>
This commit is contained in:
Ronald Cron 2024-03-01 16:01:27 +01:00
parent ae6f9a58a9
commit 7c07aab72e
1 changed files with 6 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tests/suites/test_suite_ssl.function
View File

@ -4506,8 +4506,8 @@ void tls13_cli_max_early_data_size(int max_early_data_size_arg)
/* /*
* (max_early_data_size + 1024) for the size of the socket buffers for the * (max_early_data_size + 1024) for the size of the socket buffers for the
* server one to be able to contain the maximum number of early data bytes * server one to be able to contain the maximum number of early data bytes
* plus the first flight client messages. Needed because we cannot initiate * plus the first flight of client messages. Needed because we cannot
* the handshake on server side before doing all the calls to * initiate the handshake on server side before doing all the calls to
* mbedtls_ssl_write_early_data() we want to test. See below for more * mbedtls_ssl_write_early_data() we want to test. See below for more
* information. * information.
*/ */
@ -4547,10 +4547,8 @@ void tls13_cli_max_early_data_size(int max_early_data_size_arg)
} }
written_early_data_size += buf_size; written_early_data_size += buf_size;
} }
TEST_EQUAL(client_ep.ssl.total_early_data_size, max_early_data_size);
/* In case we reached exactly the limit in the loop above, do another one
* byte early data write.
*/
ret = mbedtls_ssl_write_early_data(&(client_ep.ssl), buf, 1); ret = mbedtls_ssl_write_early_data(&(client_ep.ssl), buf, 1);
TEST_EQUAL(ret, MBEDTLS_ERR_SSL_CANNOT_WRITE_EARLY_DATA); TEST_EQUAL(ret, MBEDTLS_ERR_SSL_CANNOT_WRITE_EARLY_DATA);
TEST_EQUAL(client_ep.ssl.total_early_data_size, max_early_data_size); TEST_EQUAL(client_ep.ssl.total_early_data_size, max_early_data_size);
@ -4584,11 +4582,9 @@ void tls13_cli_max_early_data_size(int max_early_data_size_arg)
ret = mbedtls_ssl_handshake(&(server_ep.ssl)); ret = mbedtls_ssl_handshake(&(server_ep.ssl));
TEST_EQUAL(ret, MBEDTLS_ERR_SSL_WANT_READ); TEST_EQUAL(ret, MBEDTLS_ERR_SSL_WANT_READ);
ret = mbedtls_ssl_write_early_data(&(client_ep.ssl), buf, 1); TEST_ASSERT(mbedtls_test_move_handshake_to_state(
TEST_EQUAL(ret, MBEDTLS_ERR_SSL_CANNOT_WRITE_EARLY_DATA); &(client_ep.ssl), &(server_ep.ssl), MBEDTLS_SSL_HANDSHAKE_OVER)
TEST_EQUAL(client_ep.ssl.total_early_data_size, max_early_data_size); == 0);
TEST_EQUAL(client_ep.ssl.early_data_status,
MBEDTLS_SSL_EARLY_DATA_STATUS_CAN_WRITE);
exit: exit:
mbedtls_test_ssl_endpoint_free(&client_ep, NULL); mbedtls_test_ssl_endpoint_free(&client_ep, NULL);