mirrors/gst-plugins-rs
mirrors/gst-plugins-rs
1
1
Fork 0
mirror of https://gitlab.freedesktop.org/gstreamer/gst-plugins-rs.git synced 2025-02-20 14:46:24 +00:00
Code Issues Projects Releases Wiki Activity

Implement parsing of URIs in Rust

Browse source
This commit is contained in:
Sebastian Dröge 2016-05-14 12:34:50 +03:00
parent 5b6a05b548
commit 15865ab86b
4 changed files with 66 additions and 77 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
Cargo.toml
View file

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

1
src/lib.rs
View file

@ -1,6 +1,7 @@
#![crate_type="dylib"]
extern crate libc;
extern crate url;
pub mod rsfilesrc;
pub mod rsfilesink;

75
src/rsfilesrc.c
View file

@ -7,8 +7,8 @@
extern void * filesrc_new (void);
extern void filesrc_drop (void * filesrc);
extern GstFlowReturn filesrc_fill (void * filesrc, uint64_t offset, void * data, size_t * data_len);
extern void filesrc_set_location (void * filesrc, const char *location);
extern char * filesrc_get_location (void * filesrc);
extern gboolean filesrc_set_uri (void * filesrc, const char *uri);
extern char * filesrc_get_uri (void * filesrc);
extern uint64_t filesrc_get_size (void * filesrc);
extern gboolean filesrc_is_seekable (void * filesrc);
extern gboolean filesrc_start (void * filesrc);
@ -25,7 +25,7 @@ static GstStaticPadTemplate src_template = GST_STATIC_PAD_TEMPLATE ("src",
enum
{
PROP_0,
PROP_LOCATION
PROP_URI
};
static void gst_rsfile_src_uri_handler_init (gpointer g_iface,
@ -66,9 +66,9 @@ gst_rsfile_src_class_init (GstRsfileSrcClass * klass)
gobject_class->set_property = gst_rsfile_src_set_property;
gobject_class->get_property = gst_rsfile_src_get_property;
g_object_class_install_property (gobject_class, PROP_LOCATION,
g_param_spec_string ("location", "File Location",
"Location of the file to read", NULL,
g_object_class_install_property (gobject_class, PROP_URI,
g_param_spec_string ("uri", "URI",
"URI of the file to read", NULL,
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS |
GST_PARAM_MUTABLE_READY));
@ -113,8 +113,8 @@ gst_rsfile_src_set_property (GObject * object, guint prop_id,
GstRsfileSrc *src = GST_RSFILE_SRC (object);
switch (prop_id) {
case PROP_LOCATION:
filesrc_set_location (src->instance, g_value_get_string (value));
case PROP_URI:
filesrc_set_uri (src->instance, g_value_get_string (value));
break;
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
@ -129,8 +129,8 @@ gst_rsfile_src_get_property (GObject * object, guint prop_id, GValue * value,
GstRsfileSrc *src = GST_RSFILE_SRC (object);
switch (prop_id) {
case PROP_LOCATION:
g_value_take_string (value, filesrc_get_location (src->instance));
case PROP_URI:
g_value_take_string (value, filesrc_get_uri (src->instance));
break;
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
@ -211,68 +211,23 @@ static gchar *
gst_rsfile_src_uri_get_uri (GstURIHandler * handler)
{
GstRsfileSrc *src = GST_RSFILE_SRC (handler);
gchar *location, *uri;
location = filesrc_get_location (src->instance);
if (!location)
return NULL;
uri = gst_filename_to_uri (location, NULL);
g_free (location);
return uri;
return filesrc_get_uri (src->instance);
}
static gboolean
gst_rsfile_src_uri_set_uri (GstURIHandler * handler, const gchar * uri,
GError ** err)
{
gchar *location, *hostname = NULL;
gboolean ret = FALSE;
GstRsfileSrc *src = GST_RSFILE_SRC (handler);
if (strcmp (uri, "file://") == 0) {
/* Special case for "file://" as this is used by some applications
* to test with gst_element_make_from_uri if there's an element
* that supports the URI protocol. */
filesrc_set_location (src->instance, location);
return TRUE;
}
location = g_filename_from_uri (uri, &hostname, err);
if (!location || (err != NULL && *err != NULL)) {
GST_WARNING_OBJECT (src, "Invalid URI '%s' for filesrc: %s", uri,
(err != NULL && *err != NULL) ? (*err)->message : "unknown error");
goto beach;
}
if ((hostname) && (strcmp (hostname, "localhost"))) {
/* Only 'localhost' is permitted */
GST_WARNING_OBJECT (src, "Invalid hostname '%s' for filesrc", hostname);
if (!filesrc_set_uri (src->instance, uri)) {
g_set_error (err, GST_URI_ERROR, GST_URI_ERROR_BAD_URI,
"File URI with invalid hostname '%s'", hostname);
goto beach;
"Can't handle URI '%s'", uri);
return FALSE;
}
#ifdef G_OS_WIN32
/* Unfortunately, g_filename_from_uri() doesn't handle some UNC paths
* correctly on windows, it leaves them with an extra backslash
* at the start if they're of the mozilla-style file://///host/path/file
* form. Correct this.
*/
if (location[0] == '\\' && location[1] == '\\' && location[2] == '\\')
memmove (location, location + 1, strlen (location + 1) + 1);
#endif
ret = TRUE;
filesrc_set_location (src->instance, location);
beach:
if (location)
g_free (location);
if (hostname)
g_free (hostname);
return ret;
return TRUE;
}
static void

66
src/rsfilesrc.rs
View file

@ -5,7 +5,8 @@ use std::u64;
use std::slice;
use std::io::{Read, Seek, SeekFrom};
use std::fs::File;
use std::path::Path;
use std::path::PathBuf;
use url::Url;
use std::io::Write;
@ -43,7 +44,7 @@ impl GBoolean {
#[derive(Debug)]
pub struct FileSrc {
location: Option<String>,
location: Option<PathBuf>,
file: Option<File>,
position: u64,
}
@ -53,12 +54,43 @@ impl FileSrc {
FileSrc { location: None, file: None, position: 0 }
}
fn set_location(&mut self, location: &Option<String>) {
self.location = location.clone();
fn set_uri(&mut self, uri_str: &Option<String>) -> bool {
match *uri_str {
None => {
self.location = None;
return true;
},
Some(ref uri_str) => {
let uri_parsed = Url::parse(uri_str.as_str());
match uri_parsed {
Ok(u) => {
match u.to_file_path().ok() {
Some(p) => {
self.location = Some(p);
return true;
},
None => {
self.location = None;
println_err!("Unsupported file URI '{}'", uri_str);
return false;
}
}
},
Err(err) => {
self.location = None;
println_err!("Failed to parse URI '{}': {}", uri_str, err);
return false;
}
}
}
}
}
fn get_location(&self) -> &Option<String> {
&self.location
fn get_uri(&self) -> Option<String> {
match self.location {
None => None,
Some(ref location) => Url::from_file_path(&location).map(|u| u.into_string()).ok()
}
}
fn is_seekable(&self) -> bool {
@ -81,13 +113,13 @@ impl FileSrc {
match self.location {
None => return false,
Some(ref location) => {
match File::open(Path::new(&location.clone())) {
match File::open(location.as_path()) {
Ok(file) => {
self.file = Some(file);
return true;
},
Err(err) => {
println_err!("Failed to open file '{}': {}", location, err.to_string());
println_err!("Failed to open file '{}': {}", location.to_str().unwrap_or("Non-UTF8 path"), err.to_string());
return false;
},
}
@ -145,24 +177,24 @@ pub extern "C" fn filesrc_drop(ptr: *mut FileSrc) {
}
#[no_mangle]
pub extern "C" fn filesrc_set_location(ptr: *mut FileSrc, location_ptr: *const c_char) {
pub extern "C" fn filesrc_set_uri(ptr: *mut FileSrc, uri_ptr: *const c_char) -> GBoolean{
let filesrc: &mut FileSrc = unsafe { &mut *ptr };
if location_ptr.is_null() {
filesrc.set_location(&None)
if uri_ptr.is_null() {
GBoolean::from_bool(filesrc.set_uri(&None))
} else {
let location = unsafe { CStr::from_ptr(location_ptr) };
filesrc.set_location(&Some(String::from(location.to_str().unwrap())));
let uri = unsafe { CStr::from_ptr(uri_ptr) };
GBoolean::from_bool(filesrc.set_uri(&Some(String::from(uri.to_str().unwrap()))))
}
}
#[no_mangle]
pub extern "C" fn filesrc_get_location(ptr: *mut FileSrc) -> *mut c_char {
pub extern "C" fn filesrc_get_uri(ptr: *mut FileSrc) -> *mut c_char {
let filesrc: &mut FileSrc = unsafe { &mut *ptr };
match *filesrc.get_location() {
Some(ref location) =>
CString::new(location.clone().into_bytes()).unwrap().into_raw(),
match filesrc.get_uri() {
Some(ref uri) =>
CString::new(uri.clone().into_bytes()).unwrap().into_raw(),
None =>
ptr::null_mut()
}