mirror of
https://gitlab.freedesktop.org/gstreamer/gst-plugins-rs.git
synced 2024-11-21 19:11:02 +00:00
net/quinn: Add quinnquicmux to support stream multiplexing
Part-of: <https://gitlab.freedesktop.org/gstreamer/gst-plugins-rs/-/merge_requests/1634>
This commit is contained in:
Sanchayan Maity
GStreamer Marge Bot
parent
0eb3f52356
commit
1cc2682b55
3 changed files with 416 additions and 0 deletions
|
|
@ -21,6 +21,7 @@ use gst::glib;
|
|||
use gst::prelude::*;
|
||||
mod common;
|
||||
pub mod quinnquicmeta;
|
||||
mod quinnquicmux;
|
||||
pub mod quinnquicquery;
|
||||
mod quinnquicsink;
|
||||
mod quinnquicsrc;
|
||||
|
|
@ -31,6 +32,7 @@ fn plugin_init(plugin: &gst::Plugin) -> Result<(), glib::BoolError> {
|
|||
{
|
||||
common::QuinnQuicRole::static_type().mark_as_plugin_api(gst::PluginAPIFlags::empty());
|
||||
}
|
||||
quinnquicmux::register(plugin)?;
|
||||
quinnquicsink::register(plugin)?;
|
||||
quinnquicsrc::register(plugin)?;
|
||||
|
||||
|
|
|
|||
379
net/quinn/src/quinnquicmux/imp.rs
Normal file
379
net/quinn/src/quinnquicmux/imp.rs
Normal file
|
|
@ -0,0 +1,379 @@
|
|||
// Copyright (C) 2024, Asymptotic Inc.
|
||||
// Author: Sanchayan Maity <sanchayan@asymptotic.io>
|
||||
//
|
||||
// This Source Code Form is subject to the terms of the Mozilla Public License, v2.0.
|
||||
// If a copy of the MPL was not distributed with this file, You can obtain one at
|
||||
// <https://mozilla.org/MPL/2.0/>.
|
||||
//
|
||||
// SPDX-License-Identifier: MPL-2.0
|
||||
|
||||
use crate::quinnquicmeta::QuinnQuicMeta;
|
||||
use crate::quinnquicquery::*;
|
||||
use gst::{glib, prelude::*, subclass::prelude::*};
|
||||
use gst_base::prelude::*;
|
||||
use gst_base::subclass::prelude::*;
|
||||
use itertools::Itertools;
|
||||
use std::collections::HashMap;
|
||||
use std::io::Read;
|
||||
use std::sync::{LazyLock, Mutex};
|
||||
|
||||
const DEFAULT_STREAM_PRIORITY: i32 = 0;
|
||||
|
||||
static CAT: LazyLock<gst::DebugCategory> = LazyLock::new(|| {
|
||||
gst::DebugCategory::new(
|
||||
"quinnquicmux",
|
||||
gst::DebugColorFlags::empty(),
|
||||
Some("Quinn QUIC Mux"),
|
||||
)
|
||||
});
|
||||
|
||||
#[derive(Default)]
|
||||
struct QuinnQuicMuxPadSettings {
|
||||
priority: i32,
|
||||
}
|
||||
|
||||
#[derive(Default)]
|
||||
pub(crate) struct QuinnQuicMuxPad {
|
||||
settings: Mutex<QuinnQuicMuxPadSettings>,
|
||||
}
|
||||
|
||||
#[glib::object_subclass]
|
||||
impl ObjectSubclass for QuinnQuicMuxPad {
|
||||
const NAME: &'static str = "QuinnQuicMuxPad";
|
||||
type Type = super::QuinnQuicMuxPad;
|
||||
type ParentType = gst::Pad;
|
||||
}
|
||||
|
||||
impl ObjectImpl for QuinnQuicMuxPad {
|
||||
fn properties() -> &'static [glib::ParamSpec] {
|
||||
static PROPERTIES: LazyLock<Vec<glib::ParamSpec>> = LazyLock::new(|| {
|
||||
vec![glib::ParamSpecInt::builder("priority")
|
||||
.nick("Priority of the stream")
|
||||
.blurb("Priority of the stream")
|
||||
.default_value(DEFAULT_STREAM_PRIORITY)
|
||||
.readwrite()
|
||||
.build()]
|
||||
});
|
||||
|
||||
PROPERTIES.as_ref()
|
||||
}
|
||||
|
||||
fn set_property(&self, _id: usize, value: &glib::Value, pspec: &glib::ParamSpec) {
|
||||
match pspec.name() {
|
||||
"priority" => {
|
||||
let mut settings = self.settings.lock().unwrap();
|
||||
settings.priority = value.get::<i32>().expect("type checked upstream");
|
||||
}
|
||||
_ => unimplemented!(),
|
||||
}
|
||||
}
|
||||
|
||||
fn property(&self, _id: usize, pspec: &glib::ParamSpec) -> glib::Value {
|
||||
match pspec.name() {
|
||||
"priority" => {
|
||||
let settings = self.settings.lock().unwrap();
|
||||
settings.priority.to_value()
|
||||
}
|
||||
_ => unimplemented!(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl GstObjectImpl for QuinnQuicMuxPad {}
|
||||
|
||||
impl PadImpl for QuinnQuicMuxPad {}
|
||||
|
||||
impl ProxyPadImpl for QuinnQuicMuxPad {}
|
||||
|
||||
#[derive(Default)]
|
||||
struct State {
|
||||
stream_uni_conns: u64,
|
||||
datagram_requested: bool,
|
||||
stream_id_map: HashMap<gst::Pad, u64>,
|
||||
}
|
||||
|
||||
pub struct QuinnQuicMux {
|
||||
state: Mutex<State>,
|
||||
srcpad: gst::Pad,
|
||||
}
|
||||
|
||||
impl GstObjectImpl for QuinnQuicMux {}
|
||||
|
||||
impl ElementImpl for QuinnQuicMux {
|
||||
fn metadata() -> Option<&'static gst::subclass::ElementMetadata> {
|
||||
static ELEMENT_METADATA: LazyLock<gst::subclass::ElementMetadata> = LazyLock::new(|| {
|
||||
gst::subclass::ElementMetadata::new(
|
||||
"Quinn QUIC Multiplexer",
|
||||
"Source/Network/QUIC",
|
||||
"Multiplexes multiple streams and datagram for QUIC",
|
||||
"Sanchayan Maity <sanchayan@asymptotic.io>",
|
||||
)
|
||||
});
|
||||
|
||||
Some(&*ELEMENT_METADATA)
|
||||
}
|
||||
|
||||
fn pad_templates() -> &'static [gst::PadTemplate] {
|
||||
static PAD_TEMPLATES: LazyLock<Vec<gst::PadTemplate>> = LazyLock::new(|| {
|
||||
let stream_uni_pad_template = gst::PadTemplate::with_gtype(
|
||||
"stream_%u",
|
||||
gst::PadDirection::Sink,
|
||||
gst::PadPresence::Request,
|
||||
&gst::Caps::new_any(),
|
||||
super::QuinnQuicMuxPad::static_type(),
|
||||
)
|
||||
.unwrap();
|
||||
|
||||
let datagram_pad_template = gst::PadTemplate::new(
|
||||
"datagram",
|
||||
gst::PadDirection::Sink,
|
||||
gst::PadPresence::Request,
|
||||
&gst::Caps::new_any(),
|
||||
)
|
||||
.unwrap();
|
||||
|
||||
let src_pad_template = gst::PadTemplate::new(
|
||||
"src",
|
||||
gst::PadDirection::Src,
|
||||
gst::PadPresence::Always,
|
||||
&gst::Caps::new_any(),
|
||||
)
|
||||
.unwrap();
|
||||
|
||||
vec![
|
||||
datagram_pad_template,
|
||||
stream_uni_pad_template,
|
||||
src_pad_template,
|
||||
]
|
||||
});
|
||||
|
||||
PAD_TEMPLATES.as_ref()
|
||||
}
|
||||
|
||||
fn request_new_pad(
|
||||
&self,
|
||||
templ: &gst::PadTemplate,
|
||||
name: Option<&str>,
|
||||
_caps: Option<&gst::Caps>,
|
||||
) -> Option<gst::Pad> {
|
||||
let mut state = self.state.lock().unwrap();
|
||||
|
||||
match templ.name_template() {
|
||||
"stream_%u" => {
|
||||
let stream_pad_name = if let Some(pad_name) = name {
|
||||
pad_name.to_string()
|
||||
} else {
|
||||
state.stream_uni_conns += 1;
|
||||
format!("stream_{}", state.stream_uni_conns).to_string()
|
||||
};
|
||||
|
||||
gst::debug!(CAT, imp = self, "Requesting pad {}", stream_pad_name);
|
||||
|
||||
let stream_pad = gst::PadBuilder::<super::QuinnQuicMuxPad>::from_template(templ)
|
||||
.name(stream_pad_name)
|
||||
.chain_function(|pad, parent, buffer| {
|
||||
QuinnQuicMux::catch_panic_pad_function(
|
||||
parent,
|
||||
|| Err(gst::FlowError::Error),
|
||||
|this| this.stream_uni_sink_chain(pad, buffer),
|
||||
)
|
||||
})
|
||||
.flags(gst::PadFlags::FIXED_CAPS)
|
||||
.build();
|
||||
|
||||
self.obj()
|
||||
.add_pad(&stream_pad)
|
||||
.expect("Failed to add unidirectional stream pad");
|
||||
|
||||
Some(stream_pad.upcast())
|
||||
}
|
||||
"datagram" => {
|
||||
gst::debug!(CAT, imp = self, "Requesting datagram pad");
|
||||
|
||||
if state.datagram_requested {
|
||||
gst::warning!(CAT, imp = self, "datagram pad has already been requested");
|
||||
|
||||
return None;
|
||||
}
|
||||
|
||||
let datagram_pad = gst::Pad::builder_from_template(templ)
|
||||
.name("datagram")
|
||||
.chain_function(|pad, parent, buffer| {
|
||||
QuinnQuicMux::catch_panic_pad_function(
|
||||
parent,
|
||||
|| Err(gst::FlowError::Error),
|
||||
|this| this.datagram_sink_chain(pad, buffer),
|
||||
)
|
||||
})
|
||||
.flags(gst::PadFlags::FIXED_CAPS)
|
||||
.build();
|
||||
|
||||
state.datagram_requested = true;
|
||||
|
||||
self.obj()
|
||||
.add_pad(&datagram_pad)
|
||||
.expect("Failed to add datagram pad");
|
||||
|
||||
Some(datagram_pad.upcast())
|
||||
}
|
||||
_ => None,
|
||||
}
|
||||
}
|
||||
|
||||
fn release_pad(&self, pad: &gst::Pad) {
|
||||
let mut state = self.state.lock().unwrap();
|
||||
|
||||
pad.set_active(false).unwrap();
|
||||
|
||||
if pad.name() == "datagram" {
|
||||
state.datagram_requested = false;
|
||||
} else if pad.name().starts_with("stream") {
|
||||
self.close_stream_for_pad(pad, &mut state);
|
||||
}
|
||||
|
||||
self.obj().remove_pad(pad).unwrap();
|
||||
}
|
||||
|
||||
fn change_state(
|
||||
&self,
|
||||
transition: gst::StateChange,
|
||||
) -> Result<gst::StateChangeSuccess, gst::StateChangeError> {
|
||||
if let gst::StateChange::NullToReady = transition {
|
||||
for pad in self.obj().sink_pads() {
|
||||
if pad.name() == "datagram" && !request_datagram(&self.srcpad) {
|
||||
gst::warning!(CAT, imp = self, "Datagram unsupported by the peer");
|
||||
|
||||
return Err(gst::StateChangeError);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
self.parent_change_state(transition)
|
||||
}
|
||||
}
|
||||
|
||||
impl ObjectImpl for QuinnQuicMux {
|
||||
fn constructed(&self) {
|
||||
self.parent_constructed();
|
||||
|
||||
self.obj()
|
||||
.add_pad(&self.srcpad)
|
||||
.expect("Failed to add source pad");
|
||||
}
|
||||
}
|
||||
|
||||
#[glib::object_subclass]
|
||||
impl ObjectSubclass for QuinnQuicMux {
|
||||
const NAME: &'static str = "GstQuinnQuicMux";
|
||||
type Type = super::QuinnQuicMux;
|
||||
type ParentType = gst::Element;
|
||||
|
||||
fn with_class(klass: &Self::Class) -> Self {
|
||||
let templ = klass.pad_template("src").unwrap();
|
||||
let srcpad = gst::Pad::builder_from_template(&templ).build();
|
||||
|
||||
Self {
|
||||
state: Mutex::new(State::default()),
|
||||
srcpad,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl ChildProxyImpl for QuinnQuicMux {
|
||||
fn children_count(&self) -> u32 {
|
||||
let object = self.obj();
|
||||
object.num_pads() as u32
|
||||
}
|
||||
|
||||
fn child_by_name(&self, name: &str) -> Option<glib::Object> {
|
||||
let object = self.obj();
|
||||
object
|
||||
.pads()
|
||||
.into_iter()
|
||||
.find(|p| p.name() == name)
|
||||
.map(|p| p.upcast())
|
||||
}
|
||||
|
||||
fn child_by_index(&self, index: u32) -> Option<glib::Object> {
|
||||
let object = self.obj();
|
||||
object
|
||||
.pads()
|
||||
.into_iter()
|
||||
.nth(index as usize)
|
||||
.map(|p| p.upcast())
|
||||
}
|
||||
}
|
||||
|
||||
impl QuinnQuicMux {
|
||||
fn stream_uni_sink_chain(
|
||||
&self,
|
||||
pad: &super::QuinnQuicMuxPad,
|
||||
mut buffer: gst::Buffer,
|
||||
) -> Result<gst::FlowSuccess, gst::FlowError> {
|
||||
let mut state = self.state.lock().unwrap();
|
||||
let gst_pad = pad.upcast_ref::<gst::Pad>();
|
||||
|
||||
let stream_id = match state.stream_id_map.get(gst_pad) {
|
||||
Some(stream_id) => *stream_id,
|
||||
None => {
|
||||
let mux_pad_settings = pad.imp().settings.lock().unwrap();
|
||||
let priority = mux_pad_settings.priority;
|
||||
drop(mux_pad_settings);
|
||||
|
||||
gst::info!(
|
||||
CAT,
|
||||
obj = pad,
|
||||
"Requesting stream connection with priority {priority}"
|
||||
);
|
||||
|
||||
match request_stream(&self.srcpad, priority) {
|
||||
Some(stream_id) => {
|
||||
state.stream_id_map.insert(gst_pad.clone(), stream_id);
|
||||
stream_id
|
||||
}
|
||||
None => {
|
||||
gst::error!(CAT, obj = pad, "Failed to request stream");
|
||||
|
||||
return Err(gst::FlowError::Error);
|
||||
}
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
gst::trace!(
|
||||
CAT,
|
||||
obj = pad,
|
||||
"Sending buffer {buffer:?} for stream {stream_id}"
|
||||
);
|
||||
|
||||
let outbuf = buffer.make_mut();
|
||||
|
||||
QuinnQuicMeta::add(outbuf, stream_id, false);
|
||||
|
||||
self.srcpad.push(outbuf.to_owned())
|
||||
}
|
||||
|
||||
fn datagram_sink_chain(
|
||||
&self,
|
||||
pad: &gst::Pad,
|
||||
mut buffer: gst::Buffer,
|
||||
) -> Result<gst::FlowSuccess, gst::FlowError> {
|
||||
gst::trace!(CAT, obj = pad, "Handling buffer for datagram {:?}", buffer);
|
||||
|
||||
let outbuf = buffer.make_mut();
|
||||
|
||||
QuinnQuicMeta::add(outbuf, 0, true);
|
||||
|
||||
self.srcpad.push(outbuf.to_owned())
|
||||
}
|
||||
|
||||
fn close_stream_for_pad(&self, pad: &gst::Pad, state: &mut State) {
|
||||
if let Some(stream_id) = state.stream_id_map.remove(pad) {
|
||||
if close_stream(&self.srcpad, stream_id) {
|
||||
gst::info!(CAT, obj = pad, "Closed connection");
|
||||
} else {
|
||||
gst::warning!(CAT, obj = pad, "Failed to close connection");
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
35
net/quinn/src/quinnquicmux/mod.rs
Normal file
35
net/quinn/src/quinnquicmux/mod.rs
Normal file
|
|
@ -0,0 +1,35 @@
|
|||
// Copyright (C) 2024, Asymptotic Inc.
|
||||
// Author: Sanchayan Maity <sanchayan@asymptotic.io>
|
||||
//G
|
||||
// This Source Code Form is subject to the terms of the Mozilla Public License, v2.0.
|
||||
// If a copy of the MPL was not distributed with this file, You can obtain one at
|
||||
// <https://mozilla.org/MPL/2.0/>.
|
||||
//
|
||||
// SPDX-License-Identifier: MPL-2.0
|
||||
|
||||
/**
|
||||
* element-quinnquicmux:
|
||||
* @short-description: Supports stream multiplexing in QUIC
|
||||
*
|
||||
*/
|
||||
use gst::glib;
|
||||
use gst::prelude::*;
|
||||
|
||||
pub mod imp;
|
||||
|
||||
glib::wrapper! {
|
||||
pub struct QuinnQuicMux(ObjectSubclass<imp::QuinnQuicMux>) @extends gst_base::Aggregator, gst::Element, gst::Object, @implements gst::ChildProxy;
|
||||
}
|
||||
|
||||
glib::wrapper! {
|
||||
pub(crate) struct QuinnQuicMuxPad(ObjectSubclass<imp::QuinnQuicMuxPad>) @extends gst::ProxyPad, gst::Pad, gst::Object;
|
||||
}
|
||||
|
||||
pub fn register(plugin: &gst::Plugin) -> Result<(), glib::BoolError> {
|
||||
gst::Element::register(
|
||||
Some(plugin),
|
||||
"quinnquicmux",
|
||||
gst::Rank::NONE,
|
||||
QuinnQuicMux::static_type(),
|
||||
)
|
||||
}
|
||||
Loading…
Reference in a new issue