mirror of https://github.com/openssl/openssl.git
				
				
				
			
		
			
	
	
		
			633 lines
		
	
	
		
			22 KiB
		
	
	
	
		
			C
		
	
	
	
		
		
			
		
	
	
			633 lines
		
	
	
		
			22 KiB
		
	
	
	
		
			C
		
	
	
	
|  | /*
 | ||
|  |  * Copyright 2022 The OpenSSL Project Authors. All Rights Reserved. | ||
|  |  * | ||
|  |  * Licensed under the Apache License 2.0 (the "License").  You may not use | ||
|  |  * this file except in compliance with the License.  You can obtain a copy | ||
|  |  * in the file LICENSE in the source distribution or at | ||
|  |  * https://www.openssl.org/source/license.html
 | ||
|  |  */ | ||
|  | 
 | ||
|  | #include "internal/quic_fc.h"
 | ||
|  | #include "internal/quic_error.h"
 | ||
|  | #include "testutil.h"
 | ||
|  | 
 | ||
|  | static int test_txfc(int is_stream) | ||
|  | { | ||
|  |     int testresult = 0; | ||
|  |     QUIC_TXFC conn_txfc, stream_txfc, *txfc, *parent_txfc; | ||
|  | 
 | ||
|  |     if (!TEST_true(ossl_quic_txfc_init(&conn_txfc, 0))) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (is_stream && !TEST_true(ossl_quic_txfc_init(&stream_txfc, &conn_txfc))) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     txfc = is_stream ? &stream_txfc : &conn_txfc; | ||
|  |     parent_txfc = is_stream ? &conn_txfc : NULL; | ||
|  | 
 | ||
|  |     if (!TEST_true(ossl_quic_txfc_bump_cwm(txfc, 2000))) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (is_stream && !TEST_true(ossl_quic_txfc_bump_cwm(parent_txfc, 2000))) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_uint64_t_eq(ossl_quic_txfc_get_swm(txfc), 0)) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_uint64_t_eq(ossl_quic_txfc_get_cwm(txfc), 2000)) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_uint64_t_eq(ossl_quic_txfc_get_credit_local(txfc), 2000)) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (is_stream && !TEST_uint64_t_eq(ossl_quic_txfc_get_credit(txfc), | ||
|  |                                        2000)) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_false(ossl_quic_txfc_has_become_blocked(txfc, 0))) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_true(ossl_quic_txfc_consume_credit(txfc, 500))) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_uint64_t_eq(ossl_quic_txfc_get_credit_local(txfc), 1500)) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (is_stream && !TEST_uint64_t_eq(ossl_quic_txfc_get_credit(txfc), | ||
|  |                                        1500)) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_false(ossl_quic_txfc_has_become_blocked(txfc, 0))) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_uint64_t_eq(ossl_quic_txfc_get_swm(txfc), 500)) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_true(ossl_quic_txfc_consume_credit(txfc, 100))) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_uint64_t_eq(ossl_quic_txfc_get_swm(txfc), 600)) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_uint64_t_eq(ossl_quic_txfc_get_credit_local(txfc), 1400)) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (is_stream && !TEST_uint64_t_eq(ossl_quic_txfc_get_credit(txfc), | ||
|  |                                        1400)) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_false(ossl_quic_txfc_has_become_blocked(txfc, 0))) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_true(ossl_quic_txfc_consume_credit(txfc, 1400))) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_uint64_t_eq(ossl_quic_txfc_get_credit_local(txfc), 0)) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (is_stream && !TEST_uint64_t_eq(ossl_quic_txfc_get_credit(txfc), | ||
|  |                                        0)) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_uint64_t_eq(ossl_quic_txfc_get_swm(txfc), 2000)) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_true(ossl_quic_txfc_has_become_blocked(txfc, 0))) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_true(ossl_quic_txfc_has_become_blocked(txfc, 0))) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_true(ossl_quic_txfc_has_become_blocked(txfc, 1))) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_false(ossl_quic_txfc_has_become_blocked(txfc, 0))) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_false(ossl_quic_txfc_has_become_blocked(txfc, 0))) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_false(ossl_quic_txfc_consume_credit(txfc, 1))) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_uint64_t_eq(ossl_quic_txfc_get_cwm(txfc), 2000)) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_uint64_t_eq(ossl_quic_txfc_get_swm(txfc), 2000)) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_false(ossl_quic_txfc_bump_cwm(txfc, 2000))) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_true(ossl_quic_txfc_bump_cwm(txfc, 2500))) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (is_stream && !TEST_true(ossl_quic_txfc_bump_cwm(parent_txfc, 2400))) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_uint64_t_eq(ossl_quic_txfc_get_cwm(txfc), 2500)) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_uint64_t_eq(ossl_quic_txfc_get_swm(txfc), 2000)) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (!TEST_uint64_t_eq(ossl_quic_txfc_get_credit_local(txfc), 500)) | ||
|  |         goto err; | ||
|  | 
 | ||
