2016-05-15 15:54:09 +00:00
|
|
|
// Copyright (C) 2016 Sebastian Dröge <sebastian@centricular.com>
|
|
|
|
//
|
|
|
|
// This library is free software; you can redistribute it and/or
|
|
|
|
// modify it under the terms of the GNU Library General Public
|
|
|
|
// License as published by the Free Software Foundation; either
|
|
|
|
// version 2 of the License, or (at your option) any later version.
|
|
|
|
//
|
|
|
|
// This library is distributed in the hope that it will be useful,
|
|
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
|
|
// Library General Public License for more details.
|
|
|
|
//
|
|
|
|
// You should have received a copy of the GNU Library General Public
|
|
|
|
// License along with this library; if not, write to the
|
|
|
|
// Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
|
|
|
|
// Boston, MA 02110-1301, USA.
|
|
|
|
|
2016-05-15 14:43:13 +00:00
|
|
|
use libc::c_char;
|
2016-07-20 09:46:03 +00:00
|
|
|
use std::os::raw::c_void;
|
2016-05-14 11:44:49 +00:00
|
|
|
use std::ffi::{CStr, CString};
|
|
|
|
use std::slice;
|
|
|
|
use std::ptr;
|
2016-05-24 20:24:05 +00:00
|
|
|
|
|
|
|
use url::Url;
|
2016-05-14 11:44:49 +00:00
|
|
|
|
|
|
|
use utils::*;
|
2016-08-27 08:16:17 +00:00
|
|
|
use error::*;
|
|
|
|
|
|
|
|
#[derive(Debug)]
|
|
|
|
pub enum SourceError {
|
|
|
|
Failure,
|
|
|
|
OpenFailed,
|
|
|
|
NotFound,
|
|
|
|
ReadFailed,
|
|
|
|
SeekFailed,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl ToGError for SourceError {
|
|
|
|
fn to_gerror(&self) -> (u32, i32) {
|
|
|
|
match *self {
|
|
|
|
SourceError::Failure => (gst_library_error_domain(), 1),
|
|
|
|
SourceError::OpenFailed => (gst_resource_error_domain(), 5),
|
|
|
|
SourceError::NotFound => (gst_resource_error_domain(), 3),
|
|
|
|
SourceError::ReadFailed => (gst_resource_error_domain(), 9),
|
|
|
|
SourceError::SeekFailed => (gst_resource_error_domain(), 11),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2016-05-14 11:44:49 +00:00
|
|
|
|
2016-07-20 09:46:03 +00:00
|
|
|
#[derive(Debug)]
|
|
|
|
pub struct SourceController {
|
|
|
|
source: *mut c_void,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl SourceController {
|
|
|
|
fn new(source: *mut c_void) -> SourceController {
|
|
|
|
SourceController { source: source }
|
|
|
|
}
|
2016-08-27 08:16:17 +00:00
|
|
|
|
|
|
|
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);
|
|
|
|
}
|
|
|
|
}
|
2016-07-20 09:46:03 +00:00
|
|
|
}
|
|
|
|
|
2016-05-15 08:55:52 +00:00
|
|
|
pub trait Source: Sync + Send {
|
2016-08-24 21:08:06 +00:00
|
|
|
fn get_controller(&self) -> &SourceController;
|
|
|
|
|
2016-05-23 18:15:43 +00:00
|
|
|
// Called from any thread at any time
|
2016-08-24 21:47:56 +00:00
|
|
|
fn set_uri(&self, uri: Option<Url>) -> Result<(), UriError>;
|
2016-05-24 20:24:05 +00:00
|
|
|
fn get_uri(&self) -> Option<Url>;
|
2016-05-23 18:15:43 +00:00
|
|
|
|
|
|
|
// Called from any thread between start/stop
|
2016-05-14 11:44:49 +00:00
|
|
|
fn is_seekable(&self) -> bool;
|
2016-05-23 18:15:43 +00:00
|
|
|
|
|
|
|
// Called from the streaming thread only
|
2016-08-27 08:16:17 +00:00
|
|
|
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>;
|
2016-05-23 18:15:43 +00:00
|
|
|
fn get_size(&self) -> u64;
|
2016-05-14 11:44:49 +00:00
|
|
|
}
|
|
|
|
|
2016-07-20 09:46:03 +00:00
|
|
|
#[no_mangle]
|
|
|
|
pub extern "C" fn source_new(source: *mut c_void,
|
2016-08-10 16:27:38 +00:00
|
|
|
create_instance: fn(controller: SourceController) -> Box<Source>)
|
2016-07-20 09:46:03 +00:00
|
|
|
-> *mut Box<Source> {
|
2016-08-10 16:27:38 +00:00
|
|
|
Box::into_raw(Box::new(create_instance(SourceController::new(source))))
|
2016-07-20 09:46:03 +00:00
|
|
|
}
|
|
|
|
|
2016-05-14 11:44:49 +00:00
|
|
|
#[no_mangle]
|
2016-08-22 20:03:06 +00:00
|
|
|
pub unsafe extern "C" fn source_drop(ptr: *mut Box<Source>) {
|
|
|
|
Box::from_raw(ptr);
|
2016-05-14 11:44:49 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
#[no_mangle]
|
2016-08-22 21:07:44 +00:00
|
|
|
pub unsafe extern "C" fn source_set_uri(ptr: *mut Box<Source>,
|
|
|
|
uri_ptr: *const c_char,
|
|
|
|
cerr: *mut c_void)
|
|
|
|
-> GBoolean {
|
2016-08-22 20:03:06 +00:00
|
|
|
let source: &mut Box<Source> = &mut *ptr;
|
2016-05-14 11:44:49 +00:00
|
|
|
|
|
|
|
if uri_ptr.is_null() {
|
2016-08-24 21:47:56 +00:00
|
|
|
if let Err(err) = source.set_uri(None) {
|
|
|
|
err.into_gerror(cerr);
|
2016-08-22 21:07:44 +00:00
|
|
|
GBoolean::False
|
|
|
|
} else {
|
|
|
|
GBoolean::True
|
|
|
|
}
|
2016-05-14 11:44:49 +00:00
|
|
|
} else {
|
2016-08-22 20:03:06 +00:00
|
|
|
let uri_str = CStr::from_ptr(uri_ptr).to_str().unwrap();
|
|
|
|
|
2016-05-24 20:24:05 +00:00
|
|
|
match Url::parse(uri_str) {
|
2016-08-22 21:07:44 +00:00
|
|
|
Ok(uri) => {
|
2016-08-24 21:47:56 +00:00
|
|
|
if let Err(err) = source.set_uri(Some(uri)) {
|
|
|
|
err.into_gerror(cerr);
|
2016-08-22 21:07:44 +00:00
|
|
|
GBoolean::False
|
|
|
|
} else {
|
|
|
|
GBoolean::True
|
|
|
|
}
|
|
|
|
}
|
2016-05-24 20:24:05 +00:00
|
|
|
Err(err) => {
|
2016-08-22 21:07:44 +00:00
|
|
|
let _ = source.set_uri(None);
|
2016-08-24 21:47:56 +00:00
|
|
|
UriError::new(UriErrorKind::BadUri,
|
|
|
|
Some(format!("Failed to parse URI '{}': {}", uri_str, err)))
|
|
|
|
.into_gerror(cerr);
|
2016-05-24 20:24:05 +00:00
|
|
|
GBoolean::False
|
|
|
|
}
|
|
|
|
}
|
2016-05-14 11:44:49 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[no_mangle]
|
2016-08-22 20:03:06 +00:00
|
|
|
pub unsafe extern "C" fn source_get_uri(ptr: *mut Box<Source>) -> *mut c_char {
|
|
|
|
let source: &mut Box<Source> = &mut *ptr;
|
2016-05-14 11:44:49 +00:00
|
|
|
|
|
|
|
match source.get_uri() {
|
2016-07-20 08:28:58 +00:00
|
|
|
Some(uri) => CString::new(uri.into_string().into_bytes()).unwrap().into_raw(),
|
|
|
|
None => ptr::null_mut(),
|
2016-05-14 11:44:49 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[no_mangle]
|
2016-08-22 20:03:06 +00:00
|
|
|
pub unsafe extern "C" fn source_fill(ptr: *mut Box<Source>,
|
|
|
|
offset: u64,
|
|
|
|
data_ptr: *mut u8,
|
|
|
|
data_len_ptr: *mut usize)
|
|
|
|
-> GstFlowReturn {
|
|
|
|
let source: &mut Box<Source> = &mut *ptr;
|
|
|
|
|
|
|
|
let mut data_len: &mut usize = &mut *data_len_ptr;
|
|
|
|
let mut data = slice::from_raw_parts_mut(data_ptr, *data_len);
|
|
|
|
|
2016-05-14 11:44:49 +00:00
|
|
|
match source.fill(offset, data) {
|
|
|
|
Ok(actual_len) => {
|
|
|
|
*data_len = actual_len;
|
|
|
|
GstFlowReturn::Ok
|
2016-07-20 08:28:58 +00:00
|
|
|
}
|
2016-08-27 08:16:17 +00:00
|
|
|
Err(flow_error) => {
|
|
|
|
match flow_error {
|
|
|
|
FlowError::NotNegotiated(ref msg) |
|
|
|
|
FlowError::Error(ref msg) => source.get_controller().error(msg),
|
|
|
|
_ => (),
|
|
|
|
}
|
|
|
|
flow_error.to_native()
|
|
|
|
}
|
2016-05-14 11:44:49 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[no_mangle]
|
2016-08-22 20:03:06 +00:00
|
|
|
pub unsafe extern "C" fn source_get_size(ptr: *const Box<Source>) -> u64 {
|
|
|
|
let source: &Box<Source> = &*ptr;
|
2016-05-14 11:44:49 +00:00
|
|
|
|
2016-08-22 20:03:06 +00:00
|
|
|
source.get_size()
|
2016-05-14 11:44:49 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
#[no_mangle]
|
2016-08-22 20:03:06 +00:00
|
|
|
pub unsafe extern "C" fn source_start(ptr: *mut Box<Source>) -> GBoolean {
|
|
|
|
let source: &mut Box<Source> = &mut *ptr;
|
2016-05-14 11:44:49 +00:00
|
|
|
|
2016-08-27 08:16:17 +00:00
|
|
|
match source.start() {
|
|
|
|
Ok(..) => GBoolean::True,
|
|
|
|
Err(ref msg) => {
|
|
|
|
source.get_controller().error(msg);
|
|
|
|
GBoolean::False
|
|
|
|
}
|
|
|
|
}
|
2016-05-14 11:44:49 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
#[no_mangle]
|
2016-08-22 20:03:06 +00:00
|
|
|
pub unsafe extern "C" fn source_stop(ptr: *mut Box<Source>) -> GBoolean {
|
|
|
|
let source: &mut Box<Source> = &mut *ptr;
|
2016-05-14 11:44:49 +00:00
|
|
|
|
2016-08-27 08:16:17 +00:00
|
|
|
match source.stop() {
|
|
|
|
Ok(..) => GBoolean::True,
|
|
|
|
Err(ref msg) => {
|
|
|
|
source.get_controller().error(msg);
|
|
|
|
GBoolean::False
|
|
|
|
}
|
|
|
|
}
|
2016-05-14 11:44:49 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
#[no_mangle]
|
2016-08-22 20:03:06 +00:00
|
|
|
pub unsafe extern "C" fn source_is_seekable(ptr: *const Box<Source>) -> GBoolean {
|
|
|
|
let source: &Box<Source> = &*ptr;
|
2016-05-14 11:44:49 +00:00
|
|
|
|
|
|
|
GBoolean::from_bool(source.is_seekable())
|
|
|
|
}
|
|
|
|
|
2016-05-14 14:57:25 +00:00
|
|
|
#[no_mangle]
|
2016-08-22 20:03:06 +00:00
|
|
|
pub unsafe extern "C" fn source_do_seek(ptr: *mut Box<Source>, start: u64, stop: u64) -> GBoolean {
|
|
|
|
let source: &mut Box<Source> = &mut *ptr;
|
2016-05-14 14:57:25 +00:00
|
|
|
|
2016-08-27 08:16:17 +00:00
|
|
|
match source.do_seek(start, stop) {
|
|
|
|
Ok(..) => GBoolean::True,
|
|
|
|
Err(ref msg) => {
|
|
|
|
source.get_controller().error(msg);
|
|
|
|
GBoolean::False
|
|
|
|
}
|
|
|
|
}
|
2016-05-14 14:57:25 +00:00
|
|
|
}
|