Move thread-handling out of the actual Source/Sink implementation

This commit is contained in:
Sebastian Dröge 2016-09-02 00:52:28 +03:00
parent af6d6f80b4
commit 4afdd2c985
9 changed files with 371 additions and 467 deletions

View file

@ -25,6 +25,8 @@ use std::fmt::{Display, Formatter};
use std::fmt::Error as FmtError; use std::fmt::Error as FmtError;
use std::borrow::Cow; use std::borrow::Cow;
use url::Url;
use utils::*; use utils::*;
macro_rules! error_msg( macro_rules! error_msg(
@ -115,6 +117,49 @@ impl ErrorMessage {
line: line, line: line,
} }
} }
pub unsafe fn post(&self, element: *mut c_void) {
extern "C" {
fn gst_rs_element_error(sink: *mut c_void,
error_domain: u32,
error_code: i32,
message: *const c_char,
debug: *const c_char,
filename: *const c_char,
function: *const c_char,
line: u32);
}
let ErrorMessage { error_domain,
error_code,
ref message,
ref debug,
filename,
function,
line } = *self;
let message_cstr = message.as_ref().map(|m| CString::new(m.as_bytes()).unwrap());
let message_ptr = message_cstr.as_ref().map_or(ptr::null(), |m| m.as_ptr());
let debug_cstr = debug.as_ref().map(|m| CString::new(m.as_bytes()).unwrap());
let debug_ptr = debug_cstr.as_ref().map_or(ptr::null(), |m| m.as_ptr());
let file_cstr = CString::new(filename.as_bytes()).unwrap();
let file_ptr = file_cstr.as_ptr();
let function_cstr = CString::new(function.as_bytes()).unwrap();
let function_ptr = function_cstr.as_ptr();
gst_rs_element_error(element,
error_domain,
error_code,
message_ptr,
debug_ptr,
file_ptr,
function_ptr,
line);
}
} }
#[derive(Debug)] #[derive(Debug)]
@ -246,3 +291,5 @@ impl Error for UriError {
} }
} }
} }
pub type UriValidator = Fn(&Url) -> Result<(), UriError>;

View file

@ -43,3 +43,12 @@ GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
rsplugin, rsplugin,
"Rust Plugin", plugin_init, VERSION, "LGPL", GST_PACKAGE_NAME, "Rust Plugin", plugin_init, VERSION, "LGPL", GST_PACKAGE_NAME,
GST_PACKAGE_ORIGIN); GST_PACKAGE_ORIGIN);
void
gst_rs_element_error (GstElement * element, GQuark error_domain,
gint error_code, const gchar * message, const gchar * debug,
const gchar * file, const gchar * function, guint line)
{
gst_element_message_full (element, GST_MESSAGE_ERROR, error_domain,
error_code, g_strdup (message), g_strdup (debug), file, function, line);
}

View file

