Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 12 additions & 10 deletions neqo-transport/src/connection/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -2656,6 +2656,15 @@ impl Connection {
let grease_quic_bit = self.can_grease_quic_bit();
let version = self.version();

// Determine the size limit and padding for this UDP datagram.
let limit = if path.borrow().pmtud().needs_probe() {
needs_padding = true;
debug_assert!(path.borrow().pmtud().probe_size() >= profile.limit());
path.borrow().pmtud().probe_size()
} else {
profile.limit()
};

// Frames for different epochs must go in different packets, but then these
// packets can go in a single datagram
for space in PacketNumberSpace::iter() {
Expand All @@ -2668,15 +2677,6 @@ impl Connection {

let header_start = encoder.len();

// Configure the limits and padding for this packet.
let limit = if path.borrow().pmtud().needs_probe() {
needs_padding = true;
debug_assert!(path.borrow().pmtud().probe_size() >= profile.limit());
path.borrow().pmtud().probe_size() - aead_expansion
} else {
profile.limit() - aead_expansion
};

let (pt, mut builder) = Self::build_packet_header(
&path.borrow(),
epoch,
Expand All @@ -2685,7 +2685,9 @@ impl Connection {
&self.address_validation,
version,
grease_quic_bit,
limit,
// Limit the packet builder further to leave space for AEAD
// expansion added in `builder.build` below.
limit - aead_expansion,
);
let pn = Self::add_packet_number(
&mut builder,
Expand Down
Loading