Add initial HTTP source based on Hyper

No seeking yet
This commit is contained in:
Sebastian Dröge 2016-05-14 17:04:53 +03:00
parent ec01d11393
commit b2b8332043
4 changed files with 160 additions and 2 deletions

View file

@ -7,6 +7,7 @@ build = "build.rs"
[dependencies] [dependencies]
libc = "0.2" libc = "0.2"
url = "1.1" url = "1.1"
hyper = "0.9"
[build-dependencies] [build-dependencies]
gcc = "0.3" gcc = "0.3"

View file

@ -2,16 +2,19 @@
extern crate libc; extern crate libc;
extern crate url; extern crate url;
extern crate hyper;
#[macro_use] #[macro_use]
pub mod utils; pub mod utils;
pub mod rssource; pub mod rssource;
pub mod rsfilesrc; pub mod rsfilesrc;
pub mod rshttpsrc;
pub mod rsfilesink; pub mod rsfilesink;
use utils::*; use utils::*;
use rssource::Source; use rssource::Source;
use rsfilesrc::FileSrc; use rsfilesrc::FileSrc;
use rshttpsrc::HttpSrc;
use std::os::raw::c_void; use std::os::raw::c_void;
use libc::{c_char}; use libc::{c_char};
@ -42,6 +45,17 @@ pub extern "C" fn sources_register(plugin: *const c_void) -> GBoolean {
256 + 100, 256 + 100,
FileSrc::new_ptr, FileSrc::new_ptr,
CString::new("file").unwrap().as_ptr()); CString::new("file").unwrap().as_ptr());
gst_rs_source_register(plugin,
CString::new("rshttpsrc").unwrap().as_ptr(),
CString::new("HTTP Source").unwrap().as_ptr(),
CString::new("Read HTTP/HTTPS files").unwrap().as_ptr(),
CString::new("Source/Network/HTTP").unwrap().as_ptr(),
CString::new("Sebastian Dröge <sebastian@centricular.com>").unwrap().as_ptr(),
256 + 100,
HttpSrc::new_ptr,
CString::new("http:https").unwrap().as_ptr());
} }
return GBoolean::True; return GBoolean::True;
} }

142
src/rshttpsrc.rs Normal file
View file

@ -0,0 +1,142 @@
use std::u64;
use std::io::{Read, Seek, SeekFrom};
use url::Url;
use hyper::header::ContentLength;
use hyper::client::Client;
use hyper::client::response::Response;
use std::io::Write;
use utils::*;
use rssource::*;
#[derive(Debug)]
pub struct HttpSrc {
url: Option<Url>,
client: Client,
response: Option<Response>,
position: u64,
size: u64,
}
impl HttpSrc {
fn new() -> HttpSrc {
HttpSrc { url: None, client: Client::new(), response: None, position: 0, size: u64::MAX }
}
fn new_source() -> Box<Source> {
Box::new(HttpSrc::new())
}
pub extern "C" fn new_ptr() -> *mut Box<Source> {
let instance = Box::new(HttpSrc::new_source());
return Box::into_raw(instance);
}
}
impl Source for HttpSrc {
fn set_uri(&mut self, uri_str: &Option<String>) -> bool {
match *uri_str {
None => {
self.url = None;
return true;
},
Some(ref uri_str) => {
let uri_parsed = Url::parse(uri_str.as_str());
match uri_parsed {
Ok(u) => {
if u.scheme() == "http" ||
u.scheme() == "https" {
self.url = Some(u);
return true;
} else {
self.url = None;
println_err!("Unsupported file URI '{}'", uri_str);
return false;
}
},
Err(err) => {
self.url = None;
println_err!("Failed to parse URI '{}': {}", uri_str, err);
return false;
}
}
}
}
}
fn get_uri(&self) -> Option<String> {
self.url.clone().map(|u| u.into_string())
}
fn is_seekable(&self) -> bool {
false
}
fn get_size(&self) -> u64 {
self.size
}
fn start(&mut self) -> bool {
self.response = None;
self.position = 0;
self.size = u64::MAX;
match self.url {
None => return false,
Some(ref url) => {
let response = self.client.get(url.clone()).send().unwrap();
if response.status.is_success() {
match response.headers.get::<ContentLength>() {
Some(&ContentLength(size)) => self.size = size,
_ => self.size = u64::MAX
}
self.response = Some(response);
return true;
} else {
println_err!("Failed to fetch {}: {}", url, response.status);
return false;
}
},
}
}
fn stop(&mut self) -> bool {
self.position = 0;
self.size = u64::MAX;
match self.response {
Some(ref mut response) => drop(response),
None => ()
}
self.response = None;
true
}
fn fill(&mut self, offset: u64, data: &mut [u8]) -> Result<usize, GstFlowReturn> {
match self.response {
None => return Err(GstFlowReturn::Error),
Some(ref mut r) => {
if self.position != offset {
println_err!("Failed to seek to {}", offset);
return Err(GstFlowReturn::Error);
}
match r.read(data) {
Ok(size) => {
if size == 0 {
return Err(GstFlowReturn::Eos);
}
self.position += size as u64;
return Ok(size)
},
Err(err) => {
println_err!("Failed to read at {}: {}", offset, err.to_string());
return Err(GstFlowReturn::Error);
},
}
},
}
}
}

View file

@ -214,9 +214,10 @@ gst_rs_src_uri_get_type (GType type)
static const gchar *const * static const gchar *const *
gst_rs_src_uri_get_protocols (GType type) gst_rs_src_uri_get_protocols (GType type)
{ {
static const gchar *protocols[] = { "file", NULL }; ElementData * data = g_hash_table_lookup (sources, GSIZE_TO_POINTER (type));
g_assert (data != NULL);
return protocols; return (const gchar * const *) data->protocols;
} }
static gchar * static gchar *