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

… come
hyperium · Oct 13, 2024 · 18d8dbe · 18d8dbe
1 parent 6591cea
commit 18d8dbe
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)?;
 
         let mut pending_open = false;
         if counts.peer().is_local_init(frame.stream_id()) && !stream.is_pending_push {

diff --git a/src/proto/streams/state.rs b/src/proto/streams/state.rs
@@ -88,8 +88,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 => {