@ -17,21 +17,14 @@
// Boston, MA 02110-1301, USA. // Boston, MA 02110-1301, USA.
use std::fs::File; use std::fs::File;
use std::path::PathBuf;
use url::Url; use url::Url;
use std::io::Write; use std::io::Write;
use std::sync::Mutex;
use std::convert::From; use std::convert::From;
use error::*; use error::*;
use rssink::*; use rssink::*;
#[derive(Debug)]
struct Settings {
location: Option<PathBuf>,
}
#[derive(Debug)] #[derive(Debug)]
enum StreamingState { enum StreamingState {
Stopped, Stopped,
@ -40,68 +33,44 @@ enum StreamingState {
#[derive(Debug)] #[derive(Debug)]
pub struct FileSink { pub struct FileSink {
controller: SinkController, streaming_state: StreamingState,
settings: Mutex<Settings>,
streaming_state: Mutex<StreamingState>,
} }
unsafe impl Sync for FileSink {}
unsafe impl Send for FileSink {}
impl FileSink { impl FileSink {
pub fn new(controller: SinkController) -> FileSink { pub fn new() -> FileSink {
FileSink { FileSink { streaming_state: StreamingState::Stopped }
controller: controller,
settings: Mutex::new(Settings { location: None }),
streaming_state: Mutex::new(StreamingState::Stopped),
}
} }
pub fn new_boxed(controller: SinkController) -> Box<Sink> { pub fn new_boxed() -> Box<Sink> {
Box::new(FileSink::new(controller)) Box::new(FileSink::new())
} }
} }
fn validate_uri(uri: &Url) -> Result<(), UriError> {
let _ = try!(uri.to_file_path()
.or_else(|_| {
Err(UriError::new(UriErrorKind::UnsupportedProtocol,
Some(format!("Unsupported file URI '{}'", uri.as_str()))))
}));
Ok(())
}
impl Sink for FileSink { impl Sink for FileSink {
fn get_controller(&self) -> &SinkController { fn uri_validator(&self) -> Box<UriValidator> {
&self.controller Box::new(validate_uri)
} }
fn set_uri(&self, uri: Option<Url>) -> Result<(), UriError> { fn start(&mut self, uri: &Url) -> Result<(), ErrorMessage> {
let location = &mut self.settings.lock().unwrap().location; if let StreamingState::Started { .. } = self.streaming_state {
*location = None;
match uri {
None => Ok(()),
Some(ref uri) => {
*location = Some(try!(uri.to_file_path()
.or_else(|_| {
Err(UriError::new(UriErrorKind::UnsupportedProtocol,
Some(format!("Unsupported file URI '{}'", uri.as_str()))))
})));
Ok(())
}
}
}
fn get_uri(&self) -> Option<Url> {
let location = &self.settings.lock().unwrap().location;
location.as_ref()
.map(|l| Url::from_file_path(l).ok())
.and_then(|i| i) // join()
}
fn start(&self) -> Result<(), ErrorMessage> {
let location = &self.settings.lock().unwrap().location;
let mut streaming_state = self.streaming_state.lock().unwrap();
if let StreamingState::Started { .. } = *streaming_state {
return Err(error_msg!(SinkError::Failure, ["Sink already started"])); return Err(error_msg!(SinkError::Failure, ["Sink already started"]));
} }
let location = &try!(location.as_ref().ok_or_else(|| { let location = try!(uri.to_file_path()
error_msg!(SinkError::Failure, ["No URI provided"]) .or_else(|_| {
})); Err(error_msg!(SinkError::Failure,
["Unsupported file URI '{}'", uri.as_str()]))
}));
let file = try!(File::create(location.as_path()).or_else(|err| { let file = try!(File::create(location.as_path()).or_else(|err| {
Err(error_msg!(SinkError::OpenFailed, Err(error_msg!(SinkError::OpenFailed,
@ -110,7 +79,7 @@ impl Sink for FileSink {
err.to_string()])) err.to_string()]))
})); }));
*streaming_state = StreamingState::Started { self.streaming_state = StreamingState::Started {
file: file, file: file,
position: 0, position: 0,
}; };
@ -118,17 +87,14 @@ impl Sink for FileSink {
Ok(()) Ok(())
} }
fn stop(&self) -> Result<(), ErrorMessage> { fn stop(&mut self) -> Result<(), ErrorMessage> {
let mut streaming_state = self.streaming_state.lock().unwrap(); self.streaming_state = StreamingState::Stopped;
*streaming_state = StreamingState::Stopped;
Ok(()) Ok(())
} }
fn render(&self, data: &[u8]) -> Result<(), FlowError> { fn render(&mut self, data: &[u8]) -> Result<(), FlowError> {
let mut streaming_state = self.streaming_state.lock().unwrap(); let (file, position) = match self.streaming_state {
let (file, position) = match *streaming_state {
StreamingState::Started { ref mut file, ref mut position } => (file, position), StreamingState::Started { ref mut file, ref mut position } => (file, position),
StreamingState::Stopped => { StreamingState::Stopped => {
return Err(FlowError::Error(error_msg!(SinkError::Failure, ["Not started yet"]))); return Err(FlowError::Error(error_msg!(SinkError::Failure, ["Not started yet"])));
@ -140,6 +106,7 @@ impl Sink for FileSink {
})); }));
*position += data.len() as u64; *position += data.len() as u64;
Ok(()) Ok(())
} }
} }

View file

@ -18,18 +18,11 @@
use std::u64; use std::u64;
use std::io::{Read, Seek, SeekFrom}; use std::io::{Read, Seek, SeekFrom};
use std::fs::File; use std::fs::File;
use std::path::PathBuf;
use std::sync::Mutex;
use url::Url; use url::Url;
use error::*; use error::*;
use rssource::*; use rssource::*;
#[derive(Debug)]
struct Settings {
location: Option<PathBuf>,
}
#[derive(Debug)] #[derive(Debug)]
enum StreamingState { enum StreamingState {
Stopped, Stopped,
@ -38,58 +31,31 @@ enum StreamingState {
#[derive(Debug)] #[derive(Debug)]
pub struct FileSrc { pub struct FileSrc {
controller: SourceController, streaming_state: StreamingState,
settings: Mutex<Settings>,
streaming_state: Mutex<StreamingState>,
} }
unsafe impl Sync for FileSrc {}
unsafe impl Send for FileSrc {}
impl FileSrc { impl FileSrc {
pub fn new(controller: SourceController) -> FileSrc { pub fn new() -> FileSrc {
FileSrc { FileSrc { streaming_state: StreamingState::Stopped }
controller: controller,
settings: Mutex::new(Settings { location: None }),
streaming_state: Mutex::new(StreamingState::Stopped),
}
} }
pub fn new_boxed(controller: SourceController) -> Box<Source> { pub fn new_boxed() -> Box<Source> {
Box::new(FileSrc::new(controller)) Box::new(FileSrc::new())
} }
} }
fn validate_uri(uri: &Url) -> Result<(), UriError> {
let _ = try!(uri.to_file_path()
.or_else(|_| {
Err(UriError::new(UriErrorKind::UnsupportedProtocol,
Some(format!("Unsupported file URI '{}'", uri.as_str()))))
}));
Ok(())
}
impl Source for FileSrc { impl Source for FileSrc {
fn get_controller(&self) -> &SourceController { fn uri_validator(&self) -> Box<UriValidator> {
&self.controller Box::new(validate_uri)
}
fn set_uri(&self, uri: Option<Url>) -> Result<(), UriError> {
let location = &mut self.settings.lock().unwrap().location;
match uri {
None => {
*location = None;
Ok(())
}
Some(ref uri) => {
*location = Some(try!(uri.to_file_path()
.or_else(|_| {
Err(UriError::new(UriErrorKind::UnsupportedProtocol,
Some(format!("Unsupported file URI '{}'", uri.as_str()))))
})));
Ok(())
}
}
}
fn get_uri(&self) -> Option<Url> {
let location = &self.settings.lock().unwrap().location;
location.as_ref()
.map(|l| Url::from_file_path(l).ok())
.and_then(|i| i) // join()
} }
fn is_seekable(&self) -> bool { fn is_seekable(&self) -> bool {
@ -97,9 +63,7 @@ impl Source for FileSrc {
} }
fn get_size(&self) -> u64 { fn get_size(&self) -> u64 {
let streaming_state = self.streaming_state.lock().unwrap(); if let StreamingState::Started { ref file, .. } = self.streaming_state {
if let StreamingState::Started { ref file, .. } = *streaming_state {
file.metadata() file.metadata()
.ok() .ok()
.map_or(u64::MAX, |m| m.len()) .map_or(u64::MAX, |m| m.len())
@ -108,16 +72,16 @@ impl Source for FileSrc {
} }
} }
fn start(&self) -> Result<(), ErrorMessage> { fn start(&mut self, uri: &Url) -> Result<(), ErrorMessage> {
let location = &self.settings.lock().unwrap().location; if let StreamingState::Started { .. } = self.streaming_state {
let mut streaming_state = self.streaming_state.lock().unwrap();
if let StreamingState::Started { .. } = *streaming_state {
return Err(error_msg!(SourceError::Failure, ["Source already started"])); return Err(error_msg!(SourceError::Failure, ["Source already started"]));
} }
let location = &try!(location.as_ref() let location = try!(uri.to_file_path()
.ok_or_else(|| error_msg!(SourceError::Failure, ["No URI provided"]))); .or_else(|_| {
Err(error_msg!(SourceError::Failure,
["Unsupported file URI '{}'", uri.as_str()]))
}));
let file = try!(File::open(location.as_path()).or_else(|err| { let file = try!(File::open(location.as_path()).or_else(|err| {
Err(error_msg!(SourceError::OpenFailed, Err(error_msg!(SourceError::OpenFailed,
@ -126,7 +90,7 @@ impl Source for FileSrc {
err.to_string()])) err.to_string()]))
})); }));
*streaming_state = StreamingState::Started { self.streaming_state = StreamingState::Started {
file: file, file: file,
position: 0, position: 0,
}; };
@ -134,18 +98,14 @@ impl Source for FileSrc {
Ok(()) Ok(())
} }
fn stop(&self) -> Result<(), ErrorMessage> { fn stop(&mut self) -> Result<(), ErrorMessage> {
let mut streaming_state = self.streaming_state.lock().unwrap(); self.streaming_state = StreamingState::Stopped;
*streaming_state = StreamingState::Stopped;
Ok(()) Ok(())
} }
fn fill(&self, offset: u64, data: &mut [u8]) -> Result<usize, FlowError> { fn fill(&mut self, offset: u64, data: &mut [u8]) -> Result<usize, FlowError> {
let mut streaming_state = self.streaming_state.lock().unwrap(); let (file, position) = match self.streaming_state {
let (file, position) = match *streaming_state {
StreamingState::Started { ref mut file, ref mut position } => (file, position), StreamingState::Started { ref mut file, ref mut position } => (file, position),
StreamingState::Stopped => { StreamingState::Stopped => {
return Err(FlowError::Error(error_msg!(SourceError::Failure, ["Not started yet"]))); return Err(FlowError::Error(error_msg!(SourceError::Failure, ["Not started yet"])));
@ -172,7 +132,7 @@ impl Source for FileSrc {
Ok(size) Ok(size)
} }
fn do_seek(&self, _: u64, _: u64) -> Result<(), ErrorMessage> { fn seek(&mut self, _: u64, _: u64) -> Result<(), ErrorMessage> {
Ok(()) Ok(())
} }
} }

View file

@ -23,20 +23,14 @@ use hyper::header::{ContentLength, ContentRange, ContentRangeSpec, Range, ByteRa
use hyper::client::Client; use hyper::client::Client;
use hyper::client::response::Response; use hyper::client::response::Response;
use std::sync::Mutex;
use error::*; use error::*;
use rssource::*; use rssource::*;
#[derive(Debug)]
struct Settings {
url: Option<Url>,
}
#[derive(Debug)] #[derive(Debug)]
enum StreamingState { enum StreamingState {
Stopped, Stopped,
Started { Started {
uri: Url,
response: Response, response: Response,
seekable: bool, seekable: bool,
position: u64, position: u64,
@ -48,37 +42,24 @@ enum StreamingState {
#[derive(Debug)] #[derive(Debug)]
pub struct HttpSrc { pub struct HttpSrc {
controller: SourceController, streaming_state: StreamingState,
settings: Mutex<Settings>,
streaming_state: Mutex<StreamingState>,
client: Client, client: Client,
} }
unsafe impl Sync for HttpSrc {}
unsafe impl Send for HttpSrc {}
impl HttpSrc { impl HttpSrc {
pub fn new(controller: SourceController) -> HttpSrc { pub fn new() -> HttpSrc {
HttpSrc { HttpSrc {
controller: controller, streaming_state: StreamingState::Stopped,
settings: Mutex::new(Settings { url: None }),
streaming_state: Mutex::new(StreamingState::Stopped),
client: Client::new(), client: Client::new(),
} }
} }
pub fn new_boxed(controller: SourceController) -> Box<Source> { pub fn new_boxed() -> Box<Source> {
Box::new(HttpSrc::new(controller)) Box::new(HttpSrc::new())
} }
fn do_request(&self, start: u64, stop: u64) -> Result<StreamingState, ErrorMessage> { fn do_request(&self, uri: Url, start: u64, stop: u64) -> Result<StreamingState, ErrorMessage> {
let url = &self.settings.lock().unwrap().url; let mut req = self.client.get(uri.clone());
let url = try!(url.as_ref()
.ok_or_else(|| error_msg!(SourceError::Failure, ["No URI provided"])));
let mut req = self.client.get(url.clone());
if start != 0 || stop != u64::MAX { if start != 0 || stop != u64::MAX {
req = if stop == u64::MAX { req = if stop == u64::MAX {
@ -90,12 +71,12 @@ impl HttpSrc {
let response = try!(req.send().or_else(|err| { let response = try!(req.send().or_else(|err| {
Err(error_msg!(SourceError::ReadFailed, Err(error_msg!(SourceError::ReadFailed,
["Failed to fetch {}: {}", url, err.to_string()])) ["Failed to fetch {}: {}", uri, err.to_string()]))
})); }));
if !response.status.is_success() { if !response.status.is_success() {
return Err(error_msg!(SourceError::ReadFailed, return Err(error_msg!(SourceError::ReadFailed,
["Failed to fetch {}: {}", url, response.status])); ["Failed to fetch {}: {}", uri, response.status]));
} }
let size = if let Some(&ContentLength(content_length)) = response.headers.get() { let size = if let Some(&ContentLength(content_length)) = response.headers.get() {
@ -128,6 +109,7 @@ impl HttpSrc {
} }
Ok(StreamingState::Started { Ok(StreamingState::Started {
uri: uri,
response: response, response: response,
seekable: seekable, seekable: seekable,
position: 0, position: 0,
@ -138,95 +120,69 @@ impl HttpSrc {
} }
} }
fn validate_uri(uri: &Url) -> Result<(), UriError> {
if uri.scheme() != "http" && uri.scheme() != "https" {
return Err(UriError::new(UriErrorKind::UnsupportedProtocol,
Some(format!("Unsupported URI '{}'", uri.as_str()))));
}
Ok(())
}
impl Source for HttpSrc { impl Source for HttpSrc {
fn get_controller(&self) -> &SourceController { fn uri_validator(&self) -> Box<UriValidator> {
&self.controller Box::new(validate_uri)
}
fn set_uri(&self, uri: Option<Url>) -> Result<(), UriError> {
let url = &mut self.settings.lock().unwrap().url;
match uri {
None => {
*url = None;
Ok(())
}
Some(uri) => {
if uri.scheme() != "http" && uri.scheme() != "https" {
*url = None;
return Err(UriError::new(UriErrorKind::UnsupportedProtocol,
Some(format!("Unsupported URI '{}'", uri.as_str()))));
}
*url = Some(uri);
Ok(())
}
}
}
fn get_uri(&self) -> Option<Url> {
let url = &self.settings.lock().unwrap().url;
url.as_ref().cloned()
} }
fn is_seekable(&self) -> bool { fn is_seekable(&self) -> bool {
let streaming_state = self.streaming_state.lock().unwrap(); match self.streaming_state {
match *streaming_state {
StreamingState::Started { seekable, .. } => seekable, StreamingState::Started { seekable, .. } => seekable,
_ => false, _ => false,
} }
} }
fn get_size(&self) -> u64 { fn get_size(&self) -> u64 {
let streaming_state = self.streaming_state.lock().unwrap(); match self.streaming_state {
match *streaming_state {
StreamingState::Started { size, .. } => size, StreamingState::Started { size, .. } => size,
_ => u64::MAX, _ => u64::MAX,
} }
} }
fn start(&self) -> Result<(), ErrorMessage> { fn start(&mut self, uri: &Url) -> Result<(), ErrorMessage> {
let mut streaming_state = self.streaming_state.lock().unwrap(); self.streaming_state = StreamingState::Stopped;
*streaming_state = StreamingState::Stopped; self.streaming_state = try!(self.do_request(uri.clone(), 0, u64::MAX));
let new_state = try!(self.do_request(0, u64::MAX));
*streaming_state = new_state;
Ok(())
}
fn stop(&self) -> Result<(), ErrorMessage> {
let mut streaming_state = self.streaming_state.lock().unwrap();
*streaming_state = StreamingState::Stopped;
Ok(()) Ok(())
} }
fn do_seek(&self, start: u64, stop: u64) -> Result<(), ErrorMessage> { fn stop(&mut self) -> Result<(), ErrorMessage> {
let mut streaming_state = self.streaming_state.lock().unwrap(); self.streaming_state = StreamingState::Stopped;
*streaming_state = StreamingState::Stopped;
let new_state = try!(self.do_request(start, stop));
*streaming_state = new_state;
Ok(()) Ok(())
} }
fn fill(&self, offset: u64, data: &mut [u8]) -> Result<usize, FlowError> { fn seek(&mut self, start: u64, stop: u64) -> Result<(), ErrorMessage> {
let mut streaming_state = self.streaming_state.lock().unwrap(); let (position, old_stop, uri) = match self.streaming_state {
StreamingState::Started { position, stop, ref uri, .. } => {
if let StreamingState::Started { position, stop, .. } = *streaming_state { (position, stop, uri.clone())
if position != offset {
*streaming_state = StreamingState::Stopped;
let new_state = try!(self.do_request(offset, stop)
.or_else(|err| Err(FlowError::Error(err))));
*streaming_state = new_state;
} }
StreamingState::Stopped => {
return Err(error_msg!(SourceError::Failure, ["Not started yet"]));
}
};
if position == start && old_stop == stop {
return Ok(());
} }
let (response, position) = match *streaming_state { self.streaming_state = StreamingState::Stopped;
self.streaming_state = try!(self.do_request(uri, start, stop));
Ok(())
}
fn fill(&mut self, offset: u64, data: &mut [u8]) -> Result<usize, FlowError> {
let (response, position) = match self.streaming_state {
StreamingState::Started { ref mut response, ref mut position, .. } => { StreamingState::Started { ref mut response, ref mut position, .. } => {
(response, position) (response, position)
} }
@ -235,6 +191,13 @@ impl Source for HttpSrc {
} }
}; };
if *position != offset {
return Err(FlowError::Error(error_msg!(SourceError::SeekFailed,
["Got unexpected offset {}, expected {}",
offset,
position])));
}
let size = try!(response.read(data).or_else(|err| { let size = try!(response.read(data).or_else(|err| {
Err(FlowError::Error(error_msg!(SourceError::ReadFailed, Err(FlowError::Error(error_msg!(SourceError::ReadFailed,
["Failed to read at {}: {}", offset, err.to_string()]))) ["Failed to read at {}: {}", offset, err.to_string()])))
@ -245,6 +208,7 @@ impl Source for HttpSrc {
} }
*position += size as u64; *position += size as u64;
Ok(size) Ok(size)
} }
} }

View file

@ -246,15 +246,6 @@ gst_rs_sink_uri_handler_init (gpointer g_iface, gpointer iface_data)
iface->set_uri = gst_rs_sink_uri_set_uri; iface->set_uri = gst_rs_sink_uri_set_uri;
} }
void
gst_rs_sink_error (GstRsSink * sink, GQuark error_domain, gint error_code,
const gchar * message, const gchar * debug, const gchar * file,
const gchar * function, guint line)
{
gst_element_message_full (GST_ELEMENT (sink), GST_MESSAGE_ERROR, error_domain,
error_code, g_strdup (message), g_strdup (debug), file, function, line);
}
gboolean gboolean
gst_rs_sink_plugin_init (GstPlugin * plugin) gst_rs_sink_plugin_init (GstPlugin * plugin)
{ {

View file

@ -22,6 +22,8 @@ use std::ffi::{CStr, CString};
use std::slice; use std::slice;
use std::ptr; use std::ptr;
use std::sync::Mutex;
use url::Url; use url::Url;
use utils::*; use utils::*;
@ -48,117 +50,82 @@ impl ToGError for SinkError {
} }
} }
#[derive(Debug)] pub struct SinkWrapper {
pub struct SinkController { sink_raw: *mut c_void,
sink: *mut c_void, uri: Mutex<(Option<Url>, bool)>,
uri_validator: Box<UriValidator>,
sink: Mutex<Box<Sink>>,
} }
impl SinkController { pub trait Sink {
fn new(sink: *mut c_void) -> SinkController { fn uri_validator(&self) -> Box<UriValidator>;
SinkController { sink: sink }
}
pub fn error(&self, error: &ErrorMessage) { fn start(&mut self, uri: &Url) -> Result<(), ErrorMessage>;
extern "C" { fn stop(&mut self) -> Result<(), ErrorMessage>;
fn gst_rs_sink_error(sink: *mut c_void,
error_domain: u32,
error_code: i32,
message: *const c_char,
debug: *const c_char,
filename: *const c_char,
function: *const c_char,
line: u32);
}
let ErrorMessage { error_domain, fn render(&mut self, data: &[u8]) -> Result<(), FlowError>;
error_code,
ref message,
ref debug,
filename,
function,
line } = *error;
let message_cstr = message.as_ref().map(|m| CString::new(m.as_bytes()).unwrap());
let message_ptr = message_cstr.as_ref().map_or(ptr::null(), |m| m.as_ptr());
let debug_cstr = debug.as_ref().map(|m| CString::new(m.as_bytes()).unwrap());
let debug_ptr = debug_cstr.as_ref().map_or(ptr::null(), |m| m.as_ptr());
let file_cstr = CString::new(filename.as_bytes()).unwrap();
let file_ptr = file_cstr.as_ptr();
let function_cstr = CString::new(function.as_bytes()).unwrap();
let function_ptr = function_cstr.as_ptr();
unsafe {
gst_rs_sink_error(self.sink,
error_domain,
error_code,
message_ptr,
debug_ptr,
file_ptr,
function_ptr,
line);
}
}
} }
pub trait Sink: Sync + Send { impl SinkWrapper {
fn get_controller(&self) -> &SinkController; fn new(sink_raw: *mut c_void, sink: Box<Sink>) -> SinkWrapper {
SinkWrapper {
// Called from any thread at any time sink_raw: sink_raw,
fn set_uri(&self, uri: Option<Url>) -> Result<(), UriError>; uri: Mutex::new((None, false)),
fn get_uri(&self) -> Option<Url>; uri_validator: sink.uri_validator(),
sink: Mutex::new(sink),
// Called from the streaming thread only }
fn start(&self) -> Result<(), ErrorMessage>; }
fn stop(&self) -> Result<(), ErrorMessage>;
fn render(&self, data: &[u8]) -> Result<(), FlowError>;
} }
#[no_mangle] #[no_mangle]
pub extern "C" fn sink_new(sink: *mut c_void, pub extern "C" fn sink_new(sink: *mut c_void,
create_instance: fn(controller: SinkController) -> Box<Sink>) create_instance: fn() -> Box<Sink>)
-> *mut Box<Sink> { -> *mut SinkWrapper {
Box::into_raw(Box::new(create_instance(SinkController::new(sink)))) Box::into_raw(Box::new(SinkWrapper::new(sink, create_instance())))
} }
#[no_mangle] #[no_mangle]
pub unsafe extern "C" fn sink_drop(ptr: *mut Box<Sink>) { pub unsafe extern "C" fn sink_drop(ptr: *mut SinkWrapper) {
Box::from_raw(ptr); Box::from_raw(ptr);
} }
#[no_mangle] #[no_mangle]
pub unsafe extern "C" fn sink_set_uri(ptr: *mut Box<Sink>, pub unsafe extern "C" fn sink_set_uri(ptr: *mut SinkWrapper,
uri_ptr: *const c_char, uri_ptr: *const c_char,
cerr: *mut c_void) cerr: *mut c_void)
-> GBoolean { -> GBoolean {
let sink: &mut Box<Sink> = &mut *ptr; let wrap: &mut SinkWrapper = &mut *ptr;
let uri_storage = &mut wrap.uri.lock().unwrap();
if uri_storage.1 {
UriError::new(UriErrorKind::BadState, Some("Already started".to_string()))
.into_gerror(cerr);
return GBoolean::False;
}
uri_storage.0 = None;
if uri_ptr.is_null() { if uri_ptr.is_null() {
if let Err(err) = sink.set_uri(None) { GBoolean::True
err.into_gerror(cerr);
GBoolean::False
} else {
GBoolean::True
}
} else { } else {
let uri_str = CStr::from_ptr(uri_ptr).to_str().unwrap(); let uri_str = CStr::from_ptr(uri_ptr).to_str().unwrap();
match Url::parse(uri_str) { match Url::parse(uri_str) {
Ok(uri) => { Ok(uri) => {
if let Err(err) = sink.set_uri(Some(uri)) { if let Err(err) = (*wrap.uri_validator)(&uri) {
err.into_gerror(cerr); err.into_gerror(cerr);
GBoolean::False GBoolean::False
} else { } else {
uri_storage.0 = Some(uri);
GBoolean::True GBoolean::True
} }
} }
Err(err) => { Err(err) => {
let _ = sink.set_uri(None);
UriError::new(UriErrorKind::BadUri, UriError::new(UriErrorKind::BadUri,
Some(format!("Failed to parse URI '{}': {}", uri_str, err))) Some(format!("Failed to parse URI '{}': {}", uri_str, err)))
.into_gerror(cerr); .into_gerror(cerr);
GBoolean::False GBoolean::False
} }
} }
@ -166,21 +133,23 @@ pub unsafe extern "C" fn sink_set_uri(ptr: *mut Box<Sink>,
} }
#[no_mangle] #[no_mangle]
pub unsafe extern "C" fn sink_get_uri(ptr: *const Box<Sink>) -> *mut c_char { pub unsafe extern "C" fn sink_get_uri(ptr: *const SinkWrapper) -> *mut c_char {
let sink: &Box<Sink> = &*ptr; let wrap: &SinkWrapper = &*ptr;
let uri_storage = &mut wrap.uri.lock().unwrap();
match sink.get_uri() { match uri_storage.0 {
Some(uri) => CString::new(uri.into_string().into_bytes()).unwrap().into_raw(), Some(ref uri) => CString::new(uri.as_ref().as_bytes()).unwrap().into_raw(),
None => ptr::null_mut(), None => ptr::null_mut(),
} }
} }
#[no_mangle] #[no_mangle]
pub unsafe extern "C" fn sink_render(ptr: *mut Box<Sink>, pub unsafe extern "C" fn sink_render(ptr: *mut SinkWrapper,
data_ptr: *const u8, data_ptr: *const u8,
data_len: usize) data_len: usize)
-> GstFlowReturn { -> GstFlowReturn {
let sink: &mut Box<Sink> = &mut *ptr; let wrap: &mut SinkWrapper = &mut *ptr;
let sink = &mut wrap.sink.lock().unwrap();
let data = slice::from_raw_parts(data_ptr, data_len); let data = slice::from_raw_parts(data_ptr, data_len);
match sink.render(data) { match sink.render(data) {
@ -188,7 +157,7 @@ pub unsafe extern "C" fn sink_render(ptr: *mut Box<Sink>,
Err(flow_error) => { Err(flow_error) => {
match flow_error { match flow_error {
FlowError::NotNegotiated(ref msg) | FlowError::NotNegotiated(ref msg) |
FlowError::Error(ref msg) => sink.get_controller().error(msg), FlowError::Error(ref msg) => msg.post(wrap.sink_raw),
_ => (), _ => (),
} }
flow_error.to_native() flow_error.to_native()
@ -197,26 +166,45 @@ pub unsafe extern "C" fn sink_render(ptr: *mut Box<Sink>,
} }
#[no_mangle] #[no_mangle]
pub unsafe extern "C" fn sink_start(ptr: *mut Box<Sink>) -> GBoolean { pub unsafe extern "C" fn sink_start(ptr: *mut SinkWrapper) -> GBoolean {
let sink: &mut Box<Sink> = &mut *ptr; let wrap: &mut SinkWrapper = &mut *ptr;
let sink = &mut wrap.sink.lock().unwrap();
let uri_storage = &mut wrap.uri.lock().unwrap();
match sink.start() { let (uri, started) = match **uri_storage {
Ok(..) => GBoolean::True, (Some(ref uri), ref mut started) => (uri, started),
(None, _) => {
error_msg!(SinkError::OpenFailed, ["No URI given"]).post(wrap.sink_raw);
return GBoolean::False;
}
};
match sink.start(uri) {
Ok(..) => {
*started = true;
GBoolean::True
}
Err(ref msg) => { Err(ref msg) => {
sink.get_controller().error(msg); msg.post(wrap.sink_raw);
GBoolean::False GBoolean::False
} }
} }
} }
#[no_mangle] #[no_mangle]
pub unsafe extern "C" fn sink_stop(ptr: *mut Box<Sink>) -> GBoolean { pub unsafe extern "C" fn sink_stop(ptr: *mut SinkWrapper) -> GBoolean {
let sink: &mut Box<Sink> = &mut *ptr; let wrap: &mut SinkWrapper = &mut *ptr;
let sink = &mut wrap.sink.lock().unwrap();
let uri_storage = &mut wrap.uri.lock().unwrap();
match sink.stop() { match sink.stop() {
Ok(..) => GBoolean::True, Ok(..) => {
uri_storage.1 = false;
GBoolean::True
}
Err(ref msg) => { Err(ref msg) => {
sink.get_controller().error(msg); msg.post(wrap.sink_raw);
GBoolean::False GBoolean::False
} }
} }

View file

@ -38,7 +38,7 @@ extern void *source_new (GstRsSrc * source, void *create_instance);
extern void source_drop (void *rssource); extern void source_drop (void *rssource);
extern GstFlowReturn source_fill (void *rssource, extern GstFlowReturn source_fill (void *rssource,
uint64_t offset, void *data, size_t * data_len); uint64_t offset, void *data, size_t * data_len);
extern gboolean source_do_seek (void *rssource, uint64_t start, uint64_t stop); extern gboolean source_seek (void *rssource, uint64_t start, uint64_t stop);
extern gboolean source_set_uri (void *rssource, const char *uri, GError ** err); extern gboolean source_set_uri (void *rssource, const char *uri, GError ** err);
extern char *source_get_uri (void *rssource); extern char *source_get_uri (void *rssource);
extern uint64_t source_get_size (void *rssource); extern uint64_t source_get_size (void *rssource);
@ -238,7 +238,7 @@ gst_rs_src_do_seek (GstBaseSrc * basesrc, GstSegment * segment)
GstRsSrc *src = GST_RS_SRC (basesrc); GstRsSrc *src = GST_RS_SRC (basesrc);
gboolean ret; gboolean ret;
ret = source_do_seek (src->instance, segment->start, segment->stop); ret = source_seek (src->instance, segment->start, segment->stop);
if (!ret) if (!ret)
return FALSE; return FALSE;
@ -291,15 +291,6 @@ gst_rs_src_uri_handler_init (gpointer g_iface, gpointer iface_data)
iface->set_uri = gst_rs_src_uri_set_uri; iface->set_uri = gst_rs_src_uri_set_uri;
} }
void
gst_rs_source_error (GstRsSrc * src, GQuark error_domain, gint error_code,
const gchar * message, const gchar * debug, const gchar * file,
const gchar * function, guint line)
{
gst_element_message_full (GST_ELEMENT (src), GST_MESSAGE_ERROR, error_domain,
error_code, g_strdup (message), g_strdup (debug), file, function, line);
}
gboolean gboolean
gst_rs_source_plugin_init (GstPlugin * plugin) gst_rs_source_plugin_init (GstPlugin * plugin)
{ {

View file

@ -21,6 +21,8 @@ use std::ffi::{CStr, CString};
use std::slice; use std::slice;
use std::ptr; use std::ptr;
use std::sync::Mutex;
use url::Url; use url::Url;
use utils::*; use utils::*;
@ -47,122 +49,85 @@ impl ToGError for SourceError {
} }
} }
#[derive(Debug)] pub struct SourceWrapper {
pub struct SourceController { source_raw: *mut c_void,
source: *mut c_void, uri: Mutex<(Option<Url>, bool)>,
uri_validator: Box<UriValidator>,
source: Mutex<Box<Source>>,
} }
impl SourceController { pub trait Source {
fn new(source: *mut c_void) -> SourceController { fn uri_validator(&self) -> Box<UriValidator>;
SourceController { source: source }
}
pub fn error(&self, error: &ErrorMessage) {
extern "C" {
fn gst_rs_source_error(source: *mut c_void,
error_domain: u32,
error_code: i32,
message: *const c_char,
debug: *const c_char,
filename: *const c_char,
function: *const c_char,
line: u32);
}
let ErrorMessage { error_domain,
error_code,
ref message,
ref debug,
filename,
function,
line } = *error;
let message_cstr = message.as_ref().map(|m| CString::new(m.as_bytes()).unwrap());
let message_ptr = message_cstr.as_ref().map_or(ptr::null(), |m| m.as_ptr());
let debug_cstr = debug.as_ref().map(|m| CString::new(m.as_bytes()).unwrap());
let debug_ptr = debug_cstr.as_ref().map_or(ptr::null(), |m| m.as_ptr());
let file_cstr = CString::new(filename.as_bytes()).unwrap();
let file_ptr = file_cstr.as_ptr();
let function_cstr = CString::new(function.as_bytes()).unwrap();
let function_ptr = function_cstr.as_ptr();
unsafe {
gst_rs_source_error(self.source,
error_domain,
error_code,
message_ptr,
debug_ptr,
file_ptr,
function_ptr,
line);
}
}
}
pub trait Source: Sync + Send {
fn get_controller(&self) -> &SourceController;
// Called from any thread at any time
fn set_uri(&self, uri: Option<Url>) -> Result<(), UriError>;
fn get_uri(&self) -> Option<Url>;
// Called from any thread between start/stop
fn is_seekable(&self) -> bool; fn is_seekable(&self) -> bool;
// Called from the streaming thread only
fn start(&self) -> Result<(), ErrorMessage>;
fn stop(&self) -> Result<(), ErrorMessage>;
fn fill(&self, offset: u64, data: &mut [u8]) -> Result<usize, FlowError>;
fn do_seek(&self, start: u64, stop: u64) -> Result<(), ErrorMessage>;
fn get_size(&self) -> u64; fn get_size(&self) -> u64;
fn start(&mut self, uri: &Url) -> Result<(), ErrorMessage>;
fn stop(&mut self) -> Result<(), ErrorMessage>;
fn fill(&mut self, offset: u64, data: &mut [u8]) -> Result<usize, FlowError>;
fn seek(&mut self, start: u64, stop: u64) -> Result<(), ErrorMessage>;
}
impl SourceWrapper {
fn new(source_raw: *mut c_void, source: Box<Source>) -> SourceWrapper {
SourceWrapper {
source_raw: source_raw,
uri: Mutex::new((None, false)),
uri_validator: source.uri_validator(),
source: Mutex::new(source),
}
}
} }
#[no_mangle] #[no_mangle]
pub extern "C" fn source_new(source: *mut c_void, pub extern "C" fn source_new(source: *mut c_void,
create_instance: fn(controller: SourceController) -> Box<Source>) create_instance: fn() -> Box<Source>)
-> *mut Box<Source> { -> *mut SourceWrapper {
Box::into_raw(Box::new(create_instance(SourceController::new(source)))) Box::into_raw(Box::new(SourceWrapper::new(source, create_instance())))
} }
#[no_mangle] #[no_mangle]
pub unsafe extern "C" fn source_drop(ptr: *mut Box<Source>) { pub unsafe extern "C" fn source_drop(ptr: *mut SourceWrapper) {
Box::from_raw(ptr); Box::from_raw(ptr);
} }
#[no_mangle] #[no_mangle]
pub unsafe extern "C" fn source_set_uri(ptr: *mut Box<Source>, pub unsafe extern "C" fn source_set_uri(ptr: *mut SourceWrapper,
uri_ptr: *const c_char, uri_ptr: *const c_char,
cerr: *mut c_void) cerr: *mut c_void)
-> GBoolean { -> GBoolean {
let source: &mut Box<Source> = &mut *ptr; let wrap: &mut SourceWrapper = &mut *ptr;
let uri_storage = &mut wrap.uri.lock().unwrap();
if uri_storage.1 {
UriError::new(UriErrorKind::BadState, Some("Already started".to_string()))
.into_gerror(cerr);
return GBoolean::False;
}
uri_storage.0 = None;
if uri_ptr.is_null() { if uri_ptr.is_null() {
if let Err(err) = source.set_uri(None) { GBoolean::True
err.into_gerror(cerr);
GBoolean::False
} else {
GBoolean::True
}
} else { } else {
let uri_str = CStr::from_ptr(uri_ptr).to_str().unwrap(); let uri_str = CStr::from_ptr(uri_ptr).to_str().unwrap();
match Url::parse(uri_str) { match Url::parse(uri_str) {
Ok(uri) => { Ok(uri) => {
if let Err(err) = source.set_uri(Some(uri)) { if let Err(err) = (*wrap.uri_validator)(&uri) {
err.into_gerror(cerr); err.into_gerror(cerr);
GBoolean::False GBoolean::False
} else { } else {
uri_storage.0 = Some(uri);
GBoolean::True GBoolean::True
} }
} }
Err(err) => { Err(err) => {
let _ = source.set_uri(None);
UriError::new(UriErrorKind::BadUri, UriError::new(UriErrorKind::BadUri,
Some(format!("Failed to parse URI '{}': {}", uri_str, err))) Some(format!("Failed to parse URI '{}': {}", uri_str, err)))
.into_gerror(cerr); .into_gerror(cerr);
GBoolean::False GBoolean::False
} }
} }
@ -170,23 +135,84 @@ pub unsafe extern "C" fn source_set_uri(ptr: *mut Box<Source>,
} }
#[no_mangle] #[no_mangle]
pub unsafe extern "C" fn source_get_uri(ptr: *mut Box<Source>) -> *mut c_char { pub unsafe extern "C" fn source_get_uri(ptr: *mut SourceWrapper) -> *mut c_char {
let source: &mut Box<Source> = &mut *ptr; let wrap: &SourceWrapper = &*ptr;
let uri_storage = &mut wrap.uri.lock().unwrap();
match source.get_uri() { match uri_storage.0 {
Some(uri) => CString::new(uri.into_string().into_bytes()).unwrap().into_raw(), Some(ref uri) => CString::new(uri.as_ref().as_bytes()).unwrap().into_raw(),
None => ptr::null_mut(), None => ptr::null_mut(),
} }
} }
#[no_mangle] #[no_mangle]
pub unsafe extern "C" fn source_fill(ptr: *mut Box<Source>, pub unsafe extern "C" fn source_is_seekable(ptr: *const SourceWrapper) -> GBoolean {
let wrap: &SourceWrapper = &*ptr;
let source = &wrap.source.lock().unwrap();
GBoolean::from_bool(source.is_seekable())
}
#[no_mangle]
pub unsafe extern "C" fn source_get_size(ptr: *const SourceWrapper) -> u64 {
let wrap: &SourceWrapper = &*ptr;
let source = &wrap.source.lock().unwrap();
source.get_size()
}
#[no_mangle]
pub unsafe extern "C" fn source_start(ptr: *mut SourceWrapper) -> GBoolean {
let wrap: &mut SourceWrapper = &mut *ptr;
let source = &mut wrap.source.lock().unwrap();
let uri_storage = &mut wrap.uri.lock().unwrap();
let (uri, started) = match **uri_storage {
(Some(ref uri), ref mut started) => (uri, started),
(None, _) => {
error_msg!(SourceError::OpenFailed, ["No URI given"]).post(wrap.source_raw);
return GBoolean::False;
}
};
match source.start(uri) {
Ok(..) => {
*started = true;
GBoolean::True
}
Err(ref msg) => {
msg.post(wrap.source_raw);
GBoolean::False
}
}
}
#[no_mangle]
pub unsafe extern "C" fn source_stop(ptr: *mut SourceWrapper) -> GBoolean {
let wrap: &mut SourceWrapper = &mut *ptr;
let source = &mut wrap.source.lock().unwrap();
let uri_storage = &mut wrap.uri.lock().unwrap();
match source.stop() {
Ok(..) => {
uri_storage.1 = false;
GBoolean::True
}
Err(ref msg) => {
msg.post(wrap.source_raw);
GBoolean::False
}
}
}
#[no_mangle]
pub unsafe extern "C" fn source_fill(ptr: *mut SourceWrapper,
offset: u64, offset: u64,
data_ptr: *mut u8, data_ptr: *mut u8,
data_len_ptr: *mut usize) data_len_ptr: *mut usize)
-> GstFlowReturn { -> GstFlowReturn {
let source: &mut Box<Source> = &mut *ptr; let wrap: &mut SourceWrapper = &mut *ptr;
let source = &mut wrap.source.lock().unwrap();
let mut data_len: &mut usize = &mut *data_len_ptr; let mut data_len: &mut usize = &mut *data_len_ptr;
let mut data = slice::from_raw_parts_mut(data_ptr, *data_len); let mut data = slice::from_raw_parts_mut(data_ptr, *data_len);
@ -198,7 +224,7 @@ pub unsafe extern "C" fn source_fill(ptr: *mut Box<Source>,
Err(flow_error) => { Err(flow_error) => {
match flow_error { match flow_error {
FlowError::NotNegotiated(ref msg) | FlowError::NotNegotiated(ref msg) |
FlowError::Error(ref msg) => source.get_controller().error(msg), FlowError::Error(ref msg) => msg.post(wrap.source_raw),
_ => (), _ => (),
} }
flow_error.to_native() flow_error.to_native()
@ -207,53 +233,14 @@ pub unsafe extern "C" fn source_fill(ptr: *mut Box<Source>,
} }
#[no_mangle] #[no_mangle]
pub unsafe extern "C" fn source_get_size(ptr: *const Box<Source>) -> u64 { pub unsafe extern "C" fn source_seek(ptr: *mut SourceWrapper, start: u64, stop: u64) -> GBoolean {
let source: &Box<Source> = &*ptr; let wrap: &mut SourceWrapper = &mut *ptr;
let source = &mut wrap.source.lock().unwrap();
source.get_size() match source.seek(start, stop) {
}
#[no_mangle]
pub unsafe extern "C" fn source_start(ptr: *mut Box<Source>) -> GBoolean {
let source: &mut Box<Source> = &mut *ptr;
match source.start() {
Ok(..) => GBoolean::True, Ok(..) => GBoolean::True,
Err(ref msg) => { Err(ref msg) => {
source.get_controller().error(msg); msg.post(wrap.source_raw);
GBoolean::False
}
}
}
#[no_mangle]
pub unsafe extern "C" fn source_stop(ptr: *mut Box<Source>) -> GBoolean {
let source: &mut Box<Source> = &mut *ptr;
match source.stop() {
Ok(..) => GBoolean::True,
Err(ref msg) => {
source.get_controller().error(msg);
GBoolean::False
}
}
}
#[no_mangle]
pub unsafe extern "C" fn source_is_seekable(ptr: *const Box<Source>) -> GBoolean {
let source: &Box<Source> = &*ptr;
GBoolean::from_bool(source.is_seekable())
}
#[no_mangle]
pub unsafe extern "C" fn source_do_seek(ptr: *mut Box<Source>, start: u64, stop: u64) -> GBoolean {
let source: &mut Box<Source> = &mut *ptr;
match source.do_seek(start, stop) {
Ok(..) => GBoolean::True,
Err(ref msg) => {
source.get_controller().error(msg);
GBoolean::False GBoolean::False
} }
} }