ext/soup/gstsouphttpsrc.c

   1 /* GStreamer
   2  * Copyright (C) 2007-2008 Wouter Cloetens <wouter@mind.be>
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Library 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  * Library General Public License for more
  13  */
  14
  15 /**
  16  * SECTION:element-souphttpsrc
  17  *
  18  * This plugin reads data from a remote location specified by a URI.
  19  * Supported protocols are 'http', 'https'.
  20  *
  21  * An HTTP proxy must be specified by its URL.
  22  * If the "http_proxy" environment variable is set, its value is used.
  23  * If built with libsoup's GNOME integration features, the GNOME proxy
  24  * configuration will be used, or failing that, proxy autodetection.
  25  * The #GstSoupHTTPSrc:proxy property can be used to override the default.
  26  *
  27  * In case the #GstSoupHTTPSrc:iradio-mode property is set and the location is
  28  * an HTTP resource, souphttpsrc will send special Icecast HTTP headers to the
  29  * server to request additional Icecast meta-information.
  30  * If the server is not an Icecast server, it will behave as if the
  31  * #GstSoupHTTPSrc:iradio-mode property were not set. If it is, souphttpsrc will
  32  * output data with a media type of application/x-icy, in which case you will
  33  * need to use the #ICYDemux element as follow-up element to extract the Icecast
  34  * metadata and to determine the underlying media type.
  35  *
  36  * <refsect2>
  37  * <title>Example launch line</title>
  38  * |[
  39  * gst-launch-1.0 -v souphttpsrc location=https://some.server.org/index.html
  40  *     ! filesink location=/home/joe/server.html
  41  * ]| The above pipeline reads a web page from a server using the HTTPS protocol
  42  * and writes it to a local file.
  43  * |[
  44  * gst-launch-1.0 -v souphttpsrc user-agent="FooPlayer 0.99 beta"
  45  *     automatic-redirect=false proxy=http://proxy.intranet.local:8080
  46  *     location=http://music.foobar.com/demo.mp3 ! mad ! audioconvert
  47  *     ! audioresample ! alsasink
  48  * ]| The above pipeline will read and decode and play an mp3 file from a
  49  * web server using the HTTP protocol. If the server sends redirects,
  50  * the request fails instead of following the redirect. The specified
  51  * HTTP proxy server is used. The User-Agent HTTP request header
  52  * is set to a custom string instead of "GStreamer souphttpsrc."
  53  * |[
  54  * gst-launch-1.0 -v souphttpsrc location=http://10.11.12.13/mjpeg
  55  *     do-timestamp=true ! multipartdemux
  56  *     ! image/jpeg,width=640,height=480 ! matroskamux
  57  *     ! filesink location=mjpeg.mkv
  58  * ]| The above pipeline reads a motion JPEG stream from an IP camera
  59  * using the HTTP protocol, encoded as mime/multipart image/jpeg
  60  * parts, and writes a Matroska motion JPEG file. The width and
  61  * height properties are set in the caps to provide the Matroska
  62  * multiplexer with the information to set this in the header.
  63  * Timestamps are set on the buffers as they arrive from the camera.
  64  * These are used by the mime/multipart demultiplexer to emit timestamps
  65  * on the JPEG-encoded video frame buffers. This allows the Matroska
  66  * multiplexer to timestamp the frames in the resulting file.
  67  * </refsect2>
  68  */
  69
  70 #ifdef HAVE_CONFIG_H
  71 #include "config.h"
  72 #endif
  73
  74 #include <string.h>
  75 #ifdef HAVE_STDLIB_H
  76 #include <stdlib.h>             /* atoi() */
  77 #endif
  78 #include <gst/gstelement.h>
  79 #include <gst/gst-i18n-plugin.h>
  80 #include <libsoup/soup.h>
  81 #include "gstsouphttpsrc.h"
  82
  83 #include <gst/tag/tag.h>
  84
  85 GST_DEBUG_CATEGORY_STATIC (souphttpsrc_debug);
  86 #define GST_CAT_DEFAULT souphttpsrc_debug
  87
  88 static GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src",
  89     GST_PAD_SRC,
  90     GST_PAD_ALWAYS,
  91     GST_STATIC_CAPS_ANY);
  92
  93 enum
  94 {
  95   PROP_0,
  96   PROP_LOCATION,
  97   PROP_IS_LIVE,
  98   PROP_USER_AGENT,
  99   PROP_AUTOMATIC_REDIRECT,
 100   PROP_PROXY,
 101   PROP_USER_ID,
 102   PROP_USER_PW,
 103   PROP_PROXY_ID,
 104   PROP_PROXY_PW,
 105   PROP_COOKIES,
 106   PROP_IRADIO_MODE,
 107   PROP_TIMEOUT,
 108   PROP_EXTRA_HEADERS
 109 };
 110
 111 #define DEFAULT_USER_AGENT           "GStreamer souphttpsrc "
 112 #define DEFAULT_IRADIO_MODE          TRUE
 113
 114 static void gst_soup_http_src_uri_handler_init (gpointer g_iface,
 115     gpointer iface_data);
 116 static void gst_soup_http_src_finalize (GObject * gobject);
 117
 118 static void gst_soup_http_src_set_property (GObject * object, guint prop_id,
 119     const GValue * value, GParamSpec * pspec);
 120 static void gst_soup_http_src_get_property (GObject * object, guint prop_id,
 121     GValue * value, GParamSpec * pspec);
 122
 123 static GstFlowReturn gst_soup_http_src_create (GstPushSrc * psrc,
 124     GstBuffer ** outbuf);
 125 static gboolean gst_soup_http_src_start (GstBaseSrc * bsrc);
 126 static gboolean gst_soup_http_src_stop (GstBaseSrc * bsrc);
 127 static gboolean gst_soup_http_src_get_size (GstBaseSrc * bsrc, guint64 * size);
 128 static gboolean gst_soup_http_src_is_seekable (GstBaseSrc * bsrc);
 129 static gboolean gst_soup_http_src_do_seek (GstBaseSrc * bsrc,
 130     GstSegment * segment);
 131 static gboolean gst_soup_http_src_query (GstBaseSrc * bsrc, GstQuery * query);
 132 static gboolean gst_soup_http_src_unlock (GstBaseSrc * bsrc);
 133 static gboolean gst_soup_http_src_unlock_stop (GstBaseSrc * bsrc);
 134 static gboolean gst_soup_http_src_set_location (GstSoupHTTPSrc * src,
 135     const gchar * uri, GError ** error);
 136 static gboolean gst_soup_http_src_set_proxy (GstSoupHTTPSrc * src,
 137     const gchar * uri);
 138 static char *gst_soup_http_src_unicodify (const char *str);
 139 static gboolean gst_soup_http_src_build_message (GstSoupHTTPSrc * src);
 140 static void gst_soup_http_src_cancel_message (GstSoupHTTPSrc * src);
 141 static void gst_soup_http_src_queue_message (GstSoupHTTPSrc * src);
 142 static gboolean gst_soup_http_src_add_range_header (GstSoupHTTPSrc * src,
 143     guint64 offset);
 144 static void gst_soup_http_src_session_unpause_message (GstSoupHTTPSrc * src);
 145 static void gst_soup_http_src_session_pause_message (GstSoupHTTPSrc * src);
 146 static void gst_soup_http_src_session_close (GstSoupHTTPSrc * src);
 147 static void gst_soup_http_src_parse_status (SoupMessage * msg,
 148     GstSoupHTTPSrc * src);
 149 static void gst_soup_http_src_chunk_free (gpointer gstbuf);
 150 static SoupBuffer *gst_soup_http_src_chunk_allocator (SoupMessage * msg,
 151     gsize max_len, gpointer user_data);
 152 static void gst_soup_http_src_got_chunk_cb (SoupMessage * msg,
 153     SoupBuffer * chunk, GstSoupHTTPSrc * src);
 154 static void gst_soup_http_src_response_cb (SoupSession * session,
 155     SoupMessage * msg, GstSoupHTTPSrc * src);
 156 static void gst_soup_http_src_got_headers_cb (SoupMessage * msg,
 157     GstSoupHTTPSrc * src);
 158 static void gst_soup_http_src_got_body_cb (SoupMessage * msg,
 159     GstSoupHTTPSrc * src);
 160 static void gst_soup_http_src_finished_cb (SoupMessage * msg,
 161     GstSoupHTTPSrc * src);
 162 static void gst_soup_http_src_authenticate_cb (SoupSession * session,
 163     SoupMessage * msg, SoupAuth * auth, gboolean retrying,
 164     GstSoupHTTPSrc * src);
 165
 166 #define gst_soup_http_src_parent_class parent_class
 167 G_DEFINE_TYPE_WITH_CODE (GstSoupHTTPSrc, gst_soup_http_src, GST_TYPE_PUSH_SRC,
 168     G_IMPLEMENT_INTERFACE (GST_TYPE_URI_HANDLER,
 169         gst_soup_http_src_uri_handler_init));
 170
 171 static void
 172 gst_soup_http_src_class_init (GstSoupHTTPSrcClass * klass)
 173 {
 174   GObjectClass *gobject_class;
 175   GstElementClass *gstelement_class;
 176   GstBaseSrcClass *gstbasesrc_class;
 177   GstPushSrcClass *gstpushsrc_class;
 178
 179   gobject_class = (GObjectClass *) klass;
 180   gstelement_class = (GstElementClass *) klass;
 181   gstbasesrc_class = (GstBaseSrcClass *) klass;
 182   gstpushsrc_class = (GstPushSrcClass *) klass;
 183
 184   gobject_class->set_property = gst_soup_http_src_set_property;
 185   gobject_class->get_property = gst_soup_http_src_get_property;
 186   gobject_class->finalize = gst_soup_http_src_finalize;
 187
 188   g_object_class_install_property (gobject_class,
 189       PROP_LOCATION,
 190       g_param_spec_string ("location", "Location",
 191           "Location to read from", "",
 192           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 193   g_object_class_install_property (gobject_class,
 194       PROP_USER_AGENT,
 195       g_param_spec_string ("user-agent", "User-Agent",
 196           "Value of the User-Agent HTTP request header field",
 197           DEFAULT_USER_AGENT, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 198   g_object_class_install_property (gobject_class,
 199       PROP_AUTOMATIC_REDIRECT,
 200       g_param_spec_boolean ("automatic-redirect", "automatic-redirect",
 201           "Automatically follow HTTP redirects (HTTP Status Code 3xx)",
 202           TRUE, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 203   g_object_class_install_property (gobject_class,
 204       PROP_PROXY,
 205       g_param_spec_string ("proxy", "Proxy",
 206           "HTTP proxy server URI", "",
 207           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 208   g_object_class_install_property (gobject_class,
 209       PROP_USER_ID,
 210       g_param_spec_string ("user-id", "user-id",
 211           "HTTP location URI user id for authentication", "",
 212           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 213   g_object_class_install_property (gobject_class, PROP_USER_PW,
 214       g_param_spec_string ("user-pw", "user-pw",
 215           "HTTP location URI user password for authentication", "",
 216           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 217   g_object_class_install_property (gobject_class, PROP_PROXY_ID,
 218       g_param_spec_string ("proxy-id", "proxy-id",
 219           "HTTP proxy URI user id for authentication", "",
 220           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 221   g_object_class_install_property (gobject_class, PROP_PROXY_PW,
 222       g_param_spec_string ("proxy-pw", "proxy-pw",
 223           "HTTP proxy URI user password for authentication", "",
 224           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 225   g_object_class_install_property (gobject_class, PROP_COOKIES,
 226       g_param_spec_boxed ("cookies", "Cookies", "HTTP request cookies",
 227           G_TYPE_STRV, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 228   g_object_class_install_property (gobject_class, PROP_IS_LIVE,
 229       g_param_spec_boolean ("is-live", "is-live", "Act like a live source",
 230           FALSE, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 231   g_object_class_install_property (gobject_class, PROP_TIMEOUT,
 232       g_param_spec_uint ("timeout", "timeout",
 233           "Value in seconds to timeout a blocking I/O (0 = No timeout).", 0,
 234           3600, 0, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 235   g_object_class_install_property (gobject_class, PROP_EXTRA_HEADERS,
 236       g_param_spec_boxed ("extra-headers", "Extra Headers",
 237           "Extra headers to append to the HTTP request",
 238           GST_TYPE_STRUCTURE, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 239   g_object_class_install_property (gobject_class, PROP_IRADIO_MODE,
 240       g_param_spec_boolean ("iradio-mode", "iradio-mode",
 241           "Enable internet radio mode (ask server to send shoutcast/icecast "
 242           "metadata interleaved with the actual stream data)",
 243           DEFAULT_IRADIO_MODE, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 244
 245   gst_element_class_add_pad_template (gstelement_class,
 246       gst_static_pad_template_get (&srctemplate));
 247
 248   gst_element_class_set_static_metadata (gstelement_class, "HTTP client source",
 249       "Source/Network",
 250       "Receive data as a client over the network via HTTP using SOUP",
 251       "Wouter Cloetens <wouter@mind.be>");
 252
 253   gstbasesrc_class->start = GST_DEBUG_FUNCPTR (gst_soup_http_src_start);
 254   gstbasesrc_class->stop = GST_DEBUG_FUNCPTR (gst_soup_http_src_stop);
 255   gstbasesrc_class->unlock = GST_DEBUG_FUNCPTR (gst_soup_http_src_unlock);
 256   gstbasesrc_class->unlock_stop =
 257       GST_DEBUG_FUNCPTR (gst_soup_http_src_unlock_stop);
 258   gstbasesrc_class->get_size = GST_DEBUG_FUNCPTR (gst_soup_http_src_get_size);
 259   gstbasesrc_class->is_seekable =
 260       GST_DEBUG_FUNCPTR (gst_soup_http_src_is_seekable);
 261   gstbasesrc_class->do_seek = GST_DEBUG_FUNCPTR (gst_soup_http_src_do_seek);
 262   gstbasesrc_class->query = GST_DEBUG_FUNCPTR (gst_soup_http_src_query);
 263
 264   gstpushsrc_class->create = GST_DEBUG_FUNCPTR (gst_soup_http_src_create);
 265
 266   GST_DEBUG_CATEGORY_INIT (souphttpsrc_debug, "souphttpsrc", 0,
 267       "SOUP HTTP src");
 268 }
 269
 270 static void
 271 gst_soup_http_src_reset (GstSoupHTTPSrc * src)
 272 {
 273   src->interrupted = FALSE;
 274   src->retry = FALSE;
 275   src->have_size = FALSE;
 276   src->seekable = FALSE;
 277   src->read_position = 0;
 278   src->request_position = 0;
 279   src->content_size = 0;
 280
 281   gst_caps_replace (&src->src_caps, NULL);
 282   g_free (src->iradio_name);
 283   src->iradio_name = NULL;
 284   g_free (src->iradio_genre);
 285   src->iradio_genre = NULL;
 286   g_free (src->iradio_url);
 287   src->iradio_url = NULL;
 288 }
 289
 290 static void
 291 gst_soup_http_src_init (GstSoupHTTPSrc * src)
 292 {
 293   const gchar *proxy;
 294
 295   src->location = NULL;
 296   src->automatic_redirect = TRUE;
 297   src->user_agent = g_strdup (DEFAULT_USER_AGENT);
 298   src->user_id = NULL;
 299   src->user_pw = NULL;
 300   src->proxy_id = NULL;
 301   src->proxy_pw = NULL;
 302   src->cookies = NULL;
 303   src->iradio_mode = DEFAULT_IRADIO_MODE;
 304   src->loop = NULL;
 305   src->context = NULL;
 306   src->session = NULL;
 307   src->msg = NULL;
 308   proxy = g_getenv ("http_proxy");
 309   if (proxy && !gst_soup_http_src_set_proxy (src, proxy)) {
 310     GST_WARNING_OBJECT (src,
 311         "The proxy in the http_proxy env var (\"%s\") cannot be parsed.",
 312         proxy);
 313   }
 314
 315   gst_soup_http_src_reset (src);
 316 }
 317
 318 static void
 319 gst_soup_http_src_finalize (GObject * gobject)
 320 {
 321   GstSoupHTTPSrc *src = GST_SOUP_HTTP_SRC (gobject);
 322
 323   GST_DEBUG_OBJECT (src, "finalize");
 324
 325   g_free (src->location);
 326   g_free (src->user_agent);
 327   if (src->proxy != NULL) {
 328     soup_uri_free (src->proxy);
 329   }
 330   g_free (src->user_id);
 331   g_free (src->user_pw);
 332   g_free (src->proxy_id);
 333   g_free (src->proxy_pw);
 334   g_strfreev (src->cookies);
 335
 336   G_OBJECT_CLASS (parent_class)->finalize (gobject);
 337 }
 338
 339 static void
 340 gst_soup_http_src_set_property (GObject * object, guint prop_id,
 341     const GValue * value, GParamSpec * pspec)
 342 {
 343   GstSoupHTTPSrc *src = GST_SOUP_HTTP_SRC (object);
 344
 345   switch (prop_id) {
 346     case PROP_LOCATION:
 347     {
 348       const gchar *location;
 349
 350       location = g_value_get_string (value);
 351
 352       if (location == NULL) {
 353         GST_WARNING ("location property cannot be NULL");
 354         goto done;
 355       }
 356       if (!gst_soup_http_src_set_location (src, location, NULL)) {
 357         GST_WARNING ("badly formatted location");
 358         goto done;
 359       }
 360       break;
 361     }
 362     case PROP_USER_AGENT:
 363       if (src->user_agent)
 364         g_free (src->user_agent);
 365       src->user_agent = g_value_dup_string (value);
 366       break;
 367     case PROP_IRADIO_MODE:
 368       src->iradio_mode = g_value_get_boolean (value);
 369       break;
 370     case PROP_AUTOMATIC_REDIRECT:
 371       src->automatic_redirect = g_value_get_boolean (value);
 372       break;
 373     case PROP_PROXY:
 374     {
 375       const gchar *proxy;
 376
 377       proxy = g_value_get_string (value);
 378
 379       if (proxy == NULL) {
 380         GST_WARNING ("proxy property cannot be NULL");
 381         goto done;
 382       }
 383       if (!gst_soup_http_src_set_proxy (src, proxy)) {
 384         GST_WARNING ("badly formatted proxy URI");
 385         goto done;
 386       }
 387       break;
 388     }
 389     case PROP_COOKIES:
 390       g_strfreev (src->cookies);
 391       src->cookies = g_strdupv (g_value_get_boxed (value));
 392       break;
 393     case PROP_IS_LIVE:
 394       gst_base_src_set_live (GST_BASE_SRC (src), g_value_get_boolean (value));
 395       break;
 396     case PROP_USER_ID:
 397       if (src->user_id)
 398         g_free (src->user_id);
 399       src->user_id = g_value_dup_string (value);
 400       break;
 401     case PROP_USER_PW:
 402       if (src->user_pw)
 403         g_free (src->user_pw);
 404       src->user_pw = g_value_dup_string (value);
 405       break;
 406     case PROP_PROXY_ID:
 407       if (src->proxy_id)
 408         g_free (src->proxy_id);
 409       src->proxy_id = g_value_dup_string (value);
 410       break;
 411     case PROP_PROXY_PW:
 412       if (src->proxy_pw)
 413         g_free (src->proxy_pw);
 414       src->proxy_pw = g_value_dup_string (value);
 415       break;
 416     case PROP_TIMEOUT:
 417       src->timeout = g_value_get_uint (value);
 418       break;
 419     case PROP_EXTRA_HEADERS:{
 420       const GstStructure *s = gst_value_get_structure (value);
 421
 422       if (src->extra_headers)
 423         gst_structure_free (src->extra_headers);
 424
 425       src->extra_headers = s ? gst_structure_copy (s) : NULL;
 426       break;
 427     }
 428     default:
 429       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 430       break;
 431   }
 432 done:
 433   return;
 434 }
 435
 436 static void
 437 gst_soup_http_src_get_property (GObject * object, guint prop_id,
 438     GValue * value, GParamSpec * pspec)
 439 {
 440   GstSoupHTTPSrc *src = GST_SOUP_HTTP_SRC (object);
 441
 442   switch (prop_id) {
 443     case PROP_LOCATION:
 444       g_value_set_string (value, src->location);
 445       break;
 446     case PROP_USER_AGENT:
 447       g_value_set_string (value, src->user_agent);
 448       break;
 449     case PROP_AUTOMATIC_REDIRECT:
 450       g_value_set_boolean (value, src->automatic_redirect);
 451       break;
 452     case PROP_PROXY:
 453       if (src->proxy == NULL)
 454         g_value_set_static_string (value, "");
 455       else {
 456         char *proxy = soup_uri_to_string (src->proxy, FALSE);
 457
 458         g_value_set_string (value, proxy);
 459         g_free (proxy);
 460       }
 461       break;
 462     case PROP_COOKIES:
 463       g_value_set_boxed (value, g_strdupv (src->cookies));
 464       break;
 465     case PROP_IS_LIVE:
 466       g_value_set_boolean (value, gst_base_src_is_live (GST_BASE_SRC (src)));
 467       break;
 468     case PROP_IRADIO_MODE:
 469       g_value_set_boolean (value, src->iradio_mode);
 470       break;
 471     case PROP_USER_ID:
 472       g_value_set_string (value, src->user_id);
 473       break;
 474     case PROP_USER_PW:
 475       g_value_set_string (value, src->user_pw);
 476       break;
 477     case PROP_PROXY_ID:
 478       g_value_set_string (value, src->proxy_id);
 479       break;
 480     case PROP_PROXY_PW:
 481       g_value_set_string (value, src->proxy_pw);
 482       break;
 483     case PROP_TIMEOUT:
 484       g_value_set_uint (value, src->timeout);
 485       break;
 486     case PROP_EXTRA_HEADERS:
 487       gst_value_set_structure (value, src->extra_headers);
 488       break;
 489     default:
 490       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 491       break;
 492   }
 493 }
 494
 495 static gchar *
 496 gst_soup_http_src_unicodify (const gchar * str)
 497 {
 498   const gchar *env_vars[] = { "GST_ICY_TAG_ENCODING",
 499     "GST_TAG_ENCODING", NULL
 500   };
 501
 502   return gst_tag_freeform_string_to_utf8 (str, -1, env_vars);
 503 }
 504
 505 static void
 506 gst_soup_http_src_cancel_message (GstSoupHTTPSrc * src)
 507 {
 508   if (src->msg != NULL) {
 509     src->session_io_status = GST_SOUP_HTTP_SRC_SESSION_IO_STATUS_CANCELLED;
 510     soup_session_cancel_message (src->session, src->msg, SOUP_STATUS_CANCELLED);
 511   }
 512   src->session_io_status = GST_SOUP_HTTP_SRC_SESSION_IO_STATUS_IDLE;
 513   src->msg = NULL;
 514 }
 515
 516 static void
 517 gst_soup_http_src_queue_message (GstSoupHTTPSrc * src)
 518 {
 519   soup_session_queue_message (src->session, src->msg,
 520       (SoupSessionCallback) gst_soup_http_src_response_cb, src);
 521   src->session_io_status = GST_SOUP_HTTP_SRC_SESSION_IO_STATUS_QUEUED;
 522 }
 523
 524 static gboolean
 525 gst_soup_http_src_add_range_header (GstSoupHTTPSrc * src, guint64 offset)
 526 {
 527   gchar buf[64];
 528
 529   gint rc;
 530
 531   soup_message_headers_remove (src->msg->request_headers, "Range");
 532   if (offset) {
 533     rc = g_snprintf (buf, sizeof (buf), "bytes=%" G_GUINT64_FORMAT "-", offset);
 534     if (rc > sizeof (buf) || rc < 0)
 535       return FALSE;
 536     soup_message_headers_append (src->msg->request_headers, "Range", buf);
 537   }
 538   src->read_position = offset;
 539   return TRUE;
 540 }
 541
 542 static gboolean
 543 _append_extra_header (GQuark field_id, const GValue * value, gpointer user_data)
 544 {
 545   GstSoupHTTPSrc *src = GST_SOUP_HTTP_SRC (user_data);
 546   const gchar *field_name = g_quark_to_string (field_id);
 547   gchar *field_content = NULL;
 548
 549   if (G_VALUE_TYPE (value) == G_TYPE_STRING) {
 550     field_content = g_value_dup_string (value);
 551   } else {
 552     GValue dest = { 0, };
 553
 554     g_value_init (&dest, G_TYPE_STRING);
 555     if (g_value_transform (value, &dest)) {
 556       field_content = g_value_dup_string (&dest);
 557     }
 558   }
 559
 560   if (field_content == NULL) {
 561     GST_ERROR_OBJECT (src, "extra-headers field '%s' contains no value "
 562         "or can't be converted to a string", field_name);
 563     return FALSE;
 564   }
 565
 566   GST_DEBUG_OBJECT (src, "Appending extra header: \"%s: %s\"", field_name,
 567       field_content);
 568   soup_message_headers_append (src->msg->request_headers, field_name,
 569       field_content);
 570
 571   g_free (field_content);
 572
 573   return TRUE;
 574 }
 575
 576 static gboolean
 577 _append_extra_headers (GQuark field_id, const GValue * value,
 578     gpointer user_data)
 579 {
 580   if (G_VALUE_TYPE (value) == GST_TYPE_ARRAY) {
 581     guint n = gst_value_array_get_size (value);
 582     guint i;
 583
 584     for (i = 0; i < n; i++) {
 585       const GValue *v = gst_value_array_get_value (value, i);
 586
 587       if (!_append_extra_header (field_id, v, user_data))
 588         return FALSE;
 589     }
 590   } else if (G_VALUE_TYPE (value) == GST_TYPE_LIST) {
 591     guint n = gst_value_list_get_size (value);
 592     guint i;
 593
 594     for (i = 0; i < n; i++) {
 595       const GValue *v = gst_value_list_get_value (value, i);
 596
 597       if (!_append_extra_header (field_id, v, user_data))
 598         return FALSE;
 599     }
 600   } else {
 601     return _append_extra_header (field_id, value, user_data);
 602   }
 603
 604   return TRUE;
 605 }
 606
 607
 608 static gboolean
 609 gst_soup_http_src_add_extra_headers (GstSoupHTTPSrc * src)
 610 {
 611   if (!src->extra_headers)
 612     return TRUE;
 613
 614   return gst_structure_foreach (src->extra_headers, _append_extra_headers, src);
 615 }
 616
 617
 618 static void
 619 gst_soup_http_src_session_unpause_message (GstSoupHTTPSrc * src)
 620 {
 621   soup_session_unpause_message (src->session, src->msg);
 622 }
 623
 624 static void
 625 gst_soup_http_src_session_pause_message (GstSoupHTTPSrc * src)
 626 {
 627   soup_session_pause_message (src->session, src->msg);
 628 }
 629
 630 static void
 631 gst_soup_http_src_session_close (GstSoupHTTPSrc * src)
 632 {
 633   if (src->session) {
 634     soup_session_abort (src->session);  /* This unrefs the message. */
 635     g_object_unref (src->session);
 636     src->session = NULL;
 637     src->msg = NULL;
 638   }
 639 }
 640
 641 static void
 642 gst_soup_http_src_authenticate_cb (SoupSession * session, SoupMessage * msg,
 643     SoupAuth * auth, gboolean retrying, GstSoupHTTPSrc * src)
 644 {
 645   if (!retrying) {
 646     /* First time authentication only, if we fail and are called again with retry true fall through */
 647     if (msg->status_code == SOUP_STATUS_UNAUTHORIZED) {
 648       if (src->user_id && src->user_pw)
 649         soup_auth_authenticate (auth, src->user_id, src->user_pw);
 650     } else if (msg->status_code == SOUP_STATUS_PROXY_AUTHENTICATION_REQUIRED) {
 651       if (src->proxy_id && src->proxy_pw)
 652         soup_auth_authenticate (auth, src->proxy_id, src->proxy_pw);
 653     }
 654   }
 655 }
 656
 657 static void
 658 gst_soup_http_src_headers_foreach (const gchar * name, const gchar * val,
 659     gpointer src)
 660 {
 661   GST_DEBUG_OBJECT (src, " %s: %s", name, val);
 662 }
 663
 664 static void
 665 gst_soup_http_src_got_headers_cb (SoupMessage * msg, GstSoupHTTPSrc * src)
 666 {
 667   const char *value;
 668   GstTagList *tag_list;
 669   GstBaseSrc *basesrc;
 670   guint64 newsize;
 671   GHashTable *params = NULL;
 672
 673   GST_DEBUG_OBJECT (src, "got headers:");
 674   soup_message_headers_foreach (msg->response_headers,
 675       gst_soup_http_src_headers_foreach, src);
 676
 677   if (msg->status_code == 407 && src->proxy_id && src->proxy_pw)
 678     return;
 679
 680   if (src->automatic_redirect && SOUP_STATUS_IS_REDIRECTION (msg->status_code)) {
 681     GST_DEBUG_OBJECT (src, "%u redirect to \"%s\"", msg->status_code,
 682         soup_message_headers_get_one (msg->response_headers, "Location"));
 683     return;
 684   }
 685
 686   if (msg->status_code == SOUP_STATUS_UNAUTHORIZED)
 687     return;
 688
 689   src->session_io_status = GST_SOUP_HTTP_SRC_SESSION_IO_STATUS_RUNNING;
 690
 691   /* Parse Content-Length. */
 692   if (soup_message_headers_get_encoding (msg->response_headers) ==
 693       SOUP_ENCODING_CONTENT_LENGTH) {
 694     newsize = src->request_position +
 695         soup_message_headers_get_content_length (msg->response_headers);
 696     if (!src->have_size || (src->content_size != newsize)) {
 697       src->content_size = newsize;
 698       src->have_size = TRUE;
 699       src->seekable = TRUE;
 700       GST_DEBUG_OBJECT (src, "size = %" G_GUINT64_FORMAT, src->content_size);
 701
 702       basesrc = GST_BASE_SRC_CAST (src);
 703       basesrc->segment.duration = src->content_size;
 704       gst_element_post_message (GST_ELEMENT (src),
 705           gst_message_new_duration_changed (GST_OBJECT (src)));
 706     }
 707   }
 708
 709   /* Icecast stuff */
 710   tag_list = gst_tag_list_new_empty ();
 711
 712   if ((value =
 713           soup_message_headers_get_one (msg->response_headers,
 714               "icy-metaint")) != NULL) {
 715     gint icy_metaint = atoi (value);
 716
 717     GST_DEBUG_OBJECT (src, "icy-metaint: %s (parsed: %d)", value, icy_metaint);
 718     if (icy_metaint > 0) {
 719       if (src->src_caps)
 720         gst_caps_unref (src->src_caps);
 721
 722       src->src_caps = gst_caps_new_simple ("application/x-icy",
 723           "metadata-interval", G_TYPE_INT, icy_metaint, NULL);
 724
 725       gst_base_src_set_caps (GST_BASE_SRC (src), src->src_caps);
 726     }
 727   }
 728   if ((value =
 729           soup_message_headers_get_content_type (msg->response_headers,
 730               &params)) != NULL) {
 731     GST_DEBUG_OBJECT (src, "Content-Type: %s", value);
 732     if (g_ascii_strcasecmp (value, "audio/L16") == 0) {
 733       gint channels = 2;
 734       gint rate = 44100;
 735       char *param;
 736
 737       if (src->src_caps)
 738         gst_caps_unref (src->src_caps);
 739
 740       param = g_hash_table_lookup (params, "channels");
 741       if (param != NULL)
 742         channels = atol (param);
 743
 744       param = g_hash_table_lookup (params, "rate");
 745       if (param != NULL)
 746         rate = atol (param);
 747
 748       src->src_caps = gst_caps_new_simple ("audio/x-raw",
 749           "format", G_TYPE_STRING, "S16BE",
 750           "layout", G_TYPE_STRING, "interleaved",
 751           "channels", G_TYPE_INT, channels, "rate", G_TYPE_INT, rate, NULL);
 752
 753       gst_base_src_set_caps (GST_BASE_SRC (src), src->src_caps);
 754     } else {
 755       /* Set the Content-Type field on the caps */
 756       if (src->src_caps) {
 757         src->src_caps = gst_caps_make_writable (src->src_caps);
 758         gst_caps_set_simple (src->src_caps, "content-type", G_TYPE_STRING,
 759             value, NULL);
 760         gst_base_src_set_caps (GST_BASE_SRC (src), src->src_caps);
 761       }
 762     }
 763   }
 764
 765   if (params != NULL)
 766     g_hash_table_destroy (params);
 767
 768   if ((value =
 769           soup_message_headers_get_one (msg->response_headers,
 770               "icy-name")) != NULL) {
 771     g_free (src->iradio_name);
 772     src->iradio_name = gst_soup_http_src_unicodify (value);
 773     if (src->iradio_name) {
 774       gst_tag_list_add (tag_list, GST_TAG_MERGE_REPLACE, GST_TAG_ORGANIZATION,
 775           src->iradio_name, NULL);
 776     }
 777   }
 778   if ((value =
 779           soup_message_headers_get_one (msg->response_headers,
 780               "icy-genre")) != NULL) {
 781     g_free (src->iradio_genre);
 782     src->iradio_genre = gst_soup_http_src_unicodify (value);
 783     if (src->iradio_genre) {
 784       gst_tag_list_add (tag_list, GST_TAG_MERGE_REPLACE, GST_TAG_GENRE,
 785           src->iradio_genre, NULL);
 786     }
 787   }
 788   if ((value = soup_message_headers_get_one (msg->response_headers, "icy-url"))
 789       != NULL) {
 790     g_free (src->iradio_url);
 791     src->iradio_url = gst_soup_http_src_unicodify (value);
 792     if (src->iradio_url) {
 793       gst_tag_list_add (tag_list, GST_TAG_MERGE_REPLACE, GST_TAG_LOCATION,
 794           src->iradio_url, NULL);
 795     }
 796   }
 797   if (!gst_tag_list_is_empty (tag_list)) {
 798     GST_DEBUG_OBJECT (src,
 799         "calling gst_element_found_tags with %" GST_PTR_FORMAT, tag_list);
 800     gst_pad_push_event (GST_BASE_SRC_PAD (src), gst_event_new_tag (tag_list));
 801   } else {
 802     gst_tag_list_unref (tag_list);
 803   }
 804
 805   /* Handle HTTP errors. */
 806   gst_soup_http_src_parse_status (msg, src);
 807
 808   /* Check if Range header was respected. */
 809   if (src->ret == GST_FLOW_CUSTOM_ERROR &&
 810       src->read_position && msg->status_code != SOUP_STATUS_PARTIAL_CONTENT) {
 811     src->seekable = FALSE;
 812     GST_ELEMENT_ERROR (src, RESOURCE, SEEK,
 813         (_("Server does not support seeking.")),
 814         ("Server does not accept Range HTTP header, URL: %s", src->location));
 815     src->ret = GST_FLOW_ERROR;
 816   }
 817
 818   /* If we are going to error out, stop all processing right here, so we
 819    * don't output any data (such as an error html page), and return
 820    * GST_FLOW_ERROR from the create function instead of having
 821    * got_chunk_cb overwrite src->ret with FLOW_OK again. */
 822   if (src->ret == GST_FLOW_ERROR) {
 823     gst_soup_http_src_session_pause_message (src);
 824
 825     if (src->loop)
 826       g_main_loop_quit (src->loop);
 827   }
 828 }
 829
 830 /* Have body. Signal EOS. */
 831 static void
 832 gst_soup_http_src_got_body_cb (SoupMessage * msg, GstSoupHTTPSrc * src)
 833 {
 834   if (G_UNLIKELY (msg != src->msg)) {
 835     GST_DEBUG_OBJECT (src, "got body, but not for current message");
 836     return;
 837   }
 838   if (G_UNLIKELY (src->session_io_status !=
 839           GST_SOUP_HTTP_SRC_SESSION_IO_STATUS_RUNNING)) {
 840     /* Probably a redirect. */
 841     return;
 842   }
 843   GST_DEBUG_OBJECT (src, "got body");
 844   src->ret = GST_FLOW_EOS;
 845   if (src->loop)
 846     g_main_loop_quit (src->loop);
 847   gst_soup_http_src_session_pause_message (src);
 848 }
 849
 850 /* Finished. Signal EOS. */
 851 static void
 852 gst_soup_http_src_finished_cb (SoupMessage * msg, GstSoupHTTPSrc * src)
 853 {
 854   if (G_UNLIKELY (msg != src->msg)) {
 855     GST_DEBUG_OBJECT (src, "finished, but not for current message");
 856     return;
 857   }
 858   GST_DEBUG_OBJECT (src, "finished");
 859   src->ret = GST_FLOW_EOS;
 860   if (src->session_io_status == GST_SOUP_HTTP_SRC_SESSION_IO_STATUS_CANCELLED) {
 861     /* gst_soup_http_src_cancel_message() triggered this; probably a seek
 862      * that occurred in the QUEUEING state; i.e. before the connection setup
 863      * was complete. Do nothing */
 864   } else if (src->session_io_status ==
 865       GST_SOUP_HTTP_SRC_SESSION_IO_STATUS_RUNNING && src->read_position > 0) {
 866     /* The server disconnected while streaming. Reconnect and seeking to the
 867      * last location. */
 868     src->retry = TRUE;
 869     src->ret = GST_FLOW_CUSTOM_ERROR;
 870   } else if (G_UNLIKELY (src->session_io_status !=
 871           GST_SOUP_HTTP_SRC_SESSION_IO_STATUS_RUNNING)) {
 872     /* FIXME: reason_phrase is not translated, add proper error message */
 873     GST_ELEMENT_ERROR (src, RESOURCE, NOT_FOUND,
 874         ("%s", msg->reason_phrase),
 875         ("libsoup status code %d", msg->status_code));
 876   }
 877   if (src->loop)
 878     g_main_loop_quit (src->loop);
 879 }
 880
 881 /* Buffer lifecycle management.
 882  *
 883  * gst_soup_http_src_create() runs the GMainLoop for this element, to let
 884  * Soup take control.
 885  * A GstBuffer is allocated in gst_soup_http_src_chunk_allocator() and
 886  * associated with a SoupBuffer.
 887  * Soup reads HTTP data in the GstBuffer's data buffer.
 888  * The gst_soup_http_src_got_chunk_cb() is then called with the SoupBuffer.
 889  * That sets gst_soup_http_src_create()'s return argument to the GstBuffer,
 890  * increments its refcount (to 2), pauses the flow of data from the HTTP
 891  * source to prevent gst_soup_http_src_got_chunk_cb() from being called
 892  * again and breaks out of the GMainLoop.
 893  * Because the SOUP_MESSAGE_OVERWRITE_CHUNKS flag is set, Soup frees the
 894  * SoupBuffer and calls gst_soup_http_src_chunk_free(), which decrements the
 895  * refcount (to 1).
 896  * gst_soup_http_src_create() returns the GstBuffer. It will be freed by a
 897  * downstream element.
 898  * If Soup fails to read HTTP data, it does not call
 899  * gst_soup_http_src_got_chunk_cb(), but still frees the SoupBuffer and
 900  * calls gst_soup_http_src_chunk_free(), which decrements the GstBuffer's
 901  * refcount to 0, freeing it.
 902  */
 903
 904 typedef struct
 905 {
 906   GstBuffer *buffer;
 907   GstMapInfo map;
 908 } SoupGstChunk;
 909
 910 static void
 911 gst_soup_http_src_chunk_free (gpointer user_data)
 912 {
 913   SoupGstChunk *chunk = (SoupGstChunk *) user_data;
 914
 915   gst_buffer_unmap (chunk->buffer, &chunk->map);
 916   gst_buffer_unref (chunk->buffer);
 917   g_slice_free (SoupGstChunk, chunk);
 918 }
 919
 920 static SoupBuffer *
 921 gst_soup_http_src_chunk_allocator (SoupMessage * msg, gsize max_len,
 922     gpointer user_data)
 923 {
 924   GstSoupHTTPSrc *src = (GstSoupHTTPSrc *) user_data;
 925   GstBaseSrc *basesrc = GST_BASE_SRC_CAST (src);
 926   GstBuffer *gstbuf;
 927   SoupBuffer *soupbuf;
 928   gsize length;
 929   GstFlowReturn rc;
 930   SoupGstChunk *chunk;
 931
 932   if (max_len)
 933     length = MIN (basesrc->blocksize, max_len);
 934   else
 935     length = basesrc->blocksize;
 936   GST_DEBUG_OBJECT (src, "alloc %" G_GSIZE_FORMAT " bytes <= %" G_GSIZE_FORMAT,
 937       length, max_len);
 938
 939   rc = GST_BASE_SRC_CLASS (parent_class)->alloc (basesrc, -1, length, &gstbuf);
 940   if (G_UNLIKELY (rc != GST_FLOW_OK)) {
 941     /* Failed to allocate buffer. Stall SoupSession and return error code
 942      * to create(). */
 943     src->ret = rc;
 944     g_main_loop_quit (src->loop);
 945     return NULL;
 946   }
 947
 948   chunk = g_slice_new0 (SoupGstChunk);
 949   chunk->buffer = gstbuf;
 950   gst_buffer_map (gstbuf, &chunk->map, GST_MAP_READWRITE);
 951
 952   soupbuf = soup_buffer_new_with_owner (chunk->map.data, chunk->map.size,
 953       chunk, gst_soup_http_src_chunk_free);
 954
 955   return soupbuf;
 956 }
 957
 958 static void
 959 gst_soup_http_src_got_chunk_cb (SoupMessage * msg, SoupBuffer * chunk,
 960     GstSoupHTTPSrc * src)
 961 {
 962   GstBaseSrc *basesrc;
 963   guint64 new_position;
 964   SoupGstChunk *gchunk;
 965
 966   if (G_UNLIKELY (msg != src->msg)) {
 967     GST_DEBUG_OBJECT (src, "got chunk, but not for current message");
 968     return;
 969   }
 970   if (G_UNLIKELY (src->session_io_status !=
 971           GST_SOUP_HTTP_SRC_SESSION_IO_STATUS_RUNNING)) {
 972     /* Probably a redirect. */
 973     return;
 974   }
 975   basesrc = GST_BASE_SRC_CAST (src);
 976   GST_DEBUG_OBJECT (src, "got chunk of %" G_GSIZE_FORMAT " bytes",
 977       chunk->length);
 978
 979   /* Extract the GstBuffer from the SoupBuffer and set its fields. */
 980   gchunk = (SoupGstChunk *) soup_buffer_get_owner (chunk);
 981   *src->outbuf = gchunk->buffer;
 982
 983   gst_buffer_resize (*src->outbuf, 0, chunk->length);
 984   GST_BUFFER_OFFSET (*src->outbuf) = basesrc->segment.position;
 985
 986   gst_buffer_ref (*src->outbuf);
 987
 988   new_position = src->read_position + chunk->length;
 989   if (G_LIKELY (src->request_position == src->read_position))
 990     src->request_position = new_position;
 991   src->read_position = new_position;
 992
 993   src->ret = GST_FLOW_OK;
 994   g_main_loop_quit (src->loop);
 995   gst_soup_http_src_session_pause_message (src);
 996 }
 997
 998 static void
 999 gst_soup_http_src_response_cb (SoupSession * session, SoupMessage * msg,
1000     GstSoupHTTPSrc * src)
1001 {
1002   if (G_UNLIKELY (msg != src->msg)) {
1003     GST_DEBUG_OBJECT (src, "got response %d: %s, but not for current message",
1004         msg->status_code, msg->reason_phrase);
1005     return;
1006   }
1007   if (G_UNLIKELY (src->session_io_status !=
1008           GST_SOUP_HTTP_SRC_SESSION_IO_STATUS_RUNNING)
1009       && SOUP_STATUS_IS_REDIRECTION (msg->status_code)) {
1010     /* Ignore redirections. */
1011     return;
1012   }
1013   GST_DEBUG_OBJECT (src, "got response %d: %s", msg->status_code,
1014       msg->reason_phrase);
1015   if (src->session_io_status == GST_SOUP_HTTP_SRC_SESSION_IO_STATUS_RUNNING &&
1016       src->read_position > 0) {
1017     /* The server disconnected while streaming. Reconnect and seeking to the
1018      * last location. */
1019     src->retry = TRUE;
1020   } else
1021     gst_soup_http_src_parse_status (msg, src);
1022   /* The session's SoupMessage object expires after this callback returns. */
1023   src->msg = NULL;
1024   g_main_loop_quit (src->loop);
1025 }
1026
1027 #define SOUP_HTTP_SRC_ERROR(src,soup_msg,cat,code,error_message)     \
1028   GST_ELEMENT_ERROR ((src), cat, code, ("%s", error_message),        \
1029       ("%s (%d), URL: %s", (soup_msg)->reason_phrase,                \
1030           (soup_msg)->status_code, (src)->location));
1031
1032 static void
1033 gst_soup_http_src_parse_status (SoupMessage * msg, GstSoupHTTPSrc * src)
1034 {
1035   if (SOUP_STATUS_IS_TRANSPORT_ERROR (msg->status_code)) {
1036     switch (msg->status_code) {
1037       case SOUP_STATUS_CANT_RESOLVE:
1038       case SOUP_STATUS_CANT_RESOLVE_PROXY:
1039         SOUP_HTTP_SRC_ERROR (src, msg, RESOURCE, NOT_FOUND,
1040             _("Could not resolve server name."));
1041         src->ret = GST_FLOW_ERROR;
1042         break;
1043       case SOUP_STATUS_CANT_CONNECT:
1044       case SOUP_STATUS_CANT_CONNECT_PROXY:
1045         SOUP_HTTP_SRC_ERROR (src, msg, RESOURCE, OPEN_READ,
1046             _("Could not establish connection to server."));
1047         src->ret = GST_FLOW_ERROR;
1048         break;
1049       case SOUP_STATUS_SSL_FAILED:
1050         SOUP_HTTP_SRC_ERROR (src, msg, RESOURCE, OPEN_READ,
1051             _("Secure connection setup failed."));
1052         src->ret = GST_FLOW_ERROR;
1053         break;
1054       case SOUP_STATUS_IO_ERROR:
1055         SOUP_HTTP_SRC_ERROR (src, msg, RESOURCE, READ,
1056             _("A network error occured, or the server closed the connection "
1057                 "unexpectedly."));
1058         src->ret = GST_FLOW_ERROR;
1059         break;
1060       case SOUP_STATUS_MALFORMED:
1061         SOUP_HTTP_SRC_ERROR (src, msg, RESOURCE, READ,
1062             _("Server sent bad data."));
1063         src->ret = GST_FLOW_ERROR;
1064         break;
1065       case SOUP_STATUS_CANCELLED:
1066         /* No error message when interrupted by program. */
1067         break;
1068     }
1069   } else if (SOUP_STATUS_IS_CLIENT_ERROR (msg->status_code) ||
1070       SOUP_STATUS_IS_REDIRECTION (msg->status_code) ||
1071       SOUP_STATUS_IS_SERVER_ERROR (msg->status_code)) {
1072     /* Report HTTP error. */
1073     /* FIXME: reason_phrase is not translated and not suitable for user
1074      * error dialog according to libsoup documentation.
1075      * FIXME: error code (OPEN_READ vs. READ) should depend on http status? */
1076     GST_ELEMENT_ERROR (src, RESOURCE, OPEN_READ,
1077         ("%s", msg->reason_phrase),
1078         ("%s (%d), URL: %s", msg->reason_phrase, msg->status_code,
1079             src->location));
1080     src->ret = GST_FLOW_ERROR;
1081   }
1082 }
1083
1084 static gboolean
1085 gst_soup_http_src_build_message (GstSoupHTTPSrc * src)
1086 {
1087   src->msg = soup_message_new (SOUP_METHOD_GET, src->location);
1088   if (!src->msg) {
1089     GST_ELEMENT_ERROR (src, RESOURCE, OPEN_READ,
1090         ("Error parsing URL."), ("URL: %s", src->location));
1091     return FALSE;
1092   }
1093   src->session_io_status = GST_SOUP_HTTP_SRC_SESSION_IO_STATUS_IDLE;
1094   soup_message_headers_append (src->msg->request_headers, "Connection",
1095       "close");
1096   if (src->iradio_mode) {
1097     soup_message_headers_append (src->msg->request_headers, "icy-metadata",
1098         "1");
1099   }
1100   if (src->cookies) {
1101     gchar **cookie;
1102
1103     for (cookie = src->cookies; *cookie != NULL; cookie++) {
1104       soup_message_headers_append (src->msg->request_headers, "Cookie",
1105           *cookie);
1106     }
1107   }
1108   src->retry = FALSE;
1109
1110   g_signal_connect (src->msg, "got_headers",
1111       G_CALLBACK (gst_soup_http_src_got_headers_cb), src);
1112   g_signal_connect (src->msg, "got_body",
1113       G_CALLBACK (gst_soup_http_src_got_body_cb), src);
1114   g_signal_connect (src->msg, "finished",
1115       G_CALLBACK (gst_soup_http_src_finished_cb), src);
1116   g_signal_connect (src->msg, "got_chunk",
1117       G_CALLBACK (gst_soup_http_src_got_chunk_cb), src);
1118   soup_message_set_flags (src->msg, SOUP_MESSAGE_OVERWRITE_CHUNKS |
1119       (src->automatic_redirect ? 0 : SOUP_MESSAGE_NO_REDIRECT));
1120   soup_message_set_chunk_allocator (src->msg,
1121       gst_soup_http_src_chunk_allocator, src, NULL);
1122   gst_soup_http_src_add_range_header (src, src->request_position);
1123
1124   gst_soup_http_src_add_extra_headers (src);
1125
1126   GST_DEBUG_OBJECT (src, "request headers:");
1127   soup_message_headers_foreach (src->msg->request_headers,
1128       gst_soup_http_src_headers_foreach, src);
1129
1130   return TRUE;
1131 }
1132
1133 static GstFlowReturn
1134 gst_soup_http_src_create (GstPushSrc * psrc, GstBuffer ** outbuf)
1135 {
1136   GstSoupHTTPSrc *src;
1137
1138   src = GST_SOUP_HTTP_SRC (psrc);
1139
1140   if (src->msg && (src->request_position != src->read_position)) {
1141     if (src->content_size != 0 && src->request_position >= src->content_size) {
1142       GST_WARNING_OBJECT (src, "Seeking behind the end of file -- EOS");
1143       return GST_FLOW_EOS;
1144     } else if (src->session_io_status ==
1145         GST_SOUP_HTTP_SRC_SESSION_IO_STATUS_IDLE) {
1146       gst_soup_http_src_add_range_header (src, src->request_position);
1147     } else {
1148       GST_DEBUG_OBJECT (src, "Seek from position %" G_GUINT64_FORMAT
1149           " to %" G_GUINT64_FORMAT ": requeueing connection request",
1150           src->read_position, src->request_position);
1151       gst_soup_http_src_cancel_message (src);
1152     }
1153   }
1154   if (!src->msg)
1155     if (!gst_soup_http_src_build_message (src))
1156       return GST_FLOW_ERROR;
1157
1158   src->ret = GST_FLOW_CUSTOM_ERROR;
1159   src->outbuf = outbuf;
1160   do {
1161     if (src->interrupted) {
1162       GST_DEBUG_OBJECT (src, "interrupted");
1163       break;
1164     }
1165     if (src->retry) {
1166       GST_DEBUG_OBJECT (src, "Reconnecting");
1167       if (!gst_soup_http_src_build_message (src))
1168         return GST_FLOW_ERROR;
1169       src->retry = FALSE;
1170       continue;
1171     }
1172     if (!src->msg) {
1173       GST_DEBUG_OBJECT (src, "EOS reached");
1174       break;
1175     }
1176
1177     switch (src->session_io_status) {
1178       case GST_SOUP_HTTP_SRC_SESSION_IO_STATUS_IDLE:
1179         GST_DEBUG_OBJECT (src, "Queueing connection request");
1180         gst_soup_http_src_queue_message (src);
1181         break;
1182       case GST_SOUP_HTTP_SRC_SESSION_IO_STATUS_QUEUED:
1183         break;
1184       case GST_SOUP_HTTP_SRC_SESSION_IO_STATUS_RUNNING:
1185         gst_soup_http_src_session_unpause_message (src);
1186         break;
1187       case GST_SOUP_HTTP_SRC_SESSION_IO_STATUS_CANCELLED:
1188         /* Impossible. */
1189         break;
1190     }
1191
1192     if (src->ret == GST_FLOW_CUSTOM_ERROR)
1193       g_main_loop_run (src->loop);
1194   } while (src->ret == GST_FLOW_CUSTOM_ERROR);
1195
1196   if (src->ret == GST_FLOW_CUSTOM_ERROR)
1197     src->ret = GST_FLOW_EOS;
1198   return src->ret;
1199 }
1200
1201 static gboolean
1202 gst_soup_http_src_start (GstBaseSrc * bsrc)
1203 {
1204   GstSoupHTTPSrc *src = GST_SOUP_HTTP_SRC (bsrc);
1205
1206   GST_DEBUG_OBJECT (src, "start(\"%s\")", src->location);
1207
1208   if (!src->location) {
1209     GST_ELEMENT_ERROR (src, RESOURCE, OPEN_READ, (_("No URL set.")),
1210         ("Missing location property"));
1211     return FALSE;
1212   }
1213
1214   src->context = g_main_context_new ();
1215
1216   src->loop = g_main_loop_new (src->context, TRUE);
1217   if (!src->loop) {
1218     GST_ELEMENT_ERROR (src, LIBRARY, INIT,
1219         (NULL), ("Failed to start GMainLoop"));
1220     g_main_context_unref (src->context);
1221     return FALSE;
1222   }
1223
1224   if (src->proxy == NULL) {
1225     src->session =
1226         soup_session_async_new_with_options (SOUP_SESSION_ASYNC_CONTEXT,
1227         src->context, SOUP_SESSION_USER_AGENT, src->user_agent,
1228         SOUP_SESSION_TIMEOUT, src->timeout,
1229         SOUP_SESSION_ADD_FEATURE_BY_TYPE, SOUP_TYPE_PROXY_RESOLVER_DEFAULT,
1230         NULL);
1231   } else {
1232     src->session =
1233         soup_session_async_new_with_options (SOUP_SESSION_ASYNC_CONTEXT,
1234         src->context, SOUP_SESSION_PROXY_URI, src->proxy,
1235         SOUP_SESSION_TIMEOUT, src->timeout,
1236         SOUP_SESSION_USER_AGENT, src->user_agent, NULL);
1237   }
1238
1239   if (!src->session) {
1240     GST_ELEMENT_ERROR (src, LIBRARY, INIT,
1241         (NULL), ("Failed to create async session"));
1242     return FALSE;
1243   }
1244
1245   g_signal_connect (src->session, "authenticate",
1246       G_CALLBACK (gst_soup_http_src_authenticate_cb), src);
1247   return TRUE;
1248 }
1249
1250 static gboolean
1251 gst_soup_http_src_stop (GstBaseSrc * bsrc)
1252 {
1253   GstSoupHTTPSrc *src;
1254
1255   src = GST_SOUP_HTTP_SRC (bsrc);
1256   GST_DEBUG_OBJECT (src, "stop()");
1257   gst_soup_http_src_session_close (src);
1258   if (src->loop) {
1259     g_main_loop_unref (src->loop);
1260     g_main_context_unref (src->context);
1261     src->loop = NULL;
1262     src->context = NULL;
1263   }
1264   if (src->extra_headers) {
1265     gst_structure_free (src->extra_headers);
1266     src->extra_headers = NULL;
1267   }
1268
1269   gst_soup_http_src_reset (src);
1270   return TRUE;
1271 }
1272
1273 /* Interrupt a blocking request. */
1274 static gboolean
1275 gst_soup_http_src_unlock (GstBaseSrc * bsrc)
1276 {
1277   GstSoupHTTPSrc *src;
1278
1279   src = GST_SOUP_HTTP_SRC (bsrc);
1280   GST_DEBUG_OBJECT (src, "unlock()");
1281
1282   src->interrupted = TRUE;
1283   if (src->loop)
1284     g_main_loop_quit (src->loop);
1285   return TRUE;
1286 }
1287
1288 /* Interrupt interrupt. */
1289 static gboolean
1290 gst_soup_http_src_unlock_stop (GstBaseSrc * bsrc)
1291 {
1292   GstSoupHTTPSrc *src;
1293
1294   src = GST_SOUP_HTTP_SRC (bsrc);
1295   GST_DEBUG_OBJECT (src, "unlock_stop()");
1296
1297   src->interrupted = FALSE;
1298   return TRUE;
1299 }
1300
1301 static gboolean
1302 gst_soup_http_src_get_size (GstBaseSrc * bsrc, guint64 * size)
1303 {
1304   GstSoupHTTPSrc *src;
1305
1306   src = GST_SOUP_HTTP_SRC (bsrc);
1307
1308   if (src->have_size) {
1309     GST_DEBUG_OBJECT (src, "get_size() = %" G_GUINT64_FORMAT,
1310         src->content_size);
1311     *size = src->content_size;
1312     return TRUE;
1313   }
1314   GST_DEBUG_OBJECT (src, "get_size() = FALSE");
1315   return FALSE;
1316 }
1317
1318 static gboolean
1319 gst_soup_http_src_is_seekable (GstBaseSrc * bsrc)
1320 {
1321   GstSoupHTTPSrc *src = GST_SOUP_HTTP_SRC (bsrc);
1322
1323   return src->seekable;
1324 }
1325
1326 static gboolean
1327 gst_soup_http_src_do_seek (GstBaseSrc * bsrc, GstSegment * segment)
1328 {
1329   GstSoupHTTPSrc *src = GST_SOUP_HTTP_SRC (bsrc);
1330
1331   GST_DEBUG_OBJECT (src, "do_seek(%" G_GUINT64_FORMAT ")", segment->start);
1332
1333   if (src->read_position == segment->start &&
1334       src->request_position == src->read_position) {
1335     GST_DEBUG_OBJECT (src, "Seek to current read position and no seek pending");
1336     return TRUE;
1337   }
1338
1339   if (!src->seekable) {
1340     GST_WARNING_OBJECT (src, "Not seekable");
1341     return FALSE;
1342   }
1343
1344   if (segment->rate < 0.0 || segment->format != GST_FORMAT_BYTES) {
1345     GST_WARNING_OBJECT (src, "Invalid seek segment");
1346     return FALSE;
1347   }
1348
1349   if (src->content_size != 0 && segment->start >= src->content_size) {
1350     GST_WARNING_OBJECT (src, "Seeking behind end of file, will go to EOS soon");
1351   }
1352
1353   /* Wait for create() to handle the jump in offset. */
1354   src->request_position = segment->start;
1355   return TRUE;
1356 }
1357
1358 static gboolean
1359 gst_soup_http_src_query (GstBaseSrc * bsrc, GstQuery * query)
1360 {
1361   GstSoupHTTPSrc *src = GST_SOUP_HTTP_SRC (bsrc);
1362   gboolean ret;
1363   GstSchedulingFlags flags;
1364   gint minsize, maxsize, align;
1365
1366   switch (GST_QUERY_TYPE (query)) {
1367     case GST_QUERY_URI:
1368       gst_query_set_uri (query, src->location);
1369       ret = TRUE;
1370       break;
1371     default:
1372       ret = FALSE;
1373       break;
1374   }
1375
1376   if (!ret)
1377     ret = GST_BASE_SRC_CLASS (parent_class)->query (bsrc, query);
1378
1379   switch (GST_QUERY_TYPE (query)) {
1380     case GST_QUERY_SCHEDULING:
1381       gst_query_parse_scheduling (query, &flags, &minsize, &maxsize, &align);
1382       flags |= GST_SCHEDULING_FLAG_BANDWIDTH_LIMITED;
1383       gst_query_set_scheduling (query, flags, minsize, maxsize, align);
1384       break;
1385     default:
1386       break;
1387   }
1388
1389   return ret;
1390 }
1391
1392 static gboolean
1393 gst_soup_http_src_set_location (GstSoupHTTPSrc * src, const gchar * uri,
1394     GError ** error)
1395 {
1396   const char *alt_schemes[] = { "icy://", "icyx://" };
1397   guint i;
1398
1399   if (src->location) {
1400     g_free (src->location);
1401     src->location = NULL;
1402   }
1403
1404   if (uri == NULL)
1405     return FALSE;
1406
1407   for (i = 0; i < G_N_ELEMENTS (alt_schemes); i++) {
1408     if (g_str_has_prefix (uri, alt_schemes[i])) {
1409       src->location =
1410           g_strdup_printf ("http://%s", uri + strlen (alt_schemes[i]));
1411       return TRUE;
1412     }
1413   }
1414
1415   src->location = g_strdup (uri);
1416
1417   return TRUE;
1418 }
1419
1420 static gboolean
1421 gst_soup_http_src_set_proxy (GstSoupHTTPSrc * src, const gchar * uri)
1422 {
1423   if (src->proxy) {
1424     soup_uri_free (src->proxy);
1425     src->proxy = NULL;
1426   }
1427   if (g_str_has_prefix (uri, "http://")) {
1428     src->proxy = soup_uri_new (uri);
1429   } else {
1430     gchar *new_uri = g_strconcat ("http://", uri, NULL);
1431
1432     src->proxy = soup_uri_new (new_uri);
1433     g_free (new_uri);
1434   }
1435
1436   return TRUE;
1437 }
1438
1439 static guint
1440 gst_soup_http_src_uri_get_type (GType type)
1441 {
1442   return GST_URI_SRC;
1443 }
1444
1445 static const gchar *const *
1446 gst_soup_http_src_uri_get_protocols (GType type)
1447 {
1448   static const gchar *protocols[] = { "http", "https", "icy", "icyx", NULL };
1449
1450   return protocols;
1451 }
1452
1453 static gchar *
1454 gst_soup_http_src_uri_get_uri (GstURIHandler * handler)
1455 {
1456   GstSoupHTTPSrc *src = GST_SOUP_HTTP_SRC (handler);
1457
1458   /* FIXME: make thread-safe */
1459   return g_strdup (src->location);
1460 }
1461
1462 static gboolean
1463 gst_soup_http_src_uri_set_uri (GstURIHandler * handler, const gchar * uri,
1464     GError ** error)
1465 {
1466   GstSoupHTTPSrc *src = GST_SOUP_HTTP_SRC (handler);
1467
1468   return gst_soup_http_src_set_location (src, uri, error);
1469 }
1470
1471 static void
1472 gst_soup_http_src_uri_handler_init (gpointer g_iface, gpointer iface_data)
1473 {
1474   GstURIHandlerInterface *iface = (GstURIHandlerInterface *) g_iface;
1475
1476   iface->get_type = gst_soup_http_src_uri_get_type;
1477   iface->get_protocols = gst_soup_http_src_uri_get_protocols;
1478   iface->get_uri = gst_soup_http_src_uri_get_uri;
1479   iface->set_uri = gst_soup_http_src_uri_set_uri;
1480 }