@@ -161,7 +161,7 @@ impl Recv {
|
||||
counts: &mut Counts,
|
||||
) -> Result<(), RecvHeaderBlockError<Option<frame::Headers>>> {
|
||||
tracing::trace!("opening stream; init_window={}", self.init_window_sz);
|
||||
let is_initial = stream.state.recv_open(frame.is_end_stream())?;
|
||||
let is_initial = stream.state.recv_open(&frame)?;
|
||||
|
||||
if is_initial {
|
||||
// TODO: be smarter about this logic
|
||||
@@ -226,15 +226,17 @@ impl Recv {
|
||||
|
||||
let stream_id = frame.stream_id();
|
||||
let (pseudo, fields) = frame.into_parts();
|
||||
let message = counts
|
||||
.peer()
|
||||
.convert_poll_message(pseudo, fields, stream_id)?;
|
||||
if !pseudo.is_informational() {
|
||||
let message = counts
|
||||
.peer()
|
||||
.convert_poll_message(pseudo, fields, stream_id)?;
|
||||
|
||||
// Push the frame onto the stream's recv buffer
|
||||
stream
|
||||
.pending_recv
|
||||
.push_back(&mut self.buffer, Event::Headers(message));
|
||||
stream.notify_recv();
|
||||
// Push the frame onto the stream's recv buffer
|
||||
stream
|
||||
.pending_recv
|
||||
.push_back(&mut self.buffer, Event::Headers(message));
|
||||
stream.notify_recv();
|
||||
}
|
||||
|
||||
// Only servers can receive a headers frame that initiates the stream.
|
||||
// This is verified in `Streams` before calling this function.
|
||||
|
||||
@@ -2,7 +2,7 @@ use std::io;
|
||||
|
||||
use crate::codec::UserError::*;
|
||||
use crate::codec::{RecvError, UserError};
|
||||
use crate::frame::Reason;
|
||||
use crate::frame::{self, Reason};
|
||||
use crate::proto::{self, PollReset};
|
||||
|
||||
use self::Inner::*;
|
||||
@@ -132,10 +132,13 @@ impl State {
|
||||
|
||||
/// Opens the receive-half of the stream when a HEADERS frame is received.
|
||||
///
|
||||
/// is_informational: whether received a 1xx status code
|
||||
///
|
||||
/// Returns true if this transitions the state to Open.
|
||||
pub fn recv_open(&mut self, eos: bool) -> Result<bool, RecvError> {
|
||||
pub fn recv_open(&mut self, frame: &frame::Headers) -> Result<bool, RecvError> {
|
||||
let remote = Streaming;
|
||||
let mut initial = false;
|
||||
let eos = frame.is_end_stream();
|
||||
|
||||
self.inner = match self.inner {
|
||||
Idle => {
|
||||
@@ -172,6 +175,9 @@ impl State {
|
||||
HalfClosedLocal(AwaitingHeaders) => {
|
||||
if eos {
|
||||
Closed(Cause::EndStream)
|
||||
} else if frame.is_informational() {
|
||||
tracing::trace!("skipping 1xx response headers");
|
||||
HalfClosedLocal(AwaitingHeaders)
|
||||
} else {
|
||||
HalfClosedLocal(remote)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user