QUIC CHANNEL: Add idle timeout configuration and stream count APIs

Reviewed-by: Neil Horman <nhorman@openssl.org>
Reviewed-by: Matt Caswell <matt@openssl.org>
Reviewed-by: Tomas Mraz <tomas@openssl.org>
(Merged from https://github.com/openssl/openssl/pull/23360)
This commit is contained in:
Hugo Landau 2024-01-22 13:16:25 +00:00
parent a774062466
commit ecff7ca4c1
3 changed files with 98 additions and 5 deletions

View File

@ -422,6 +422,29 @@ void ossl_quic_channel_get_diag_local_cid(QUIC_CHANNEL *ch, QUIC_CONN_ID *cid);
*/
int ossl_quic_channel_is_new_local_stream_admissible(QUIC_CHANNEL *ch, int is_uni);
/*
* Returns the number of additional streams that can currently be created based
* on flow control.
*/
uint64_t ossl_quic_channel_get_local_stream_count_avail(const QUIC_CHANNEL *ch,
int is_uni);
uint64_t ossl_quic_channel_get_remote_stream_count_avail(const QUIC_CHANNEL *ch,
int is_uni);
/*
* Returns 1 if we have generated our local transport parameters yet.
*/
int ossl_quic_channel_have_generated_transport_params(const QUIC_CHANNEL *ch);
/* Configures the idle timeout to request from peer (milliseconds, 0=no timeout). */
void ossl_quic_channel_set_max_idle_timeout_request(QUIC_CHANNEL *ch, uint64_t ms);
/* Get the configured idle timeout to request from peer. */
uint64_t ossl_quic_channel_get_max_idle_timeout_request(const QUIC_CHANNEL *ch);
/* Get the idle timeout requested by the peer. */
uint64_t ossl_quic_channel_get_max_idle_timeout_peer_request(const QUIC_CHANNEL *ch);
/* Get the idle timeout actually negotiated. */
uint64_t ossl_quic_channel_get_max_idle_timeout_actual(const QUIC_CHANNEL *ch);
# endif
#endif

View File

@ -331,11 +331,14 @@ static int ch_init(QUIC_CHANNEL *ch)
ch->rx_max_ack_delay = QUIC_DEFAULT_MAX_ACK_DELAY;
ch->rx_ack_delay_exp = QUIC_DEFAULT_ACK_DELAY_EXP;
ch->rx_active_conn_id_limit = QUIC_MIN_ACTIVE_CONN_ID_LIMIT;
ch->max_idle_timeout = QUIC_DEFAULT_IDLE_TIMEOUT;
ch->tx_enc_level = QUIC_ENC_LEVEL_INITIAL;
ch->rx_enc_level = QUIC_ENC_LEVEL_INITIAL;
ch->txku_threshold_override = UINT64_MAX;
ch->max_idle_timeout_local_req = QUIC_DEFAULT_IDLE_TIMEOUT;
ch->max_idle_timeout_remote_req = 0;
ch->max_idle_timeout = ch->max_idle_timeout_local_req;
ossl_ackm_set_tx_max_ack_delay(ch->ackm, ossl_ms2time(ch->tx_max_ack_delay));
ossl_ackm_set_rx_max_ack_delay(ch->ackm, ossl_ms2time(ch->rx_max_ack_delay));
@ -1429,6 +1432,8 @@ static int ch_on_transport_params(const unsigned char *params,
goto malformed;
}
ch->max_idle_timeout_remote_req = v;
if (v > 0 && v < ch->max_idle_timeout)
ch->max_idle_timeout = v;
@ -1691,7 +1696,7 @@ static int ch_generate_transport_params(QUIC_CHANNEL *ch)
int wpkt_valid = 0;
size_t buf_len = 0;
if (ch->local_transport_params != NULL)
if (ch->local_transport_params != NULL || ch->got_local_transport_params)
goto err;
if ((buf_mem = BUF_MEM_new()) == NULL)
@ -1722,7 +1727,7 @@ static int ch_generate_transport_params(QUIC_CHANNEL *ch)
}
if (!ossl_quic_wire_encode_transport_param_int(&wpkt, QUIC_TPARAM_MAX_IDLE_TIMEOUT,
ch->max_idle_timeout))
ch->max_idle_timeout_local_req))
goto err;
if (!ossl_quic_wire_encode_transport_param_int(&wpkt, QUIC_TPARAM_MAX_UDP_PAYLOAD_SIZE,
@ -1807,6 +1812,8 @@ static int ch_generate_transport_params(QUIC_CHANNEL *ch)
QLOG_EVENT_END()
#endif
ch->got_local_transport_params = 1;
ok = 1;
err:
if (wpkt_valid)
@ -3430,21 +3437,54 @@ static uint64_t *ch_get_local_stream_next_ordinal_ptr(QUIC_CHANNEL *ch,
: &ch->next_local_stream_ordinal_bidi;
}
static const uint64_t *ch_get_local_stream_max_ptr(const QUIC_CHANNEL *ch,
int is_uni)
{
return is_uni ? &ch->max_local_streams_uni
: &ch->max_local_streams_bidi;
}
static const QUIC_RXFC *ch_get_remote_stream_count_rxfc(const QUIC_CHANNEL *ch,
int is_uni)
{
return is_uni ? &ch->max_streams_uni_rxfc
: &ch->max_streams_bidi_rxfc;
}
int ossl_quic_channel_is_new_local_stream_admissible(QUIC_CHANNEL *ch,
int is_uni)
{
uint64_t *p_next_ordinal = ch_get_local_stream_next_ordinal_ptr(ch, is_uni);
const uint64_t *p_next_ordinal = ch_get_local_stream_next_ordinal_ptr(ch, is_uni);
return ossl_quic_stream_map_is_local_allowed_by_stream_limit(&ch->qsm,
*p_next_ordinal,
is_uni);
}
uint64_t ossl_quic_channel_get_local_stream_count_avail(const QUIC_CHANNEL *ch,
int is_uni)
{
const uint64_t *p_next_ordinal, *p_max;
p_next_ordinal = ch_get_local_stream_next_ordinal_ptr((QUIC_CHANNEL *)ch,
is_uni);
p_max = ch_get_local_stream_max_ptr(ch, is_uni);
return *p_max - *p_next_ordinal;
}
uint64_t ossl_quic_channel_get_remote_stream_count_avail(const QUIC_CHANNEL *ch,
int is_uni)
{
return ossl_quic_rxfc_get_credit(ch_get_remote_stream_count_rxfc(ch, is_uni));
}
QUIC_STREAM *ossl_quic_channel_new_stream_local(QUIC_CHANNEL *ch, int is_uni)
{
QUIC_STREAM *qs;
int type;
uint64_t stream_id, *p_next_ordinal;
uint64_t stream_id;
uint64_t *p_next_ordinal;
type = ch->is_server ? QUIC_STREAM_INITIATOR_SERVER
: QUIC_STREAM_INITIATOR_CLIENT;
@ -3616,3 +3656,27 @@ void ossl_quic_channel_get_diag_local_cid(QUIC_CHANNEL *ch, QUIC_CONN_ID *cid)
{
*cid = ch->cur_local_cid;
}
int ossl_quic_channel_have_generated_transport_params(const QUIC_CHANNEL *ch)
{
return ch->got_local_transport_params;
}
void ossl_quic_channel_set_max_idle_timeout_request(QUIC_CHANNEL *ch, uint64_t ms)
{
ch->max_idle_timeout_local_req = ms;
}
uint64_t ossl_quic_channel_get_max_idle_timeout_request(const QUIC_CHANNEL *ch)
{
return ch->max_idle_timeout_local_req;
}
uint64_t ossl_quic_channel_get_max_idle_timeout_peer_request(const QUIC_CHANNEL *ch)
{
return ch->max_idle_timeout_remote_req;
}
uint64_t ossl_quic_channel_get_max_idle_timeout_actual(const QUIC_CHANNEL *ch)
{
return ch->max_idle_timeout;
}

View File

@ -155,6 +155,10 @@ struct quic_channel_st {
uint64_t max_local_streams_bidi;
uint64_t max_local_streams_uni;
/* The idle timeout values we and our peer requested. */
uint64_t max_idle_timeout_local_req;
uint64_t max_idle_timeout_remote_req;
/* The negotiated maximum idle timeout in milliseconds. */
uint64_t max_idle_timeout;
@ -288,6 +292,8 @@ struct quic_channel_st {
/* We have received transport parameters from the peer. */
unsigned int got_remote_transport_params : 1;
/* We have generated our local transport parameters. */
unsigned int got_local_transport_params : 1;
/*
* This monotonically transitions to 1 once the TLS state machine is