2016-05-13 15:02:19 +00:00
|
|
|
use libc::{c_char};
|
|
|
|
use std::ffi::{CStr, CString};
|
2016-05-13 15:53:35 +00:00
|
|
|
use std::ptr;
|
2016-05-13 15:16:49 +00:00
|
|
|
use std::u64;
|
2016-05-13 15:38:28 +00:00
|
|
|
use std::slice;
|
2016-05-14 08:31:13 +00:00
|
|
|
use std::io::{Read, Seek, SeekFrom};
|
2016-05-13 16:04:10 +00:00
|
|
|
use std::fs::File;
|
|
|
|
use std::path::Path;
|
2016-05-13 15:16:49 +00:00
|
|
|
|
2016-05-14 08:31:13 +00:00
|
|
|
use std::io::Write;
|
|
|
|
|
|
|
|
macro_rules! println_err(
|
|
|
|
($($arg:tt)*) => { {
|
|
|
|
let r = writeln!(&mut ::std::io::stderr(), $($arg)*);
|
|
|
|
r.expect("failed printing to stderr");
|
|
|
|
} }
|
|
|
|
);
|
|
|
|
|
2016-05-13 15:16:49 +00:00
|
|
|
#[repr(C)]
|
|
|
|
pub enum GstFlowReturn {
|
|
|
|
Ok = 0,
|
|
|
|
NotLinked = -1,
|
|
|
|
Flushing = -2,
|
|
|
|
Eos = -3,
|
|
|
|
NotNegotiated = -4,
|
|
|
|
Error = -5,
|
|
|
|
}
|
|
|
|
|
|
|
|
#[repr(C)]
|
|
|
|
pub enum GBoolean {
|
|
|
|
False = 0,
|
|
|
|
True = 1,
|
|
|
|
}
|
|
|
|
|
2016-05-13 15:53:35 +00:00
|
|
|
impl GBoolean {
|
|
|
|
fn from_bool(v: bool) -> GBoolean {
|
|
|
|
match v {
|
|
|
|
true => GBoolean::True,
|
|
|
|
false => GBoolean::False,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Debug)]
|
|
|
|
pub struct FileSrc {
|
|
|
|
location: Option<String>,
|
2016-05-13 16:04:10 +00:00
|
|
|
file: Option<File>,
|
2016-05-14 08:31:13 +00:00
|
|
|
position: u64,
|
2016-05-13 15:53:35 +00:00
|
|
|
}
|
|
|
|
|
2016-05-13 15:16:49 +00:00
|
|
|
impl FileSrc {
|
|
|
|
fn new() -> FileSrc {
|
2016-05-14 08:31:13 +00:00
|
|
|
FileSrc { location: None, file: None, position: 0 }
|
2016-05-13 15:16:49 +00:00
|
|
|
}
|
2016-05-13 15:53:35 +00:00
|
|
|
|
|
|
|
fn set_location(&mut self, location: &Option<String>) {
|
|
|
|
self.location = location.clone();
|
|
|
|
}
|
|
|
|
|
|
|
|
fn get_location(&self) -> &Option<String> {
|
|
|
|
&self.location
|
|
|
|
}
|
|
|
|
|
|
|
|
fn is_seekable(&self) -> bool {
|
|
|
|
true
|
|
|
|
}
|
|
|
|
|
|
|
|
fn get_size(&self) -> u64 {
|
2016-05-13 16:06:06 +00:00
|
|
|
match self.file {
|
|
|
|
None => return u64::MAX,
|
|
|
|
Some(ref f) => {
|
2016-05-14 08:31:13 +00:00
|
|
|
return f.metadata().map(|m| m.len()).unwrap_or(u64::MAX);
|
2016-05-13 16:06:06 +00:00
|
|
|
},
|
|
|
|
}
|
2016-05-13 15:53:35 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
fn start(&mut self) -> bool {
|
2016-05-14 08:31:13 +00:00
|
|
|
self.file = None;
|
|
|
|
self.position = 0;
|
|
|
|
|
|
|
|
match self.location {
|
|
|
|
None => return false,
|
|
|
|
Some(ref location) => {
|
|
|
|
match File::open(Path::new(&location.clone())) {
|
|
|
|
Ok(file) => {
|
|
|
|
self.file = Some(file);
|
|
|
|
return true;
|
|
|
|
},
|
|
|
|
Err(err) => {
|
|
|
|
println_err!("Failed to open file '{}': {}", location, err.to_string());
|
|
|
|
return false;
|
|
|
|
},
|
|
|
|
}
|
|
|
|
},
|
|
|
|
}
|
2016-05-13 15:53:35 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
fn stop(&mut self) -> bool {
|
2016-05-13 16:04:10 +00:00
|
|
|
self.file = None;
|
2016-05-14 08:31:13 +00:00
|
|
|
self.position = 0;
|
2016-05-13 16:04:10 +00:00
|
|
|
|
2016-05-13 15:53:35 +00:00
|
|
|
true
|
|
|
|
}
|
|
|
|
|
2016-05-14 08:31:13 +00:00
|
|
|
fn fill(&mut self, offset: u64, data: &mut [u8]) -> Result<usize, GstFlowReturn> {
|
2016-05-13 16:04:10 +00:00
|
|
|
match self.file {
|
2016-05-14 08:09:48 +00:00
|
|
|
None => return Err(GstFlowReturn::Error),
|
2016-05-13 16:04:10 +00:00
|
|
|
Some(ref mut f) => {
|
2016-05-14 08:31:13 +00:00
|
|
|
if self.position != offset {
|
|
|
|
match f.seek(SeekFrom::Start(offset)) {
|
|
|
|
Ok(_) => {
|
|
|
|
self.position = offset;
|
|
|
|
},
|
|
|
|
Err(err) => {
|
|
|
|
println_err!("Failed to seek to {}: {}", offset, err.to_string());
|
|
|
|
return Err(GstFlowReturn::Error);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
match f.read(data) {
|
|
|
|
Ok(size) => {
|
|
|
|
self.position += size as u64;
|
|
|
|
return Ok(size)
|
|
|
|
},
|
|
|
|
Err(err) => {
|
|
|
|
println_err!("Failed to read at {}: {}", offset, err.to_string());
|
|
|
|
return Err(GstFlowReturn::Error);
|
|
|
|
},
|
|
|
|
}
|
2016-05-13 16:04:10 +00:00
|
|
|
},
|
2016-05-13 15:53:35 +00:00
|
|
|
}
|
|
|
|
}
|
2016-05-13 15:16:49 +00:00
|
|
|
}
|
|
|
|
|
2016-05-13 13:35:48 +00:00
|
|
|
#[no_mangle]
|
2016-05-13 14:35:09 +00:00
|
|
|
pub extern "C" fn filesrc_new() -> *mut FileSrc {
|
2016-05-13 14:43:32 +00:00
|
|
|
let instance = Box::new(FileSrc::new());
|
|
|
|
return Box::into_raw(instance);
|
2016-05-13 14:35:09 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
#[no_mangle]
|
|
|
|
pub extern "C" fn filesrc_drop(ptr: *mut FileSrc) {
|
2016-05-13 14:43:32 +00:00
|
|
|
unsafe { Box::from_raw(ptr) };
|
2016-05-13 14:35:09 +00:00
|
|
|
}
|
|
|
|
|
2016-05-13 15:02:19 +00:00
|
|
|
#[no_mangle]
|
|
|
|
pub extern "C" fn filesrc_set_location(ptr: *mut FileSrc, location_ptr: *const c_char) {
|
|
|
|
let filesrc: &mut FileSrc = unsafe { &mut *ptr };
|
|
|
|
|
|
|
|
if location_ptr.is_null() {
|
2016-05-13 15:53:35 +00:00
|
|
|
filesrc.set_location(&None)
|
2016-05-13 15:02:19 +00:00
|
|
|
} else {
|
|
|
|
let location = unsafe { CStr::from_ptr(location_ptr) };
|
2016-05-13 15:53:35 +00:00
|
|
|
filesrc.set_location(&Some(String::from(location.to_str().unwrap())));
|
2016-05-13 15:02:19 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[no_mangle]
|
|
|
|
pub extern "C" fn filesrc_get_location(ptr: *mut FileSrc) -> *mut c_char {
|
|
|
|
let filesrc: &mut FileSrc = unsafe { &mut *ptr };
|
|
|
|
|
2016-05-13 15:53:35 +00:00
|
|
|
match *filesrc.get_location() {
|
2016-05-13 15:02:19 +00:00
|
|
|
Some(ref location) =>
|
|
|
|
CString::new(location.clone().into_bytes()).unwrap().into_raw(),
|
|
|
|
None =>
|
|
|
|
ptr::null_mut()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-05-13 14:35:09 +00:00
|
|
|
#[no_mangle]
|
2016-05-14 08:31:13 +00:00
|
|
|
pub extern "C" fn filesrc_fill(ptr: *mut FileSrc, offset: u64, data_ptr: *mut u8, data_len_ptr: *mut usize) -> GstFlowReturn {
|
2016-05-13 14:35:09 +00:00
|
|
|
let filesrc: &mut FileSrc = unsafe { &mut *ptr };
|
|
|
|
|
2016-05-14 08:09:48 +00:00
|
|
|
let mut data_len: &mut usize = unsafe { &mut *data_len_ptr };
|
|
|
|
let mut data = unsafe { slice::from_raw_parts_mut(data_ptr, *data_len) };
|
2016-05-14 08:31:13 +00:00
|
|
|
match filesrc.fill(offset, data) {
|
2016-05-14 08:09:48 +00:00
|
|
|
Ok(actual_len) => {
|
|
|
|
*data_len = actual_len;
|
|
|
|
GstFlowReturn::Ok
|
|
|
|
},
|
|
|
|
Err(ret) => ret,
|
|
|
|
}
|
2016-05-13 14:35:09 +00:00
|
|
|
}
|
|
|
|
|
2016-05-13 15:16:49 +00:00
|
|
|
#[no_mangle]
|
|
|
|
pub extern "C" fn filesrc_get_size(ptr: *mut FileSrc) -> u64 {
|
|
|
|
let filesrc: &mut FileSrc = unsafe { &mut *ptr };
|
|
|
|
|
2016-05-13 15:53:35 +00:00
|
|
|
return filesrc.get_size();
|
2016-05-13 14:35:09 +00:00
|
|
|
}
|
|
|
|
|
2016-05-13 15:16:49 +00:00
|
|
|
#[no_mangle]
|
|
|
|
pub extern "C" fn filesrc_start(ptr: *mut FileSrc) -> GBoolean {
|
|
|
|
let filesrc: &mut FileSrc = unsafe { &mut *ptr };
|
|
|
|
|
2016-05-13 15:53:35 +00:00
|
|
|
GBoolean::from_bool(filesrc.start())
|
2016-05-13 14:35:09 +00:00
|
|
|
}
|
|
|
|
|
2016-05-13 15:16:49 +00:00
|
|
|
#[no_mangle]
|
|
|
|
pub extern "C" fn filesrc_stop(ptr: *mut FileSrc) -> GBoolean {
|
|
|
|
let filesrc: &mut FileSrc = unsafe { &mut *ptr };
|
|
|
|
|
2016-05-13 15:53:35 +00:00
|
|
|
GBoolean::from_bool(filesrc.stop())
|
2016-05-13 13:35:48 +00:00
|
|
|
}
|
2016-05-13 15:16:49 +00:00
|
|
|
|
|
|
|
#[no_mangle]
|
|
|
|
pub extern "C" fn filesrc_is_seekable(ptr: *mut FileSrc) -> GBoolean {
|
|
|
|
let filesrc: &mut FileSrc = unsafe { &mut *ptr };
|
|
|
|
|
2016-05-13 15:53:35 +00:00
|
|
|
GBoolean::from_bool(filesrc.is_seekable())
|
2016-05-13 15:16:49 +00:00
|
|
|
}
|
|
|
|
|