static gchar *
gst_data_uri_src_get_uri (GstURIHandler * handler)
{
GstDataURISrc *src = GST_DATA_URI_SRC (handler);
gchar *src_uri = NULL;
GST_OBJECT_LOCK (src);
src_uri = g_strdup (src->uri);
GST_OBJECT_UNLOCK (src);
return src_uri;
}
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 GstCaps *
gst_data_uri_src_get_caps (GstBaseSrc * basesrc, GstCaps * filter)
{
GstDataURISrc *src = GST_DATA_URI_SRC (basesrc);
GstCaps *caps;
GST_OBJECT_LOCK (src);
caps = gst_pad_get_current_caps (GST_BASE_SRC_PAD (basesrc));
if (!caps)
caps = gst_caps_new_any ();
GST_OBJECT_UNLOCK (src);
return caps;
}
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 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_get_size (src->buffer)) {
ret = GST_FLOW_EOS;
} else if (*buf != NULL) {
GstMapInfo src_info;
GstMapInfo dest_info;
gsize fill_size;
gst_buffer_map (src->buffer, &src_info, GST_MAP_READ);
gst_buffer_map (*buf, &dest_info, GST_MAP_WRITE);
fill_size = gst_buffer_fill (*buf, 0, src_info.data + offset, size);
gst_buffer_unmap (*buf, &dest_info);
gst_buffer_unmap (src->buffer, &src_info);
gst_buffer_set_size (*buf, fill_size);
ret = GST_FLOW_OK;
} else {
*buf =
gst_buffer_copy_region (src->buffer, GST_BUFFER_COPY_ALL, offset, size);
ret = GST_FLOW_OK;
}
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_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_get_size (src->buffer);
}
GST_OBJECT_UNLOCK (src);
return ret;
}
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 gboolean
gst_data_uri_src_set_uri (GstURIHandler * handler, const gchar * uri,
GError ** error)
{
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;
GstBuffer *buffer;
gboolean base64 = FALSE;
gchar *charset = NULL;
gpointer bdata;
gsize bsize;
GST_OBJECT_LOCK (src);
if (GST_STATE (src) >= GST_STATE_PAUSED)
goto wrong_state;
GST_OBJECT_UNLOCK (src);
/* 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) {
bdata = g_base64_decode (data_start, &bsize);
} else {
/* URI encoded, i.e. "percent" encoding */
bdata = g_uri_unescape_string (data_start, NULL);
if (bdata == NULL)
goto invalid_uri_encoded_data;
bsize = strlen (bdata) + 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;
gpointer data;
data =
g_convert_with_fallback (bdata, -1, "UTF-8", charset, (char *) "*",
&read, &written, NULL);
g_free (bdata);
bdata = data;
bsize = written;
}
buffer = gst_buffer_new_wrapped (bdata, bsize);
caps = gst_type_find_helper_for_buffer (GST_OBJECT (src), buffer, NULL);
if (!caps)
caps = gst_caps_new_empty_simple (mimetype);
gst_base_src_set_caps (GST_BASE_SRC_CAST (src), caps);
gst_caps_unref (caps);
GST_OBJECT_LOCK (src);
gst_buffer_replace (&src->buffer, buffer);
gst_buffer_unref (buffer);
g_free (src->uri);
src->uri = g_strdup (orig_uri);
GST_OBJECT_UNLOCK (src);
ret = TRUE;
out:
g_free (mimetype);
g_free (charset);
return ret;
wrong_state:
{
GST_WARNING_OBJECT (src, "Can't set URI in %s state",
gst_element_state_get_name (GST_STATE (src)));
GST_OBJECT_UNLOCK (src);
g_set_error (error, GST_URI_ERROR, GST_URI_ERROR_BAD_STATE,
"Changing the 'uri' property on dataurisrc while it is running "
"is not supported");
goto out;
}
invalid_uri:
{
GST_WARNING_OBJECT (src, "invalid URI '%s'", uri);
g_set_error (error, GST_URI_ERROR, GST_URI_ERROR_BAD_URI,
"Invalid data URI");
goto out;
}
invalid_uri_encoded_data:
{
GST_WARNING_OBJECT (src, "Failed to parse data encoded in URI '%s'", uri);
g_set_error (error, GST_URI_ERROR, GST_URI_ERROR_BAD_URI,
"Could not parse data encoded in data URI");
goto out;
}
}
