Skip to content

Commit

Permalink
proto: remove panic-on-drop from Chunks
Browse files Browse the repository at this point in the history
  • Loading branch information
abonander committed Sep 7, 2024
1 parent 29d8eb2 commit 092e554
Showing 1 changed file with 23 additions and 9 deletions.
32 changes: 23 additions & 9 deletions quinn-proto/src/connection/streams/recv.rs
Original file line number Diff line number Diff line change
Expand Up @@ -220,7 +220,17 @@ impl Recv {
}
}

/// Chunks
/// Chunks returned from [`RecvStream::read()`][crate::RecvStream::read].
///
/// ### Note: Finalization Needed
/// Bytes read from the stream are not released from the congestion window until
/// either [`Self::finalize()`] is called, or this type is dropped.
///
/// It is recommended that you call [`Self::finalize()`] because it returns a flag
/// telling you whether reading from the stream has resulted in the need to transmit a packet.
///
/// If this type is leaked, the stream will remain blocked on the remote peer until
/// another read from the stream is done.
pub struct Chunks<'a> {
id: StreamId,
ordered: bool,
Expand Down Expand Up @@ -302,17 +312,21 @@ impl<'a> Chunks<'a> {
}
}

/// Finalize
/// Mark the read data as consumed from the stream.
///
/// The number of read bytes will be released from the congestion window,
/// allowing the remote peer to send more data if it was previously blocked.
///
/// If this returns `ShouldTransmit(true)`, a packet needs to be sent to the peer
/// informing them that the stream is unblocked. This means that you should call
/// [`Connection::poll_transmit()`][crate::Connection::poll_transmit]
/// and send the returned packet as soon as is reasonable to unblock the remote peer.
pub fn finalize(mut self) -> ShouldTransmit {
self.finalize_inner(false)
self.finalize_inner()
}

fn finalize_inner(&mut self, drop: bool) -> ShouldTransmit {
fn finalize_inner(&mut self) -> ShouldTransmit {
let state = mem::replace(&mut self.state, ChunksState::Finalized);
debug_assert!(
!drop || matches!(state, ChunksState::Finalized),
"finalize must be called before drop"
);
if let ChunksState::Finalized = state {
// Noop on repeated calls
return ShouldTransmit(false);
Expand Down Expand Up @@ -344,7 +358,7 @@ impl<'a> Chunks<'a> {

impl<'a> Drop for Chunks<'a> {
fn drop(&mut self) {
let _ = self.finalize_inner(true);
let _ = self.finalize_inner();
}
}

Expand Down

0 comments on commit 092e554

Please sign in to comment.