From 2aa4a82499d4becd2284cdb482213d541b8804dd Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@progress-linux.org>
Date: Sun, 28 Apr 2024 16:29:10 +0200
Subject: Adding upstream version 86.0.1.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
---
 .../rust/neqo-http3/src/control_stream_local.rs    | 71 ++++++++++++++++++++++
 1 file changed, 71 insertions(+)
 create mode 100644 third_party/rust/neqo-http3/src/control_stream_local.rs

(limited to 'third_party/rust/neqo-http3/src/control_stream_local.rs')

diff --git a/third_party/rust/neqo-http3/src/control_stream_local.rs b/third_party/rust/neqo-http3/src/control_stream_local.rs
new file mode 100644
index 0000000000..2d6e693327
--- /dev/null
+++ b/third_party/rust/neqo-http3/src/control_stream_local.rs
@@ -0,0 +1,71 @@
+// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
+// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
+// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
+// option. This file may not be copied, modified, or distributed
+// except according to those terms.
+
+use crate::hframe::HFrame;
+use crate::Res;
+use neqo_common::{qtrace, Encoder};
+use neqo_transport::{Connection, StreamType};
+use std::convert::TryFrom;
+
+pub const HTTP3_UNI_STREAM_TYPE_CONTROL: u64 = 0x0;
+
+// The local control stream, responsible for encoding frames and sending them
+#[derive(Debug)]
+pub(crate) struct ControlStreamLocal {
+    stream_id: Option<u64>,
+    buf: Vec<u8>,
+}
+
+impl ::std::fmt::Display for ControlStreamLocal {
+    fn fmt(&self, f: &mut ::std::fmt::Formatter) -> ::std::fmt::Result {
+        write!(f, "Local control stream {:?}", self.stream_id)
+    }
+}
+
+impl ControlStreamLocal {
+    pub fn new() -> Self {
+        Self {
+            stream_id: None,
+            buf: vec![u8::try_from(HTTP3_UNI_STREAM_TYPE_CONTROL).unwrap()],
+        }
+    }
+
+    /// Add a new frame that needs to be send.
+    pub fn queue_frame(&mut self, f: &HFrame) {
+        let mut enc = Encoder::default();
+        f.encode(&mut enc);
+        self.buf.append(&mut enc.into());
+    }
+
+    /// Send control data if available.
+    pub fn send(&mut self, conn: &mut Connection) -> Res<()> {
+        if let Some(stream_id) = self.stream_id {
+            if !self.buf.is_empty() {
+                qtrace!([self], "sending data.");
+                let sent = conn.stream_send(stream_id, &self.buf[..])?;
+                if sent == self.buf.len() {
+                    self.buf.clear();
+                } else {
+                    let b = self.buf.split_off(sent);
+                    self.buf = b;
+                }
+            }
+        }
+        Ok(())
+    }
+
+    /// Create a control stream.
+    pub fn create(&mut self, conn: &mut Connection) -> Res<()> {
+        qtrace!([self], "Create a control stream.");
+        self.stream_id = Some(conn.stream_create(StreamType::UniDi)?);
+        Ok(())
+    }
+
+    #[must_use]
+    pub fn stream_id(&self) -> Option<u64> {
+        self.stream_id
+    }
+}
-- 
cgit v1.2.3