|  |     if (is_stream) | ||
|  |         ossl_quic_txfc_has_become_blocked(parent_txfc, 1); | ||
|  | 
 | ||
|  |     if (is_stream) { | ||
|  |         if (!TEST_true(ossl_quic_txfc_consume_credit(txfc, 399))) | ||
|  |             goto err; | ||
|  | 
 | ||
|  |         if (!TEST_false(ossl_quic_txfc_has_become_blocked(txfc, 0))) | ||
|  |             goto err; | ||
|  | 
 | ||
|  |         if (!TEST_uint64_t_eq(ossl_quic_txfc_get_credit(txfc), 1)) | ||
|  |             goto err; | ||
|  | 
 | ||
|  |         if (!TEST_true(ossl_quic_txfc_consume_credit(txfc, 1))) | ||
|  |             goto err; | ||
|  | 
 | ||
|  |         if (!TEST_true(ossl_quic_txfc_has_become_blocked(parent_txfc, 0))) | ||
|  |             goto err; | ||
|  | 
 | ||
|  |         if (!TEST_true(ossl_quic_txfc_has_become_blocked(parent_txfc, 1))) | ||
|  |             goto err; | ||
|  | 
 | ||
|  |         if (!TEST_false(ossl_quic_txfc_has_become_blocked(parent_txfc, 0))) | ||
|  |             goto err; | ||
|  |     } else { | ||
|  |         if (!TEST_true(ossl_quic_txfc_consume_credit(txfc, 499))) | ||
|  |             goto err; | ||
|  | 
 | ||
|  |         if (!TEST_false(ossl_quic_txfc_has_become_blocked(txfc, 0))) | ||
|  |             goto err; | ||
|  | 
 | ||
|  |         if (is_stream && !TEST_false(ossl_quic_txfc_has_become_blocked(parent_txfc, 0))) | ||
|  |             goto err; | ||
|  | 
 | ||
|  |         if (!TEST_true(ossl_quic_txfc_consume_credit(txfc, 1))) | ||
|  |             goto err; | ||
|  | 
 | ||
|  |         if (!TEST_true(ossl_quic_txfc_has_become_blocked(txfc, 0))) | ||
|  |             goto err; | ||
|  | 
 | ||
|  |         if (!TEST_true(ossl_quic_txfc_has_become_blocked(txfc, 1))) | ||
|  |             goto err; | ||
|  | 
 | ||
|  |         if (!TEST_false(ossl_quic_txfc_has_become_blocked(txfc, 0))) | ||
|  |             goto err; | ||
|  |     } | ||
|  | 
 | ||
|  |     testresult = 1; | ||
|  | err: | ||
|  |     return testresult; | ||
|  | } | ||
|  | 
 | ||
|  | static OSSL_TIME cur_time; | ||
|  | 
 | ||
|  | static OSSL_TIME fake_now(void *arg) | ||
|  | { | ||
|  |     return cur_time; | ||
|  | } | ||
|  | 
 | ||
|  | #define RX_OPC_END                    0
 | ||
|  | #define RX_OPC_INIT_CONN              1 /* arg0=initial window, arg1=max window */
 | ||
|  | #define RX_OPC_INIT_STREAM            2 /* arg0=initial window, arg1=max window */
 | ||
|  | #define RX_OPC_RX                     3 /* arg0=end, arg1=is_fin */
 | ||
|  | #define RX_OPC_RETIRE                 4 /* arg0=num_bytes, arg1=rtt in OSSL_TIME ticks, expect_fail */
 | ||
|  | #define RX_OPC_CHECK_CWM_CONN         5 /* arg0=expected */
 | ||
|  | #define RX_OPC_CHECK_CWM_STREAM       6 /* arg0=expected */
 | ||
|  | #define RX_OPC_CHECK_SWM_CONN         7 /* arg0=expected */
 | ||
|  | #define RX_OPC_CHECK_SWM_STREAM       8 /* arg0=expected */
 | ||
|  | #define RX_OPC_CHECK_RWM_CONN         9 /* arg0=expected */
 | ||
|  | #define RX_OPC_CHECK_RWM_STREAM      10 /* arg0=expected */
 | ||
|  | #define RX_OPC_CHECK_CHANGED_CONN    11 /* arg0=expected, arg1=clear */
 | ||
|  | #define RX_OPC_CHECK_CHANGED_STREAM  12 /* arg0=expected, arg1=clear */
 | ||
|  | #define RX_OPC_CHECK_ERROR_CONN      13 /* arg0=expected, arg1=clear */
 | ||
|  | #define RX_OPC_CHECK_ERROR_STREAM    14 /* arg0=expected, arg1=clear */
 | ||
|  | #define RX_OPC_STEP_TIME             15 /* arg0=OSSL_TIME ticks to advance */
 | ||
|  | #define RX_OPC_MSG                   16
 | ||
|  | 
 | ||
