State::send_open:Add final_response indicating if more headers will…

… come
hyperium · Feb 6, 2022 · 9dde7e1 · 9dde7e1
1 parent 657bcb4
commit 9dde7e1
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/src/proto/streams/send.rs b/src/proto/streams/send.rs
@@ -138,10 +138,11 @@ impl Send {
 
         Self::check_headers(frame.fields())?;
 
+        let final_response = !frame.is_informational();
         let end_stream = frame.is_end_stream();
 
         // Update the state
-        stream.state.send_open(end_stream)?;
+        stream.state.send_open(final_response, end_stream)?;
 
         if counts.peer().is_local_init(frame.stream_id()) {
             // If we're waiting on a PushPromise anyway

diff --git a/src/proto/streams/state.rs b/src/proto/streams/state.rs
@@ -87,8 +87,12 @@ enum Cause {
 
 impl State {
     /// Opens the send-half of a stream if it is not already open.
-    pub fn send_open(&mut self, eos: bool) -> Result<(), UserError> {
-        let local = Streaming;
+    pub fn send_open(&mut self, final_response: bool, eos: bool) -> Result<(), UserError> {
+        let local = if final_response {
+            Streaming
+        } else {
+            AwaitingHeaders
+        };
 
         self.inner = match self.inner {
             Idle => {