Unreviewed, rolling out r140420.
[WebKit-https.git] / Source / WebCore / platform / graphics / gstreamer / WebKitWebSourceGStreamer.cpp
1 /*
2  *  Copyright (C) 2009, 2010 Sebastian Dröge <sebastian.droege@collabora.co.uk>
3  *
4  *  This library is free software; you can redistribute it and/or
5  *  modify it under the terms of the GNU Lesser General Public
6  *  License as published by the Free Software Foundation; either
7  *  version 2 of the License, or (at your option) any later version.
8  *
9  *  This library is distributed in the hope that it will be useful,
10  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
11  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  *  Lesser General Public License for more details.
13  *
14  *  You should have received a copy of the GNU Lesser General Public
15  *  License along with this library; if not, write to the Free Software
16  *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
17  */
18
19 #include "config.h"
20 #include "WebKitWebSourceGStreamer.h"
21
22 #if ENABLE(VIDEO) && USE(GSTREAMER)
23
24 #include "Document.h"
25 #include "Frame.h"
26 #include "GRefPtrGStreamer.h"
27 #include "GStreamerVersioning.h"
28 #include "MediaPlayer.h"
29 #include "NetworkingContext.h"
30 #include "NotImplemented.h"
31 #include "ResourceHandleClient.h"
32 #include "ResourceHandleInternal.h"
33 #include "ResourceRequest.h"
34 #include "ResourceResponse.h"
35  
36 #include <gst/app/gstappsrc.h>
37 #include <gst/pbutils/missing-plugins.h>
38
39 #include <wtf/Noncopyable.h>
40 #include <wtf/gobject/GOwnPtr.h>
41 #include <wtf/gobject/GRefPtr.h>
42 #include <wtf/text/CString.h>
43
44 using namespace WebCore;
45
46 class StreamingClient : public ResourceHandleClient {
47     WTF_MAKE_NONCOPYABLE(StreamingClient); WTF_MAKE_FAST_ALLOCATED;
48     public:
49         StreamingClient(WebKitWebSrc*);
50         virtual ~StreamingClient();
51
52         virtual void willSendRequest(ResourceHandle*, ResourceRequest&, const ResourceResponse&);
53         virtual void didReceiveResponse(ResourceHandle*, const ResourceResponse&);
54         virtual void didReceiveData(ResourceHandle*, const char*, int, int);
55         virtual void didFinishLoading(ResourceHandle*, double /*finishTime*/);
56         virtual void didFail(ResourceHandle*, const ResourceError&);
57         virtual void wasBlocked(ResourceHandle*);
58         virtual void cannotShowURL(ResourceHandle*);
59
60     private:
61         WebKitWebSrc* m_src;
62 };
63
64 #define WEBKIT_WEB_SRC_GET_PRIVATE(obj) (G_TYPE_INSTANCE_GET_PRIVATE((obj), WEBKIT_TYPE_WEB_SRC, WebKitWebSrcPrivate))
65 struct _WebKitWebSrcPrivate {
66     GstAppSrc* appsrc;
67     GstPad* srcpad;
68     gchar* uri;
69
70     RefPtr<WebCore::Frame> frame;
71     WebCore::MediaPlayer* player;
72
73     StreamingClient* client;
74     RefPtr<ResourceHandle> resourceHandle;
75
76     guint64 offset;
77     guint64 size;
78     gboolean seekable;
79     gboolean paused;
80
81     guint64 requestedOffset;
82
83     guint needDataID;
84     guint enoughDataID;
85     guint seekID;
86
87     // icecast stuff
88     gboolean iradioMode;
89     gchar* iradioName;
90     gchar* iradioGenre;
91     gchar* iradioUrl;
92     gchar* iradioTitle;
93
94     // TRUE if appsrc's version is >= 0.10.27, see
95     // https://bugzilla.gnome.org/show_bug.cgi?id=609423
96     gboolean haveAppSrc27;
97 };
98
99 enum {
100     PROP_IRADIO_MODE = 1,
101     PROP_IRADIO_NAME,
102     PROP_IRADIO_GENRE,
103     PROP_IRADIO_URL,
104     PROP_IRADIO_TITLE,
105     PROP_LOCATION
106 };
107
108 static GstStaticPadTemplate srcTemplate = GST_STATIC_PAD_TEMPLATE("src",
109                                                                   GST_PAD_SRC,
110                                                                   GST_PAD_ALWAYS,
111                                                                   GST_STATIC_CAPS_ANY);
112
113 GST_DEBUG_CATEGORY_STATIC(webkit_web_src_debug);
114 #define GST_CAT_DEFAULT webkit_web_src_debug
115
116 static void webKitWebSrcUriHandlerInit(gpointer gIface, gpointer ifaceData);
117
118 static void webKitWebSrcFinalize(GObject*);
119 static void webKitWebSrcSetProperty(GObject*, guint propertyID, const GValue*, GParamSpec*);
120 static void webKitWebSrcGetProperty(GObject*, guint propertyID, GValue*, GParamSpec*);
121 static GstStateChangeReturn webKitWebSrcChangeState(GstElement*, GstStateChange);
122
123 static gboolean webKitWebSrcQueryWithParent(GstPad*, GstObject*, GstQuery*);
124 #ifndef GST_API_VERSION_1
125 static gboolean webKitWebSrcQuery(GstPad*, GstQuery*);
126 #endif
127
128 static void webKitWebSrcNeedDataCb(GstAppSrc*, guint length, gpointer userData);
129 static void webKitWebSrcEnoughDataCb(GstAppSrc*, gpointer userData);
130 static gboolean webKitWebSrcSeekDataCb(GstAppSrc*, guint64 offset, gpointer userData);
131
132 static void webKitWebSrcStop(WebKitWebSrc*, bool);
133
134 static GstAppSrcCallbacks appsrcCallbacks = {
135     webKitWebSrcNeedDataCb,
136     webKitWebSrcEnoughDataCb,
137     webKitWebSrcSeekDataCb,
138     { 0 }
139 };
140
141 #define webkit_web_src_parent_class parent_class
142 // We split this out into another macro to avoid a check-webkit-style error.
143 #define WEBKIT_WEB_SRC_CATEGORY_INIT GST_DEBUG_CATEGORY_INIT(webkit_web_src_debug, "webkitwebsrc", 0, "websrc element");
144 G_DEFINE_TYPE_WITH_CODE(WebKitWebSrc, webkit_web_src, GST_TYPE_BIN,
145                          G_IMPLEMENT_INTERFACE(GST_TYPE_URI_HANDLER, webKitWebSrcUriHandlerInit);
146                          WEBKIT_WEB_SRC_CATEGORY_INIT);
147
148 static void webkit_web_src_class_init(WebKitWebSrcClass* klass)
149 {
150     GObjectClass* oklass = G_OBJECT_CLASS(klass);
151     GstElementClass* eklass = GST_ELEMENT_CLASS(klass);
152
153     oklass->finalize = webKitWebSrcFinalize;
154     oklass->set_property = webKitWebSrcSetProperty;
155     oklass->get_property = webKitWebSrcGetProperty;
156
157     gst_element_class_add_pad_template(eklass,
158                                        gst_static_pad_template_get(&srcTemplate));
159     setGstElementClassMetadata(eklass, "WebKit Web source element", "Source", "Handles HTTP/HTTPS uris",
160                                "Sebastian Dröge <sebastian.droege@collabora.co.uk>");
161
162     // icecast stuff
163     g_object_class_install_property(oklass,
164                                     PROP_IRADIO_MODE,
165                                     g_param_spec_boolean("iradio-mode",
166                                                          "iradio-mode",
167                                                          "Enable internet radio mode (extraction of shoutcast/icecast metadata)",
168                                                          FALSE,
169                                                          (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
170
171     g_object_class_install_property(oklass,
172                                     PROP_IRADIO_NAME,
173                                     g_param_spec_string("iradio-name",
174                                                         "iradio-name",
175                                                         "Name of the stream",
176                                                         0,
177                                                         (GParamFlags) (G_PARAM_READABLE | G_PARAM_STATIC_STRINGS)));
178
179     g_object_class_install_property(oklass,
180                                     PROP_IRADIO_GENRE,
181                                     g_param_spec_string("iradio-genre",
182                                                         "iradio-genre",
183                                                         "Genre of the stream",
184                                                         0,
185                                                         (GParamFlags) (G_PARAM_READABLE | G_PARAM_STATIC_STRINGS)));
186                                                         
187     g_object_class_install_property(oklass,
188                                     PROP_IRADIO_URL,
189                                     g_param_spec_string("iradio-url",
190                                                         "iradio-url",
191                                                         "Homepage URL for radio stream",
192                                                         0,
193                                                         (GParamFlags) (G_PARAM_READABLE | G_PARAM_STATIC_STRINGS)));
194
195     g_object_class_install_property(oklass,
196                                     PROP_IRADIO_TITLE,
197                                     g_param_spec_string("iradio-title",
198                                                         "iradio-title",
199                                                         "Name of currently playing song",
200                                                         0,
201                                                         (GParamFlags) (G_PARAM_READABLE | G_PARAM_STATIC_STRINGS)));
202
203
204     /* Allows setting the uri using the 'location' property, which is used
205      * for example by gst_element_make_from_uri() */
206     g_object_class_install_property(oklass,
207                                     PROP_LOCATION,
208                                     g_param_spec_string("location",
209                                                         "location",
210                                                         "Location to read from",
211                                                         0,
212                                                         (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
213     eklass->change_state = webKitWebSrcChangeState;
214
215     g_type_class_add_private(klass, sizeof(WebKitWebSrcPrivate));
216 }
217
218 static void webkit_web_src_init(WebKitWebSrc* src)
219 {
220     WebKitWebSrcPrivate* priv = WEBKIT_WEB_SRC_GET_PRIVATE(src);
221
222     src->priv = priv;
223
224     priv->client = new StreamingClient(src);
225
226     priv->appsrc = GST_APP_SRC(gst_element_factory_make("appsrc", 0));
227     if (!priv->appsrc) {
228         GST_ERROR_OBJECT(src, "Failed to create appsrc");
229         return;
230     }
231
232     GstElementFactory* factory = GST_ELEMENT_FACTORY(GST_ELEMENT_GET_CLASS(priv->appsrc)->elementfactory);
233     priv->haveAppSrc27 = gst_plugin_feature_check_version(GST_PLUGIN_FEATURE(factory), 0, 10, 27);
234
235     gst_bin_add(GST_BIN(src), GST_ELEMENT(priv->appsrc));
236
237
238     GRefPtr<GstPad> targetPad = adoptGRef(gst_element_get_static_pad(GST_ELEMENT(priv->appsrc), "src"));
239     priv->srcpad = webkitGstGhostPadFromStaticTemplate(&srcTemplate, "src", targetPad.get());
240
241     gst_element_add_pad(GST_ELEMENT(src), priv->srcpad);
242
243 #ifdef GST_API_VERSION_1
244     GST_OBJECT_FLAG_SET(priv->srcpad, GST_PAD_FLAG_NEED_PARENT);
245     gst_pad_set_query_function(priv->srcpad, webKitWebSrcQueryWithParent);
246 #else
247     gst_pad_set_query_function(priv->srcpad, webKitWebSrcQuery);
248 #endif
249
250     gst_app_src_set_callbacks(priv->appsrc, &appsrcCallbacks, src, 0);
251     gst_app_src_set_emit_signals(priv->appsrc, FALSE);
252     gst_app_src_set_stream_type(priv->appsrc, GST_APP_STREAM_TYPE_SEEKABLE);
253
254     // 512k is a abitrary number but we should choose a value
255     // here to not pause/unpause the SoupMessage too often and
256     // to make sure there's always some data available for
257     // GStreamer to handle.
258     gst_app_src_set_max_bytes(priv->appsrc, 512 * 1024);
259
260     // Emit the need-data signal if the queue contains less
261     // than 20% of data. Without this the need-data signal
262     // is emitted when the queue is empty, we then dispatch
263     // the soup message unpausing to the main loop and from
264     // there unpause the soup message. This already takes
265     // quite some time and libsoup even needs some more time
266     // to actually provide data again. If we do all this
267     // already if the queue is 20% empty, it's much more
268     // likely that libsoup already provides new data before
269     // the queue is really empty.
270     // This might need tweaking for ports not using libsoup.
271     if (priv->haveAppSrc27)
272         g_object_set(priv->appsrc, "min-percent", 20, NULL);
273
274     webKitWebSrcStop(src, false);
275 }
276
277 static void webKitWebSrcFinalize(GObject* object)
278 {
279     WebKitWebSrc* src = WEBKIT_WEB_SRC(object);
280     WebKitWebSrcPrivate* priv = src->priv;
281
282     delete priv->client;
283
284     g_free(priv->uri);
285
286     GST_CALL_PARENT(G_OBJECT_CLASS, finalize, ((GObject* )(src)));
287 }
288
289 static void webKitWebSrcSetProperty(GObject* object, guint propID, const GValue* value, GParamSpec* pspec)
290 {
291     WebKitWebSrc* src = WEBKIT_WEB_SRC(object);
292     WebKitWebSrcPrivate* priv = src->priv;
293
294     switch (propID) {
295     case PROP_IRADIO_MODE:
296         priv->iradioMode = g_value_get_boolean(value);
297         break;
298     case PROP_LOCATION:
299 #ifdef GST_API_VERSION_1
300         gst_uri_handler_set_uri(reinterpret_cast<GstURIHandler*>(src), g_value_get_string(value), 0);
301 #else
302         gst_uri_handler_set_uri(reinterpret_cast<GstURIHandler*>(src), g_value_get_string(value));
303 #endif
304         break;
305     default:
306         G_OBJECT_WARN_INVALID_PROPERTY_ID(object, propID, pspec);
307         break;
308     }
309 }
310
311 static void webKitWebSrcGetProperty(GObject* object, guint propID, GValue* value, GParamSpec* pspec)
312 {
313     WebKitWebSrc* src = WEBKIT_WEB_SRC(object);
314     WebKitWebSrcPrivate* priv = src->priv;
315
316     switch (propID) {
317     case PROP_IRADIO_MODE:
318         g_value_set_boolean(value, priv->iradioMode);
319         break;
320     case PROP_IRADIO_NAME:
321         g_value_set_string(value, priv->iradioName);
322         break;
323     case PROP_IRADIO_GENRE:
324         g_value_set_string(value, priv->iradioGenre);
325         break;
326     case PROP_IRADIO_URL:
327         g_value_set_string(value, priv->iradioUrl);
328         break;
329     case PROP_IRADIO_TITLE:
330         g_value_set_string(value, priv->iradioTitle);
331         break;
332     case PROP_LOCATION:
333         g_value_set_string(value, priv->uri);
334         break;
335     default:
336         G_OBJECT_WARN_INVALID_PROPERTY_ID(object, propID, pspec);
337         break;
338     }
339 }
340
341
342 static void webKitWebSrcStop(WebKitWebSrc* src, bool seeking)
343 {
344     WebKitWebSrcPrivate* priv = src->priv;
345
346     if (priv->resourceHandle) {
347         priv->resourceHandle->cancel();
348         priv->resourceHandle.release();
349     }
350     priv->resourceHandle = 0;
351
352     if (priv->frame && !seeking)
353         priv->frame.clear();
354
355     priv->player = 0;
356
357     GST_OBJECT_LOCK(src);
358     if (priv->needDataID)
359         g_source_remove(priv->needDataID);
360     priv->needDataID = 0;
361
362     if (priv->enoughDataID)
363         g_source_remove(priv->enoughDataID);
364     priv->enoughDataID = 0;
365
366     if (priv->seekID)
367         g_source_remove(priv->seekID);
368     priv->seekID = 0;
369
370     priv->paused = FALSE;
371     GST_OBJECT_UNLOCK(src);
372
373     g_free(priv->iradioName);
374     priv->iradioName = 0;
375
376     g_free(priv->iradioGenre);
377     priv->iradioGenre = 0;
378
379     g_free(priv->iradioUrl);
380     priv->iradioUrl = 0;
381
382     g_free(priv->iradioTitle);
383     priv->iradioTitle = 0;
384
385     if (priv->appsrc) {
386         gst_app_src_set_caps(priv->appsrc, 0);
387         if (!seeking)
388             gst_app_src_set_size(priv->appsrc, -1);
389     }
390
391     priv->offset = 0;
392     priv->seekable = FALSE;
393
394     if (!seeking) {
395         priv->size = 0;
396         priv->requestedOffset = 0;
397     }
398
399     GST_DEBUG_OBJECT(src, "Stopped request");
400 }
401
402 static bool webKitWebSrcStart(WebKitWebSrc* src)
403 {
404     WebKitWebSrcPrivate* priv = src->priv;
405
406     if (!priv->uri) {
407         GST_ERROR_OBJECT(src, "No URI provided");
408         return false;
409     }
410     
411     KURL url = KURL(KURL(), priv->uri);
412
413     ResourceRequest request(url);
414     request.setAllowCookies(true);
415
416     NetworkingContext* context = 0;
417     FrameLoader* loader = priv->frame ? priv->frame->loader() : 0;
418     if (loader) {
419         loader->addExtraFieldsToSubresourceRequest(request);
420         context = loader->networkingContext();
421     }
422
423     if (priv->player)
424         request.setHTTPReferrer(priv->player->referrer());
425
426     // Let Apple web servers know we want to access their nice movie trailers.
427     if (!g_ascii_strcasecmp("movies.apple.com", url.host().utf8().data())
428         || !g_ascii_strcasecmp("trailers.apple.com", url.host().utf8().data()))
429         request.setHTTPUserAgent("Quicktime/7.6.6");
430
431     if (priv->requestedOffset) {
432         GOwnPtr<gchar> val;
433
434         val.set(g_strdup_printf("bytes=%" G_GUINT64_FORMAT "-", priv->requestedOffset));
435         request.setHTTPHeaderField("Range", val.get());
436     }
437
438     if (priv->iradioMode)
439         request.setHTTPHeaderField("icy-metadata", "1");
440
441     // Needed to use DLNA streaming servers
442     request.setHTTPHeaderField("transferMode.dlna", "Streaming");
443
444     priv->resourceHandle = ResourceHandle::create(context, request, priv->client, false, false);
445     if (!priv->resourceHandle) {
446         GST_ERROR_OBJECT(src, "Failed to create ResourceHandle");
447         return false;
448     }
449
450     GST_DEBUG_OBJECT(src, "Started request");
451
452     return true;
453 }
454
455 static GstStateChangeReturn webKitWebSrcChangeState(GstElement* element, GstStateChange transition)
456 {
457     GstStateChangeReturn ret = GST_STATE_CHANGE_SUCCESS;
458     WebKitWebSrc* src = WEBKIT_WEB_SRC(element);
459     WebKitWebSrcPrivate* priv = src->priv;
460
461     switch (transition) {
462     case GST_STATE_CHANGE_NULL_TO_READY:
463         if (!priv->appsrc) {
464             gst_element_post_message(element,
465                                      gst_missing_element_message_new(element, "appsrc"));
466             GST_ELEMENT_ERROR(src, CORE, MISSING_PLUGIN, (0), ("no appsrc"));
467             return GST_STATE_CHANGE_FAILURE;
468         }
469         break;
470     default:
471         break;
472     }
473
474     ret = GST_ELEMENT_CLASS(parent_class)->change_state(element, transition);
475     if (G_UNLIKELY(ret == GST_STATE_CHANGE_FAILURE)) {
476         GST_DEBUG_OBJECT(src, "State change failed");
477         return ret;
478     }
479
480     switch (transition) {
481     case GST_STATE_CHANGE_READY_TO_PAUSED:
482         GST_DEBUG_OBJECT(src, "READY->PAUSED");
483         if (!webKitWebSrcStart(src))
484             ret = GST_STATE_CHANGE_FAILURE;
485         break;
486     case GST_STATE_CHANGE_PAUSED_TO_READY:
487         GST_DEBUG_OBJECT(src, "PAUSED->READY");
488         webKitWebSrcStop(src, false);
489         break;
490     default:
491         break;
492     }
493
494     return ret;
495 }
496
497 static gboolean webKitWebSrcQueryWithParent(GstPad* pad, GstObject* parent, GstQuery* query)
498 {
499     WebKitWebSrc* webkitSrc = WEBKIT_WEB_SRC(GST_ELEMENT(parent));
500     gboolean result = FALSE;
501
502     switch (GST_QUERY_TYPE(query)) {
503     case GST_QUERY_DURATION: {
504         GstFormat format;
505
506         gst_query_parse_duration(query, &format, NULL);
507
508         GST_DEBUG_OBJECT(webkitSrc, "duration query in format %s", gst_format_get_name(format));
509         if ((format == GST_FORMAT_BYTES) && (webkitSrc->priv->size > 0)) {
510             gst_query_set_duration(query, format, webkitSrc->priv->size);
511             result = TRUE;
512         }
513         break;
514     }
515     case GST_QUERY_URI: {
516         gst_query_set_uri(query, webkitSrc->priv->uri);
517         result = TRUE;
518         break;
519     }
520     default: {
521         GRefPtr<GstPad> target = adoptGRef(gst_ghost_pad_get_target(GST_GHOST_PAD_CAST(pad)));
522
523         // Forward the query to the proxy target pad.
524         if (target)
525             result = gst_pad_query(target.get(), query);
526         break;
527     }
528     }
529
530     return result;
531 }
532
533 #ifndef GST_API_VERSION_1
534 static gboolean webKitWebSrcQuery(GstPad* pad, GstQuery* query)
535 {
536     GRefPtr<GstElement> src = adoptGRef(gst_pad_get_parent_element(pad));
537     return webKitWebSrcQueryWithParent(pad, GST_OBJECT(src.get()), query);
538 }
539 #endif
540
541 // uri handler interface
542
543 #ifdef GST_API_VERSION_1
544 static GstURIType webKitWebSrcUriGetType(GType)
545 {
546     return GST_URI_SRC;
547 }
548
549 const gchar* const* webKitWebSrcGetProtocols(GType)
550 {
551     static const char* protocols[] = {"http", "https", 0 };
552     return protocols;
553 }
554
555 static gchar* webKitWebSrcGetUri(GstURIHandler* handler)
556 {
557     return g_strdup(WEBKIT_WEB_SRC(handler)->priv->uri);
558 }
559
560 static gboolean webKitWebSrcSetUri(GstURIHandler* handler, const gchar* uri, GError** error)
561 {
562     WebKitWebSrc* src = WEBKIT_WEB_SRC(handler);
563     WebKitWebSrcPrivate* priv = src->priv;
564
565     if (GST_STATE(src) >= GST_STATE_PAUSED) {
566         GST_ERROR_OBJECT(src, "URI can only be set in states < PAUSED");
567         return FALSE;
568     }
569
570     g_free(priv->uri);
571     priv->uri = 0;
572
573     if (!uri)
574         return TRUE;
575
576     KURL url(KURL(), uri);
577
578     if (!url.isValid() || !url.protocolIsInHTTPFamily()) {
579         g_set_error(error, GST_URI_ERROR, GST_URI_ERROR_BAD_URI, "Invalid URI '%s'", uri);
580         return FALSE;
581     }
582
583     priv->uri = g_strdup(url.string().utf8().data());
584     return TRUE;
585 }
586
587 #else
588 static GstURIType webKitWebSrcUriGetType(void)
589 {
590     return GST_URI_SRC;
591 }
592
593 static gchar** webKitWebSrcGetProtocols(void)
594 {
595     static gchar* protocols[] = {(gchar*) "http", (gchar*) "https", 0 };
596     return protocols;
597 }
598
599 static const gchar* webKitWebSrcGetUri(GstURIHandler* handler)
600 {
601     return g_strdup(WEBKIT_WEB_SRC(handler)->priv->uri);
602 }
603
604 static gboolean webKitWebSrcSetUri(GstURIHandler* handler, const gchar* uri)
605 {
606     WebKitWebSrc* src = WEBKIT_WEB_SRC(handler);
607     WebKitWebSrcPrivate* priv = src->priv;
608
609     if (GST_STATE(src) >= GST_STATE_PAUSED) {
610         GST_ERROR_OBJECT(src, "URI can only be set in states < PAUSED");
611         return FALSE;
612     }
613
614     g_free(priv->uri);
615     priv->uri = 0;
616
617     if (!uri)
618         return TRUE;
619
620     KURL url(KURL(), uri);
621
622     if (!url.isValid() || !url.protocolIsInHTTPFamily()) {
623         GST_ERROR_OBJECT(src, "Invalid URI '%s'", uri);
624         return FALSE;
625     }
626
627     priv->uri = g_strdup(url.string().utf8().data());
628     return TRUE;
629 }
630 #endif
631
632 static void webKitWebSrcUriHandlerInit(gpointer gIface, gpointer)
633 {
634     GstURIHandlerInterface* iface = (GstURIHandlerInterface *) gIface;
635
636     iface->get_type = webKitWebSrcUriGetType;
637     iface->get_protocols = webKitWebSrcGetProtocols;
638     iface->get_uri = webKitWebSrcGetUri;
639     iface->set_uri = webKitWebSrcSetUri;
640 }
641
642 // appsrc callbacks
643
644 static gboolean webKitWebSrcNeedDataMainCb(WebKitWebSrc* src)
645 {
646     WebKitWebSrcPrivate* priv = src->priv;
647
648     priv->resourceHandle->setDefersLoading(false);
649
650     GST_OBJECT_LOCK(src);
651     priv->paused = FALSE;
652     priv->needDataID = 0;
653     GST_OBJECT_UNLOCK(src);
654     return FALSE;
655 }
656
657 static void webKitWebSrcNeedDataCb(GstAppSrc*, guint length, gpointer userData)
658 {
659     WebKitWebSrc* src = WEBKIT_WEB_SRC(userData);
660     WebKitWebSrcPrivate* priv = src->priv;
661
662     GST_DEBUG_OBJECT(src, "Need more data: %u", length);
663
664     GST_OBJECT_LOCK(src);
665     if (priv->needDataID || !priv->paused) {
666         GST_OBJECT_UNLOCK(src);
667         return;
668     }
669
670     priv->needDataID = g_timeout_add_full(G_PRIORITY_DEFAULT, 0, (GSourceFunc) webKitWebSrcNeedDataMainCb, gst_object_ref(src), (GDestroyNotify) gst_object_unref);
671     GST_OBJECT_UNLOCK(src);
672 }
673
674 static gboolean webKitWebSrcEnoughDataMainCb(WebKitWebSrc* src)
675 {
676     WebKitWebSrcPrivate* priv = src->priv;
677
678     priv->resourceHandle->setDefersLoading(true);
679
680     GST_OBJECT_LOCK(src);
681     priv->paused = TRUE;
682     priv->enoughDataID = 0;
683     GST_OBJECT_UNLOCK(src);
684
685     return FALSE;
686 }
687
688 static void webKitWebSrcEnoughDataCb(GstAppSrc*, gpointer userData)
689 {
690     WebKitWebSrc* src = WEBKIT_WEB_SRC(userData);
691     WebKitWebSrcPrivate* priv = src->priv;
692
693     GST_DEBUG_OBJECT(src, "Have enough data");
694
695     GST_OBJECT_LOCK(src);
696     if (priv->enoughDataID || priv->paused) {
697         GST_OBJECT_UNLOCK(src);
698         return;
699     }
700
701     priv->enoughDataID = g_timeout_add_full(G_PRIORITY_DEFAULT, 0, (GSourceFunc) webKitWebSrcEnoughDataMainCb, gst_object_ref(src), (GDestroyNotify) gst_object_unref);
702     GST_OBJECT_UNLOCK(src);
703 }
704
705 static gboolean webKitWebSrcSeekMainCb(WebKitWebSrc* src)
706 {
707     webKitWebSrcStop(src, true);
708     webKitWebSrcStart(src);
709
710     return FALSE;
711 }
712
713 static gboolean webKitWebSrcSeekDataCb(GstAppSrc*, guint64 offset, gpointer userData)
714 {
715     WebKitWebSrc* src = WEBKIT_WEB_SRC(userData);
716     WebKitWebSrcPrivate* priv = src->priv;
717
718     GST_DEBUG_OBJECT(src, "Seeking to offset: %" G_GUINT64_FORMAT, offset);
719     if (offset == priv->offset)
720         return TRUE;
721
722     if (!priv->seekable)
723         return FALSE;
724     if (offset > priv->size)
725         return FALSE;
726
727     GST_DEBUG_OBJECT(src, "Doing range-request seek");
728     priv->requestedOffset = offset;
729
730     GST_OBJECT_LOCK(src);
731     if (priv->seekID)
732         g_source_remove(priv->seekID);
733     priv->seekID = g_timeout_add_full(G_PRIORITY_DEFAULT, 0, (GSourceFunc) webKitWebSrcSeekMainCb, gst_object_ref(src), (GDestroyNotify) gst_object_unref);
734     GST_OBJECT_UNLOCK(src);
735     
736     return TRUE;
737 }
738
739 void webKitWebSrcSetMediaPlayer(WebKitWebSrc* src, WebCore::MediaPlayer* player)
740 {
741     WebKitWebSrcPrivate* priv = src->priv;
742     WebCore::Frame* frame = 0;
743
744     WebCore::Document* document = player->mediaPlayerClient()->mediaPlayerOwningDocument();
745     if (document)
746         frame = document->frame();
747
748     priv->frame = frame;
749     priv->player = player;
750 }
751
752 StreamingClient::StreamingClient(WebKitWebSrc* src) : m_src(src)
753 {
754
755 }
756
757 StreamingClient::~StreamingClient()
758 {
759
760 }
761
762 void StreamingClient::willSendRequest(ResourceHandle*, ResourceRequest&, const ResourceResponse&)
763 {
764 }
765
766 void StreamingClient::didReceiveResponse(ResourceHandle*, const ResourceResponse& response)
767 {
768     WebKitWebSrcPrivate* priv = m_src->priv;
769
770     GST_DEBUG_OBJECT(m_src, "Received response: %d", response.httpStatusCode());
771
772     // If we seeked we need 206 == PARTIAL_CONTENT
773     if (priv->requestedOffset && response.httpStatusCode() != 206) {
774         GST_ELEMENT_ERROR(m_src, RESOURCE, READ, (0), (0));
775         gst_app_src_end_of_stream(priv->appsrc);
776         webKitWebSrcStop(m_src, false);
777         return;
778     }
779
780     long long length = response.expectedContentLength();
781     if (length > 0) {
782         length += priv->requestedOffset;
783         gst_app_src_set_size(priv->appsrc, length);
784
785 #ifndef GST_API_VERSION_1
786         if (!priv->haveAppSrc27) {
787             gst_segment_set_duration(&GST_BASE_SRC(priv->appsrc)->segment, GST_FORMAT_BYTES, length);
788             gst_element_post_message(GST_ELEMENT(priv->appsrc),
789                                      gst_message_new_duration(GST_OBJECT(priv->appsrc),
790                                                               GST_FORMAT_BYTES, length));
791         }
792 #endif
793     }
794
795     priv->size = length >= 0 ? length : 0;
796     priv->seekable = length > 0 && g_ascii_strcasecmp("none", response.httpHeaderField("Accept-Ranges").utf8().data());
797
798     // icecast stuff
799     String value = response.httpHeaderField("icy-metaint");
800     if (!value.isEmpty()) {
801         gchar* endptr = 0;
802         gint64 icyMetaInt = g_ascii_strtoll(value.utf8().data(), &endptr, 10);
803             
804         if (endptr && *endptr == '\0' && icyMetaInt > 0) {
805             GRefPtr<GstCaps> caps = adoptGRef(gst_caps_new_simple("application/x-icy", "metadata-interval", G_TYPE_INT, (gint) icyMetaInt, NULL));
806
807             gst_app_src_set_caps(priv->appsrc, caps.get());
808         }
809     }
810
811 #ifdef GST_API_VERSION_1
812     GstTagList* tags = gst_tag_list_new_empty();
813 #else
814     GstTagList* tags = gst_tag_list_new();
815 #endif
816     value = response.httpHeaderField("icy-name");
817     if (!value.isEmpty()) {
818         g_free(priv->iradioName);
819         priv->iradioName = g_strdup(value.utf8().data());
820         g_object_notify(G_OBJECT(m_src), "iradio-name");
821         gst_tag_list_add(tags, GST_TAG_MERGE_REPLACE, GST_TAG_ORGANIZATION, priv->iradioName, NULL);
822     }
823     value = response.httpHeaderField("icy-genre");
824     if (!value.isEmpty()) {
825         g_free(priv->iradioGenre);
826         priv->iradioGenre = g_strdup(value.utf8().data());
827         g_object_notify(G_OBJECT(m_src), "iradio-genre");
828         gst_tag_list_add(tags, GST_TAG_MERGE_REPLACE, GST_TAG_GENRE, priv->iradioGenre, NULL);
829     }
830     value = response.httpHeaderField("icy-url");
831     if (!value.isEmpty()) {
832         g_free(priv->iradioUrl);
833         priv->iradioUrl = g_strdup(value.utf8().data());
834         g_object_notify(G_OBJECT(m_src), "iradio-url");
835         gst_tag_list_add(tags, GST_TAG_MERGE_REPLACE, GST_TAG_LOCATION, priv->iradioUrl, NULL);
836     }
837     value = response.httpHeaderField("icy-title");
838     if (!value.isEmpty()) {
839         g_free(priv->iradioTitle);
840         priv->iradioTitle = g_strdup(value.utf8().data());
841         g_object_notify(G_OBJECT(m_src), "iradio-title");
842         gst_tag_list_add(tags, GST_TAG_MERGE_REPLACE, GST_TAG_TITLE, priv->iradioTitle, NULL);
843     }
844
845     if (gst_tag_list_is_empty(tags))
846 #ifdef GST_API_VERSION_1
847         gst_tag_list_unref(tags);
848 #else
849         gst_tag_list_free(tags);
850 #endif
851     else
852         notifyGstTagsOnPad(GST_ELEMENT(m_src), m_src->priv->srcpad, tags);
853 }
854
855 void StreamingClient::didReceiveData(ResourceHandle* handle, const char* data, int length, int)
856 {
857     WebKitWebSrcPrivate* priv = m_src->priv;
858
859     GST_LOG_OBJECT(m_src, "Have %d bytes of data", length);
860
861     if (priv->seekID || handle != priv->resourceHandle) {
862         GST_DEBUG_OBJECT(m_src, "Seek in progress, ignoring data");
863         return;
864     }
865
866     GstBuffer* buffer = gst_buffer_new_and_alloc(length);
867
868 #ifdef GST_API_VERSION_1
869     gst_buffer_fill(buffer, 0, data, length);
870 #else
871     memcpy(GST_BUFFER_DATA(buffer), data, length);
872 #endif
873     GST_BUFFER_OFFSET(buffer) = priv->offset;
874     priv->offset += length;
875     GST_BUFFER_OFFSET_END(buffer) = priv->offset;
876
877     GstFlowReturn ret = gst_app_src_push_buffer(priv->appsrc, buffer);
878 #ifdef GST_API_VERSION_1
879     if (ret != GST_FLOW_OK && ret != GST_FLOW_EOS)
880 #else
881     if (ret != GST_FLOW_OK && ret != GST_FLOW_UNEXPECTED)
882 #endif
883         GST_ELEMENT_ERROR(m_src, CORE, FAILED, (0), (0));
884 }
885
886 void StreamingClient::didFinishLoading(ResourceHandle*, double)
887 {
888     WebKitWebSrcPrivate* priv = m_src->priv;
889
890     GST_DEBUG_OBJECT(m_src, "Have EOS");
891
892     if (!priv->seekID)
893         gst_app_src_end_of_stream(m_src->priv->appsrc);
894 }
895
896 void StreamingClient::didFail(ResourceHandle*, const ResourceError& error)
897 {
898     GST_ERROR_OBJECT(m_src, "Have failure: %s", error.localizedDescription().utf8().data());
899     GST_ELEMENT_ERROR(m_src, RESOURCE, FAILED, ("%s", error.localizedDescription().utf8().data()), (0));
900     gst_app_src_end_of_stream(m_src->priv->appsrc);
901 }
902
903 void StreamingClient::wasBlocked(ResourceHandle*)
904 {
905     GST_ERROR_OBJECT(m_src, "Request was blocked");
906     GST_ELEMENT_ERROR(m_src, RESOURCE, OPEN_READ, ("Access to \"%s\" was blocked", m_src->priv->uri), (0));
907 }
908
909 void StreamingClient::cannotShowURL(ResourceHandle*)
910 {
911     GST_ERROR_OBJECT(m_src, "Cannot show URL");
912     GST_ELEMENT_ERROR(m_src, RESOURCE, OPEN_READ, ("Can't show \"%s\"", m_src->priv->uri), (0));
913 }
914
915 #endif // USE(GSTREAMER)
916