|  | struct rx_test_op { | ||
|  |     unsigned char   op; | ||
|  |     size_t          stream_idx; | ||
|  |     uint64_t        arg0, arg1; | ||
|  |     unsigned char   expect_fail; | ||
|  |     const char     *msg; | ||
|  | }; | ||
|  | 
 | ||
|  | #define RX_OP_END \
 | ||
|  |     { RX_OPC_END } | ||
|  | #define RX_OP_INIT_CONN(init_window_size, max_window_size) \
 | ||
|  |     { RX_OPC_INIT_CONN, 0, (init_window_size), (max_window_size) }, | ||
|  | #define RX_OP_INIT_STREAM(stream_idx, init_window_size, max_window_size) \
 | ||
|  |     { RX_OPC_INIT_STREAM, (stream_idx), (init_window_size), (max_window_size) }, | ||
|  | #define RX_OP_RX(stream_idx, end, is_fin) \
 | ||
|  |     { RX_OPC_RX, (stream_idx), (end), (is_fin) }, | ||
|  | #define RX_OP_RETIRE(stream_idx, num_bytes, rtt, expect_fail) \
 | ||
|  |     { RX_OPC_RETIRE, (stream_idx), (num_bytes), (rtt), (expect_fail) }, | ||
|  | #define RX_OP_CHECK_CWM_CONN(expected) \
 | ||
|  |     { RX_OPC_CHECK_CWM_CONN, 0, (expected) }, | ||
|  | #define RX_OP_CHECK_CWM_STREAM(stream_id, expected) \
 | ||
|  |     { RX_OPC_CHECK_CWM_STREAM, (stream_id), (expected) }, | ||
|  | #define RX_OP_CHECK_SWM_CONN(expected) \
 | ||
|  |     { RX_OPC_CHECK_SWM_CONN, 0, (expected) }, | ||
|  | #define RX_OP_CHECK_SWM_STREAM(stream_id, expected) \
 | ||
|  |     { RX_OPC_CHECK_SWM_STREAM, (stream_id), (expected) }, | ||
|  | #define RX_OP_CHECK_RWM_CONN(expected) \
 | ||
|  |     { RX_OPC_CHECK_RWM_CONN, 0, (expected) }, | ||
|  | #define RX_OP_CHECK_RWM_STREAM(stream_id, expected) \
 | ||
|  |     { RX_OPC_CHECK_RWM_STREAM, (stream_id), (expected) }, | ||
|  | #define RX_OP_CHECK_CHANGED_CONN(expected, clear) \
 | ||
|  |     { RX_OPC_CHECK_CHANGED_CONN, 0, (expected), (clear) }, | ||
|  | #define RX_OP_CHECK_CHANGED_STREAM(stream_id, expected, clear) \
 | ||
|  |     { RX_OPC_CHECK_CHANGED_STREAM, (stream_id), (expected), (clear) }, | ||
|  | #define RX_OP_CHECK_ERROR_CONN(expected, clear) \
 | ||
|  |     { RX_OPC_CHECK_ERROR_CONN, 0, (expected), (clear) }, | ||
|  | #define RX_OP_CHECK_ERROR_STREAM(stream_id, expected, clear) \
 | ||
|  |     { RX_OPC_CHECK_ERROR_STREAM, (stream_id), (expected), (clear) }, | ||
|  | #define RX_OP_STEP_TIME(t) \
 | ||
|  |     { RX_OPC_STEP_TIME, 0, (t) }, | ||
|  | #define RX_OP_MSG(msg) \
 | ||
|  |     { RX_OPC_MSG, 0, 0, 0, 0, (msg) }, | ||
|  | 
 | ||
|  | #define RX_OP_INIT(init_window_size, max_window_size) \
 | ||
|  |     RX_OP_INIT_CONN(init_window_size, max_window_size) \ | ||
|  |     RX_OP_INIT_STREAM(0, init_window_size, max_window_size) | ||
|  | #define RX_OP_CHECK_CWM(expected) \
 | ||
|  |     RX_OP_CHECK_CWM_CONN(expected) \ | ||
|  |     RX_OP_CHECK_CWM_STREAM(0, expected) | ||
|  | #define RX_OP_CHECK_SWM(expected) \
 | ||
|  |     RX_OP_CHECK_SWM_CONN(expected) \ | ||
|  |     RX_OP_CHECK_SWM_STREAM(0, expected) | ||
|  | #define RX_OP_CHECK_RWM(expected) \
 | ||
|  |     RX_OP_CHECK_RWM_CONN(expected) \ | ||
|  |     RX_OP_CHECK_RWM_STREAM(0, expected) | ||
|  | #define RX_OP_CHECK_CHANGED(expected, clear) \
 | ||
|  |     RX_OP_CHECK_CHANGED_CONN(expected, clear) \ | ||
|  |     RX_OP_CHECK_CHANGED_STREAM(0, expected, clear) | ||
|  | #define RX_OP_CHECK_ERROR(expected, clear) \
 | ||
|  |     RX_OP_CHECK_ERROR_CONN(expected, clear) \ | ||
|  |     RX_OP_CHECK_ERROR_STREAM(0, expected, clear) | ||
|  | 
 | ||
|  | #define INIT_WINDOW_SIZE (1 * 1024 * 1024)
 | ||
|  | #define INIT_S_WINDOW_SIZE (384 * 1024)
 | ||
|  | 
 | ||
|  | /* 1. Basic RXFC Tests (stream window == connection window) */ | ||
|  | static const struct rx_test_op rx_script_1[] = { | ||
|  |     RX_OP_STEP_TIME(1000 * OSSL_TIME_MS) | ||
|  |     RX_OP_INIT(INIT_WINDOW_SIZE, 10 * INIT_WINDOW_SIZE) | ||
|  |     /* Check initial state. */ | ||
|  |     RX_OP_CHECK_CWM(INIT_WINDOW_SIZE) | ||
|  |     RX_OP_CHECK_ERROR(0, 0) | ||
|  |     RX_OP_CHECK_CHANGED(0, 0) | ||
|  |     /* We cannot retire what we have not received. */ | ||
|  |     RX_OP_RETIRE(0, 1, 0, 1) | ||
|  |     /* Zero bytes is a no-op and always valid. */ | ||
|  |     RX_OP_RETIRE(0, 0, 0, 0) | ||
|  |     /* Consume some window. */ | ||
|  |     RX_OP_RX(0, 50, 0) | ||
|  |     /* CWM has not changed. */ | ||
|  |     RX_OP_CHECK_CWM(INIT_WINDOW_SIZE) | ||
|  |     RX_OP_CHECK_SWM(50) | ||
|  | 
 | ||
|  |     /* RX, Partial retire */ | ||
|  |     RX_OP_RX(0, 60, 0) | ||
|  |     RX_OP_CHECK_SWM(60) | ||
|  |     RX_OP_RETIRE(0, 20, 50 * OSSL_TIME_MS, 0) | ||
|  |     RX_OP_CHECK_RWM(20) | ||
|  |     RX_OP_CHECK_SWM(60) | ||
|  |     RX_OP_CHECK_CWM(INIT_WINDOW_SIZE) | ||
|  |     RX_OP_CHECK_CHANGED(0, 0) | ||
|  |     RX_OP_CHECK_ERROR(0, 0) | ||
|  | 
 | ||
|  |     /* Fully retired */ | ||
|  |     RX_OP_RETIRE(0, 41, 0, 1) | ||
|  |     RX_OP_RETIRE(0, 40, 0, 0) | ||
|  |     RX_OP_CHECK_SWM(60) | ||
|  |     RX_OP_CHECK_RWM(60) | ||
|  |     RX_OP_CHECK_CWM(INIT_WINDOW_SIZE) | ||
|  |     RX_OP_CHECK_CHANGED(0, 0) | ||
|  |     RX_OP_CHECK_ERROR(0, 0) | ||
|  | 
 | ||
|  |     /* Exhaustion of window - we do not enlarge the window this epoch */ | ||
|  |     RX_OP_STEP_TIME(201 * OSSL_TIME_MS) | ||
|  |     RX_OP_RX(0, INIT_WINDOW_SIZE, 0) | ||
|  |     RX_OP_RETIRE(0, INIT_WINDOW_SIZE - 60, 50 * OSSL_TIME_MS, 0) | ||
|  |     RX_OP_CHECK_SWM(INIT_WINDOW_SIZE) | ||
|  |     RX_OP_CHECK_CHANGED(1, 0) | ||
|  |     RX_OP_CHECK_CHANGED(1, 1) | ||
|  |     RX_OP_CHECK_CHANGED(0, 0) | ||
|  |     RX_OP_CHECK_ERROR(0, 0) | ||
|  |     RX_OP_CHECK_CWM(INIT_WINDOW_SIZE * 2) | ||
|  | 
 | ||
|  |     /* Second epoch - we still do not enlarge the window this epoch */ | ||
|  |     RX_OP_RX(0, INIT_WINDOW_SIZE + 1, 0) | ||
|  |     RX_OP_STEP_TIME(201 * OSSL_TIME_MS) | ||
|  |     RX_OP_RX(0, INIT_WINDOW_SIZE * 2, 0) | ||
|  |     RX_OP_RETIRE(0, INIT_WINDOW_SIZE, 50 * OSSL_TIME_MS, 0) | ||
|  |     RX_OP_CHECK_SWM(INIT_WINDOW_SIZE * 2) | ||
|  |     RX_OP_CHECK_CHANGED(1, 0) | ||
|  |     RX_OP_CHECK_CHANGED(1, 1) | ||
|  |     RX_OP_CHECK_CHANGED(0, 0) | ||
|  |     RX_OP_CHECK_ERROR(0, 0) | ||
|  |     RX_OP_CHECK_CWM(INIT_WINDOW_SIZE * 3) | ||
|  | 
 | ||
|  |     /* Third epoch - we enlarge the window */ | ||
|  |     RX_OP_RX(0, INIT_WINDOW_SIZE * 2 + 1, 0) | ||
|  |     RX_OP_STEP_TIME(199 * OSSL_TIME_MS) | ||
|  |     RX_OP_RX(0, INIT_WINDOW_SIZE * 3, 0) | ||
|  |     RX_OP_RETIRE(0, INIT_WINDOW_SIZE, 50 * OSSL_TIME_MS, 0) | ||
|  |     RX_OP_CHECK_SWM(INIT_WINDOW_SIZE * 3) | ||
|  |     RX_OP_CHECK_CHANGED(1, 0) | ||
|  |     RX_OP_CHECK_CHANGED(1, 1) | ||
|  |     RX_OP_CHECK_CHANGED(0, 0) | ||
|  |     RX_OP_CHECK_ERROR(0, 0) | ||
|  |     RX_OP_CHECK_CWM(INIT_WINDOW_SIZE * 5) | ||
|  | 
 | ||
|  |     /* Fourth epoch - peer violates flow control */ | ||
|  |     RX_OP_RX(0, INIT_WINDOW_SIZE * 5 - 5, 0) | ||
|  |     RX_OP_STEP_TIME(250 * OSSL_TIME_MS) | ||
|  |     RX_OP_RX(0, INIT_WINDOW_SIZE * 5 + 1, 0) | ||
|  |     RX_OP_CHECK_SWM(INIT_WINDOW_SIZE * 5) | ||
|  |     RX_OP_CHECK_ERROR(QUIC_ERR_FLOW_CONTROL_ERROR, 0) | ||
|  |     RX_OP_CHECK_ERROR(QUIC_ERR_FLOW_CONTROL_ERROR, 1) | ||
|  |     RX_OP_CHECK_ERROR(0, 0) | ||
|  |     RX_OP_CHECK_CWM(INIT_WINDOW_SIZE * 5) | ||
|  |     /* 
 | ||
|  |      * No window expansion due to flow control violation; window expansion is | ||
|  |      * triggered by retirement only. | ||
|  |      */ | ||
|  |     RX_OP_CHECK_CHANGED(0, 0) | ||
|  | 
 | ||
|  |     RX_OP_END | ||
|  | }; | ||
|  | 
 | ||
