gstreamer/gst/dataurisrc/gstdataurisrc.c

482 lines
13 KiB
C

/* GStreamer
*
* Copyright (C) 2009 Igalia S.L
* Copyright (C) 2009 Sebastian Dröge <sebastian.droege@collabora.co.uk>
*
* 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., 59 Temple Place - Suite 330,
* Boston, MA 02111-1307, USA.
*/
/**
* SECTION:element-dataurisrc
*
* dataurisrc handles data: URIs, see <ulink url="http://tools.ietf.org/html/rfc2397">RFC 2397</ulink> for more information.
*
* <refsect2>
* <title>Example launch line</title>
* |[
* gst-launch-0.10 -v dataurisrc uri="data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAfElEQVQ4je2MwQnAIAxFgziA4EnczIsO4MEROo/gzZWc4xdTbe1R6LGRR74heYS7iKElzfcMiRnt4hf8gk8EayB6luefue/HzlJfCA50XsNjYRxprZmenXNIKSGEsC+QUqK1hhgj521BzhnWWiilUGvdF5RS4L2HMQZCCJy8sHMm2TYdJAAAAABJRU5ErkJggg==" ! pngdec ! ffmpegcolorspace ! freeze ! ffmpegcolorspace ! autovideosink
* ]| This pipeline displays a small 16x16 PNG image from the data URI.
* </refsect2>
*/
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include "gstdataurisrc.h"
#include <string.h>
#include <gst/base/gsttypefindhelper.h>
GST_DEBUG_CATEGORY (data_uri_src_debug);
#define GST_CAT_DEFAULT (data_uri_src_debug)
static GstStaticPadTemplate src_template = GST_STATIC_PAD_TEMPLATE ("src",
GST_PAD_SRC,
GST_PAD_ALWAYS,
GST_STATIC_CAPS_ANY);
enum
{
PROP_0,
PROP_URI,
};
static void gst_data_uri_src_finalize (GObject * object);
static void gst_data_uri_src_set_property (GObject * object,
guint prop_id, const GValue * value, GParamSpec * pspec);
static void gst_data_uri_src_get_property (GObject * object,
guint prop_id, GValue * value, GParamSpec * pspec);
static GstCaps *gst_data_uri_src_get_caps (GstBaseSrc * src);
static gboolean gst_data_uri_src_get_size (GstBaseSrc * src, guint64 * size);
static gboolean gst_data_uri_src_is_seekable (GstBaseSrc * src);
static GstFlowReturn gst_data_uri_src_create (GstBaseSrc * src, guint64 offset,
guint size, GstBuffer ** buf);
static gboolean gst_data_uri_src_check_get_range (GstBaseSrc * src);
static gboolean gst_data_uri_src_start (GstBaseSrc * src);
static void gst_data_uri_src_handler_init (gpointer g_iface,
gpointer iface_data);
static GstURIType gst_data_uri_src_get_uri_type (void);
static gchar **gst_data_uri_src_get_protocols (void);
static const gchar *gst_data_uri_src_get_uri (GstURIHandler * handler);
static gboolean gst_data_uri_src_set_uri (GstURIHandler * handler,
const gchar * uri);
static void
_do_init (GType gtype)
{
static const GInterfaceInfo urihandler_info = {
gst_data_uri_src_handler_init,
0, 0
};
GST_DEBUG_CATEGORY_INIT (data_uri_src_debug, "dataurisrc", 0,
"data: URI source");
g_type_add_interface_static (gtype, GST_TYPE_URI_HANDLER, &urihandler_info);
}
GST_BOILERPLATE_FULL (GstDataURISrc, gst_data_uri_src, GstBaseSrc,
GST_TYPE_BASE_SRC, _do_init);
static void
gst_data_uri_src_base_init (gpointer klass)
{
GstElementClass *element_class = (GstElementClass *) (klass);
gst_element_class_add_static_pad_template (element_class, &src_template);
gst_element_class_set_details_simple (element_class,
"data: URI source element", "Source", "Handles data: uris",
"Philippe Normand <pnormand@igalia.com>, "
"Sebastian Dröge <sebastian.droege@collabora.co.uk>");
}
static void
gst_data_uri_src_class_init (GstDataURISrcClass * klass)
{
GObjectClass *gobject_class = (GObjectClass *) klass;
GstBaseSrcClass *basesrc_class = (GstBaseSrcClass *) klass;
gobject_class->finalize = gst_data_uri_src_finalize;
gobject_class->set_property = gst_data_uri_src_set_property;
gobject_class->get_property = gst_data_uri_src_get_property;
g_object_class_install_property (gobject_class, PROP_URI,
g_param_spec_string ("uri",
"URI",
"URI that should be used",
NULL, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
basesrc_class->get_caps = GST_DEBUG_FUNCPTR (gst_data_uri_src_get_caps);
basesrc_class->get_size = GST_DEBUG_FUNCPTR (gst_data_uri_src_get_size);
basesrc_class->is_seekable = GST_DEBUG_FUNCPTR (gst_data_uri_src_is_seekable);
basesrc_class->create = GST_DEBUG_FUNCPTR (gst_data_uri_src_create);
basesrc_class->check_get_range =
GST_DEBUG_FUNCPTR (gst_data_uri_src_check_get_range);
basesrc_class->start = GST_DEBUG_FUNCPTR (gst_data_uri_src_start);
}
static void
gst_data_uri_src_init (GstDataURISrc * src, GstDataURISrcClass * g_class)
{
gst_base_src_set_format (GST_BASE_SRC (src), GST_FORMAT_BYTES);
}
static void
gst_data_uri_src_finalize (GObject * object)
{
GstDataURISrc *src = GST_DATA_URI_SRC (object);
g_free (src->uri);
src->uri = NULL;
if (src->buffer)
gst_buffer_unref (src->buffer);
src->buffer = NULL;
G_OBJECT_CLASS (parent_class)->finalize (object);
}
static void
gst_data_uri_src_set_property (GObject * object, guint prop_id,
const GValue * value, GParamSpec * pspec)
{
GstDataURISrc *src = GST_DATA_URI_SRC (object);
switch (prop_id) {
case PROP_URI:
gst_data_uri_src_set_uri (GST_URI_HANDLER (src),
g_value_get_string (value));
break;
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
break;
}
}
static void
gst_data_uri_src_get_property (GObject * object,
guint prop_id, GValue * value, GParamSpec * pspec)
{
GstDataURISrc *src = GST_DATA_URI_SRC (object);
switch (prop_id) {
case PROP_URI:
g_value_set_string (value,
gst_data_uri_src_get_uri (GST_URI_HANDLER (src)));
break;
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
break;
}
}
static GstCaps *
gst_data_uri_src_get_caps (GstBaseSrc * basesrc)
{
GstDataURISrc *src = GST_DATA_URI_SRC (basesrc);
GstCaps *caps;
GST_OBJECT_LOCK (src);
if (!src->buffer || !GST_BUFFER_CAPS (src->buffer))
caps = gst_caps_new_empty ();
else
caps = gst_buffer_get_caps (src->buffer);
GST_OBJECT_UNLOCK (src);
return caps;
}
static gboolean
gst_data_uri_src_get_size (GstBaseSrc * basesrc, guint64 * size)
{
GstDataURISrc *src = GST_DATA_URI_SRC (basesrc);
gboolean ret;
GST_OBJECT_LOCK (src);
if (!src->buffer) {
ret = FALSE;
*size = -1;
} else {
ret = TRUE;
*size = GST_BUFFER_SIZE (src->buffer);
}
GST_OBJECT_UNLOCK (src);
return ret;
}
static gboolean
gst_data_uri_src_is_seekable (GstBaseSrc * basesrc)
{
return TRUE;
}
static GstFlowReturn
gst_data_uri_src_create (GstBaseSrc * basesrc, guint64 offset, guint size,
GstBuffer ** buf)
{
GstDataURISrc *src = GST_DATA_URI_SRC (basesrc);
GstFlowReturn ret;
GST_OBJECT_LOCK (src);
if (!src->buffer)
goto no_buffer;
/* This is only correct because GstBaseSrc already clips size for us to be no
* larger than the max. available size if a segment at the end is requested */
if (offset + size > GST_BUFFER_SIZE (src->buffer)) {
ret = GST_FLOW_UNEXPECTED;
} else {
ret = GST_FLOW_OK;
*buf = gst_buffer_create_sub (src->buffer, offset, size);
gst_buffer_set_caps (*buf, GST_BUFFER_CAPS (src->buffer));
}
GST_OBJECT_UNLOCK (src);
return ret;
/* ERRORS */
no_buffer:
{
GST_OBJECT_UNLOCK (src);
GST_ELEMENT_ERROR (src, RESOURCE, NOT_FOUND, (NULL), (NULL));
return GST_FLOW_NOT_NEGOTIATED;
}
}
static gboolean
gst_data_uri_src_check_get_range (GstBaseSrc * basesrc)
{
return TRUE;
}
static gboolean
gst_data_uri_src_start (GstBaseSrc * basesrc)
{
GstDataURISrc *src = GST_DATA_URI_SRC (basesrc);
GST_OBJECT_LOCK (src);
if (src->uri == NULL || *src->uri == '\0' || src->buffer == NULL)
goto no_uri;
GST_OBJECT_UNLOCK (src);
return TRUE;
/* ERRORS */
no_uri:
{
GST_OBJECT_UNLOCK (src);
GST_ELEMENT_ERROR (src, RESOURCE, OPEN_READ,
("No valid data URI specified, or the data URI could not be parsed."),
("%s", src->uri));
return FALSE;
}
}
static void
gst_data_uri_src_handler_init (gpointer g_iface, gpointer iface_data)
{
GstURIHandlerInterface *iface = (GstURIHandlerInterface *) g_iface;
iface->get_type = gst_data_uri_src_get_uri_type;
iface->get_protocols = gst_data_uri_src_get_protocols;
iface->get_uri = gst_data_uri_src_get_uri;
iface->set_uri = gst_data_uri_src_set_uri;
}
static GstURIType
gst_data_uri_src_get_uri_type (void)
{
return GST_URI_SRC;
}
static gchar **
gst_data_uri_src_get_protocols (void)
{
static gchar *protocols[] = { (char *) "data", 0 };
return protocols;
}
static const gchar *
gst_data_uri_src_get_uri (GstURIHandler * handler)
{
GstDataURISrc *src = GST_DATA_URI_SRC (handler);
return src->uri;
}
static gboolean
gst_data_uri_src_set_uri (GstURIHandler * handler, const gchar * uri)
{
GstDataURISrc *src = GST_DATA_URI_SRC (handler);
gboolean ret = FALSE;
gchar *mimetype = NULL;
const gchar *parameters_start;
const gchar *data_start;
const gchar *orig_uri = uri;
GstCaps *caps;
gboolean base64 = FALSE;
gchar *charset = NULL;
GST_OBJECT_LOCK (src);
if (GST_STATE (src) >= GST_STATE_PAUSED)
goto wrong_state;
/* uri must be an URI as defined in RFC 2397
* data:[<mediatype>][;base64],<data>
*/
if (strncmp ("data:", uri, 5) != 0)
goto invalid_uri;
uri += 5;
parameters_start = strchr (uri, ';');
data_start = strchr (uri, ',');
if (data_start == NULL)
goto invalid_uri;
if (data_start != uri && parameters_start != uri)
mimetype =
g_strndup (uri,
(parameters_start ? parameters_start : data_start) - uri);
else
mimetype = g_strdup ("text/plain");
GST_DEBUG_OBJECT (src, "Mimetype: %s", mimetype);
if (parameters_start != NULL) {
gchar **walk;
gchar *parameters =
g_strndup (parameters_start + 1, data_start - parameters_start - 1);
gchar **parameters_strv;
parameters_strv = g_strsplit (parameters, ";", -1);
GST_DEBUG_OBJECT (src, "Parameters: ");
walk = parameters_strv;
while (*walk) {
GST_DEBUG_OBJECT (src, "\t %s", *walk);
if (strcmp ("base64", *walk) == 0) {
base64 = TRUE;
} else if (strncmp ("charset=", *walk, 8) == 0) {
charset = g_strdup (*walk + 8);
}
walk++;
}
g_free (parameters);
g_strfreev (parameters_strv);
}
/* Skip comma */
data_start += 1;
if (base64) {
gsize bsize;
src->buffer = gst_buffer_new ();
GST_BUFFER_DATA (src->buffer) =
(guint8 *) g_base64_decode (data_start, &bsize);
GST_BUFFER_MALLOCDATA (src->buffer) = GST_BUFFER_DATA (src->buffer);
GST_BUFFER_SIZE (src->buffer) = bsize;
} else {
gchar *data;
/* URI encoded, i.e. "percent" encoding */
data = g_uri_unescape_string (data_start, NULL);
if (data == NULL)
goto invalid_uri_encoded_data;
src->buffer = gst_buffer_new ();
GST_BUFFER_DATA (src->buffer) = (guint8 *) data;
GST_BUFFER_MALLOCDATA (src->buffer) = GST_BUFFER_DATA (src->buffer);
GST_BUFFER_SIZE (src->buffer) = strlen (data) + 1;
}
/* Convert to UTF8 */
if (strcmp ("text/plain", mimetype) == 0 &&
charset && g_ascii_strcasecmp ("US-ASCII", charset) != 0
&& g_ascii_strcasecmp ("UTF-8", charset) != 0) {
gsize read;
gsize written;
gchar *old_data = (gchar *) GST_BUFFER_DATA (src->buffer);
gchar *data;
data =
g_convert_with_fallback (old_data, -1, "UTF-8", charset, (char *) "*",
&read, &written, NULL);
g_free (old_data);
GST_BUFFER_DATA (src->buffer) = GST_BUFFER_MALLOCDATA (src->buffer) =
(guint8 *) data;
GST_BUFFER_SIZE (src->buffer) = written;
}
caps = gst_type_find_helper_for_buffer (GST_OBJECT (src), src->buffer, NULL);
if (!caps)
caps = gst_caps_new_simple (mimetype, NULL);
gst_buffer_set_caps (src->buffer, caps);
gst_caps_unref (caps);
g_free (src->uri);
src->uri = g_strdup (orig_uri);
ret = TRUE;
out:
GST_OBJECT_UNLOCK (src);
g_free (mimetype);
g_free (charset);
return ret;
invalid_uri:
{
GST_WARNING_OBJECT (src, "invalid URI '%s'", uri);
goto out;
}
wrong_state:
{
GST_WARNING_OBJECT (src, "Can't set URI in %s state",
gst_element_state_get_name (GST_STATE (src)));
goto out;
}
invalid_uri_encoded_data:
{
GST_WARNING_OBJECT (src, "Failed to parse data encoded in URI '%s'", uri);
goto out;
}
}
static gboolean
plugin_init (GstPlugin * plugin)
{
return gst_element_register (plugin, "dataurisrc",
GST_RANK_PRIMARY, GST_TYPE_DATA_URI_SRC);
}
GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
GST_VERSION_MINOR,
"dataurisrc",
"data: URI source",
plugin_init, VERSION, GST_LICENSE, GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN);