mirror of
https://gitlab.freedesktop.org/gstreamer/gst-plugins-rs.git
synced 2024-11-25 13:01:07 +00:00
whipsink: A GstBin implementation for WHIP
Fixes https://gitlab.freedesktop.org/gstreamer/gst-plugins-bad/-/issues/1410 Created a new plugin 'webrtchttp' to implement all the WebRTC HTTP protocols under /net/webrtc-http directory. WhipSink wraps around 'webrtcbin' with HTTP capabilites to exchange SDP offer/answer so an ICE/DTLS session can be established between the encoder/media producer (WHIP client) and the broadcasting ingestion endpoint (Media Server). Once the ICE/DTLS session is set up, the media will flow unidirectionally from the WHIP client to the broadcasting ingestion endpoint (Media Server). Spec: https://www.ietf.org/archive/id/draft-ietf-wish-whip-04.html
This commit is contained in:
parent
827099d22d
commit
67e9ba8286
9 changed files with 1011 additions and 2 deletions
|
@ -17,6 +17,7 @@ members = [
|
|||
"net/onvif",
|
||||
"net/reqwest",
|
||||
"net/aws",
|
||||
"net/webrtc-http",
|
||||
"utils/fallbackswitch",
|
||||
"utils/togglerecord",
|
||||
"utils/tracers",
|
||||
|
@ -53,6 +54,7 @@ default-members = [
|
|||
"net/raptorq",
|
||||
"net/reqwest",
|
||||
"net/aws",
|
||||
"net/webrtc-http",
|
||||
"utils/fallbackswitch",
|
||||
"utils/togglerecord",
|
||||
"utils/tracers",
|
||||
|
|
|
@ -5,7 +5,8 @@ DIRS = ['audio', 'generic', 'net', 'text', 'utils', 'video']
|
|||
RS_PREFIXED = ['audiofx', 'closedcaption',
|
||||
'dav1d', 'file', 'json', 'onvif', 'regex', 'webp']
|
||||
OVERRIDE = {'wrap': 'rstextwrap', 'flavors': 'rsflv',
|
||||
'ahead': 'textahead', 'tracers': 'rstracers'}
|
||||
'ahead': 'textahead', 'tracers': 'rstracers',
|
||||
'webrtc-http': 'webrtchttp'}
|
||||
|
||||
|
||||
def iterate_plugins():
|
||||
|
|
|
@ -6704,5 +6704,80 @@
|
|||
"source": "gst-plugin-videofx",
|
||||
"tracers": {},
|
||||
"url": "https://gitlab.freedesktop.org/gstreamer/gst-plugins-rs"
|
||||
},
|
||||
"webrtchttp": {
|
||||
"description": "Plugin for WebRTC HTTP protocols",
|
||||
"elements": {
|
||||
"whipsink": {
|
||||
"author": "Taruntej Kanakamalla <taruntej@asymptotic.io>",
|
||||
"description": "A bin to stream media using the WebRTC HTTP Ingestion Protocol (WHIP)",
|
||||
"hierarchy": [
|
||||
"WhipSink",
|
||||
"GstBin",
|
||||
"GstElement",
|
||||
"GstObject",
|
||||
"GInitiallyUnowned",
|
||||
"GObject"
|
||||
],
|
||||
"interfaces": [
|
||||
"GstChildProxy"
|
||||
],
|
||||
"klass": "Sink/Network/WebRTC",
|
||||
"long-name": "WHIP Sink Bin",
|
||||
"pad-templates": {
|
||||
"sink_%%u": {
|
||||
"caps": "application/x-rtp:\n",
|
||||
"direction": "sink",
|
||||
"presence": "request"
|
||||
}
|
||||
},
|
||||
"properties": {
|
||||
"auth-token": {
|
||||
"blurb": "Authentication token to use, will be sent in the HTTP Header as 'Bearer <auth-token>'",
|
||||
"conditionally-available": false,
|
||||
"construct": false,
|
||||
"construct-only": false,
|
||||
"controllable": false,
|
||||
"default": "NULL",
|
||||
"mutable": "ready",
|
||||
"readable": true,
|
||||
"type": "gchararray",
|
||||
"writable": true
|
||||
},
|
||||
"use-link-headers": {
|
||||
"blurb": "Use link headers to configure ice-servers from the WHIP server response to the POST or OPTIONS request.\n If set to TRUE and the WHIP server returns valid ice-servers,\n this property overrides the ice-servers values set using the stun-server and turn-server properties.",
|
||||
"conditionally-available": false,
|
||||
"construct": false,
|
||||
"construct-only": false,
|
||||
"controllable": false,
|
||||
"default": "false",
|
||||
"mutable": "ready",
|
||||
"readable": true,
|
||||
"type": "gboolean",
|
||||
"writable": true
|
||||
},
|
||||
"whip-endpoint": {
|
||||
"blurb": "The WHIP server endpoint to POST SDP offer to.\n e.g.: https://example.com/whip/endpoint/room1234",
|
||||
"conditionally-available": false,
|
||||
"construct": false,
|
||||
"construct-only": false,
|
||||
"controllable": false,
|
||||
"default": "NULL",
|
||||
"mutable": "ready",
|
||||
"readable": true,
|
||||
"type": "gchararray",
|
||||
"writable": true
|
||||
}
|
||||
},
|
||||
"rank": "marginal"
|
||||
}
|
||||
},
|
||||
"filename": "gstwebrtchttp",
|
||||
"license": "MPL",
|
||||
"other-types": {},
|
||||
"package": "gst-plugin-webrtchttp",
|
||||
"source": "gst-plugin-webrtchttp",
|
||||
"tracers": {},
|
||||
"url": "https://gitlab.freedesktop.org/gstreamer/gst-plugins-rs"
|
||||
}
|
||||
}
|
|
@ -64,6 +64,7 @@ plugins = {
|
|||
'gst-plugin-textahead': 'libgsttextahead',
|
||||
'gst-plugin-onvif': 'libgstrsonvif',
|
||||
'gst-plugin-tracers': 'libgstrstracers',
|
||||
'gst-plugin-webrtchttp': 'libgstwebrtchttp'
|
||||
}
|
||||
|
||||
extra_env = {}
|
||||
|
@ -144,7 +145,7 @@ endforeach
|
|||
# before gstreamer-rs when building with gst-build.
|
||||
# Custom targets can't depend on dependency() objects so we have to depend
|
||||
# on the library variable from the subproject instead.
|
||||
gst_req = '>= 1.14.0'
|
||||
gst_req = '>= 1.18.0'
|
||||
depends = []
|
||||
|
||||
deps = [
|
||||
|
@ -157,6 +158,8 @@ deps = [
|
|||
['gstreamer-net-1.0', 'gstreamer', 'gst_net_dep', 'gst_net'],
|
||||
['gstreamer-rtp-1.0', 'gst-plugins-base', 'rtp_dep', 'gst_rtp'],
|
||||
['gstreamer-video-1.0', 'gst-plugins-base', 'video_dep', 'gstvideo'],
|
||||
['gstreamer-sdp-1.0', 'gst-plugins-base', 'sdp_dep', 'gstsdp_dep'],
|
||||
['gstreamer-webrtc-1.0', 'gst-plugins-bad', 'gstwebrtc_dep', 'gstwebrtc_dep'],
|
||||
]
|
||||
|
||||
foreach d: deps
|
||||
|
|
51
net/webrtc-http/Cargo.toml
Normal file
51
net/webrtc-http/Cargo.toml
Normal file
|
@ -0,0 +1,51 @@
|
|||
[package]
|
||||
name = "gst-plugin-webrtchttp"
|
||||
version = "0.1.0"
|
||||
authors = ["Taruntej Kanakamalla <taruntej@asymptotic.io"]
|
||||
repository = "https://gitlab.freedesktop.org/gstreamer/gst-plugins-rs"
|
||||
license = "MPL-2.0"
|
||||
edition = "2021"
|
||||
description = "Plugin for WebRTC HTTP protocols"
|
||||
|
||||
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
||||
|
||||
[dependencies]
|
||||
gst = { package = "gstreamer", git = "https://gitlab.freedesktop.org/gstreamer/gstreamer-rs" }
|
||||
gst-base = { package = "gstreamer-base", git = "https://gitlab.freedesktop.org/gstreamer/gstreamer-rs" }
|
||||
gst-webrtc = { package = "gstreamer-webrtc", git="https://gitlab.freedesktop.org/gstreamer/gstreamer-rs", features = ["v1_18"] }
|
||||
gst-sdp = { package = "gstreamer-sdp", git="https://gitlab.freedesktop.org/gstreamer/gstreamer-rs" }
|
||||
gst-rtp = { package = "gstreamer-rtp", git="https://gitlab.freedesktop.org/gstreamer/gstreamer-rs" }
|
||||
reqwest = { version = "0.11", features = ["default-tls"] }
|
||||
once_cell = "1.0"
|
||||
parse_link_header = {version = "0.3", features = ["url"]}
|
||||
tokio = { version = "1.20.1", default-features = false, features = ["time", "rt-multi-thread"] }
|
||||
futures = "0.3.23"
|
||||
|
||||
[dev-dependencies.gst-check]
|
||||
git = "https://gitlab.freedesktop.org/gstreamer/gstreamer-rs"
|
||||
package = "gstreamer-check"
|
||||
|
||||
[lib]
|
||||
name = "gstwebrtchttp"
|
||||
crate-type = ["cdylib", "rlib"]
|
||||
path = "src/lib.rs"
|
||||
|
||||
[build-dependencies]
|
||||
gst-plugin-version-helper = { path="../../version-helper" }
|
||||
|
||||
[features]
|
||||
static = []
|
||||
capi = []
|
||||
|
||||
[package.metadata.capi]
|
||||
min_version = "0.8.0"
|
||||
|
||||
[package.metadata.capi.header]
|
||||
enabled = false
|
||||
|
||||
[package.metadata.capi.library]
|
||||
install_subdir = "gstreamer-1.0"
|
||||
versioning = false
|
||||
|
||||
[package.metadata.capi.pkg_config]
|
||||
requires_private = "gstreamer-1.0, gstreamer-base-1.0, gobject-2.0, glib-2.0, gstreamer-sdp-1.0, gstreamer-webrtc-1.0"
|
3
net/webrtc-http/build.rs
Normal file
3
net/webrtc-http/build.rs
Normal file
|
@ -0,0 +1,3 @@
|
|||
fn main() {
|
||||
gst_plugin_version_helper::info()
|
||||
}
|
28
net/webrtc-http/src/lib.rs
Normal file
28
net/webrtc-http/src/lib.rs
Normal file
|
@ -0,0 +1,28 @@
|
|||
// Copyright (C) 2022, Asymptotic Inc.
|
||||
// Author: Taruntej Kanakamalla <taruntej@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
|
||||
#![allow(clippy::non_send_fields_in_send_ty)]
|
||||
|
||||
use gst::glib;
|
||||
mod whipsink;
|
||||
|
||||
fn plugin_init(plugin: &gst::Plugin) -> Result<(), glib::BoolError> {
|
||||
whipsink::register(plugin)
|
||||
}
|
||||
|
||||
gst::plugin_define!(
|
||||
webrtchttp,
|
||||
env!("CARGO_PKG_DESCRIPTION"),
|
||||
plugin_init,
|
||||
concat!(env!("CARGO_PKG_VERSION"), "-", env!("COMMIT_ID")),
|
||||
"MPL",
|
||||
env!("CARGO_PKG_NAME"),
|
||||
env!("CARGO_PKG_NAME"),
|
||||
env!("CARGO_PKG_REPOSITORY"),
|
||||
env!("BUILD_REL_DATE")
|
||||
);
|
820
net/webrtc-http/src/whipsink/imp.rs
Normal file
820
net/webrtc-http/src/whipsink/imp.rs
Normal file
|
@ -0,0 +1,820 @@
|
|||
// Copyright (C) 2022, Asymptotic Inc.
|
||||
// Author: Taruntej Kanakamalla <taruntej@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 futures::future;
|
||||
use futures::prelude::*;
|
||||
use gst::glib;
|
||||
use gst::prelude::*;
|
||||
use gst::subclass::prelude::*;
|
||||
use gst::ErrorMessage;
|
||||
use gst_sdp::*;
|
||||
use gst_webrtc::*;
|
||||
use once_cell::sync::Lazy;
|
||||
use parse_link_header;
|
||||
use reqwest::header::HeaderMap;
|
||||
use reqwest::header::HeaderValue;
|
||||
use reqwest::redirect::Policy;
|
||||
use reqwest::StatusCode;
|
||||
use std::fmt::Display;
|
||||
use std::sync::Mutex;
|
||||
use tokio::runtime;
|
||||
|
||||
static CAT: Lazy<gst::DebugCategory> = Lazy::new(|| {
|
||||
gst::DebugCategory::new("whipsink", gst::DebugColorFlags::empty(), Some("WHIP Sink"))
|
||||
});
|
||||
|
||||
static RUNTIME: Lazy<runtime::Runtime> = Lazy::new(|| {
|
||||
runtime::Builder::new_multi_thread()
|
||||
.enable_all()
|
||||
.worker_threads(1)
|
||||
.build()
|
||||
.unwrap()
|
||||
});
|
||||
|
||||
const MAX_REDIRECTS: u8 = 10;
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
struct Settings {
|
||||
whip_endpoint: Option<String>,
|
||||
use_link_headers: bool,
|
||||
auth_token: Option<String>,
|
||||
}
|
||||
|
||||
#[allow(clippy::derivable_impls)]
|
||||
impl Default for Settings {
|
||||
fn default() -> Self {
|
||||
Self {
|
||||
whip_endpoint: None,
|
||||
use_link_headers: false,
|
||||
auth_token: None,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
enum State {
|
||||
Stopped,
|
||||
Options { redirects: u8 },
|
||||
Post { redirects: u8 },
|
||||
Running { whip_resource_url: String },
|
||||
}
|
||||
|
||||
impl Default for State {
|
||||
fn default() -> Self {
|
||||
Self::Stopped
|
||||
}
|
||||
}
|
||||
|
||||
pub struct WhipSink {
|
||||
settings: Mutex<Settings>,
|
||||
state: Mutex<State>,
|
||||
webrtcbin: gst::Element,
|
||||
canceller: Mutex<Option<future::AbortHandle>>,
|
||||
}
|
||||
|
||||
impl Default for WhipSink {
|
||||
fn default() -> Self {
|
||||
let webrtcbin = gst::ElementFactory::make("webrtcbin", Some("whip-webrtcbin"))
|
||||
.expect("Failed to create webrtcbin");
|
||||
Self {
|
||||
settings: Mutex::new(Settings::default()),
|
||||
state: Mutex::new(State::default()),
|
||||
webrtcbin,
|
||||
canceller: Mutex::new(None),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl BinImpl for WhipSink {}
|
||||
|
||||
impl GstObjectImpl for WhipSink {}
|
||||
|
||||
impl ElementImpl for WhipSink {
|
||||
fn metadata() -> Option<&'static gst::subclass::ElementMetadata> {
|
||||
static ELEMENT_METADATA: Lazy<gst::subclass::ElementMetadata> = Lazy::new(|| {
|
||||
gst::subclass::ElementMetadata::new(
|
||||
"WHIP Sink Bin",
|
||||
"Sink/Network/WebRTC",
|
||||
"A bin to stream media using the WebRTC HTTP Ingestion Protocol (WHIP)",
|
||||
"Taruntej Kanakamalla <taruntej@asymptotic.io>",
|
||||
)
|
||||
});
|
||||
Some(&*ELEMENT_METADATA)
|
||||
}
|
||||
|
||||
fn pad_templates() -> &'static [gst::PadTemplate] {
|
||||
static PAD_TEMPLATES: Lazy<Vec<gst::PadTemplate>> = Lazy::new(|| {
|
||||
let sink_caps = gst::Caps::builder("application/x-rtp").build();
|
||||
let sink_pad_template = gst::PadTemplate::new(
|
||||
"sink_%u",
|
||||
gst::PadDirection::Sink,
|
||||
gst::PadPresence::Request,
|
||||
&sink_caps,
|
||||
)
|
||||
.unwrap();
|
||||
|
||||
vec![sink_pad_template]
|
||||
});
|
||||
|
||||
PAD_TEMPLATES.as_ref()
|
||||
}
|
||||
|
||||
fn request_new_pad(
|
||||
&self,
|
||||
element: &Self::Type,
|
||||
templ: &gst::PadTemplate,
|
||||
name: Option<String>,
|
||||
caps: Option<&gst::Caps>,
|
||||
) -> Option<gst::Pad> {
|
||||
let wb_sink_pad = self.webrtcbin.request_pad(templ, name.as_deref(), caps)?;
|
||||
let sink_pad = gst::GhostPad::new(Some(&wb_sink_pad.name()), gst::PadDirection::Sink);
|
||||
|
||||
sink_pad.set_target(Some(&wb_sink_pad)).unwrap();
|
||||
element.add_pad(&sink_pad).unwrap();
|
||||
|
||||
Some(sink_pad.upcast())
|
||||
}
|
||||
|
||||
fn change_state(
|
||||
&self,
|
||||
element: &Self::Type,
|
||||
transition: gst::StateChange,
|
||||
) -> Result<gst::StateChangeSuccess, gst::StateChangeError> {
|
||||
let ret = self.parent_change_state(element, transition);
|
||||
if transition == gst::StateChange::PausedToReady {
|
||||
// Interrupt requests in progress, if any
|
||||
if let Some(canceller) = &*self.canceller.lock().unwrap() {
|
||||
canceller.abort();
|
||||
}
|
||||
|
||||
let state = self.state.lock().unwrap();
|
||||
if let State::Running { .. } = *state {
|
||||
// Release server-side resources
|
||||
drop(state);
|
||||
self.terminate_session();
|
||||
}
|
||||
}
|
||||
|
||||
ret
|
||||
}
|
||||
}
|
||||
|
||||
impl ObjectImpl for WhipSink {
|
||||
fn properties() -> &'static [glib::ParamSpec] {
|
||||
static PROPERTIES: Lazy<Vec<glib::ParamSpec>> = Lazy::new(|| {
|
||||
vec![glib::ParamSpecString::builder("whip-endpoint")
|
||||
.nick("WHIP Endpoint")
|
||||
.blurb("The WHIP server endpoint to POST SDP offer to.
|
||||
e.g.: https://example.com/whip/endpoint/room1234")
|
||||
.mutable_ready()
|
||||
.build(),
|
||||
|
||||
glib::ParamSpecBoolean::builder("use-link-headers")
|
||||
.nick("Use Link Headers")
|
||||
.blurb("Use link headers to configure ice-servers from the WHIP server response to the POST or OPTIONS request.
|
||||
If set to TRUE and the WHIP server returns valid ice-servers,
|
||||
this property overrides the ice-servers values set using the stun-server and turn-server properties.")
|
||||
.mutable_ready()
|
||||
.build(),
|
||||
|
||||
glib::ParamSpecString::builder("auth-token")
|
||||
.nick("Authorization Token")
|
||||
.blurb("Authentication token to use, will be sent in the HTTP Header as 'Bearer <auth-token>'")
|
||||
.mutable_ready()
|
||||
.build()
|
||||
]
|
||||
});
|
||||
PROPERTIES.as_ref()
|
||||
}
|
||||
|
||||
fn set_property(
|
||||
&self,
|
||||
_obj: &Self::Type,
|
||||
_id: usize,
|
||||
value: &glib::Value,
|
||||
pspec: &glib::ParamSpec,
|
||||
) {
|
||||
match pspec.name() {
|
||||
"whip-endpoint" => {
|
||||
let mut settings = self.settings.lock().unwrap();
|
||||
settings.whip_endpoint = value.get().expect("WHIP endpoint should be a string");
|
||||
}
|
||||
"use-link-headers" => {
|
||||
let mut settings = self.settings.lock().unwrap();
|
||||
settings.use_link_headers = value
|
||||
.get()
|
||||
.expect("use-link-headers should be a boolean value");
|
||||
}
|
||||
"auth-token" => {
|
||||
let mut settings = self.settings.lock().unwrap();
|
||||
settings.auth_token = value.get().expect("Auth token should be a string");
|
||||
}
|
||||
_ => unimplemented!(),
|
||||
}
|
||||
}
|
||||
|
||||
fn property(&self, _obj: &Self::Type, _id: usize, pspec: &glib::ParamSpec) -> glib::Value {
|
||||
match pspec.name() {
|
||||
"whip-endpoint" => {
|
||||
let settings = self.settings.lock().unwrap();
|
||||
settings.whip_endpoint.to_value()
|
||||
}
|
||||
"use-link-headers" => {
|
||||
let settings = self.settings.lock().unwrap();
|
||||
settings.use_link_headers.to_value()
|
||||
}
|
||||
"auth-token" => {
|
||||
let settings = self.settings.lock().unwrap();
|
||||
settings.auth_token.to_value()
|
||||
}
|
||||
_ => unimplemented!(),
|
||||
}
|
||||
}
|
||||
|
||||
fn constructed(&self, obj: &Self::Type) {
|
||||
self.parent_constructed(obj);
|
||||
obj.set_suppressed_flags(gst::ElementFlags::SINK | gst::ElementFlags::SOURCE);
|
||||
obj.set_element_flags(gst::ElementFlags::SINK);
|
||||
obj.add(&self.webrtcbin).unwrap();
|
||||
|
||||
// The spec requires all m= lines to be bundled (section 4.2)
|
||||
self.webrtcbin
|
||||
.set_property("bundle-policy", gst_webrtc::WebRTCBundlePolicy::MaxBundle);
|
||||
|
||||
self.webrtcbin.connect("on-negotiation-needed", false, {
|
||||
move |args| {
|
||||
let webrtcbin = args[0].get::<gst::Element>().unwrap();
|
||||
let ele = match webrtcbin
|
||||
.parent()
|
||||
.map(|p| p.downcast::<Self::Type>().unwrap())
|
||||
{
|
||||
Some(e) => e,
|
||||
None => return None,
|
||||
};
|
||||
|
||||
let whipsink = ele.imp();
|
||||
let settings = whipsink.settings.lock().unwrap();
|
||||
if settings.whip_endpoint.is_none() {
|
||||
gst::element_error!(
|
||||
ele,
|
||||
gst::ResourceError::NotFound,
|
||||
["Endpoint URL must be set"]
|
||||
);
|
||||
return None;
|
||||
}
|
||||
|
||||
let endpoint =
|
||||
reqwest::Url::parse(settings.whip_endpoint.as_ref().unwrap().as_str());
|
||||
if let Err(e) = endpoint {
|
||||
gst::element_error!(
|
||||
ele,
|
||||
gst::ResourceError::Failed,
|
||||
["Could not parse endpoint URL :{}", e]
|
||||
);
|
||||
return None;
|
||||
}
|
||||
|
||||
drop(settings);
|
||||
let mut state = whipsink.state.lock().unwrap();
|
||||
*state = State::Options { redirects: 0 };
|
||||
drop(state);
|
||||
|
||||
if let Err(e) = whipsink.lookup_ice_servers(endpoint.unwrap()) {
|
||||
gst::element_error!(
|
||||
ele,
|
||||
gst::ResourceError::Failed,
|
||||
["Error in 'lookup_ice_servers' - {}", e.to_string()]
|
||||
);
|
||||
}
|
||||
|
||||
// Promise for 'create-offer' signal emitted to webrtcbin
|
||||
// Closure is called when the promise is fulfilled
|
||||
let promise = gst::Promise::with_change_func(move |reply| {
|
||||
let ele = match webrtcbin
|
||||
.parent()
|
||||
.map(|p| p.downcast::<Self::Type>().unwrap())
|
||||
{
|
||||
Some(ele) => ele,
|
||||
None => return,
|
||||
};
|
||||
let whipsink = ele.imp();
|
||||
|
||||
let offer_sdp = match reply {
|
||||
Ok(Some(sdp)) => sdp
|
||||
.value("offer")
|
||||
.expect("structure must have an offer key")
|
||||
.get::<gst_webrtc::WebRTCSessionDescription>()
|
||||
.expect("offer must be an SDP"),
|
||||
Ok(None) => {
|
||||
gst::element_error!(
|
||||
ele,
|
||||
gst::LibraryError::Failed,
|
||||
["create-offer::Promise returned with no reply"]
|
||||
);
|
||||
return;
|
||||
}
|
||||
Err(e) => {
|
||||
gst::element_error!(
|
||||
ele,
|
||||
gst::LibraryError::Failed,
|
||||
["create-offer::Promise returned with error {:?}", e]
|
||||
);
|
||||
return;
|
||||
}
|
||||
};
|
||||
if let Err(e) = whipsink.send_offer(offer_sdp) {
|
||||
gst::element_error!(
|
||||
ele,
|
||||
gst::ResourceError::Failed,
|
||||
["Error in 'send_offer' - {}", e.to_string()]
|
||||
);
|
||||
}
|
||||
});
|
||||
|
||||
whipsink
|
||||
.webrtcbin
|
||||
.emit_by_name::<()>("create-offer", &[&None::<gst::Structure>, &promise]);
|
||||
|
||||
None
|
||||
}
|
||||
});
|
||||
|
||||
self.webrtcbin.connect("on-new-transceiver", false, {
|
||||
move |args| {
|
||||
let trans = args[1].get::<gst_webrtc::WebRTCRTPTransceiver>().unwrap();
|
||||
// We only ever send data
|
||||
trans.set_direction(gst_webrtc::WebRTCRTPTransceiverDirection::Sendonly);
|
||||
None
|
||||
}
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
#[glib::object_subclass]
|
||||
impl ObjectSubclass for WhipSink {
|
||||
const NAME: &'static str = "WhipSink";
|
||||
type Type = super::WhipSink;
|
||||
type ParentType = gst::Bin;
|
||||
}
|
||||
impl WhipSink {
|
||||
fn lookup_ice_servers(&self, endpoint: reqwest::Url) -> Result<(), ErrorMessage> {
|
||||
let settings = self.settings.lock().unwrap();
|
||||
let state = self.state.lock().unwrap();
|
||||
|
||||
let redirects = match *state {
|
||||
State::Options { redirects } => redirects,
|
||||
_ => {
|
||||
return Err(gst::error_msg!(
|
||||
gst::ResourceError::Failed,
|
||||
["Trying to do OPTIONS in unexpected state"]
|
||||
));
|
||||
}
|
||||
};
|
||||
drop(state);
|
||||
|
||||
if !settings.use_link_headers {
|
||||
// We're not configured to use OPTIONS, so we're done
|
||||
return Ok(());
|
||||
}
|
||||
|
||||
// We'll handle redirects manually since the default redirect handler does not
|
||||
// reuse the authentication token on the redirected server
|
||||
let pol = reqwest::redirect::Policy::none();
|
||||
let client = build_reqwest_client(pol);
|
||||
|
||||
let mut headermap = HeaderMap::new();
|
||||
if let Some(token) = &settings.auth_token {
|
||||
let bearer_token = "Bearer ".to_owned() + token;
|
||||
drop(settings);
|
||||
headermap.insert(
|
||||
reqwest::header::AUTHORIZATION,
|
||||
HeaderValue::from_str(&bearer_token)
|
||||
.expect("Auth token should only contain characters valid for an HTTP header"),
|
||||
);
|
||||
}
|
||||
|
||||
let future = client
|
||||
.request(reqwest::Method::OPTIONS, endpoint.as_ref())
|
||||
.headers(headermap)
|
||||
.send();
|
||||
|
||||
let resp = match self.wait(future) {
|
||||
Ok(r) => r,
|
||||
Err(e) => {
|
||||
return Err(e);
|
||||
}
|
||||
};
|
||||
|
||||
match resp.status() {
|
||||
StatusCode::NO_CONTENT => {
|
||||
self.set_ice_servers(resp.headers());
|
||||
Ok(())
|
||||
}
|
||||
status if status.is_redirection() => {
|
||||
if redirects < MAX_REDIRECTS {
|
||||
let mut state = self.state.lock().unwrap();
|
||||
*state = State::Options {
|
||||
redirects: redirects + 1,
|
||||
};
|
||||
drop(state);
|
||||
|
||||
match parse_redirect_location(resp.headers(), &endpoint) {
|
||||
Ok(redirect_url) => {
|
||||
gst::debug!(CAT, "Redirecting endpoint to {}", redirect_url.as_str());
|
||||
self.lookup_ice_servers(redirect_url)
|
||||
}
|
||||
Err(e) => Err(e),
|
||||
}
|
||||
} else {
|
||||
Err(gst::error_msg!(
|
||||
gst::ResourceError::Failed,
|
||||
["Too many redirects. Unable to connect to do OPTIONS request"]
|
||||
))
|
||||
}
|
||||
}
|
||||
status => Err(gst::error_msg!(
|
||||
gst::ResourceError::Failed,
|
||||
[
|
||||
"lookup_ice_servers - Unexpected response {} {:?}",
|
||||
status,
|
||||
self.wait(resp.bytes()).unwrap()
|
||||
]
|
||||
)),
|
||||
}
|
||||
}
|
||||
|
||||
fn set_ice_servers(&self, headermap: &HeaderMap) {
|
||||
for link in headermap.get_all("link").iter() {
|
||||
let link = link
|
||||
.to_str()
|
||||
.expect("Header value should contain only visible ASCII strings");
|
||||
// FIXME: The Demo WHIP Server appends an extra ; at the end of each link
|
||||
// but not needed as per https://datatracker.ietf.org/doc/html/rfc8288#section-3.5
|
||||
let link = link.trim_matches(';');
|
||||
|
||||
let item_map = parse_link_header::parse_with_rel(link);
|
||||
if let Err(e) = item_map {
|
||||
gst::error!(CAT, "set_ice_servers {} - Error {:?}", link, e);
|
||||
continue;
|
||||
}
|
||||
|
||||
let item_map = item_map.unwrap();
|
||||
if !item_map.contains_key("ice-server") {
|
||||
// Not a link header we care about
|
||||
continue;
|
||||
}
|
||||
|
||||
let link = item_map.get("ice-server").unwrap();
|
||||
|
||||
// Note: webrtcbin needs ice servers to be in the below format
|
||||
// <scheme>://<user:pass>@<url>
|
||||
// and the ice-servers (link headers) received from the whip server might be
|
||||
// in the format <scheme>:<host> with username and password as separate params.
|
||||
// Constructing these with 'url' crate also require a format/parse
|
||||
// for changing <scheme>:<host> to <scheme>://<user>:<password>@<host>.
|
||||
// So preferred to use the String rather
|
||||
|
||||
let mut ice_server_url;
|
||||
|
||||
// check if uri has ://
|
||||
if link.uri.has_authority() {
|
||||
// use raw_uri as is
|
||||
// username and password in the link.uri.params ignored
|
||||
ice_server_url = link.raw_uri.as_str().to_string();
|
||||
} else {
|
||||
// construct url as '<scheme>://<user:pass>@<url>'
|
||||
ice_server_url = format!("{}://", link.uri.scheme());
|
||||
if let Some(user) = link.params.get("username") {
|
||||
ice_server_url += user.as_str();
|
||||
if let Some(pass) = link.params.get("credential") {
|
||||
ice_server_url = ice_server_url + ":" + pass.as_str();
|
||||
}
|
||||
ice_server_url += "@";
|
||||
}
|
||||
|
||||
// the raw_uri contains the ice-server in the form <scheme>:<url>
|
||||
// so strip the scheme and the ':' from the beginning of raw_uri and use
|
||||
// the rest of raw_uri to append it the url which will be in the form
|
||||
// <scheme>://<user:pass>@<url> as expected
|
||||
ice_server_url += link
|
||||
.raw_uri
|
||||
.strip_prefix((link.uri.scheme().to_owned() + ":").as_str())
|
||||
.expect("strip 'scheme:' from raw uri");
|
||||
}
|
||||
|
||||
gst::debug!(CAT, "Setting STUN/TURN server {}", ice_server_url);
|
||||
|
||||
// It's icer to not collapse the `else if` and its inner `if`
|
||||
#[allow(clippy::collapsible_if)]
|
||||
if link.uri.scheme() == "stun" {
|
||||
self.webrtcbin
|
||||
.set_property_from_str("stun-server", ice_server_url.as_str());
|
||||
} else if link.uri.scheme().starts_with("turn") {
|
||||
if !self
|
||||
.webrtcbin
|
||||
.emit_by_name::<bool>("add-turn-server", &[&ice_server_url.as_str()])
|
||||
{
|
||||
gst::error!(CAT, "Falied to set turn server {}", ice_server_url);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn send_offer(
|
||||
&self,
|
||||
offer: gst_webrtc::WebRTCSessionDescription,
|
||||
) -> Result<(), gst::ErrorMessage> {
|
||||
let settings = self.settings.lock().unwrap();
|
||||
|
||||
self.webrtcbin
|
||||
.emit_by_name::<()>("set-local-description", &[&offer, &None::<gst::Promise>]);
|
||||
|
||||
if settings.whip_endpoint.is_none() {
|
||||
return Err(gst::error_msg!(
|
||||
gst::ResourceError::NotFound,
|
||||
["Endpoint URL must be set"]
|
||||
));
|
||||
}
|
||||
|
||||
let endpoint = reqwest::Url::parse(settings.whip_endpoint.as_ref().unwrap().as_str());
|
||||
if let Err(e) = endpoint {
|
||||
return Err(gst::error_msg!(
|
||||
gst::ResourceError::Failed,
|
||||
["Could not parse endpoint URL: {}", e]
|
||||
));
|
||||
}
|
||||
|
||||
drop(settings);
|
||||
let mut state = self.state.lock().unwrap();
|
||||
*state = State::Post { redirects: 0 };
|
||||
drop(state);
|
||||
|
||||
let answer = self.do_post(offer, endpoint.unwrap())?;
|
||||
|
||||
self.webrtcbin
|
||||
.emit_by_name::<()>("set-remote-description", &[&answer, &None::<gst::Promise>]);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn do_post(
|
||||
&self,
|
||||
offer: gst_webrtc::WebRTCSessionDescription,
|
||||
endpoint: reqwest::Url,
|
||||
) -> Result<gst_webrtc::WebRTCSessionDescription, gst::ErrorMessage> {
|
||||
let settings = self.settings.lock().unwrap();
|
||||
let state = self.state.lock().unwrap();
|
||||
|
||||
let redirects = match *state {
|
||||
State::Post { redirects } => redirects,
|
||||
_ => {
|
||||
return Err(gst::error_msg!(
|
||||
gst::ResourceError::Failed,
|
||||
["Trying to POST in unexpected state"]
|
||||
));
|
||||
}
|
||||
};
|
||||
drop(state);
|
||||
|
||||
// Default policy for redirect does not share the auth token to new location
|
||||
// So disable inbuilt redirecting and do a recursive call upon 3xx response code
|
||||
let pol = reqwest::redirect::Policy::none();
|
||||
let client = build_reqwest_client(pol);
|
||||
|
||||
let sdp = offer.sdp();
|
||||
let body = sdp.as_text().unwrap();
|
||||
|
||||
gst::debug!(CAT, "Using endpoint {}", endpoint.as_str());
|
||||
let mut headermap = HeaderMap::new();
|
||||
headermap.insert(
|
||||
reqwest::header::CONTENT_TYPE,
|
||||
HeaderValue::from_static("application/sdp"),
|
||||
);
|
||||
|
||||
if let Some(token) = &settings.auth_token {
|
||||
let bearer_token = "Bearer ".to_owned() + token;
|
||||
drop(settings);
|
||||
headermap.insert(
|
||||
reqwest::header::AUTHORIZATION,
|
||||
HeaderValue::from_str(bearer_token.as_str())
|
||||
.expect("Failed to set auth token to header"),
|
||||
);
|
||||
}
|
||||
|
||||
let future = client
|
||||
.request(reqwest::Method::POST, endpoint.as_ref())
|
||||
.headers(headermap)
|
||||
.body(body)
|
||||
.send();
|
||||
|
||||
let resp = match self.wait(future) {
|
||||
Ok(r) => r,
|
||||
Err(e) => {
|
||||
return Err(e);
|
||||
}
|
||||
};
|
||||
|
||||
let res = match resp.status() {
|
||||
StatusCode::OK | StatusCode::CREATED => {
|
||||
// XXX: this is a long function, we should factor it out.
|
||||
// Note: Not taking care of 'Link' headers in POST response
|
||||
// because we do a mandatory OPTIONS request before 'create-offer'
|
||||
// and update the ICE servers from response to OPTIONS
|
||||
|
||||
// Get the url of the resource from 'location' header.
|
||||
// The resource created is expected be a relative path
|
||||
// and not an absolute path
|
||||
// So we want to construct the full url of the resource
|
||||
// using the endpoint url i.e., replace the end point path with
|
||||
// resource path
|
||||
let location = resp.headers().get(reqwest::header::LOCATION).unwrap();
|
||||
let mut url = reqwest::Url::parse(endpoint.as_str()).unwrap();
|
||||
|
||||
url.set_path(location.to_str().unwrap());
|
||||
let mut state = self.state.lock().unwrap();
|
||||
*state = State::Running {
|
||||
whip_resource_url: url.to_string(),
|
||||
};
|
||||
drop(state);
|
||||
|
||||
let ans_bytes = match self.wait(resp.bytes()) {
|
||||
Ok(ans) => ans.to_vec(),
|
||||
Err(e) => return Err(e),
|
||||
};
|
||||
match sdp_message::SDPMessage::parse_buffer(&ans_bytes) {
|
||||
Ok(ans_sdp) => {
|
||||
let answer = gst_webrtc::WebRTCSessionDescription::new(
|
||||
gst_webrtc::WebRTCSDPType::Answer,
|
||||
ans_sdp,
|
||||
);
|
||||
Ok(answer)
|
||||
}
|
||||
|
||||
Err(e) => Err(gst::error_msg!(
|
||||
gst::ResourceError::Failed,
|
||||
["Could not parse answer SDP: {}", e]
|
||||
)),
|
||||
}
|
||||
}
|
||||
|
||||
s if s.is_redirection() => {
|
||||
if redirects < MAX_REDIRECTS {
|
||||
let mut state = self.state.lock().unwrap();
|
||||
*state = State::Post {
|
||||
redirects: redirects + 1,
|
||||
};
|
||||
drop(state);
|
||||
|
||||
match parse_redirect_location(resp.headers(), &endpoint) {
|
||||
Ok(redirect_url) => {
|
||||
gst::debug!(CAT, "Redirecting endpoint to {}", redirect_url.as_str());
|
||||
self.do_post(offer, redirect_url)
|
||||
}
|
||||
Err(e) => return Err(e),
|
||||
}
|
||||
} else {
|
||||
Err(gst::error_msg!(
|
||||
gst::ResourceError::Failed,
|
||||
["Too many redirects. Unable to connect to do POST"]
|
||||
))
|
||||
}
|
||||
}
|
||||
|
||||
s if s.is_server_error() => {
|
||||
// FIXME: Check and handle 'Retry-After' header in case of server error
|
||||
todo!()
|
||||
}
|
||||
|
||||
s => Err(gst::error_msg!(
|
||||
gst::ResourceError::Failed,
|
||||
[
|
||||
"Unexpected response {:?} {:?}",
|
||||
s,
|
||||
self.wait(resp.bytes()).unwrap()
|
||||
]
|
||||
)),
|
||||
};
|
||||
|
||||
res
|
||||
}
|
||||
|
||||
fn terminate_session(&self) {
|
||||
let settings = self.settings.lock().unwrap();
|
||||
let state = self.state.lock().unwrap();
|
||||
let resource_url = match *state {
|
||||
State::Running {
|
||||
ref whip_resource_url,
|
||||
} => whip_resource_url.clone(),
|
||||
_ => {
|
||||
gst::error!(CAT, "Terminated in unexpected state");
|
||||
return;
|
||||
}
|
||||
};
|
||||
drop(state);
|
||||
|
||||
let mut headermap = HeaderMap::new();
|
||||
if let Some(token) = &settings.auth_token {
|
||||
let bearer_token = "Bearer ".to_owned() + token.as_str();
|
||||
headermap.insert(
|
||||
reqwest::header::AUTHORIZATION,
|
||||
HeaderValue::from_str(bearer_token.as_str())
|
||||
.expect("Failed to set auth token to header"),
|
||||
);
|
||||
}
|
||||
|
||||
gst::debug!(CAT, "DELETE request on {}", resource_url);
|
||||
let client = build_reqwest_client(reqwest::redirect::Policy::default());
|
||||
let future = client.delete(resource_url).headers(headermap).send();
|
||||
|
||||
let res = self.wait(future);
|
||||
match res {
|
||||
Ok(r) => {
|
||||
gst::debug!(CAT, "Response to DELETE : {}", r.status());
|
||||
}
|
||||
Err(e) => {
|
||||
gst::error!(CAT, "{}", e);
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
fn wait<F, T, E>(&self, future: F) -> Result<T, ErrorMessage>
|
||||
where
|
||||
F: Send + Future<Output = Result<T, E>>,
|
||||
T: Send + 'static,
|
||||
E: Send + Display,
|
||||
{
|
||||
let mut canceller = self.canceller.lock().unwrap();
|
||||
let (abort_handle, abort_registration) = future::AbortHandle::new_pair();
|
||||
|
||||
canceller.replace(abort_handle);
|
||||
drop(canceller);
|
||||
|
||||
// make abortable
|
||||
let future = async {
|
||||
match future::Abortable::new(future, abort_registration).await {
|
||||
// Future resolved successfully
|
||||
Ok(Ok(res)) => Ok(res),
|
||||
|
||||
// Future resolved with an error
|
||||
Ok(Err(err)) => Err(gst::error_msg!(
|
||||
gst::ResourceError::Failed,
|
||||
["Future resolved with an error {}", err.to_string()]
|
||||
)),
|
||||
|
||||
// Canceller called before future resolved
|
||||
Err(future::Aborted) => Err(gst::error_msg!(
|
||||
gst::ResourceError::Failed,
|
||||
["Canceller called before future resolved"]
|
||||
)),
|
||||
}
|
||||
};
|
||||
|
||||
let res = {
|
||||
let _enter = RUNTIME.enter();
|
||||
futures::executor::block_on(future)
|
||||
};
|
||||
|
||||
/* Clear out the canceller */
|
||||
let _ = self.canceller.lock().unwrap().take();
|
||||
res
|
||||
}
|
||||
}
|
||||
|
||||
fn parse_redirect_location(
|
||||
headermap: &HeaderMap,
|
||||
old_url: &reqwest::Url,
|
||||
) -> Result<reqwest::Url, ErrorMessage> {
|
||||
let location = headermap.get(reqwest::header::LOCATION).unwrap();
|
||||
if let Err(e) = location.to_str() {
|
||||
return Err(gst::error_msg!(
|
||||
gst::ResourceError::Failed,
|
||||
[
|
||||
"Failed to convert the redirect location to string {}",
|
||||
e.to_string()
|
||||
]
|
||||
));
|
||||
}
|
||||
let location = location.to_str().unwrap();
|
||||
|
||||
if location.to_ascii_lowercase().starts_with("http") {
|
||||
// location url is an absolute path
|
||||
reqwest::Url::parse(location)
|
||||
.map_err(|e| gst::error_msg!(gst::ResourceError::Failed, ["{}", e.to_string()]))
|
||||
} else {
|
||||
// location url is a relative path
|
||||
let mut new_url = old_url.clone();
|
||||
new_url.set_path(location);
|
||||
Ok(new_url)
|
||||
}
|
||||
}
|
||||
|
||||
fn build_reqwest_client(pol: Policy) -> reqwest::Client {
|
||||
let client_builder = reqwest::Client::builder();
|
||||
client_builder.redirect(pol).build().unwrap()
|
||||
}
|
26
net/webrtc-http/src/whipsink/mod.rs
Normal file
26
net/webrtc-http/src/whipsink/mod.rs
Normal file
|
@ -0,0 +1,26 @@
|
|||
// Copyright (C) 2022, Asymptotic Inc.
|
||||
// Author: Taruntej Kanakamalla <taruntej@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 gst::glib;
|
||||
use gst::prelude::*;
|
||||
|
||||
pub mod imp;
|
||||
|
||||
glib::wrapper! {
|
||||
pub struct WhipSink(ObjectSubclass<imp::WhipSink>) @extends gst::Bin, gst::Element, gst::Object;
|
||||
}
|
||||
|
||||
pub fn register(plugin: &gst::Plugin) -> Result<(), glib::BoolError> {
|
||||
gst::Element::register(
|
||||
Some(plugin),
|
||||
"whipsink",
|
||||
gst::Rank::Marginal,
|
||||
WhipSink::static_type(),
|
||||
)
|
||||
}
|
Loading…
Reference in a new issue