|  | /* 2. Interaction between connection and stream-level flow control */ | ||
|  | static const struct rx_test_op rx_script_2[] = { | ||
|  |     RX_OP_STEP_TIME(1000 * OSSL_TIME_MS) | ||
|  |     RX_OP_INIT_CONN(INIT_WINDOW_SIZE, 10 * INIT_WINDOW_SIZE) | ||
|  |     RX_OP_INIT_STREAM(0, INIT_S_WINDOW_SIZE, 30 * INIT_S_WINDOW_SIZE) | ||
|  |     RX_OP_INIT_STREAM(1, INIT_S_WINDOW_SIZE, 30 * INIT_S_WINDOW_SIZE) | ||
|  | 
 | ||
|  |     RX_OP_RX(0, 10, 0) | ||
|  |     RX_OP_CHECK_CWM_CONN(INIT_WINDOW_SIZE) | ||
|  |     RX_OP_CHECK_CWM_STREAM(0, INIT_S_WINDOW_SIZE) | ||
|  |     RX_OP_CHECK_CWM_STREAM(1, INIT_S_WINDOW_SIZE) | ||
|  |     RX_OP_CHECK_SWM_CONN(10) | ||
|  |     RX_OP_CHECK_SWM_STREAM(0, 10) | ||
|  |     RX_OP_CHECK_SWM_STREAM(1, 0) | ||
|  |     RX_OP_CHECK_RWM_CONN(0) | ||
|  |     RX_OP_CHECK_RWM_STREAM(0, 0) | ||
|  |     RX_OP_CHECK_RWM_STREAM(1, 0) | ||
|  | 
 | ||
|  |     RX_OP_RX(1, 42, 0) | ||
|  |     RX_OP_RX(1, 42, 0) /* monotonic; equal or lower values ignored */ | ||
|  |     RX_OP_RX(1, 35, 0) | ||
|  |     RX_OP_CHECK_CWM_CONN(INIT_WINDOW_SIZE) | ||
|  |     RX_OP_CHECK_CWM_STREAM(0, INIT_S_WINDOW_SIZE) | ||
|  |     RX_OP_CHECK_CWM_STREAM(1, INIT_S_WINDOW_SIZE) | ||
|  |     RX_OP_CHECK_SWM_CONN(52) | ||
|  |     RX_OP_CHECK_SWM_STREAM(0, 10) | ||
|  |     RX_OP_CHECK_SWM_STREAM(1, 42) | ||
|  |     RX_OP_CHECK_RWM_CONN(0) | ||
|  |     RX_OP_CHECK_RWM_STREAM(0, 0) | ||
|  |     RX_OP_CHECK_RWM_STREAM(1, 0) | ||
|  | 
 | ||
|  |     RX_OP_RETIRE(0, 10, 50 * OSSL_TIME_MS, 0) | ||
|  |     RX_OP_CHECK_RWM_CONN(10) | ||
|  |     RX_OP_CHECK_RWM_STREAM(0, 10) | ||
|  |     RX_OP_CHECK_CWM_CONN(INIT_WINDOW_SIZE) | ||
|  |     RX_OP_CHECK_CWM_STREAM(0, INIT_S_WINDOW_SIZE) | ||
|  |     RX_OP_CHECK_CWM_STREAM(1, INIT_S_WINDOW_SIZE) | ||
|  | 
 | ||
|  |     RX_OP_RETIRE(1, 42, 50 * OSSL_TIME_MS, 0) | ||
|  |     RX_OP_CHECK_RWM_CONN(52) | ||
|  |     RX_OP_CHECK_RWM_STREAM(1, 42) | ||
|  |     RX_OP_CHECK_CWM_CONN(INIT_WINDOW_SIZE) | ||
|  |     RX_OP_CHECK_CWM_STREAM(0, INIT_S_WINDOW_SIZE) | ||
|  |     RX_OP_CHECK_CWM_STREAM(1, INIT_S_WINDOW_SIZE) | ||
|  | 
 | ||
|  |     RX_OP_CHECK_CHANGED_CONN(0, 0) | ||
|  | 
 | ||
|  |     /* FC limited by stream but not connection */ | ||
|  |     RX_OP_STEP_TIME(1000 * OSSL_TIME_MS) | ||
|  |     RX_OP_RX(0, INIT_S_WINDOW_SIZE, 0) | ||
|  |     RX_OP_CHECK_SWM_CONN(INIT_S_WINDOW_SIZE + 42) | ||
|  |     RX_OP_CHECK_SWM_STREAM(0, INIT_S_WINDOW_SIZE) | ||
|  |     RX_OP_CHECK_SWM_STREAM(1, 42) | ||
|  |     RX_OP_CHECK_CWM_CONN(INIT_WINDOW_SIZE) | ||
|  |     RX_OP_CHECK_CWM_STREAM(0, INIT_S_WINDOW_SIZE) | ||
|  | 
 | ||
|  |     /* We bump CWM when more than 1/4 of the window has been retired */ | ||
|  |     RX_OP_RETIRE(0, INIT_S_WINDOW_SIZE - 10, 50 * OSSL_TIME_MS, 0) | ||
|  |     RX_OP_CHECK_CWM_STREAM(0, INIT_S_WINDOW_SIZE * 2) | ||
|  |     RX_OP_CHECK_CHANGED_STREAM(0, 1, 0) | ||
|  |     RX_OP_CHECK_CHANGED_STREAM(0, 1, 1) | ||
|  |     RX_OP_CHECK_CHANGED_STREAM(0, 0, 0) | ||
|  | 
 | ||
|  |     /*
 | ||
|  |      * This is more than 1/4 of the connection window, so CWM will | ||
|  |      * be bumped here too. | ||
|  |      */ | ||
|  |     RX_OP_CHECK_CWM_CONN(INIT_S_WINDOW_SIZE + INIT_WINDOW_SIZE + 42) | ||
|  |     RX_OP_CHECK_RWM_CONN(INIT_S_WINDOW_SIZE + 42) | ||
|  |     RX_OP_CHECK_RWM_STREAM(0, INIT_S_WINDOW_SIZE) | ||
|  |     RX_OP_CHECK_RWM_STREAM(1, 42) | ||
|  |     RX_OP_CHECK_CHANGED_CONN(1, 0) | ||
|  |     RX_OP_CHECK_CHANGED_CONN(1, 1) | ||
|  |     RX_OP_CHECK_CHANGED_CONN(0, 0) | ||
|  |     RX_OP_CHECK_ERROR_CONN(0, 0) | ||
|  |     RX_OP_CHECK_ERROR_STREAM(0, 0, 0) | ||
|  |     RX_OP_CHECK_ERROR_STREAM(1, 0, 0) | ||
|  | 
 | ||
|  |     /* Test exceeding limit at stream level. */ | ||
|  |     RX_OP_RX(0, INIT_S_WINDOW_SIZE * 2 + 1, 0) | ||
|  |     RX_OP_CHECK_ERROR_STREAM(0, QUIC_ERR_FLOW_CONTROL_ERROR, 0) | ||
|  |     RX_OP_CHECK_ERROR_STREAM(0, QUIC_ERR_FLOW_CONTROL_ERROR, 1) | ||
|  |     RX_OP_CHECK_ERROR_STREAM(0, 0, 0) | ||
|  |     RX_OP_CHECK_ERROR_CONN(0, 0) /* doesn't affect conn */ | ||
|  | 
 | ||
|  |     /* Test exceeding limit at connection level. */ | ||
|  |     RX_OP_RX(0, INIT_WINDOW_SIZE * 2, 0) | ||
|  |     RX_OP_CHECK_ERROR_CONN(QUIC_ERR_FLOW_CONTROL_ERROR, 0) | ||
|  |     RX_OP_CHECK_ERROR_CONN(QUIC_ERR_FLOW_CONTROL_ERROR, 1) | ||
|  |     RX_OP_CHECK_ERROR_CONN(0, 0) | ||
|  | 
 | ||
|  |     RX_OP_END | ||
|  | }; | ||
|  | 
 | ||
