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-13 15:16:49 +00:00
|
|
|
use std::u64;
|
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;
|
2016-05-14 09:34:50 +00:00
|
|
|
use std::path::PathBuf;
|
2016-05-23 17:29:01 +00:00
|
|
|
use std::sync::Mutex;
|
2016-05-14 09:34:50 +00:00
|
|
|
use url::Url;
|
2016-05-13 15:16:49 +00:00
|
|
|
|
2016-05-14 08:31:13 +00:00
|
|
|
use std::io::Write;
|
|
|
|
|
2016-05-14 11:44:49 +00:00
|
|
|
use utils::*;
|
|
|
|
use rssource::*;
|
2016-05-14 10:34:50 +00:00
|
|
|
|
2016-05-13 15:53:35 +00:00
|
|
|
#[derive(Debug)]
|
|
|
|
pub struct FileSrc {
|
2016-05-23 17:29:01 +00:00
|
|
|
location: Mutex<Option<PathBuf>>,
|
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-15 08:55:52 +00:00
|
|
|
unsafe impl Sync for FileSrc {}
|
|
|
|
unsafe impl Send for FileSrc {}
|
|
|
|
|
2016-05-13 15:16:49 +00:00
|
|
|
impl FileSrc {
|
|
|
|
fn new() -> FileSrc {
|
2016-07-20 08:28:58 +00:00
|
|
|
FileSrc {
|
|
|
|
location: Mutex::new(None),
|
|
|
|
file: None,
|
|
|
|
position: 0,
|
|
|
|
}
|
2016-05-13 15:16:49 +00:00
|
|
|
}
|
2016-05-13 15:53:35 +00:00
|
|
|
|
2016-05-14 10:34:50 +00:00
|
|
|
fn new_source() -> Box<Source> {
|
|
|
|
Box::new(FileSrc::new())
|
|
|
|
}
|
2016-05-14 12:44:40 +00:00
|
|
|
pub extern "C" fn new_ptr() -> *mut Box<Source> {
|
|
|
|
let instance = Box::new(FileSrc::new_source());
|
|
|
|
return Box::into_raw(instance);
|
|
|
|
}
|
2016-05-14 10:34:50 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
impl Source for FileSrc {
|
2016-05-24 20:24:05 +00:00
|
|
|
fn set_uri(&mut self, uri: Option<Url>) -> bool {
|
|
|
|
match uri {
|
2016-05-14 09:34:50 +00:00
|
|
|
None => {
|
2016-05-23 17:29:01 +00:00
|
|
|
let mut location = self.location.lock().unwrap();
|
|
|
|
*location = None;
|
2016-05-14 09:34:50 +00:00
|
|
|
return true;
|
2016-07-20 08:28:58 +00:00
|
|
|
}
|
2016-05-24 20:24:05 +00:00
|
|
|
Some(uri) => {
|
|
|
|
match uri.to_file_path().ok() {
|
|
|
|
Some(p) => {
|
|
|
|
let mut location = self.location.lock().unwrap();
|
|
|
|
*location = Some(p);
|
|
|
|
return true;
|
2016-07-20 08:28:58 +00:00
|
|
|
}
|
2016-05-24 20:24:05 +00:00
|
|
|
None => {
|
2016-05-23 17:29:01 +00:00
|
|
|
let mut location = self.location.lock().unwrap();
|
|
|
|
*location = None;
|
2016-05-24 20:24:05 +00:00
|
|
|
println_err!("Unsupported file URI '{}'", uri.as_str());
|
2016-05-14 09:34:50 +00:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2016-05-13 15:53:35 +00:00
|
|
|
}
|
|
|
|
|
2016-05-24 20:24:05 +00:00
|
|
|
fn get_uri(&self) -> Option<Url> {
|
2016-05-23 17:29:01 +00:00
|
|
|
let location = self.location.lock().unwrap();
|
2016-07-20 08:28:58 +00:00
|
|
|
(*location)
|
|
|
|
.as_ref()
|
2016-05-15 13:15:58 +00:00
|
|
|
.map(|l| Url::from_file_path(l).ok())
|
|
|
|
.and_then(|i| i) // join()
|
2016-05-13 15:53:35 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
fn is_seekable(&self) -> bool {
|
|
|
|
true
|
|
|
|
}
|
|
|
|
|
|
|
|
fn get_size(&self) -> u64 {
|
2016-05-15 13:15:58 +00:00
|
|
|
self.file.as_ref()
|
|
|
|
.map(|f| f.metadata().ok())
|
|
|
|
.and_then(|i| i) // join()
|
|
|
|
.map(|m| m.len())
|
|
|
|
.unwrap_or(u64::MAX)
|
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;
|
2016-05-23 17:29:01 +00:00
|
|
|
let location = self.location.lock().unwrap();
|
2016-05-14 08:31:13 +00:00
|
|
|
|
2016-05-23 17:29:01 +00:00
|
|
|
match *location {
|
2016-05-14 08:31:13 +00:00
|
|
|
None => return false,
|
|
|
|
Some(ref location) => {
|
2016-05-14 09:34:50 +00:00
|
|
|
match File::open(location.as_path()) {
|
2016-05-14 08:31:13 +00:00
|
|
|
Ok(file) => {
|
|
|
|
self.file = Some(file);
|
|
|
|
return true;
|
2016-07-20 08:28:58 +00:00
|
|
|
}
|
2016-05-14 08:31:13 +00:00
|
|
|
Err(err) => {
|
2016-07-20 08:28:58 +00:00
|
|
|
println_err!("Failed to open file '{}': {}",
|
|
|
|
location.to_str().unwrap_or("Non-UTF8 path"),
|
|
|
|
err.to_string());
|
2016-05-14 08:31:13 +00:00
|
|
|
return false;
|
2016-07-20 08:28:58 +00:00
|
|
|
}
|
2016-05-14 08:31:13 +00:00
|
|
|
}
|
2016-07-20 08:28:58 +00:00
|
|
|
}
|
2016-05-14 08:31:13 +00:00
|
|
|
}
|
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;
|
2016-07-20 08:28:58 +00:00
|
|
|
}
|
2016-05-14 08:31:13 +00:00
|
|
|
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;
|
2016-07-20 08:28:58 +00:00
|
|
|
return Ok(size);
|
|
|
|
}
|
2016-05-14 08:31:13 +00:00
|
|
|
Err(err) => {
|
|
|
|
println_err!("Failed to read at {}: {}", offset, err.to_string());
|
|
|
|
return Err(GstFlowReturn::Error);
|
2016-07-20 08:28:58 +00:00
|
|
|
}
|
2016-05-14 08:31:13 +00:00
|
|
|
}
|
2016-07-20 08:28:58 +00:00
|
|
|
}
|
2016-05-13 15:53:35 +00:00
|
|
|
}
|
|
|
|
}
|
2016-05-22 13:23:43 +00:00
|
|
|
|
|
|
|
fn do_seek(&mut self, _: u64, _: u64) -> bool {
|
|
|
|
true
|
|
|
|
}
|
2016-05-13 15:16:49 +00:00
|
|
|
}
|