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-14 14:04:53 +00:00
|
|
|
use std::u64;
|
2016-05-15 14:34:13 +00:00
|
|
|
use std::io::Read;
|
2016-05-14 14:04:53 +00:00
|
|
|
use url::Url;
|
2016-07-20 08:28:58 +00:00
|
|
|
use hyper::header::{ContentLength, ContentRange, ContentRangeSpec, Range, ByteRangeSpec,
|
|
|
|
AcceptRanges, RangeUnit};
|
2016-05-14 14:04:53 +00:00
|
|
|
use hyper::client::Client;
|
|
|
|
use hyper::client::response::Response;
|
|
|
|
|
|
|
|
use std::io::Write;
|
2016-05-23 17:29:01 +00:00
|
|
|
use std::sync::Mutex;
|
2016-05-14 14:04:53 +00:00
|
|
|
|
|
|
|
use utils::*;
|
|
|
|
use rssource::*;
|
|
|
|
|
2016-08-22 19:25:58 +00:00
|
|
|
#[derive(Debug)]
|
|
|
|
struct Settings {
|
|
|
|
url: Option<Url>,
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Debug)]
|
|
|
|
enum StreamingState {
|
|
|
|
Stopped,
|
|
|
|
Started {
|
|
|
|
response: Response,
|
|
|
|
seekable: bool,
|
|
|
|
position: u64,
|
|
|
|
size: u64,
|
|
|
|
start: u64,
|
|
|
|
stop: u64,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
2016-05-14 14:04:53 +00:00
|
|
|
#[derive(Debug)]
|
|
|
|
pub struct HttpSrc {
|
2016-07-20 09:46:03 +00:00
|
|
|
controller: SourceController,
|
2016-08-22 19:25:58 +00:00
|
|
|
settings: Mutex<Settings>,
|
|
|
|
streaming_state: Mutex<StreamingState>,
|
2016-05-14 14:04:53 +00:00
|
|
|
client: Client,
|
|
|
|
}
|
|
|
|
|
2016-05-15 08:55:52 +00:00
|
|
|
unsafe impl Sync for HttpSrc {}
|
|
|
|
unsafe impl Send for HttpSrc {}
|
|
|
|
|
2016-05-14 14:04:53 +00:00
|
|
|
impl HttpSrc {
|
2016-08-10 16:27:38 +00:00
|
|
|
pub fn new(controller: SourceController) -> HttpSrc {
|
2016-07-20 08:28:58 +00:00
|
|
|
HttpSrc {
|
2016-07-20 09:46:03 +00:00
|
|
|
controller: controller,
|
2016-08-22 19:25:58 +00:00
|
|
|
settings: Mutex::new(Settings { url: None }),
|
|
|
|
streaming_state: Mutex::new(StreamingState::Stopped),
|
2016-07-20 08:28:58 +00:00
|
|
|
client: Client::new(),
|
|
|
|
}
|
2016-05-14 14:04:53 +00:00
|
|
|
}
|
|
|
|
|
2016-08-10 16:27:38 +00:00
|
|
|
pub fn new_boxed(controller: SourceController) -> Box<Source> {
|
|
|
|
Box::new(HttpSrc::new(controller))
|
2016-05-14 14:04:53 +00:00
|
|
|
}
|
2016-05-15 08:48:54 +00:00
|
|
|
|
2016-08-22 19:25:58 +00:00
|
|
|
fn do_request(&self, start: u64, stop: u64) -> StreamingState {
|
2016-08-22 19:35:16 +00:00
|
|
|
let url = &self.settings.lock().unwrap().url;
|
2016-05-15 08:48:54 +00:00
|
|
|
|
2016-05-23 17:29:01 +00:00
|
|
|
match *url {
|
2016-08-22 19:25:58 +00:00
|
|
|
None => StreamingState::Stopped,
|
2016-05-15 08:48:54 +00:00
|
|
|
Some(ref url) => {
|
|
|
|
let mut req = self.client.get(url.clone());
|
|
|
|
|
|
|
|
if start != 0 || stop != u64::MAX {
|
|
|
|
req = if stop == u64::MAX {
|
2016-07-20 08:28:58 +00:00
|
|
|
req.header(Range::Bytes(vec![ByteRangeSpec::AllFrom(start)]))
|
|
|
|
} else {
|
2016-08-22 19:25:58 +00:00
|
|
|
req.header(Range::Bytes(vec![ByteRangeSpec::FromTo(start, stop - 1)]))
|
2016-07-20 08:28:58 +00:00
|
|
|
};
|
2016-05-15 08:48:54 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
match req.send() {
|
|
|
|
Ok(response) => {
|
|
|
|
if response.status.is_success() {
|
2016-08-22 19:25:58 +00:00
|
|
|
let size = if let Some(&ContentLength(content_length)) =
|
|
|
|
response.headers.get() {
|
2016-05-15 08:48:54 +00:00
|
|
|
content_length + start
|
|
|
|
} else {
|
|
|
|
u64::MAX
|
|
|
|
};
|
2016-07-20 08:28:58 +00:00
|
|
|
let accept_byte_ranges = if let Some(&AcceptRanges(ref ranges)) =
|
|
|
|
response.headers.get() {
|
2016-05-15 08:48:54 +00:00
|
|
|
ranges.iter().any(|u| *u == RangeUnit::Bytes)
|
|
|
|
} else {
|
|
|
|
false
|
|
|
|
};
|
|
|
|
|
2016-08-22 19:25:58 +00:00
|
|
|
let seekable = size != u64::MAX && accept_byte_ranges;
|
2016-05-15 08:48:54 +00:00
|
|
|
|
2016-08-22 19:25:58 +00:00
|
|
|
let position = if let Some(&ContentRange(ContentRangeSpec::Bytes{range: Some((range_start, _)), ..})) = response.headers.get() {
|
2016-05-15 08:48:54 +00:00
|
|
|
range_start
|
|
|
|
} else {
|
|
|
|
start
|
|
|
|
};
|
|
|
|
|
2016-08-22 19:25:58 +00:00
|
|
|
if position != start {
|
|
|
|
println_err!("Failed to seek to {}: Got {}", start, position);
|
|
|
|
StreamingState::Stopped
|
|
|
|
} else {
|
|
|
|
StreamingState::Started {
|
|
|
|
response: response,
|
|
|
|
seekable: seekable,
|
|
|
|
position: 0,
|
|
|
|
size: size,
|
|
|
|
start: start,
|
|
|
|
stop: stop,
|
|
|
|
}
|
2016-05-15 08:48:54 +00:00
|
|
|
}
|
|
|
|
} else {
|
|
|
|
println_err!("Failed to fetch {}: {}", url, response.status);
|
2016-08-22 19:25:58 +00:00
|
|
|
StreamingState::Stopped
|
2016-05-15 08:48:54 +00:00
|
|
|
}
|
2016-07-20 08:28:58 +00:00
|
|
|
}
|
2016-05-15 08:48:54 +00:00
|
|
|
Err(err) => {
|
|
|
|
println_err!("Failed to fetch {}: {}", url, err.to_string());
|
2016-08-22 19:25:58 +00:00
|
|
|
StreamingState::Stopped
|
2016-05-15 08:48:54 +00:00
|
|
|
}
|
|
|
|
}
|
2016-07-20 08:28:58 +00:00
|
|
|
}
|
2016-05-15 08:48:54 +00:00
|
|
|
}
|
|
|
|
}
|
2016-05-14 14:04:53 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
impl Source for HttpSrc {
|
2016-08-22 19:25:58 +00:00
|
|
|
fn set_uri(&self, uri: Option<Url>) -> bool {
|
2016-08-22 19:35:16 +00:00
|
|
|
let url = &mut self.settings.lock().unwrap().url;
|
2016-05-15 08:48:54 +00:00
|
|
|
|
2016-05-24 20:24:05 +00:00
|
|
|
match uri {
|
2016-05-14 14:04:53 +00:00
|
|
|
None => {
|
2016-05-23 17:29:01 +00:00
|
|
|
*url = None;
|
2016-08-22 19:35:16 +00:00
|
|
|
true
|
2016-07-20 08:28:58 +00:00
|
|
|
}
|
2016-05-24 20:24:05 +00:00
|
|
|
Some(uri) => {
|
2016-07-20 08:28:58 +00:00
|
|
|
if uri.scheme() == "http" || uri.scheme() == "https" {
|
2016-05-24 20:24:05 +00:00
|
|
|
*url = Some(uri);
|
2016-08-22 19:35:16 +00:00
|
|
|
true
|
2016-05-24 20:24:05 +00:00
|
|
|
} else {
|
|
|
|
*url = None;
|
|
|
|
println_err!("Unsupported URI '{}'", uri.as_str());
|
2016-08-22 19:35:16 +00:00
|
|
|
false
|
2016-05-14 14:04:53 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-05-24 20:24:05 +00:00
|
|
|
fn get_uri(&self) -> Option<Url> {
|
2016-08-22 19:35:16 +00:00
|
|
|
let url = &self.settings.lock().unwrap().url;
|
|
|
|
url.as_ref().cloned()
|
2016-05-14 14:04:53 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
fn is_seekable(&self) -> bool {
|
2016-08-22 19:25:58 +00:00
|
|
|
let streaming_state = self.streaming_state.lock().unwrap();
|
|
|
|
|
|
|
|
match *streaming_state {
|
|
|
|
StreamingState::Started { seekable, .. } => seekable,
|
|
|
|
_ => false,
|
|
|
|
}
|
2016-05-14 14:04:53 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
fn get_size(&self) -> u64 {
|
2016-08-22 19:25:58 +00:00
|
|
|
let streaming_state = self.streaming_state.lock().unwrap();
|
|
|
|
match *streaming_state {
|
|
|
|
StreamingState::Started { size, .. } => size,
|
|
|
|
_ => u64::MAX,
|
|
|
|
}
|
2016-05-14 14:04:53 +00:00
|
|
|
}
|
|
|
|
|
2016-08-22 19:25:58 +00:00
|
|
|
fn start(&self) -> bool {
|
|
|
|
let mut streaming_state = self.streaming_state.lock().unwrap();
|
|
|
|
*streaming_state = self.do_request(0, u64::MAX);
|
2016-05-14 14:04:53 +00:00
|
|
|
|
2016-08-22 19:25:58 +00:00
|
|
|
if let StreamingState::Stopped = *streaming_state {
|
|
|
|
false
|
|
|
|
} else {
|
|
|
|
true
|
2016-05-14 14:04:53 +00:00
|
|
|
}
|
2016-08-22 19:25:58 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
fn stop(&self) -> bool {
|
|
|
|
let mut streaming_state = self.streaming_state.lock().unwrap();
|
|
|
|
*streaming_state = StreamingState::Stopped;
|
2016-05-14 14:04:53 +00:00
|
|
|
|
2016-08-22 19:25:58 +00:00
|
|
|
true
|
2016-05-14 14:04:53 +00:00
|
|
|
}
|
|
|
|
|
2016-08-22 19:25:58 +00:00
|
|
|
fn do_seek(&self, start: u64, stop: u64) -> bool {
|
|
|
|
let mut streaming_state = self.streaming_state.lock().unwrap();
|
|
|
|
*streaming_state = self.do_request(start, stop);
|
|
|
|
|
|
|
|
if let StreamingState::Stopped = *streaming_state {
|
|
|
|
false
|
|
|
|
} else {
|
|
|
|
true
|
|
|
|
}
|
2016-05-15 08:48:54 +00:00
|
|
|
}
|
|
|
|
|
2016-08-22 19:25:58 +00:00
|
|
|
fn fill(&self, offset: u64, data: &mut [u8]) -> Result<usize, GstFlowReturn> {
|
|
|
|
let mut streaming_state = self.streaming_state.lock().unwrap();
|
|
|
|
|
|
|
|
if let StreamingState::Stopped = *streaming_state {
|
|
|
|
return Err(GstFlowReturn::Error);
|
2016-05-15 08:48:54 +00:00
|
|
|
}
|
|
|
|
|
2016-08-22 19:25:58 +00:00
|
|
|
if let StreamingState::Started { position, stop, .. } = *streaming_state {
|
|
|
|
if position != offset {
|
|
|
|
*streaming_state = self.do_request(offset, stop);
|
|
|
|
if let StreamingState::Stopped = *streaming_state {
|
|
|
|
println_err!("Failed to seek to {}", offset);
|
|
|
|
return Err(GstFlowReturn::Error);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2016-05-14 14:04:53 +00:00
|
|
|
|
2016-08-22 19:25:58 +00:00
|
|
|
if let StreamingState::Started { ref mut response, ref mut position, .. } =
|
|
|
|
*streaming_state {
|
|
|
|
match response.read(data) {
|
|
|
|
Ok(size) => {
|
|
|
|
if size == 0 {
|
|
|
|
return Err(GstFlowReturn::Eos);
|
2016-07-20 08:28:58 +00:00
|
|
|
}
|
2016-08-22 19:25:58 +00:00
|
|
|
|
|
|
|
*position += size as u64;
|
|
|
|
Ok(size)
|
|
|
|
}
|
|
|
|
Err(err) => {
|
|
|
|
println_err!("Failed to read at {}: {}", offset, err.to_string());
|
|
|
|
Err(GstFlowReturn::Error)
|
2016-05-14 14:04:53 +00:00
|
|
|
}
|
2016-07-20 08:28:58 +00:00
|
|
|
}
|
2016-08-22 19:25:58 +00:00
|
|
|
} else {
|
|
|
|
Err(GstFlowReturn::Error)
|
2016-05-14 14:04:53 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|