|  | static const struct rx_test_op *rx_scripts[] = { | ||
|  |     rx_script_1, | ||
|  |     rx_script_2 | ||
|  | }; | ||
|  | 
 | ||
|  | static int run_rxfc_script(const struct rx_test_op *script) | ||
|  | { | ||
|  | #define MAX_STREAMS     3
 | ||
|  |     int testresult = 0; | ||
|  |     const struct rx_test_op *op = script; | ||
|  |     QUIC_RXFC conn_rxfc, stream_rxfc[MAX_STREAMS]; | ||
|  |     char stream_init_done[MAX_STREAMS] = {0}; | ||
|  |     int conn_init_done = 0; | ||
|  | 
 | ||
|  |     cur_time = ossl_time_zero(); | ||
|  | 
 | ||
|  |     for (; op->op != RX_OPC_END; ++op) { | ||
|  |         switch (op->op) { | ||
|  |             case RX_OPC_INIT_CONN: | ||
|  |                 if (!TEST_true(ossl_quic_rxfc_init(&conn_rxfc, 0, | ||
|  |                                                    op->arg0, op->arg1, | ||
|  |                                                    fake_now, NULL))) | ||
|  |                     goto err; | ||
|  | 
 | ||
|  |                 conn_init_done = 1; | ||
|  |                 break; | ||
|  | 
 | ||
|  |             case RX_OPC_INIT_STREAM: | ||
|  |                 if (!TEST_size_t_lt(op->stream_idx, OSSL_NELEM(stream_rxfc))) | ||
|  |                     goto err; | ||
|  | 
 | ||
|  |                 if (!TEST_true(ossl_quic_rxfc_init(&stream_rxfc[op->stream_idx], | ||
|  |                                                    &conn_rxfc, | ||
|  |                                                    op->arg0, op->arg1, | ||
|  |                                                    fake_now, NULL))) | ||
|  |                     goto err; | ||
|  | 
 | ||
|  |                 stream_init_done[op->stream_idx] = 1; | ||
|  |                 break; | ||
|  | 
 | ||
|  |             case RX_OPC_RX: | ||
|  |                 if (!TEST_true(conn_init_done && op->stream_idx < OSSL_NELEM(stream_rxfc) | ||
|  |                                && stream_init_done[op->stream_idx])) | ||
|  |                     goto err; | ||
|  | 
 | ||
|  |                 if (!TEST_true(ossl_quic_rxfc_on_rx_stream_frame(&stream_rxfc[op->stream_idx], | ||
|  |                                                                  op->arg0, | ||
|  |                                                                  (int)op->arg1))) | ||
|  |                     goto err; | ||
|  | 
 | ||
|  |                 break; | ||
|  | 
 | ||
|  |             case RX_OPC_RETIRE: | ||
|  |                 if (!TEST_true(conn_init_done && op->stream_idx < OSSL_NELEM(stream_rxfc) | ||
|  |                                && stream_init_done[op->stream_idx])) | ||
|  |                     goto err; | ||
|  | 
 | ||
|  |                 if (!TEST_int_eq(ossl_quic_rxfc_on_retire(&stream_rxfc[op->stream_idx], | ||
|  |                                                           op->arg0, | ||
|  |                                                           ossl_ticks2time(op->arg1)), | ||
|  |                                  !op->expect_fail)) | ||
|  |                     goto err; | ||
|  | 
 | ||
|  |                 break; | ||
|  |             case RX_OPC_CHECK_CWM_CONN: | ||
|  |                 if (!TEST_true(conn_init_done)) | ||
|  |                     goto err; | ||
|  |                 if (!TEST_uint64_t_eq(ossl_quic_rxfc_get_cwm(&conn_rxfc), | ||
|  |                                       op->arg0)) | ||
|  |                     goto err; | ||
|  |                 break; | ||
|  |             case RX_OPC_CHECK_CWM_STREAM: | ||
|  |                 if (!TEST_true(op->stream_idx < OSSL_NELEM(stream_rxfc) | ||
|  |                                && stream_init_done[op->stream_idx])) | ||
|  |                     goto err; | ||
|  |                 if (!TEST_uint64_t_eq(ossl_quic_rxfc_get_cwm(&stream_rxfc[op->stream_idx]), | ||
|  |                                       op->arg0)) | ||
|  |                     goto err; | ||
|  |                 break; | ||
|  |             case RX_OPC_CHECK_SWM_CONN: | ||
|  |                 if (!TEST_true(conn_init_done)) | ||
|  |                     goto err; | ||
|  |                 if (!TEST_uint64_t_eq(ossl_quic_rxfc_get_swm(&conn_rxfc), | ||
|  |                                       op->arg0)) | ||
|  |                     goto err; | ||
|  |                 break; | ||
|  |             case RX_OPC_CHECK_SWM_STREAM: | ||
|  |                 if (!TEST_true(op->stream_idx < OSSL_NELEM(stream_rxfc) | ||
|  |                                && stream_init_done[op->stream_idx])) | ||
|  |                     goto err; | ||
|  |                 if (!TEST_uint64_t_eq(ossl_quic_rxfc_get_swm(&stream_rxfc[op->stream_idx]), | ||
|  |                                       op->arg0)) | ||
|  |                     goto err; | ||
|  |                 break; | ||
|  |             case RX_OPC_CHECK_RWM_CONN: | ||
|  |                 if (!TEST_true(conn_init_done)) | ||
|  |                     goto err; | ||
|  |                 if (!TEST_uint64_t_eq(ossl_quic_rxfc_get_rwm(&conn_rxfc), | ||
|  |                                       op->arg0)) | ||
|  |                     goto err; | ||
|  |                 break; | ||
|  |             case RX_OPC_CHECK_RWM_STREAM: | ||
|  |                 if (!TEST_true(op->stream_idx < OSSL_NELEM(stream_rxfc) | ||
|  |                                && stream_init_done[op->stream_idx])) | ||
|  |                     goto err; | ||
|  |                 if (!TEST_uint64_t_eq(ossl_quic_rxfc_get_rwm(&stream_rxfc[op->stream_idx]), | ||
|  |                                       op->arg0)) | ||
|  |                     goto err; | ||
|  |                 break; | ||
|  |             case RX_OPC_CHECK_CHANGED_CONN: | ||
|  |                 if (!TEST_true(conn_init_done)) | ||
|  |                     goto err; | ||
|  |                 if (!TEST_int_eq(ossl_quic_rxfc_has_cwm_changed(&conn_rxfc, | ||
|  |                                                                 (int)op->arg1), | ||
|  |                                  (int)op->arg0)) | ||
|  |                     goto err; | ||
|  |                 break; | ||
|  |             case RX_OPC_CHECK_CHANGED_STREAM: | ||
|  |                 if (!TEST_true(op->stream_idx < OSSL_NELEM(stream_rxfc) | ||
|  |                                && stream_init_done[op->stream_idx])) | ||
|  |                     goto err; | ||
|  |                 if (!TEST_int_eq(ossl_quic_rxfc_has_cwm_changed(&stream_rxfc[op->stream_idx], | ||
|  |                                                                 (int)op->arg1), | ||
|  |                                  (int)op->arg0)) | ||
|  |                     goto err; | ||
|  |                 break; | ||
|  |             case RX_OPC_CHECK_ERROR_CONN: | ||
|  |                 if (!TEST_true(conn_init_done)) | ||
|  |                     goto err; | ||
|  |                 if (!TEST_int_eq(ossl_quic_rxfc_get_error(&conn_rxfc, | ||
|  |                                                           (int)op->arg1), | ||
|  |                                  (int)op->arg0)) | ||
|  |                     goto err; | ||
|  |                 break; | ||
|  |             case RX_OPC_CHECK_ERROR_STREAM: | ||
|  |                 if (!TEST_true(op->stream_idx < OSSL_NELEM(stream_rxfc) | ||
|  |                                && stream_init_done[op->stream_idx])) | ||
|  |                     goto err; | ||
|  |                 if (!TEST_int_eq(ossl_quic_rxfc_get_error(&stream_rxfc[op->stream_idx], | ||
|  |                                                           (int)op->arg1), | ||
|  |                                  (int)op->arg0)) | ||
|  |                     goto err; | ||
|  |                 break; | ||
|  |             case RX_OPC_STEP_TIME: | ||
|  |                 cur_time = ossl_time_add(cur_time, ossl_ticks2time(op->arg0)); | ||
|  |                 break; | ||
|  |             case RX_OPC_MSG: | ||
|  |                 fprintf(stderr, "# %s\n", op->msg); | ||
|  |                 break; | ||
|  |             default: | ||
|  |                 goto err; | ||
|  |         } | ||
|  |     } | ||
|  | 
 | ||
|  |     testresult = 1; | ||
|  | err: | ||
|  |     return testresult; | ||
|  | } | ||
|  | 
 | ||
|  | static int test_rxfc(int idx) | ||
|  | { | ||
|  |     return run_rxfc_script(rx_scripts[idx]); | ||
|  | } | ||
|  | 
 | ||
|  | int setup_tests(void) | ||
|  | { | ||
|  |     ADD_ALL_TESTS(test_txfc, 2); | ||
|  |     ADD_ALL_TESTS(test_rxfc, OSSL_NELEM(rx_scripts)); | ||
|  |     return 1; | ||
|  | } |