mirror of
https://gitlab.freedesktop.org/gstreamer/gst-plugins-rs.git
synced 2024-05-20 17:28:49 +00:00
Implement seeking in HTTP source
This commit is contained in:
parent
da1f869b47
commit
846f4775d9
128
src/rshttpsrc.rs
128
src/rshttpsrc.rs
|
@ -1,7 +1,7 @@
|
||||||
use std::u64;
|
use std::u64;
|
||||||
use std::io::{Read, Seek, SeekFrom};
|
use std::io::{Read, Seek, SeekFrom};
|
||||||
use url::Url;
|
use url::Url;
|
||||||
use hyper::header::ContentLength;
|
use hyper::header::{ContentLength, ContentRange, ContentRangeSpec, Range, ByteRangeSpec, AcceptRanges, RangeUnit};
|
||||||
use hyper::client::Client;
|
use hyper::client::Client;
|
||||||
use hyper::client::response::Response;
|
use hyper::client::response::Response;
|
||||||
|
|
||||||
|
@ -15,13 +15,16 @@ pub struct HttpSrc {
|
||||||
url: Option<Url>,
|
url: Option<Url>,
|
||||||
client: Client,
|
client: Client,
|
||||||
response: Option<Response>,
|
response: Option<Response>,
|
||||||
|
seekable: bool,
|
||||||
position: u64,
|
position: u64,
|
||||||
size: u64,
|
size: u64,
|
||||||
|
start: u64,
|
||||||
|
stop: u64,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl HttpSrc {
|
impl HttpSrc {
|
||||||
fn new() -> HttpSrc {
|
fn new() -> HttpSrc {
|
||||||
HttpSrc { url: None, client: Client::new(), response: None, position: 0, size: u64::MAX }
|
HttpSrc { url: None, client: Client::new(), response: None, seekable: false, position: 0, size: u64::MAX, start: 0, stop: u64::MAX }
|
||||||
}
|
}
|
||||||
|
|
||||||
fn new_source() -> Box<Source> {
|
fn new_source() -> Box<Source> {
|
||||||
|
@ -31,10 +34,81 @@ impl HttpSrc {
|
||||||
let instance = Box::new(HttpSrc::new_source());
|
let instance = Box::new(HttpSrc::new_source());
|
||||||
return Box::into_raw(instance);
|
return Box::into_raw(instance);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn do_request(&mut self, start: u64, stop: u64) -> bool {
|
||||||
|
self.response = None;
|
||||||
|
self.seekable = false;
|
||||||
|
self.position = 0;
|
||||||
|
self.size = u64::MAX;
|
||||||
|
|
||||||
|
match self.url {
|
||||||
|
None => return false,
|
||||||
|
Some(ref url) => {
|
||||||
|
let mut req = self.client.get(url.clone());
|
||||||
|
|
||||||
|
if start != 0 || stop != u64::MAX {
|
||||||
|
req = if stop == u64::MAX {
|
||||||
|
req.header(Range::Bytes(vec![ByteRangeSpec::AllFrom(start)]))
|
||||||
|
} else {
|
||||||
|
req.header(Range::Bytes(vec![ByteRangeSpec::FromTo(start, stop)]))
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
match req.send() {
|
||||||
|
Ok(response) => {
|
||||||
|
if response.status.is_success() {
|
||||||
|
self.size = if let Some(&ContentLength(content_length)) = response.headers.get() {
|
||||||
|
content_length + start
|
||||||
|
} else {
|
||||||
|
u64::MAX
|
||||||
|
};
|
||||||
|
let accept_byte_ranges = if let Some(&AcceptRanges(ref ranges)) = response.headers.get() {
|
||||||
|
ranges.iter().any(|u| *u == RangeUnit::Bytes)
|
||||||
|
} else {
|
||||||
|
false
|
||||||
|
};
|
||||||
|
|
||||||
|
self.seekable = self.size != u64::MAX && accept_byte_ranges;
|
||||||
|
|
||||||
|
self.start = start;
|
||||||
|
self.stop = stop;
|
||||||
|
|
||||||
|
self.position = if let Some(&ContentRange(ContentRangeSpec::Bytes{range: Some((range_start, _)), ..})) = response.headers.get() {
|
||||||
|
range_start
|
||||||
|
} else {
|
||||||
|
start
|
||||||
|
};
|
||||||
|
|
||||||
|
if self.position != start {
|
||||||
|
println_err!("Failed to seek to {}: Got {}", start, self.position);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
self.response = Some(response);
|
||||||
|
|
||||||
|
return true;
|
||||||
|
} else {
|
||||||
|
println_err!("Failed to fetch {}: {}", url, response.status);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
},
|
||||||
|
Err(err) => {
|
||||||
|
println_err!("Failed to fetch {}: {}", url, err.to_string());
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Source for HttpSrc {
|
impl Source for HttpSrc {
|
||||||
fn set_uri(&mut self, uri_str: &Option<String>) -> bool {
|
fn set_uri(&mut self, uri_str: &Option<String>) -> bool {
|
||||||
|
if self.response.is_some() {
|
||||||
|
println_err!("Can't set URI after starting");
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
match *uri_str {
|
match *uri_str {
|
||||||
None => {
|
None => {
|
||||||
self.url = None;
|
self.url = None;
|
||||||
|
@ -69,7 +143,7 @@ impl Source for HttpSrc {
|
||||||
}
|
}
|
||||||
|
|
||||||
fn is_seekable(&self) -> bool {
|
fn is_seekable(&self) -> bool {
|
||||||
false
|
self.seekable
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_size(&self) -> u64 {
|
fn get_size(&self) -> u64 {
|
||||||
|
@ -77,37 +151,12 @@ impl Source for HttpSrc {
|
||||||
}
|
}
|
||||||
|
|
||||||
fn start(&mut self) -> bool {
|
fn start(&mut self) -> bool {
|
||||||
self.response = None;
|
self.seekable = false;
|
||||||
self.position = 0;
|
return self.do_request(0, u64::MAX);
|
||||||
self.size = u64::MAX;
|
|
||||||
|
|
||||||
match self.url {
|
|
||||||
None => return false,
|
|
||||||
Some(ref url) => {
|
|
||||||
match self.client.get(url.clone()).send() {
|
|
||||||
Ok(response) => {
|
|
||||||
if response.status.is_success() {
|
|
||||||
self.size = match response.headers.get::<ContentLength>() {
|
|
||||||
Some(&ContentLength(size)) => size,
|
|
||||||
_ => u64::MAX
|
|
||||||
};
|
|
||||||
self.response = Some(response);
|
|
||||||
return true;
|
|
||||||
} else {
|
|
||||||
println_err!("Failed to fetch {}: {}", url, response.status);
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
},
|
|
||||||
Err(err) => {
|
|
||||||
println_err!("Failed to fetch {}: {}", url, err.to_string());
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fn stop(&mut self) -> bool {
|
fn stop(&mut self) -> bool {
|
||||||
|
self.seekable = false;
|
||||||
self.position = 0;
|
self.position = 0;
|
||||||
self.size = u64::MAX;
|
self.size = u64::MAX;
|
||||||
match self.response {
|
match self.response {
|
||||||
|
@ -119,15 +168,22 @@ impl Source for HttpSrc {
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn do_seek(&mut self, start: u64, stop: u64) -> bool {
|
||||||
|
return self.do_request(start, stop);
|
||||||
|
}
|
||||||
|
|
||||||
fn fill(&mut self, offset: u64, data: &mut [u8]) -> Result<usize, GstFlowReturn> {
|
fn fill(&mut self, offset: u64, data: &mut [u8]) -> Result<usize, GstFlowReturn> {
|
||||||
|
if self.position != offset || self.response.is_none() {
|
||||||
|
let stop = self.stop;
|
||||||
|
if !self.do_request(offset, stop) {
|
||||||
|
println_err!("Failed to seek to {}", offset);
|
||||||
|
return Err(GstFlowReturn::Error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
match self.response {
|
match self.response {
|
||||||
None => return Err(GstFlowReturn::Error),
|
None => return Err(GstFlowReturn::Error),
|
||||||
Some(ref mut r) => {
|
Some(ref mut r) => {
|
||||||
if self.position != offset {
|
|
||||||
println_err!("Failed to seek to {}", offset);
|
|
||||||
return Err(GstFlowReturn::Error);
|
|
||||||
}
|
|
||||||
|
|
||||||
match r.read(data) {
|
match r.read(data) {
|
||||||
Ok(size) => {
|
Ok(size) => {
|
||||||
if size == 0 {
|
if size == 0 {
|
||||||
|
|
Loading…
Reference in a new issue