dataurisrc: Remove unnecessary else if condition
[platform/upstream/gstreamer.git] / plugins / elements / gstdataurisrc.c
1 /* GStreamer
2  *
3  * Copyright (C) 2009 Igalia S.L
4  * Copyright (C) 2009 Sebastian Dröge <sebastian.droege@collabora.co.uk>
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Library General Public
8  * License as published by the Free Software Foundation; either
9  * version 2 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Library General Public License for more details.
15  *
16  * You should have received a copy of the GNU Library General Public
17  * License along with this library; if not, write to the
18  * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
19  * Boston, MA 02110-1301, USA.
20  */
21
22 /**
23  * SECTION:element-dataurisrc
24  *
25  * dataurisrc handles data: URIs, see <ulink url="http://tools.ietf.org/html/rfc2397">RFC 2397</ulink> for more information.
26  *
27  * <refsect2>
28  * <title>Example launch line</title>
29  * |[
30  * gst-launch-0.10 -v dataurisrc uri="data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAfElEQVQ4je2MwQnAIAxFgziA4EnczIsO4MEROo/gzZWc4xdTbe1R6LGRR74heYS7iKElzfcMiRnt4hf8gk8EayB6luefue/HzlJfCA50XsNjYRxprZmenXNIKSGEsC+QUqK1hhgj521BzhnWWiilUGvdF5RS4L2HMQZCCJy8sHMm2TYdJAAAAABJRU5ErkJggg==" ! pngdec ! videoconvert ! freeze ! videoconvert ! autovideosink
31  * ]| This pipeline displays a small 16x16 PNG image from the data URI.
32  * </refsect2>
33  */
34
35 #ifdef HAVE_CONFIG_H
36 #include "config.h"
37 #endif
38
39 #include "gstdataurisrc.h"
40
41 #include <string.h>
42 #include <gst/base/gsttypefindhelper.h>
43
44 GST_DEBUG_CATEGORY (data_uri_src_debug);
45 #define GST_CAT_DEFAULT (data_uri_src_debug)
46
47 static GstStaticPadTemplate src_template = GST_STATIC_PAD_TEMPLATE ("src",
48     GST_PAD_SRC,
49     GST_PAD_ALWAYS,
50     GST_STATIC_CAPS_ANY);
51
52 enum
53 {
54   PROP_0,
55   PROP_URI,
56 };
57
58 static void gst_data_uri_src_finalize (GObject * object);
59 static void gst_data_uri_src_set_property (GObject * object,
60     guint prop_id, const GValue * value, GParamSpec * pspec);
61 static void gst_data_uri_src_get_property (GObject * object,
62     guint prop_id, GValue * value, GParamSpec * pspec);
63
64 static GstCaps *gst_data_uri_src_get_caps (GstBaseSrc * src, GstCaps * filter);
65 static gboolean gst_data_uri_src_get_size (GstBaseSrc * src, guint64 * size);
66 static gboolean gst_data_uri_src_is_seekable (GstBaseSrc * src);
67 static GstFlowReturn gst_data_uri_src_create (GstBaseSrc * src, guint64 offset,
68     guint size, GstBuffer ** buf);
69 static gboolean gst_data_uri_src_start (GstBaseSrc * src);
70
71 static void gst_data_uri_src_handler_init (gpointer g_iface,
72     gpointer iface_data);
73 static GstURIType gst_data_uri_src_get_uri_type (GType type);
74 static const gchar *const *gst_data_uri_src_get_protocols (GType type);
75 static gchar *gst_data_uri_src_get_uri (GstURIHandler * handler);
76 static gboolean gst_data_uri_src_set_uri (GstURIHandler * handler,
77     const gchar * uri, GError ** error);
78
79
80 #define gst_data_uri_src_parent_class parent_class
81 G_DEFINE_TYPE_WITH_CODE (GstDataURISrc, gst_data_uri_src, GST_TYPE_BASE_SRC,
82     G_IMPLEMENT_INTERFACE (GST_TYPE_URI_HANDLER,
83         gst_data_uri_src_handler_init));
84
85 static void
86 gst_data_uri_src_class_init (GstDataURISrcClass * klass)
87 {
88   GObjectClass *gobject_class = (GObjectClass *) klass;
89   GstElementClass *element_class = (GstElementClass *) klass;
90   GstBaseSrcClass *basesrc_class = (GstBaseSrcClass *) klass;
91
92   gobject_class->finalize = gst_data_uri_src_finalize;
93   gobject_class->set_property = gst_data_uri_src_set_property;
94   gobject_class->get_property = gst_data_uri_src_get_property;
95
96   g_object_class_install_property (gobject_class, PROP_URI,
97       g_param_spec_string ("uri",
98           "URI",
99           "URI that should be used",
100           NULL, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
101
102   gst_element_class_add_pad_template (element_class,
103       gst_static_pad_template_get (&src_template));
104   gst_element_class_set_static_metadata (element_class,
105       "data: URI source element", "Source", "Handles data: uris",
106       "Philippe Normand <pnormand@igalia.com>, "
107       "Sebastian Dröge <sebastian.droege@collabora.co.uk>");
108
109   GST_DEBUG_CATEGORY_INIT (data_uri_src_debug, "dataurisrc", 0,
110       "data: URI source");
111
112   basesrc_class->get_caps = GST_DEBUG_FUNCPTR (gst_data_uri_src_get_caps);
113   basesrc_class->get_size = GST_DEBUG_FUNCPTR (gst_data_uri_src_get_size);
114   basesrc_class->is_seekable = GST_DEBUG_FUNCPTR (gst_data_uri_src_is_seekable);
115   basesrc_class->create = GST_DEBUG_FUNCPTR (gst_data_uri_src_create);
116   basesrc_class->start = GST_DEBUG_FUNCPTR (gst_data_uri_src_start);
117 }
118
119 static void
120 gst_data_uri_src_init (GstDataURISrc * src)
121 {
122 }
123
124 static void
125 gst_data_uri_src_finalize (GObject * object)
126 {
127   GstDataURISrc *src = GST_DATA_URI_SRC (object);
128
129   g_free (src->uri);
130   src->uri = NULL;
131
132   if (src->buffer)
133     gst_buffer_unref (src->buffer);
134   src->buffer = NULL;
135
136   G_OBJECT_CLASS (parent_class)->finalize (object);
137 }
138
139 static void
140 gst_data_uri_src_set_property (GObject * object, guint prop_id,
141     const GValue * value, GParamSpec * pspec)
142 {
143   GstDataURISrc *src = GST_DATA_URI_SRC (object);
144
145   switch (prop_id) {
146     case PROP_URI:
147       gst_data_uri_src_set_uri (GST_URI_HANDLER (src),
148           g_value_get_string (value), NULL);
149       break;
150     default:
151       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
152       break;
153   }
154 }
155
156 static void
157 gst_data_uri_src_get_property (GObject * object,
158     guint prop_id, GValue * value, GParamSpec * pspec)
159 {
160   GstDataURISrc *src = GST_DATA_URI_SRC (object);
161
162   switch (prop_id) {
163     case PROP_URI:
164       g_value_set_string (value,
165           gst_data_uri_src_get_uri (GST_URI_HANDLER (src)));
166       break;
167     default:
168       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
169       break;
170   }
171 }
172
173 static GstCaps *
174 gst_data_uri_src_get_caps (GstBaseSrc * basesrc, GstCaps * filter)
175 {
176   GstDataURISrc *src = GST_DATA_URI_SRC (basesrc);
177   GstCaps *caps;
178
179   GST_OBJECT_LOCK (src);
180   if (gst_pad_has_current_caps (GST_BASE_SRC_PAD (basesrc)))
181     caps = gst_pad_get_current_caps (GST_BASE_SRC_PAD (basesrc));
182   else
183     caps = gst_caps_new_any ();
184   GST_OBJECT_UNLOCK (src);
185
186   return caps;
187 }
188
189 static gboolean
190 gst_data_uri_src_get_size (GstBaseSrc * basesrc, guint64 * size)
191 {
192   GstDataURISrc *src = GST_DATA_URI_SRC (basesrc);
193   gboolean ret;
194
195   GST_OBJECT_LOCK (src);
196   if (!src->buffer) {
197     ret = FALSE;
198     *size = -1;
199   } else {
200     ret = TRUE;
201     *size = gst_buffer_get_size (src->buffer);
202   }
203   GST_OBJECT_UNLOCK (src);
204
205   return ret;
206 }
207
208 static gboolean
209 gst_data_uri_src_is_seekable (GstBaseSrc * basesrc)
210 {
211   return TRUE;
212 }
213
214 static GstFlowReturn
215 gst_data_uri_src_create (GstBaseSrc * basesrc, guint64 offset, guint size,
216     GstBuffer ** buf)
217 {
218   GstDataURISrc *src = GST_DATA_URI_SRC (basesrc);
219   GstFlowReturn ret;
220
221   GST_OBJECT_LOCK (src);
222
223   if (!src->buffer)
224     goto no_buffer;
225
226   /* This is only correct because GstBaseSrc already clips size for us to be no
227    * larger than the max. available size if a segment at the end is requested */
228   if (offset + size > gst_buffer_get_size (src->buffer)) {
229     ret = GST_FLOW_EOS;
230   } else {
231     /* Adding a check to unreference the buffer if *buf != NULL, 
232      * just in case a buffer is provided*/
233     if (*buf != NULL) {
234       gst_buffer_unref (*buf);
235       GST_FIXME_OBJECT (src, "Buffer is not supposed to be provided here");
236     }
237     *buf =
238         gst_buffer_copy_region (src->buffer, GST_BUFFER_COPY_ALL, offset, size);
239     ret = GST_FLOW_OK;
240   }
241   GST_OBJECT_UNLOCK (src);
242
243   return ret;
244
245 /* ERRORS */
246 no_buffer:
247   {
248     GST_OBJECT_UNLOCK (src);
249     GST_ELEMENT_ERROR (src, RESOURCE, NOT_FOUND, (NULL), (NULL));
250     return GST_FLOW_NOT_NEGOTIATED;
251   }
252 }
253
254 static gboolean
255 gst_data_uri_src_start (GstBaseSrc * basesrc)
256 {
257   GstDataURISrc *src = GST_DATA_URI_SRC (basesrc);
258
259   GST_OBJECT_LOCK (src);
260
261   if (src->uri == NULL || *src->uri == '\0' || src->buffer == NULL)
262     goto no_uri;
263
264   GST_OBJECT_UNLOCK (src);
265
266   return TRUE;
267
268 /* ERRORS */
269 no_uri:
270   {
271     GST_OBJECT_UNLOCK (src);
272     GST_ELEMENT_ERROR (src, RESOURCE, OPEN_READ,
273         ("No valid data URI specified, or the data URI could not be parsed."),
274         ("%s", src->uri));
275     return FALSE;
276   }
277 }
278
279 static GstURIType
280 gst_data_uri_src_get_uri_type (GType type)
281 {
282   return GST_URI_SRC;
283 }
284
285 static const gchar *const *
286 gst_data_uri_src_get_protocols (GType type)
287 {
288   static const gchar *protocols[] = { "data", 0 };
289
290   return protocols;
291 }
292
293 static gchar *
294 gst_data_uri_src_get_uri (GstURIHandler * handler)
295 {
296   GstDataURISrc *src = GST_DATA_URI_SRC (handler);
297   gchar *src_uri = NULL;
298
299   GST_OBJECT_LOCK (src);
300   src_uri = g_strdup (src->uri);
301   GST_OBJECT_UNLOCK (src);
302   return src_uri;
303 }
304
305 static gboolean
306 gst_data_uri_src_set_uri (GstURIHandler * handler, const gchar * uri,
307     GError ** error)
308 {
309   GstDataURISrc *src = GST_DATA_URI_SRC (handler);
310   gboolean ret = FALSE;
311   gchar *mimetype = NULL;
312   const gchar *parameters_start;
313   const gchar *data_start;
314   const gchar *orig_uri = uri;
315   GstCaps *caps;
316   GstBuffer *buffer;
317   gboolean base64 = FALSE;
318   gchar *charset = NULL;
319   gpointer bdata;
320   gsize bsize;
321
322   GST_OBJECT_LOCK (src);
323   if (GST_STATE (src) >= GST_STATE_PAUSED)
324     goto wrong_state;
325   GST_OBJECT_UNLOCK (src);
326
327   /* uri must be an URI as defined in RFC 2397
328    * data:[<mediatype>][;base64],<data>
329    */
330   if (strncmp ("data:", uri, 5) != 0)
331     goto invalid_uri;
332
333   uri += 5;
334
335   parameters_start = strchr (uri, ';');
336   data_start = strchr (uri, ',');
337   if (data_start == NULL)
338     goto invalid_uri;
339
340   if (data_start != uri && parameters_start != uri)
341     mimetype =
342         g_strndup (uri,
343         (parameters_start ? parameters_start : data_start) - uri);
344   else
345     mimetype = g_strdup ("text/plain");
346
347   GST_DEBUG_OBJECT (src, "Mimetype: %s", mimetype);
348
349   if (parameters_start != NULL) {
350     gchar **walk;
351     gchar *parameters =
352         g_strndup (parameters_start + 1, data_start - parameters_start - 1);
353     gchar **parameters_strv;
354
355     parameters_strv = g_strsplit (parameters, ";", -1);
356
357     GST_DEBUG_OBJECT (src, "Parameters: ");
358     walk = parameters_strv;
359     while (*walk) {
360       GST_DEBUG_OBJECT (src, "\t %s", *walk);
361       if (strcmp ("base64", *walk) == 0) {
362         base64 = TRUE;
363       } else if (strncmp ("charset=", *walk, 8) == 0) {
364         charset = g_strdup (*walk + 8);
365       }
366       walk++;
367     }
368     g_free (parameters);
369     g_strfreev (parameters_strv);
370   }
371
372   /* Skip comma */
373   data_start += 1;
374   if (base64) {
375     bdata = g_base64_decode (data_start, &bsize);
376   } else {
377     /* URI encoded, i.e. "percent" encoding */
378     bdata = g_uri_unescape_string (data_start, NULL);
379     if (bdata == NULL)
380       goto invalid_uri_encoded_data;
381     bsize = strlen (bdata) + 1;
382   }
383   /* Convert to UTF8 */
384   if (strcmp ("text/plain", mimetype) == 0 &&
385       charset && g_ascii_strcasecmp ("US-ASCII", charset) != 0
386       && g_ascii_strcasecmp ("UTF-8", charset) != 0) {
387     gsize read;
388     gsize written;
389     gpointer data;
390
391     data =
392         g_convert_with_fallback (bdata, -1, "UTF-8", charset, (char *) "*",
393         &read, &written, NULL);
394     g_free (bdata);
395
396     bdata = data;
397     bsize = written;
398   }
399   buffer = gst_buffer_new_wrapped (bdata, bsize);
400
401   caps = gst_type_find_helper_for_buffer (GST_OBJECT (src), buffer, NULL);
402   if (!caps)
403     caps = gst_caps_new_empty_simple (mimetype);
404   gst_base_src_set_caps (GST_BASE_SRC_CAST (src), caps);
405   gst_caps_unref (caps);
406
407   GST_OBJECT_LOCK (src);
408   gst_buffer_replace (&src->buffer, buffer);
409   gst_buffer_unref (buffer);
410   g_free (src->uri);
411   src->uri = g_strdup (orig_uri);
412   GST_OBJECT_UNLOCK (src);
413
414   ret = TRUE;
415
416 out:
417
418   g_free (mimetype);
419   g_free (charset);
420
421   return ret;
422
423 wrong_state:
424   {
425     GST_WARNING_OBJECT (src, "Can't set URI in %s state",
426         gst_element_state_get_name (GST_STATE (src)));
427     GST_OBJECT_UNLOCK (src);
428     g_set_error (error, GST_URI_ERROR, GST_URI_ERROR_BAD_STATE,
429         "Changing the 'uri' property on dataurisrc while it is running "
430         "is not supported");
431     goto out;
432   }
433 invalid_uri:
434   {
435     GST_WARNING_OBJECT (src, "invalid URI '%s'", uri);
436     g_set_error (error, GST_URI_ERROR, GST_URI_ERROR_BAD_URI,
437         "Invalid data URI");
438     goto out;
439   }
440 invalid_uri_encoded_data:
441   {
442     GST_WARNING_OBJECT (src, "Failed to parse data encoded in URI '%s'", uri);
443     g_set_error (error, GST_URI_ERROR, GST_URI_ERROR_BAD_URI,
444         "Could not parse data encoded in data URI");
445     goto out;
446   }
447 }
448
449 static void
450 gst_data_uri_src_handler_init (gpointer g_iface, gpointer iface_data)
451 {
452   GstURIHandlerInterface *iface = (GstURIHandlerInterface *) g_iface;
453
454   iface->get_type = gst_data_uri_src_get_uri_type;
455   iface->get_protocols = gst_data_uri_src_get_protocols;
456   iface->get_uri = gst_data_uri_src_get_uri;
457   iface->set_uri = gst_data_uri_src_set_uri;
458 }
459
460 static gboolean
461 plugin_init (GstPlugin * plugin)
462 {
463   return gst_element_register (plugin, "dataurisrc",
464       GST_RANK_PRIMARY, GST_TYPE_DATA_URI_SRC);
465 }
466
467 GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
468     GST_VERSION_MINOR,
469     dataurisrc,
470     "data: URI source",
471     plugin_init, VERSION, GST_LICENSE, GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN);