mirror of
https://github.com/openssl/openssl.git
synced 2024-12-03 05:41:46 +08:00
ff3a26b24f
Reviewed-by: Tomas Mraz <tomas@openssl.org> Reviewed-by: Matt Caswell <matt@openssl.org> (Merged from https://github.com/openssl/openssl/pull/22674)
429 lines
17 KiB
C
429 lines
17 KiB
C
/*
|
|
* Copyright 2022-2023 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
|
|
*/
|
|
|
|
#ifndef OSSL_INTERNAL_QUIC_STREAM_H
|
|
# define OSSL_INTERNAL_QUIC_STREAM_H
|
|
# pragma once
|
|
|
|
#include "internal/e_os.h"
|
|
#include "internal/time.h"
|
|
#include "internal/quic_types.h"
|
|
#include "internal/quic_predef.h"
|
|
#include "internal/quic_wire.h"
|
|
#include "internal/quic_record_tx.h"
|
|
#include "internal/quic_record_rx.h"
|
|
#include "internal/quic_fc.h"
|
|
#include "internal/quic_statm.h"
|
|
|
|
# ifndef OPENSSL_NO_QUIC
|
|
|
|
/*
|
|
* QUIC Send Stream
|
|
* ================
|
|
*
|
|
* The QUIC Send Stream Manager (QUIC_SSTREAM) is responsible for:
|
|
*
|
|
* - accepting octet strings of stream data;
|
|
*
|
|
* - generating corresponding STREAM frames;
|
|
*
|
|
* - receiving notifications of lost frames, in order to generate new STREAM
|
|
* frames for the lost data;
|
|
*
|
|
* - receiving notifications of acknowledged frames, in order to internally
|
|
* reuse memory used to store acknowledged stream data;
|
|
*
|
|
* - informing the caller of how much more stream data it can accept into
|
|
* its internal buffers, so as to ensure that the amount of unacknowledged
|
|
* data which can be written to a stream is not infinite and to allow the
|
|
* caller to manifest backpressure conditions to the user.
|
|
*
|
|
* The QUIC_SSTREAM is instantiated once for every stream with a send component
|
|
* (i.e., for a unidirectional send stream or for the send component of a
|
|
* bidirectional stream).
|
|
*
|
|
* Note: The terms 'TX' and 'RX' are used when referring to frames, packets and
|
|
* datagrams. The terms 'send' and 'receive' are used when referring to the
|
|
* stream abstraction. Applications send; we transmit.
|
|
*/
|
|
|
|
/*
|
|
* Instantiates a new QUIC_SSTREAM. init_buf_size specifies the initial size of
|
|
* the stream data buffer in bytes, which must be positive.
|
|
*/
|
|
QUIC_SSTREAM *ossl_quic_sstream_new(size_t init_buf_size);
|
|
|
|
/*
|
|
* Frees a QUIC_SSTREAM and associated stream data storage.
|
|
*
|
|
* Any iovecs returned by ossl_quic_sstream_get_stream_frame cease to be valid after
|
|
* calling this function.
|
|
*/
|
|
void ossl_quic_sstream_free(QUIC_SSTREAM *qss);
|
|
|
|
/*
|
|
* (For TX packetizer use.) Retrieves information about application stream data
|
|
* which is ready for transmission.
|
|
*
|
|
* *hdr is filled with the logical offset, maximum possible length of stream
|
|
* data which can be transmitted, and a pointer to the stream data to be
|
|
* transmitted. is_fin is set to 1 if hdr->offset + hdr->len is the final size
|
|
* of the stream and 0 otherwise. hdr->stream_id is not set; the caller must set
|
|
* it.
|
|
*
|
|
* The caller is not obligated to send all of the data. If the caller does not
|
|
* send all of the data, the caller must reduce hdr->len before serializing the
|
|
* header structure and must ensure that hdr->is_fin is cleared.
|
|
*
|
|
* hdr->has_explicit_len is always set. It is the caller's responsibility to
|
|
* clear this if it wants to use the optimization of omitting the length field,
|
|
* as only the caller can know when this optimization can be performed.
|
|
*
|
|
* *num_iov must be set to the size of the iov array at call time. When this
|
|
* function returns successfully, it is updated to the number of iov entries
|
|
* which have been written.
|
|
*
|
|
* The stream data may be split across up to two IOVs due to internal ring
|
|
* buffer organisation. The sum of the lengths of the IOVs and the value written
|
|
* to hdr->len will always match. If the caller decides to send less than
|
|
* hdr->len of stream data, it must adjust the IOVs accordingly. This may be
|
|
* done by updating hdr->len and then calling the utility function
|
|
* ossl_quic_sstream_adjust_iov().
|
|
*
|
|
* After committing one or more bytes returned by ossl_quic_sstream_get_stream_frame to a
|
|
* packet, call ossl_quic_sstream_mark_transmitted with the inclusive range of logical
|
|
* byte numbers of the transmitted bytes (i.e., hdr->offset, hdr->offset +
|
|
* hdr->len - 1). If you do not call ossl_quic_sstream_mark_transmitted, the next call to
|
|
* ossl_quic_sstream_get_stream_frame will return the same data (or potentially the same
|
|
* and more, if more data has been appended by the application).
|
|
*
|
|
* It is the caller's responsibility to clamp the length of data which this
|
|
* function indicates is available according to other concerns, such as
|
|
* stream-level flow control, connection-level flow control, or the applicable
|
|
* maximum datagram payload length (MDPL) for a packet under construction.
|
|
*
|
|
* The skip argument can usually be given as zero. If it is non-zero, this
|
|
* function outputs a range which would be output if it were called again after
|
|
* calling ossl_quic_sstream_mark_transmitted() with the returned range, repeated 'skip'
|
|
* times, and so on. This may be useful for callers which wish to enumerate
|
|
* available stream frames and batch their calls to ossl_quic_sstream_mark_transmitted at
|
|
* a later time.
|
|
*
|
|
* On success, this function will never write *num_iov with a value other than
|
|
* 0, 1 or 2. A *num_iov value of 0 can only occurs when hdr->is_fin is set (for
|
|
* example, when a stream is closed after all existing data has been sent, and
|
|
* without sending any more data); otherwise the function returns 0 as there is
|
|
* nothing useful to report.
|
|
*
|
|
* Returns 1 on success and 0 if there is no stream data available for
|
|
* transmission, or on other error (such as if the caller provides fewer
|
|
* than two IOVs.)
|
|
*/
|
|
int ossl_quic_sstream_get_stream_frame(QUIC_SSTREAM *qss,
|
|
size_t skip,
|
|
OSSL_QUIC_FRAME_STREAM *hdr,
|
|
OSSL_QTX_IOVEC *iov,
|
|
size_t *num_iov);
|
|
|
|
/*
|
|
* Returns 1 if there is data pending transmission. Equivalent to calling
|
|
* ossl_quic_sstream_get_stream_frame and seeing if it succeeds.
|
|
*/
|
|
int ossl_quic_sstream_has_pending(QUIC_SSTREAM *qss);
|
|
|
|
/*
|
|
* Returns the current size of the stream; i.e., the number of bytes which have
|
|
* been appended to the stream so far.
|
|
*/
|
|
uint64_t ossl_quic_sstream_get_cur_size(QUIC_SSTREAM *qss);
|
|
|
|
/*
|
|
* (For TX packetizer use.) Marks a logical range of the send stream as having
|
|
* been transmitted.
|
|
*
|
|
* 0 denotes the first byte ever sent on the stream. The start and end values
|
|
* are both inclusive, therefore all calls to this function always mark at least
|
|
* one byte as being transmitted; if no bytes have been transmitted, do not call
|
|
* this function.
|
|
*
|
|
* If the STREAM frame sent had the FIN bit set, you must also call
|
|
* ossl_quic_sstream_mark_transmitted_fin() after calling this function.
|
|
*
|
|
* If you sent a zero-length STREAM frame with the FIN bit set, you need only
|
|
* call ossl_quic_sstream_mark_transmitted_fin() and must not call this function.
|
|
*
|
|
* Returns 1 on success and 0 on error (e.g. if end < start).
|
|
*/
|
|
int ossl_quic_sstream_mark_transmitted(QUIC_SSTREAM *qss,
|
|
uint64_t start,
|
|
uint64_t end);
|
|
|
|
/*
|
|
* (For TX packetizer use.) Marks a STREAM frame with the FIN bit set as having
|
|
* been transmitted. final_size is the final size of the stream (i.e., the value
|
|
* offset + len of the transmitted STREAM frame).
|
|
*
|
|
* This function fails returning 0 if ossl_quic_sstream_fin() has not been called or if
|
|
* final_size is not correct. The final_size argument is not strictly needed by
|
|
* the QUIC_SSTREAM but is required as a sanity check.
|
|
*/
|
|
int ossl_quic_sstream_mark_transmitted_fin(QUIC_SSTREAM *qss,
|
|
uint64_t final_size);
|
|
|
|
/*
|
|
* (RX/ACKM use.) Marks a logical range of the send stream as having been lost.
|
|
* The send stream will return the lost data for retransmission on a future call
|
|
* to ossl_quic_sstream_get_stream_frame. The start and end values denote logical byte
|
|
* numbers and are inclusive.
|
|
*
|
|
* If the lost frame had the FIN bit set, you must also call
|
|
* ossl_quic_sstream_mark_lost_fin() after calling this function.
|
|
*
|
|
* Returns 1 on success and 0 on error (e.g. if end < start).
|
|
*/
|
|
int ossl_quic_sstream_mark_lost(QUIC_SSTREAM *qss,
|
|
uint64_t start,
|
|
uint64_t end);
|
|
|
|
/*
|
|
* (RX/ACKM use.) Informs the QUIC_SSTREAM that a STREAM frame with the FIN bit
|
|
* set was lost.
|
|
*
|
|
* Returns 1 on success and 0 on error.
|
|
*/
|
|
int ossl_quic_sstream_mark_lost_fin(QUIC_SSTREAM *qss);
|
|
|
|
/*
|
|
* (RX/ACKM use.) Marks a logical range of the send stream as having been
|
|
* acknowledged, meaning that the storage for the data in that range of the
|
|
* stream can be now recycled and neither that logical range of the stream nor
|
|
* any subset of it can be retransmitted again. The start and end values are
|
|
* inclusive.
|
|
*
|
|
* If the acknowledged frame had the FIN bit set, you must also call
|
|
* ossl_quic_sstream_mark_acked_fin() after calling this function.
|
|
*
|
|
* Returns 1 on success and 0 on error (e.g. if end < start).
|
|
*/
|
|
int ossl_quic_sstream_mark_acked(QUIC_SSTREAM *qss,
|
|
uint64_t start,
|
|
uint64_t end);
|
|
|
|
/*
|
|
* (RX/ACKM use.) Informs the QUIC_SSTREAM that a STREAM frame with the FIN bit
|
|
* set was acknowledged.
|
|
*
|
|
* Returns 1 on success and 0 on error.
|
|
*/
|
|
int ossl_quic_sstream_mark_acked_fin(QUIC_SSTREAM *qss);
|
|
|
|
/*
|
|
* (Front end use.) Appends user data to the stream. The data is copied into the
|
|
* stream. The amount of data consumed from buf is written to *consumed on
|
|
* success (short writes are possible). The amount of data which can be written
|
|
* can be determined in advance by calling the ossl_quic_sstream_get_buffer_avail()
|
|
* function; data is copied into an internal ring buffer of finite size.
|
|
*
|
|
* If the buffer is full, this should be materialised as a backpressure
|
|
* condition by the front end. This is not considered a failure condition;
|
|
* *consumed is written as 0 and the function returns 1.
|
|
*
|
|
* Returns 1 on success or 0 on failure.
|
|
*/
|
|
int ossl_quic_sstream_append(QUIC_SSTREAM *qss,
|
|
const unsigned char *buf,
|
|
size_t buf_len,
|
|
size_t *consumed);
|
|
|
|
/*
|
|
* Marks a stream as finished. ossl_quic_sstream_append() may not be called anymore
|
|
* after calling this.
|
|
*/
|
|
void ossl_quic_sstream_fin(QUIC_SSTREAM *qss);
|
|
|
|
/*
|
|
* If the stream has had ossl_quic_sstream_fin() called, returns 1 and writes
|
|
* the final size to *final_size. Otherwise, returns 0.
|
|
*/
|
|
int ossl_quic_sstream_get_final_size(QUIC_SSTREAM *qss, uint64_t *final_size);
|
|
|
|
/*
|
|
* Returns 1 iff all bytes (and any FIN, if any) which have been appended to the
|
|
* QUIC_SSTREAM so far, and any FIN (if any), have been both sent and acked.
|
|
*/
|
|
int ossl_quic_sstream_is_totally_acked(QUIC_SSTREAM *qss);
|
|
|
|
/*
|
|
* Resizes the internal ring buffer. All stream data is preserved safely.
|
|
*
|
|
* This can be used to expand or contract the ring buffer, but not to contract
|
|
* the ring buffer below the amount of stream data currently stored in it.
|
|
* Returns 1 on success and 0 on failure.
|
|
*
|
|
* IMPORTANT: Any buffers referenced by iovecs output by
|
|
* ossl_quic_sstream_get_stream_frame() cease to be valid after calling this function.
|
|
*/
|
|
int ossl_quic_sstream_set_buffer_size(QUIC_SSTREAM *qss, size_t num_bytes);
|
|
|
|
/*
|
|
* Gets the internal ring buffer size in bytes.
|
|
*/
|
|
size_t ossl_quic_sstream_get_buffer_size(QUIC_SSTREAM *qss);
|
|
|
|
/*
|
|
* Gets the number of bytes used in the internal ring buffer.
|
|
*/
|
|
size_t ossl_quic_sstream_get_buffer_used(QUIC_SSTREAM *qss);
|
|
|
|
/*
|
|
* Gets the number of bytes free in the internal ring buffer.
|
|
*/
|
|
size_t ossl_quic_sstream_get_buffer_avail(QUIC_SSTREAM *qss);
|
|
|
|
/*
|
|
* Utility function to ensure the length of an array of iovecs matches the
|
|
* length given as len. Trailing iovecs have their length values reduced or set
|
|
* to 0 as necessary.
|
|
*/
|
|
void ossl_quic_sstream_adjust_iov(size_t len,
|
|
OSSL_QTX_IOVEC *iov,
|
|
size_t num_iov);
|
|
|
|
/*
|
|
* Sets flag to cleanse the buffered data when it is acked.
|
|
*/
|
|
void ossl_quic_sstream_set_cleanse(QUIC_SSTREAM *qss, int cleanse);
|
|
|
|
/*
|
|
* QUIC Receive Stream Manager
|
|
* ===========================
|
|
*
|
|
* The QUIC Receive Stream Manager (QUIC_RSTREAM) is responsible for
|
|
* storing the received stream data frames until the application
|
|
* is able to read the data.
|
|
*
|
|
* The QUIC_RSTREAM is instantiated once for every stream that can receive data.
|
|
* (i.e., for a unidirectional receiving stream or for the receiving component
|
|
* of a bidirectional stream).
|
|
*/
|
|
|
|
/*
|
|
* Create a new instance of QUIC_RSTREAM with pointers to the flow
|
|
* controller and statistics module. They can be NULL for unit testing.
|
|
* If they are non-NULL, the `rxfc` is called when receive stream data
|
|
* is read by application. `statm` is queried for current rtt.
|
|
* `rbuf_size` is the initial size of the ring buffer to be used
|
|
* when ossl_quic_rstream_move_to_rbuf() is called.
|
|
*/
|
|
QUIC_RSTREAM *ossl_quic_rstream_new(QUIC_RXFC *rxfc,
|
|
OSSL_STATM *statm, size_t rbuf_size);
|
|
|
|
/*
|
|
* Frees a QUIC_RSTREAM and any associated storage.
|
|
*/
|
|
void ossl_quic_rstream_free(QUIC_RSTREAM *qrs);
|
|
|
|
/*
|
|
* Adds received stream frame data to `qrs`. The `pkt_wrap` refcount is
|
|
* incremented if the `data` is queued directly without copying.
|
|
* It can be NULL for unit-testing purposes, i.e. if `data` is static or
|
|
* never released before calling ossl_quic_rstream_free().
|
|
* The `offset` is the absolute offset of the data in the stream.
|
|
* `data_len` can be 0 - can be useful for indicating `fin` for empty stream.
|
|
* Or to indicate `fin` without any further data added to the stream.
|
|
*/
|
|
|
|
int ossl_quic_rstream_queue_data(QUIC_RSTREAM *qrs, OSSL_QRX_PKT *pkt,
|
|
uint64_t offset,
|
|
const unsigned char *data, uint64_t data_len,
|
|
int fin);
|
|
|
|
/*
|
|
* Copies the data from the stream storage to buffer `buf` of size `size`.
|
|
* `readbytes` is set to the number of bytes actually copied.
|
|
* `fin` is set to 1 if all the data from the stream were read so the
|
|
* stream is finished. It is set to 0 otherwise.
|
|
*/
|
|
int ossl_quic_rstream_read(QUIC_RSTREAM *qrs, unsigned char *buf, size_t size,
|
|
size_t *readbytes, int *fin);
|
|
|
|
/*
|
|
* Peeks at the data in the stream storage. It copies them to buffer `buf`
|
|
* of size `size` and sets `readbytes` to the number of bytes actually copied.
|
|
* `fin` is set to 1 if the copied data reach end of the stream.
|
|
* It is set to 0 otherwise.
|
|
*/
|
|
int ossl_quic_rstream_peek(QUIC_RSTREAM *qrs, unsigned char *buf, size_t size,
|
|
size_t *readbytes, int *fin);
|
|
|
|
/*
|
|
* Returns the size of the data available for reading. `fin` is set to 1 if
|
|
* after reading all the available data the stream will be finished,
|
|
* set to 0 otherwise.
|
|
*/
|
|
int ossl_quic_rstream_available(QUIC_RSTREAM *qrs, size_t *avail, int *fin);
|
|
|
|
/*
|
|
* Sets *record to the beginning of the first readable stream data chunk and
|
|
* *reclen to the size of the chunk. *fin is set to 1 if the end of the
|
|
* chunk is the last of the stream data chunks.
|
|
* If there is no record available *record is set to NULL and *rec_len to 0;
|
|
* ossl_quic_rstream_release_record() should not be called in that case.
|
|
* Returns 1 on success (including calls if no record is available, or
|
|
* after end of the stream - in that case *fin will be set to 1 and
|
|
* *rec_len to 0), 0 on error.
|
|
* It is an error to call ossl_quic_rstream_get_record() multiple times
|
|
* without calling ossl_quic_rstream_release_record() in between.
|
|
*/
|
|
int ossl_quic_rstream_get_record(QUIC_RSTREAM *qrs,
|
|
const unsigned char **record, size_t *rec_len,
|
|
int *fin);
|
|
|
|
/*
|
|
* Releases (possibly partially) the record returned by
|
|
* previous ossl_quic_rstream_get_record() call.
|
|
* read_len between previously returned *rec_len and SIZE_MAX indicates
|
|
* release of the whole record. Otherwise only part of the record is
|
|
* released. The remaining part of the record is unlocked, another
|
|
* call to ossl_quic_rstream_get_record() is needed to obtain further
|
|
* stream data.
|
|
* Returns 1 on success, 0 on error.
|
|
* It is an error to call ossl_quic_rstream_release_record() multiple
|
|
* times without calling ossl_quic_rstream_get_record() in between.
|
|
*/
|
|
int ossl_quic_rstream_release_record(QUIC_RSTREAM *qrs, size_t read_len);
|
|
|
|
/*
|
|
* Moves received frame data from decrypted packets to ring buffer.
|
|
* This should be called when there are too many decrypted packets allocated.
|
|
* Returns 1 on success, 0 when it was not possible to release all
|
|
* referenced packets due to an insufficient size of the ring buffer.
|
|
* Exception is the packet from the record returned previously by
|
|
* ossl_quic_rstream_get_record() - that one will be always skipped.
|
|
*/
|
|
int ossl_quic_rstream_move_to_rbuf(QUIC_RSTREAM *qrs);
|
|
|
|
/*
|
|
* Resizes the internal ring buffer to a new `rbuf_size` size.
|
|
* Returns 1 on success, 0 on error.
|
|
* Possible error conditions are an allocation failure, trying to resize
|
|
* the ring buffer when ossl_quic_rstream_get_record() was called and
|
|
* not yet released, or trying to resize the ring buffer to a smaller size
|
|
* than currently occupied.
|
|
*/
|
|
int ossl_quic_rstream_resize_rbuf(QUIC_RSTREAM *qrs, size_t rbuf_size);
|
|
|
|
/*
|
|
* Sets flag to cleanse the buffered data when user reads it.
|
|
*/
|
|
void ossl_quic_rstream_set_cleanse(QUIC_RSTREAM *qrs, int cleanse);
|
|
# endif
|
|
|
|
#endif
|