plugins/elements/gstqueue2.c

   1 /* GStreamer
   2  * Copyright (C) 1999,2000 Erik Walthinsen <omega@cse.ogi.edu>
   3  *                    2003 Colin Walters <cwalters@gnome.org>
   4  *                    2000,2005,2007 Wim Taymans <wim.taymans@gmail.com>
   5  *                    2007 Thiago Sousa Santos <thiagoss@lcc.ufcg.edu.br>
   6  *                 SA 2010 ST-Ericsson <benjamin.gaignard@stericsson.com>
   7  *
   8  * gstqueue2.c:
   9  *
  10  * This library is free software; you can redistribute it and/or
  11  * modify it under the terms of the GNU Library General Public
  12  * License as published by the Free Software Foundation; either
  13  * version 2 of the License, or (at your option) any later version.
  14  *
  15  * This library is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  18  * Library General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU Library General Public
  21  * License along with this library; if not, write to the
  22  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  23  * Boston, MA 02111-1307, USA.
  24  */
  25
  26 /**
  27  * SECTION:element-queue2
  28  *
  29  * Data is queued until one of the limits specified by the
  30  * #GstQueue2:max-size-buffers, #GstQueue2:max-size-bytes and/or
  31  * #GstQueue2:max-size-time properties has been reached. Any attempt to push
  32  * more buffers into the queue will block the pushing thread until more space
  33  * becomes available.
  34  *
  35  * The queue will create a new thread on the source pad to decouple the
  36  * processing on sink and source pad.
  37  *
  38  * You can query how many buffers are queued by reading the
  39  * #GstQueue2:current-level-buffers property.
  40  *
  41  * The default queue size limits are 100 buffers, 2MB of data, or
  42  * two seconds worth of data, whichever is reached first.
  43  *
  44  * If you set temp-tmpl to a value such as /tmp/gstreamer-XXXXXX, the element
  45  * will allocate a random free filename and buffer data in the file.
  46  * By using this, it will buffer the entire stream data on the file independently
  47  * of the queue size limits, they will only be used for buffering statistics.
  48  *
  49  * Since 0.10.24, setting the temp-location property with a filename is deprecated
  50  * because it's impossible to securely open a temporary file in this way. The
  51  * property will still be used to notify the application of the allocated
  52  * filename, though.
  53  *
  54  * Last reviewed on 2009-07-10 (0.10.24)
  55  */
  56
  57 #ifdef HAVE_CONFIG_H
  58 #include "config.h"
  59 #endif
  60
  61 #include "gstqueue2.h"
  62
  63 #include <glib/gstdio.h>
  64
  65 #include "gst/gst-i18n-lib.h"
  66 #include "gst/glib-compat-private.h"
  67
  68 #include <string.h>
  69
  70 #ifdef G_OS_WIN32
  71 #include <io.h>                 /* lseek, open, close, read */
  72 #undef lseek
  73 #define lseek _lseeki64
  74 #undef off_t
  75 #define off_t guint64
  76 #else
  77 #include <unistd.h>
  78 #endif
  79
  80 static GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink",
  81     GST_PAD_SINK,
  82     GST_PAD_ALWAYS,
  83     GST_STATIC_CAPS_ANY);
  84
  85 static GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src",
  86     GST_PAD_SRC,
  87     GST_PAD_ALWAYS,
  88     GST_STATIC_CAPS_ANY);
  89
  90 GST_DEBUG_CATEGORY_STATIC (queue_debug);
  91 #define GST_CAT_DEFAULT (queue_debug)
  92 GST_DEBUG_CATEGORY_STATIC (queue_dataflow);
  93
  94 enum
  95 {
  96   LAST_SIGNAL
  97 };
  98
  99 /* other defines */
 100 #define DEFAULT_BUFFER_SIZE 4096
 101 #define QUEUE_IS_USING_TEMP_FILE(queue) ((queue)->temp_location_set || (queue)->temp_template != NULL)
 102 #define QUEUE_IS_USING_RING_BUFFER(queue) ((queue)->ring_buffer_max_size != 0)  /* for consistency with the above macro */
 103 #define QUEUE_IS_USING_QUEUE(queue) (!QUEUE_IS_USING_TEMP_FILE(queue) && !QUEUE_IS_USING_RING_BUFFER (queue))
 104
 105 #define QUEUE_MAX_BYTES(queue) MIN((queue)->max_level.bytes, (queue)->ring_buffer_max_size)
 106
 107 /* default property values */
 108 #define DEFAULT_MAX_SIZE_BUFFERS   100  /* 100 buffers */
 109 #define DEFAULT_MAX_SIZE_BYTES     (2 * 1024 * 1024)    /* 2 MB */
 110 #define DEFAULT_MAX_SIZE_TIME      2 * GST_SECOND       /* 2 seconds */
 111 #define DEFAULT_USE_BUFFERING      FALSE
 112 #define DEFAULT_USE_RATE_ESTIMATE  TRUE
 113 #define DEFAULT_LOW_PERCENT        10
 114 #define DEFAULT_HIGH_PERCENT       99
 115 #define DEFAULT_TEMP_REMOVE        TRUE
 116 #define DEFAULT_RING_BUFFER_MAX_SIZE 0
 117
 118 enum
 119 {
 120   PROP_0,
 121   PROP_CUR_LEVEL_BUFFERS,
 122   PROP_CUR_LEVEL_BYTES,
 123   PROP_CUR_LEVEL_TIME,
 124   PROP_MAX_SIZE_BUFFERS,
 125   PROP_MAX_SIZE_BYTES,
 126   PROP_MAX_SIZE_TIME,
 127   PROP_USE_BUFFERING,
 128   PROP_USE_RATE_ESTIMATE,
 129   PROP_LOW_PERCENT,
 130   PROP_HIGH_PERCENT,
 131   PROP_TEMP_TEMPLATE,
 132   PROP_TEMP_LOCATION,
 133   PROP_TEMP_REMOVE,
 134   PROP_RING_BUFFER_MAX_SIZE,
 135   PROP_LAST
 136 };
 137
 138 #define GST_QUEUE2_CLEAR_LEVEL(l) G_STMT_START {         \
 139   l.buffers = 0;                                        \
 140   l.bytes = 0;                                          \
 141   l.time = 0;                                           \
 142   l.rate_time = 0;                                      \
 143 } G_STMT_END
 144
 145 #define STATUS(queue, pad, msg) \
 146   GST_CAT_LOG_OBJECT (queue_dataflow, queue, \
 147                       "(%s:%s) " msg ": %u of %u buffers, %u of %u " \
 148                       "bytes, %" G_GUINT64_FORMAT " of %" G_GUINT64_FORMAT \
 149                       " ns, %"G_GUINT64_FORMAT" items", \
 150                       GST_DEBUG_PAD_NAME (pad), \
 151                       queue->cur_level.buffers, \
 152                       queue->max_level.buffers, \
 153                       queue->cur_level.bytes, \
 154                       queue->max_level.bytes, \
 155                       queue->cur_level.time, \
 156                       queue->max_level.time, \
 157                       (guint64) (!QUEUE_IS_USING_QUEUE(queue) ? \
 158                         queue->current->writing_pos - queue->current->max_reading_pos : \
 159                         queue->queue.length))
 160
 161 #define GST_QUEUE2_MUTEX_LOCK(q) G_STMT_START {                          \
 162   g_mutex_lock (q->qlock);                                              \
 163 } G_STMT_END
 164
 165 #define GST_QUEUE2_MUTEX_LOCK_CHECK(q,res,label) G_STMT_START {         \
 166   GST_QUEUE2_MUTEX_LOCK (q);                                            \
 167   if (res != GST_FLOW_OK)                                               \
 168     goto label;                                                         \
 169 } G_STMT_END
 170
 171 #define GST_QUEUE2_MUTEX_UNLOCK(q) G_STMT_START {                        \
 172   g_mutex_unlock (q->qlock);                                            \
 173 } G_STMT_END
 174
 175 #define GST_QUEUE2_WAIT_DEL_CHECK(q, res, label) G_STMT_START {         \
 176   STATUS (queue, q->sinkpad, "wait for DEL");                           \
 177   q->waiting_del = TRUE;                                                \
 178   g_cond_wait (q->item_del, queue->qlock);                              \
 179   q->waiting_del = FALSE;                                               \
 180   if (res != GST_FLOW_OK) {                                             \
 181     STATUS (queue, q->srcpad, "received DEL wakeup");                   \
 182     goto label;                                                         \
 183   }                                                                     \
 184   STATUS (queue, q->sinkpad, "received DEL");                           \
 185 } G_STMT_END
 186
 187 #define GST_QUEUE2_WAIT_ADD_CHECK(q, res, label) G_STMT_START {         \
 188   STATUS (queue, q->srcpad, "wait for ADD");                            \
 189   q->waiting_add = TRUE;                                                \
 190   g_cond_wait (q->item_add, q->qlock);                                  \
 191   q->waiting_add = FALSE;                                               \
 192   if (res != GST_FLOW_OK) {                                             \
 193     STATUS (queue, q->srcpad, "received ADD wakeup");                   \
 194     goto label;                                                         \
 195   }                                                                     \
 196   STATUS (queue, q->srcpad, "received ADD");                            \
 197 } G_STMT_END
 198
 199 #define GST_QUEUE2_SIGNAL_DEL(q) G_STMT_START {                          \
 200   if (q->waiting_del) {                                                 \
 201     STATUS (q, q->srcpad, "signal DEL");                                \
 202     g_cond_signal (q->item_del);                                        \
 203   }                                                                     \
 204 } G_STMT_END
 205
 206 #define GST_QUEUE2_SIGNAL_ADD(q) G_STMT_START {                          \
 207   if (q->waiting_add) {                                                 \
 208     STATUS (q, q->sinkpad, "signal ADD");                               \
 209     g_cond_signal (q->item_add);                                        \
 210   }                                                                     \
 211 } G_STMT_END
 212
 213 #define _do_init(bla) \
 214     GST_DEBUG_CATEGORY_INIT (queue_debug, "queue2", 0, "queue element"); \
 215     GST_DEBUG_CATEGORY_INIT (queue_dataflow, "queue2_dataflow", 0, \
 216         "dataflow inside the queue element");
 217
 218 GST_BOILERPLATE_FULL (GstQueue2, gst_queue2, GstElement, GST_TYPE_ELEMENT,
 219     _do_init);
 220
 221 static void gst_queue2_finalize (GObject * object);
 222
 223 static void gst_queue2_set_property (GObject * object,
 224     guint prop_id, const GValue * value, GParamSpec * pspec);
 225 static void gst_queue2_get_property (GObject * object,
 226     guint prop_id, GValue * value, GParamSpec * pspec);
 227
 228 static GstFlowReturn gst_queue2_chain (GstPad * pad, GstBuffer * buffer);
 229 static GstFlowReturn gst_queue2_chain_list (GstPad * pad,
 230     GstBufferList * buffer_list);
 231 static GstFlowReturn gst_queue2_bufferalloc (GstPad * pad, guint64 offset,
 232     guint size, GstCaps * caps, GstBuffer ** buf);
 233 static GstFlowReturn gst_queue2_push_one (GstQueue2 * queue);
 234 static void gst_queue2_loop (GstPad * pad);
 235
 236 static gboolean gst_queue2_handle_sink_event (GstPad * pad, GstEvent * event);
 237
 238 static gboolean gst_queue2_handle_src_event (GstPad * pad, GstEvent * event);
 239 static gboolean gst_queue2_handle_src_query (GstPad * pad, GstQuery * query);
 240 static gboolean gst_queue2_handle_query (GstElement * element,
 241     GstQuery * query);
 242
 243 static GstCaps *gst_queue2_getcaps (GstPad * pad);
 244 static gboolean gst_queue2_acceptcaps (GstPad * pad, GstCaps * caps);
 245
 246 static GstFlowReturn gst_queue2_get_range (GstPad * pad, guint64 offset,
 247     guint length, GstBuffer ** buffer);
 248 static gboolean gst_queue2_src_checkgetrange_function (GstPad * pad);
 249
 250 static gboolean gst_queue2_src_activate_pull (GstPad * pad, gboolean active);
 251 static gboolean gst_queue2_src_activate_push (GstPad * pad, gboolean active);
 252 static gboolean gst_queue2_sink_activate_push (GstPad * pad, gboolean active);
 253 static GstStateChangeReturn gst_queue2_change_state (GstElement * element,
 254     GstStateChange transition);
 255
 256 static gboolean gst_queue2_is_empty (GstQueue2 * queue);
 257 static gboolean gst_queue2_is_filled (GstQueue2 * queue);
 258
 259 static void update_cur_level (GstQueue2 * queue, GstQueue2Range * range);
 260
 261 typedef enum
 262 {
 263   GST_QUEUE2_ITEM_TYPE_UNKNOWN = 0,
 264   GST_QUEUE2_ITEM_TYPE_BUFFER,
 265   GST_QUEUE2_ITEM_TYPE_BUFFER_LIST,
 266   GST_QUEUE2_ITEM_TYPE_EVENT
 267 } GstQueue2ItemType;
 268
 269 /* static guint gst_queue2_signals[LAST_SIGNAL] = { 0 }; */
 270
 271 static void
 272 gst_queue2_base_init (gpointer g_class)
 273 {
 274   GstElementClass *gstelement_class = GST_ELEMENT_CLASS (g_class);
 275
 276   gst_element_class_add_pad_template (gstelement_class,
 277       gst_static_pad_template_get (&srctemplate));
 278   gst_element_class_add_pad_template (gstelement_class,
 279       gst_static_pad_template_get (&sinktemplate));
 280
 281   gst_element_class_set_details_simple (gstelement_class, "Queue 2",
 282       "Generic",
 283       "Simple data queue",
 284       "Erik Walthinsen <omega@cse.ogi.edu>, "
 285       "Wim Taymans <wim.taymans@gmail.com>");
 286 }
 287
 288 static void
 289 gst_queue2_class_init (GstQueue2Class * klass)
 290 {
 291   GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
 292   GstElementClass *gstelement_class = GST_ELEMENT_CLASS (klass);
 293
 294   gobject_class->set_property = gst_queue2_set_property;
 295   gobject_class->get_property = gst_queue2_get_property;
 296
 297   /* properties */
 298   g_object_class_install_property (gobject_class, PROP_CUR_LEVEL_BYTES,
 299       g_param_spec_uint ("current-level-bytes", "Current level (kB)",
 300           "Current amount of data in the queue (bytes)",
 301           0, G_MAXUINT, 0, G_PARAM_READABLE | G_PARAM_STATIC_STRINGS));
 302   g_object_class_install_property (gobject_class, PROP_CUR_LEVEL_BUFFERS,
 303       g_param_spec_uint ("current-level-buffers", "Current level (buffers)",
 304           "Current number of buffers in the queue",
 305           0, G_MAXUINT, 0, G_PARAM_READABLE | G_PARAM_STATIC_STRINGS));
 306   g_object_class_install_property (gobject_class, PROP_CUR_LEVEL_TIME,
 307       g_param_spec_uint64 ("current-level-time", "Current level (ns)",
 308           "Current amount of data in the queue (in ns)",
 309           0, G_MAXUINT64, 0, G_PARAM_READABLE | G_PARAM_STATIC_STRINGS));
 310
 311   g_object_class_install_property (gobject_class, PROP_MAX_SIZE_BYTES,
 312       g_param_spec_uint ("max-size-bytes", "Max. size (kB)",
 313           "Max. amount of data in the queue (bytes, 0=disable)",
 314           0, G_MAXUINT, DEFAULT_MAX_SIZE_BYTES,
 315           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 316   g_object_class_install_property (gobject_class, PROP_MAX_SIZE_BUFFERS,
 317       g_param_spec_uint ("max-size-buffers", "Max. size (buffers)",
 318           "Max. number of buffers in the queue (0=disable)", 0, G_MAXUINT,
 319           DEFAULT_MAX_SIZE_BUFFERS,
 320           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 321   g_object_class_install_property (gobject_class, PROP_MAX_SIZE_TIME,
 322       g_param_spec_uint64 ("max-size-time", "Max. size (ns)",
 323           "Max. amount of data in the queue (in ns, 0=disable)", 0, G_MAXUINT64,
 324           DEFAULT_MAX_SIZE_TIME, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 325
 326   g_object_class_install_property (gobject_class, PROP_USE_BUFFERING,
 327       g_param_spec_boolean ("use-buffering", "Use buffering",
 328           "Emit GST_MESSAGE_BUFFERING based on low-/high-percent thresholds",
 329           DEFAULT_USE_BUFFERING, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 330   g_object_class_install_property (gobject_class, PROP_USE_RATE_ESTIMATE,
 331       g_param_spec_boolean ("use-rate-estimate", "Use Rate Estimate",
 332           "Estimate the bitrate of the stream to calculate time level",
 333           DEFAULT_USE_RATE_ESTIMATE,
 334           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 335   g_object_class_install_property (gobject_class, PROP_LOW_PERCENT,
 336       g_param_spec_int ("low-percent", "Low percent",
 337           "Low threshold for buffering to start. Only used if use-buffering is True",
 338           0, 100, DEFAULT_LOW_PERCENT,
 339           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 340   g_object_class_install_property (gobject_class, PROP_HIGH_PERCENT,
 341       g_param_spec_int ("high-percent", "High percent",
 342           "High threshold for buffering to finish. Only used if use-buffering is True",
 343           0, 100, DEFAULT_HIGH_PERCENT,
 344           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 345
 346   g_object_class_install_property (gobject_class, PROP_TEMP_TEMPLATE,
 347       g_param_spec_string ("temp-template", "Temporary File Template",
 348           "File template to store temporary files in, should contain directory "
 349           "and XXXXXX. (NULL == disabled)",
 350           NULL, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 351
 352   g_object_class_install_property (gobject_class, PROP_TEMP_LOCATION,
 353       g_param_spec_string ("temp-location", "Temporary File Location",
 354           "Location to store temporary files in (Deprecated: Only read this "
 355           "property, use temp-template to configure the name template)",
 356           NULL, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 357
 358   /**
 359    * GstQueue2:temp-remove
 360    *
 361    * When temp-template is set, remove the temporary file when going to READY.
 362    *
 363    * Since: 0.10.26
 364    */
 365   g_object_class_install_property (gobject_class, PROP_TEMP_REMOVE,
 366       g_param_spec_boolean ("temp-remove", "Remove the Temporary File",
 367           "Remove the temp-location after use",
 368           DEFAULT_TEMP_REMOVE, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 369
 370   /**
 371    * GstQueue2:ring-buffer-max-size
 372    *
 373    * The maximum size of the ring buffer in bytes. If set to 0, the ring
 374    * buffer is disabled. Default 0.
 375    *
 376    * Since: 0.10.31
 377    */
 378   g_object_class_install_property (gobject_class, PROP_RING_BUFFER_MAX_SIZE,
 379       g_param_spec_uint64 ("ring-buffer-max-size",
 380           "Max. ring buffer size (bytes)",
 381           "Max. amount of data in the ring buffer (bytes, 0 = disabled)",
 382           0, G_MAXUINT64, DEFAULT_RING_BUFFER_MAX_SIZE,
 383           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 384
 385   /* set several parent class virtual functions */
 386   gobject_class->finalize = gst_queue2_finalize;
 387
 388   gstelement_class->change_state = GST_DEBUG_FUNCPTR (gst_queue2_change_state);
 389   gstelement_class->query = GST_DEBUG_FUNCPTR (gst_queue2_handle_query);
 390 }
 391
 392 static void
 393 gst_queue2_init (GstQueue2 * queue, GstQueue2Class * g_class)
 394 {
 395   queue->sinkpad = gst_pad_new_from_static_template (&sinktemplate, "sink");
 396
 397   gst_pad_set_chain_function (queue->sinkpad,
 398       GST_DEBUG_FUNCPTR (gst_queue2_chain));
 399   gst_pad_set_chain_list_function (queue->sinkpad,
 400       GST_DEBUG_FUNCPTR (gst_queue2_chain_list));
 401   gst_pad_set_activatepush_function (queue->sinkpad,
 402       GST_DEBUG_FUNCPTR (gst_queue2_sink_activate_push));
 403   gst_pad_set_event_function (queue->sinkpad,
 404       GST_DEBUG_FUNCPTR (gst_queue2_handle_sink_event));
 405   gst_pad_set_getcaps_function (queue->sinkpad,
 406       GST_DEBUG_FUNCPTR (gst_queue2_getcaps));
 407   gst_pad_set_acceptcaps_function (queue->sinkpad,
 408       GST_DEBUG_FUNCPTR (gst_queue2_acceptcaps));
 409   gst_pad_set_bufferalloc_function (queue->sinkpad,
 410       GST_DEBUG_FUNCPTR (gst_queue2_bufferalloc));
 411   gst_element_add_pad (GST_ELEMENT (queue), queue->sinkpad);
 412
 413   queue->srcpad = gst_pad_new_from_static_template (&srctemplate, "src");
 414
 415   gst_pad_set_activatepull_function (queue->srcpad,
 416       GST_DEBUG_FUNCPTR (gst_queue2_src_activate_pull));
 417   gst_pad_set_activatepush_function (queue->srcpad,
 418       GST_DEBUG_FUNCPTR (gst_queue2_src_activate_push));
 419   gst_pad_set_getrange_function (queue->srcpad,
 420       GST_DEBUG_FUNCPTR (gst_queue2_get_range));
 421   gst_pad_set_checkgetrange_function (queue->srcpad,
 422       GST_DEBUG_FUNCPTR (gst_queue2_src_checkgetrange_function));
 423   gst_pad_set_getcaps_function (queue->srcpad,
 424       GST_DEBUG_FUNCPTR (gst_queue2_getcaps));
 425   gst_pad_set_acceptcaps_function (queue->srcpad,
 426       GST_DEBUG_FUNCPTR (gst_queue2_acceptcaps));
 427   gst_pad_set_event_function (queue->srcpad,
 428       GST_DEBUG_FUNCPTR (gst_queue2_handle_src_event));
 429   gst_pad_set_query_function (queue->srcpad,
 430       GST_DEBUG_FUNCPTR (gst_queue2_handle_src_query));
 431   gst_element_add_pad (GST_ELEMENT (queue), queue->srcpad);
 432
 433   /* levels */
 434   GST_QUEUE2_CLEAR_LEVEL (queue->cur_level);
 435   queue->max_level.buffers = DEFAULT_MAX_SIZE_BUFFERS;
 436   queue->max_level.bytes = DEFAULT_MAX_SIZE_BYTES;
 437   queue->max_level.time = DEFAULT_MAX_SIZE_TIME;
 438   queue->max_level.rate_time = DEFAULT_MAX_SIZE_TIME;
 439   queue->use_buffering = DEFAULT_USE_BUFFERING;
 440   queue->use_rate_estimate = DEFAULT_USE_RATE_ESTIMATE;
 441   queue->low_percent = DEFAULT_LOW_PERCENT;
 442   queue->high_percent = DEFAULT_HIGH_PERCENT;
 443
 444   gst_segment_init (&queue->sink_segment, GST_FORMAT_TIME);
 445   gst_segment_init (&queue->src_segment, GST_FORMAT_TIME);
 446
 447   queue->sinktime = GST_CLOCK_TIME_NONE;
 448   queue->srctime = GST_CLOCK_TIME_NONE;
 449   queue->sink_tainted = TRUE;
 450   queue->src_tainted = TRUE;
 451
 452   queue->srcresult = GST_FLOW_WRONG_STATE;
 453   queue->sinkresult = GST_FLOW_WRONG_STATE;
 454   queue->is_eos = FALSE;
 455   queue->in_timer = g_timer_new ();
 456   queue->out_timer = g_timer_new ();
 457
 458   queue->qlock = g_mutex_new ();
 459   queue->waiting_add = FALSE;
 460   queue->item_add = g_cond_new ();
 461   queue->waiting_del = FALSE;
 462   queue->item_del = g_cond_new ();
 463   g_queue_init (&queue->queue);
 464
 465   queue->buffering_percent = 100;
 466
 467   /* tempfile related */
 468   queue->temp_template = NULL;
 469   queue->temp_location = NULL;
 470   queue->temp_location_set = FALSE;
 471   queue->temp_remove = DEFAULT_TEMP_REMOVE;
 472
 473   queue->ring_buffer = NULL;
 474   queue->ring_buffer_max_size = DEFAULT_RING_BUFFER_MAX_SIZE;
 475
 476   GST_DEBUG_OBJECT (queue,
 477       "initialized queue's not_empty & not_full conditions");
 478 }
 479
 480 /* called only once, as opposed to dispose */
 481 static void
 482 gst_queue2_finalize (GObject * object)
 483 {
 484   GstQueue2 *queue = GST_QUEUE2 (object);
 485
 486   GST_DEBUG_OBJECT (queue, "finalizing queue");
 487
 488   while (!g_queue_is_empty (&queue->queue)) {
 489     GstMiniObject *data = g_queue_pop_head (&queue->queue);
 490
 491     gst_mini_object_unref (data);
 492   }
 493
 494   g_queue_clear (&queue->queue);
 495   g_mutex_free (queue->qlock);
 496   g_cond_free (queue->item_add);
 497   g_cond_free (queue->item_del);
 498   g_timer_destroy (queue->in_timer);
 499   g_timer_destroy (queue->out_timer);
 500
 501   /* temp_file path cleanup  */
 502   g_free (queue->temp_template);
 503   g_free (queue->temp_location);
 504
 505   G_OBJECT_CLASS (parent_class)->finalize (object);
 506 }
 507
 508 static void
 509 debug_ranges (GstQueue2 * queue)
 510 {
 511   GstQueue2Range *walk;
 512
 513   for (walk = queue->ranges; walk; walk = walk->next) {
 514     GST_DEBUG_OBJECT (queue,
 515         "range [%" G_GUINT64_FORMAT "-%" G_GUINT64_FORMAT "] (rb [%"
 516         G_GUINT64_FORMAT "-%" G_GUINT64_FORMAT "]), reading %" G_GUINT64_FORMAT
 517         " current range? %s", walk->offset, walk->writing_pos, walk->rb_offset,
 518         walk->rb_writing_pos, walk->reading_pos,
 519         walk == queue->current ? "**y**" : "  n  ");
 520   }
 521 }
 522
 523 /* clear all the downloaded ranges */
 524 static void
 525 clean_ranges (GstQueue2 * queue)
 526 {
 527   GST_DEBUG_OBJECT (queue, "clean queue ranges");
 528
 529   g_slice_free_chain (GstQueue2Range, queue->ranges, next);
 530   queue->ranges = NULL;
 531   queue->current = NULL;
 532 }
 533
 534 /* find a range that contains @offset or NULL when nothing does */
 535 static GstQueue2Range *
 536 find_range (GstQueue2 * queue, guint64 offset)
 537 {
 538   GstQueue2Range *range = NULL;
 539   GstQueue2Range *walk;
 540
 541   /* first do a quick check for the current range */
 542   for (walk = queue->ranges; walk; walk = walk->next) {
 543     if (offset >= walk->offset && offset <= walk->writing_pos) {
 544       /* we can reuse an existing range */
 545       range = walk;
 546       break;
 547     }
 548   }
 549   if (range) {
 550     GST_DEBUG_OBJECT (queue,
 551         "found range for %" G_GUINT64_FORMAT ": [%" G_GUINT64_FORMAT "-%"
 552         G_GUINT64_FORMAT "]", offset, range->offset, range->writing_pos);
 553   } else {
 554     GST_DEBUG_OBJECT (queue, "no range for %" G_GUINT64_FORMAT, offset);
 555   }
 556   return range;
 557 }
 558
 559 static void
 560 update_cur_level (GstQueue2 * queue, GstQueue2Range * range)
 561 {
 562   guint64 max_reading_pos, writing_pos;
 563
 564   writing_pos = range->writing_pos;
 565   max_reading_pos = range->max_reading_pos;
 566
 567   if (writing_pos > max_reading_pos)
 568     queue->cur_level.bytes = writing_pos - max_reading_pos;
 569   else
 570     queue->cur_level.bytes = 0;
 571 }
 572
 573 /* make a new range for @offset or reuse an existing range */
 574 static GstQueue2Range *
 575 add_range (GstQueue2 * queue, guint64 offset)
 576 {
 577   GstQueue2Range *range, *prev, *next;
 578
 579   GST_DEBUG_OBJECT (queue, "find range for %" G_GUINT64_FORMAT, offset);
 580
 581   if ((range = find_range (queue, offset))) {
 582     GST_DEBUG_OBJECT (queue,
 583         "reusing range %" G_GUINT64_FORMAT "-%" G_GUINT64_FORMAT, range->offset,
 584         range->writing_pos);
 585     range->writing_pos = offset;
 586   } else {
 587     GST_DEBUG_OBJECT (queue,
 588         "new range %" G_GUINT64_FORMAT "-%" G_GUINT64_FORMAT, offset, offset);
 589
 590     range = g_slice_new0 (GstQueue2Range);
 591     range->offset = offset;
 592     /* we want to write to the next location in the ring buffer */
 593     range->rb_offset = queue->current ? queue->current->rb_writing_pos : 0;
 594     range->writing_pos = offset;
 595     range->rb_writing_pos = range->rb_offset;
 596     range->reading_pos = offset;
 597     range->max_reading_pos = offset;
 598
 599     /* insert sorted */
 600     prev = NULL;
 601     next = queue->ranges;
 602     while (next) {
 603       if (next->offset > offset) {
 604         /* insert before next */
 605         GST_DEBUG_OBJECT (queue,
 606             "insert before range %p, offset %" G_GUINT64_FORMAT, next,
 607             next->offset);
 608         break;
 609       }
 610       /* try next */
 611       prev = next;
 612       next = next->next;
 613     }
 614     range->next = next;
 615     if (prev)
 616       prev->next = range;
 617     else
 618       queue->ranges = range;
 619   }
 620   debug_ranges (queue);
 621
 622   /* update the stats for this range */
 623   update_cur_level (queue, range);
 624
 625   return range;
 626 }
 627
 628
 629 /* clear and init the download ranges for offset 0 */
 630 static void
 631 init_ranges (GstQueue2 * queue)
 632 {
 633   GST_DEBUG_OBJECT (queue, "init queue ranges");
 634
 635   /* get rid of all the current ranges */
 636   clean_ranges (queue);
 637   /* make a range for offset 0 */
 638   queue->current = add_range (queue, 0);
 639 }
 640
 641 static gboolean
 642 gst_queue2_acceptcaps (GstPad * pad, GstCaps * caps)
 643 {
 644   GstQueue2 *queue;
 645   GstPad *otherpad;
 646   gboolean result;
 647
 648   queue = GST_QUEUE2 (GST_PAD_PARENT (pad));
 649
 650   otherpad = (pad == queue->srcpad ? queue->sinkpad : queue->srcpad);
 651   result = gst_pad_peer_accept_caps (otherpad, caps);
 652
 653   return result;
 654 }
 655
 656 static GstCaps *
 657 gst_queue2_getcaps (GstPad * pad)
 658 {
 659   GstQueue2 *queue;
 660   GstPad *otherpad;
 661   GstCaps *result;
 662
 663   queue = GST_QUEUE2 (gst_pad_get_parent (pad));
 664   if (G_UNLIKELY (queue == NULL))
 665     return gst_caps_new_any ();
 666
 667   otherpad = (pad == queue->srcpad ? queue->sinkpad : queue->srcpad);
 668   result = gst_pad_peer_get_caps (otherpad);
 669   if (result == NULL)
 670     result = gst_caps_new_any ();
 671
 672   gst_object_unref (queue);
 673
 674   return result;
 675 }
 676
 677 static GstFlowReturn
 678 gst_queue2_bufferalloc (GstPad * pad, guint64 offset, guint size,
 679     GstCaps * caps, GstBuffer ** buf)
 680 {
 681   GstQueue2 *queue;
 682   GstFlowReturn result;
 683
 684   queue = GST_QUEUE2 (GST_PAD_PARENT (pad));
 685
 686   /* Forward to src pad, without setting caps on the src pad */
 687   result = gst_pad_alloc_buffer (queue->srcpad, offset, size, caps, buf);
 688
 689   return result;
 690 }
 691
 692 /* calculate the diff between running time on the sink and src of the queue.
 693  * This is the total amount of time in the queue. */
 694 static void
 695 update_time_level (GstQueue2 * queue)
 696 {
 697   if (queue->sink_tainted) {
 698     queue->sinktime =
 699         gst_segment_to_running_time (&queue->sink_segment, GST_FORMAT_TIME,
 700         queue->sink_segment.last_stop);
 701     queue->sink_tainted = FALSE;
 702   }
 703
 704   if (queue->src_tainted) {
 705     queue->srctime =
 706         gst_segment_to_running_time (&queue->src_segment, GST_FORMAT_TIME,
 707         queue->src_segment.last_stop);
 708     queue->src_tainted = FALSE;
 709   }
 710
 711   GST_DEBUG_OBJECT (queue, "sink %" GST_TIME_FORMAT ", src %" GST_TIME_FORMAT,
 712       GST_TIME_ARGS (queue->sinktime), GST_TIME_ARGS (queue->srctime));
 713
 714   if (queue->sinktime != GST_CLOCK_TIME_NONE
 715       && queue->srctime != GST_CLOCK_TIME_NONE
 716       && queue->sinktime >= queue->srctime)
 717     queue->cur_level.time = queue->sinktime - queue->srctime;
 718   else
 719     queue->cur_level.time = 0;
 720 }
 721
 722 /* take a NEWSEGMENT event and apply the values to segment, updating the time
 723  * level of queue. */
 724 static void
 725 apply_segment (GstQueue2 * queue, GstEvent * event, GstSegment * segment,
 726     gboolean is_sink)
 727 {
 728   gboolean update;
 729   GstFormat format;
 730   gdouble rate, arate;
 731   gint64 start, stop, time;
 732
 733   gst_event_parse_new_segment_full (event, &update, &rate, &arate,
 734       &format, &start, &stop, &time);
 735
 736   GST_DEBUG_OBJECT (queue,
 737       "received NEWSEGMENT update %d, rate %lf, applied rate %lf, "
 738       "format %d, "
 739       "%" G_GINT64_FORMAT " -- %" G_GINT64_FORMAT ", time %"
 740       G_GINT64_FORMAT, update, rate, arate, format, start, stop, time);
 741
 742   if (format == GST_FORMAT_BYTES) {
 743     if (!QUEUE_IS_USING_QUEUE (queue)) {
 744       /* start is where we'll be getting from and as such writing next */
 745       queue->current = add_range (queue, start);
 746     }
 747   }
 748
 749   /* now configure the values, we use these to track timestamps on the
 750    * sinkpad. */
 751   if (format != GST_FORMAT_TIME) {
 752     /* non-time format, pretent the current time segment is closed with a
 753      * 0 start and unknown stop time. */
 754     update = FALSE;
 755     format = GST_FORMAT_TIME;
 756     start = 0;
 757     stop = -1;
 758     time = 0;
 759   }
 760   gst_segment_set_newsegment_full (segment, update,
 761       rate, arate, format, start, stop, time);
 762
 763   GST_DEBUG_OBJECT (queue,
 764       "configured NEWSEGMENT %" GST_SEGMENT_FORMAT, segment);
 765
 766   if (is_sink)
 767     queue->sink_tainted = TRUE;
 768   else
 769     queue->src_tainted = TRUE;
 770
 771   /* segment can update the time level of the queue */
 772   update_time_level (queue);
 773 }
 774
 775 /* take a buffer and update segment, updating the time level of the queue. */
 776 static void
 777 apply_buffer (GstQueue2 * queue, GstBuffer * buffer, GstSegment * segment,
 778     gboolean is_sink)
 779 {
 780   GstClockTime duration, timestamp;
 781
 782   timestamp = GST_BUFFER_TIMESTAMP (buffer);
 783   duration = GST_BUFFER_DURATION (buffer);
 784
 785   /* if no timestamp is set, assume it's continuous with the previous
 786    * time */
 787   if (timestamp == GST_CLOCK_TIME_NONE)
 788     timestamp = segment->last_stop;
 789
 790   /* add duration */
 791   if (duration != GST_CLOCK_TIME_NONE)
 792     timestamp += duration;
 793
 794   GST_DEBUG_OBJECT (queue, "last_stop updated to %" GST_TIME_FORMAT,
 795       GST_TIME_ARGS (timestamp));
 796
 797   gst_segment_set_last_stop (segment, GST_FORMAT_TIME, timestamp);
 798
 799   if (is_sink)
 800     queue->sink_tainted = TRUE;
 801   else
 802     queue->src_tainted = TRUE;
 803
 804   /* calc diff with other end */
 805   update_time_level (queue);
 806 }
 807
 808 static GstBufferListItem
 809 buffer_list_apply_time (GstBuffer ** buf, guint group, guint idx, gpointer data)
 810 {
 811   GstClockTime *timestamp = data;
 812
 813   GST_TRACE ("buffer %u in group %u has ts %" GST_TIME_FORMAT
 814       " duration %" GST_TIME_FORMAT, idx, group,
 815       GST_TIME_ARGS (GST_BUFFER_TIMESTAMP (*buf)),
 816       GST_TIME_ARGS (GST_BUFFER_DURATION (*buf)));
 817
 818   if (GST_BUFFER_TIMESTAMP_IS_VALID (*buf))
 819     *timestamp = GST_BUFFER_TIMESTAMP (*buf);
 820
 821   if (GST_BUFFER_DURATION_IS_VALID (*buf))
 822     *timestamp += GST_BUFFER_DURATION (*buf);
 823
 824   GST_TRACE ("ts now %" GST_TIME_FORMAT, GST_TIME_ARGS (*timestamp));
 825   return GST_BUFFER_LIST_CONTINUE;
 826 }
 827
 828 /* take a buffer list and update segment, updating the time level of the queue */
 829 static void
 830 apply_buffer_list (GstQueue2 * queue, GstBufferList * buffer_list,
 831     GstSegment * segment, gboolean is_sink)
 832 {
 833   GstClockTime timestamp;
 834
 835   /* if no timestamp is set, assume it's continuous with the previous time */
 836   timestamp = segment->last_stop;
 837
 838   gst_buffer_list_foreach (buffer_list, buffer_list_apply_time, &timestamp);
 839
 840   GST_DEBUG_OBJECT (queue, "last_stop updated to %" GST_TIME_FORMAT,
 841       GST_TIME_ARGS (timestamp));
 842
 843   gst_segment_set_last_stop (segment, GST_FORMAT_TIME, timestamp);
 844
 845   if (is_sink)
 846     queue->sink_tainted = TRUE;
 847   else
 848     queue->src_tainted = TRUE;
 849
 850   /* calc diff with other end */
 851   update_time_level (queue);
 852 }
 853
 854 static void
 855 update_buffering (GstQueue2 * queue)
 856 {
 857   gint64 percent;
 858   gboolean post = FALSE;
 859
 860   if (queue->high_percent <= 0)
 861     return;
 862
 863 #define GET_PERCENT(format,alt_max) ((queue->max_level.format) > 0 ? (queue->cur_level.format) * 100 / ((alt_max) > 0 ? MIN ((alt_max), (queue->max_level.format)) : (queue->max_level.format)) : 0)
 864
 865   if (queue->is_eos) {
 866     /* on EOS we are always 100% full, we set the var here so that it we can
 867      * reuse the logic below to stop buffering */
 868     percent = 100;
 869     GST_LOG_OBJECT (queue, "we are EOS");
 870   } else {
 871     /* figure out the percent we are filled, we take the max of all formats. */
 872
 873     if (!QUEUE_IS_USING_RING_BUFFER (queue)) {
 874       percent = GET_PERCENT (bytes, 0);
 875     } else {
 876       guint64 rb_size = queue->ring_buffer_max_size;
 877       percent = GET_PERCENT (bytes, rb_size);
 878     }
 879     percent = MAX (percent, GET_PERCENT (time, 0));
 880     percent = MAX (percent, GET_PERCENT (buffers, 0));
 881
 882     /* also apply the rate estimate when we need to */
 883     if (queue->use_rate_estimate)
 884       percent = MAX (percent, GET_PERCENT (rate_time, 0));
 885   }
 886
 887   if (queue->is_buffering) {
 888     post = TRUE;
 889     /* if we were buffering see if we reached the high watermark */
 890     if (percent >= queue->high_percent)
 891       queue->is_buffering = FALSE;
 892   } else {
 893     /* we were not buffering, check if we need to start buffering if we drop
 894      * below the low threshold */
 895     if (percent < queue->low_percent) {
 896       queue->is_buffering = TRUE;
 897       queue->buffering_iteration++;
 898       post = TRUE;
 899     }
 900   }
 901   if (post) {
 902     GstMessage *message;
 903     GstBufferingMode mode;
 904     gint64 buffering_left = -1;
 905
 906     /* scale to high percent so that it becomes the 100% mark */
 907     percent = percent * 100 / queue->high_percent;
 908     /* clip */
 909     if (percent > 100)
 910       percent = 100;
 911
 912     if (percent != queue->buffering_percent) {
 913       queue->buffering_percent = percent;
 914
 915       if (!QUEUE_IS_USING_QUEUE (queue)) {
 916         GstFormat fmt = GST_FORMAT_BYTES;
 917         gint64 duration;
 918
 919         if (QUEUE_IS_USING_RING_BUFFER (queue))
 920           mode = GST_BUFFERING_TIMESHIFT;
 921         else
 922           mode = GST_BUFFERING_DOWNLOAD;
 923
 924         if (queue->byte_in_rate > 0) {
 925           if (gst_pad_query_peer_duration (queue->sinkpad, &fmt, &duration))
 926             buffering_left =
 927                 (gdouble) ((duration -
 928                     queue->current->writing_pos) * 1000) / queue->byte_in_rate;
 929         } else {
 930           buffering_left = G_MAXINT64;
 931         }
 932       } else {
 933         mode = GST_BUFFERING_STREAM;
 934       }
 935
 936       GST_DEBUG_OBJECT (queue, "buffering %d percent", (gint) percent);
 937       message = gst_message_new_buffering (GST_OBJECT_CAST (queue),
 938           (gint) percent);
 939       gst_message_set_buffering_stats (message, mode,
 940           queue->byte_in_rate, queue->byte_out_rate, buffering_left);
 941
 942       gst_element_post_message (GST_ELEMENT_CAST (queue), message);
 943     }
 944   } else {
 945     GST_DEBUG_OBJECT (queue, "filled %d percent", (gint) percent);
 946   }
 947
 948 #undef GET_PERCENT
 949 }
 950
 951 static void
 952 reset_rate_timer (GstQueue2 * queue)
 953 {
 954   queue->bytes_in = 0;
 955   queue->bytes_out = 0;
 956   queue->byte_in_rate = 0.0;
 957   queue->byte_in_period = 0;
 958   queue->byte_out_rate = 0.0;
 959   queue->last_in_elapsed = 0.0;
 960   queue->last_out_elapsed = 0.0;
 961   queue->in_timer_started = FALSE;
 962   queue->out_timer_started = FALSE;
 963 }
 964
 965 /* the interval in seconds to recalculate the rate */
 966 #define RATE_INTERVAL    0.2
 967 /* Tuning for rate estimation. We use a large window for the input rate because
 968  * it should be stable when connected to a network. The output rate is less
 969  * stable (the elements preroll, queues behind a demuxer fill, ...) and should
 970  * therefore adapt more quickly.
 971  * However, initial input rate may be subject to a burst, and should therefore
 972  * initially also adapt more quickly to changes, and only later on give higher
 973  * weight to previous values. */
 974 #define AVG_IN(avg,val,w1,w2)  ((avg) * (w1) + (val) * (w2)) / ((w1) + (w2))
 975 #define AVG_OUT(avg,val) ((avg) * 3.0 + (val)) / 4.0
 976
 977 static void
 978 update_in_rates (GstQueue2 * queue)
 979 {
 980   gdouble elapsed, period;
 981   gdouble byte_in_rate;
 982
 983   if (!queue->in_timer_started) {
 984     queue->in_timer_started = TRUE;
 985     g_timer_start (queue->in_timer);
 986     return;
 987   }
 988
 989   elapsed = g_timer_elapsed (queue->in_timer, NULL);
 990
 991   /* recalc after each interval. */
 992   if (queue->last_in_elapsed + RATE_INTERVAL < elapsed) {
 993     period = elapsed - queue->last_in_elapsed;
 994
 995     GST_DEBUG_OBJECT (queue,
 996         "rates: period %f, in %" G_GUINT64_FORMAT ", global period %f",
 997         period, queue->bytes_in, queue->byte_in_period);
 998
 999     byte_in_rate = queue->bytes_in / period;
1000
1001     if (queue->byte_in_rate == 0.0)
1002       queue->byte_in_rate = byte_in_rate;
1003     else
1004       queue->byte_in_rate = AVG_IN (queue->byte_in_rate, byte_in_rate,
1005           (double) queue->byte_in_period, period);
1006
1007     /* another data point, cap at 16 for long time running average */
1008     if (queue->byte_in_period < 16 * RATE_INTERVAL)
1009       queue->byte_in_period += period;
1010
1011     /* reset the values to calculate rate over the next interval */
1012     queue->last_in_elapsed = elapsed;
1013     queue->bytes_in = 0;
1014   }
1015
1016   if (queue->byte_in_rate > 0.0) {
1017     queue->cur_level.rate_time =
1018         queue->cur_level.bytes / queue->byte_in_rate * GST_SECOND;
1019   }
1020   GST_DEBUG_OBJECT (queue, "rates: in %f, time %" GST_TIME_FORMAT,
1021       queue->byte_in_rate, GST_TIME_ARGS (queue->cur_level.rate_time));
1022 }
1023
1024 static void
1025 update_out_rates (GstQueue2 * queue)
1026 {
1027   gdouble elapsed, period;
1028   gdouble byte_out_rate;
1029
1030   if (!queue->out_timer_started) {
1031     queue->out_timer_started = TRUE;
1032     g_timer_start (queue->out_timer);
1033     return;
1034   }
1035
1036   elapsed = g_timer_elapsed (queue->out_timer, NULL);
1037
1038   /* recalc after each interval. */
1039   if (queue->last_out_elapsed + RATE_INTERVAL < elapsed) {
1040     period = elapsed - queue->last_out_elapsed;
1041
1042     GST_DEBUG_OBJECT (queue,
1043         "rates: period %f, out %" G_GUINT64_FORMAT, period, queue->bytes_out);
1044
1045     byte_out_rate = queue->bytes_out / period;
1046
1047     if (queue->byte_out_rate == 0.0)
1048       queue->byte_out_rate = byte_out_rate;
1049     else
1050       queue->byte_out_rate = AVG_OUT (queue->byte_out_rate, byte_out_rate);
1051
1052     /* reset the values to calculate rate over the next interval */
1053     queue->last_out_elapsed = elapsed;
1054     queue->bytes_out = 0;
1055   }
1056   if (queue->byte_in_rate > 0.0) {
1057     queue->cur_level.rate_time =
1058         queue->cur_level.bytes / queue->byte_in_rate * GST_SECOND;
1059   }
1060   GST_DEBUG_OBJECT (queue, "rates: out %f, time %" GST_TIME_FORMAT,
1061       queue->byte_out_rate, GST_TIME_ARGS (queue->cur_level.rate_time));
1062 }
1063
1064 static void
1065 update_cur_pos (GstQueue2 * queue, GstQueue2Range * range, guint64 pos)
1066 {
1067   guint64 reading_pos, max_reading_pos;
1068
1069   reading_pos = pos;
1070   max_reading_pos = range->max_reading_pos;
1071
1072   max_reading_pos = MAX (max_reading_pos, reading_pos);
1073
1074   GST_DEBUG_OBJECT (queue,
1075       "updating max_reading_pos from %" G_GUINT64_FORMAT " to %"
1076       G_GUINT64_FORMAT, range->max_reading_pos, max_reading_pos);
1077   range->max_reading_pos = max_reading_pos;
1078
1079   update_cur_level (queue, range);
1080 }
1081
1082 static gboolean
1083 perform_seek_to_offset (GstQueue2 * queue, guint64 offset)
1084 {
1085   GstEvent *event;
1086   gboolean res;
1087
1088   /* until we receive the FLUSH_STOP from this seek, we skip data */
1089   queue->seeking = TRUE;
1090   GST_QUEUE2_MUTEX_UNLOCK (queue);
1091
1092   GST_DEBUG_OBJECT (queue, "Seeking to %" G_GUINT64_FORMAT, offset);
1093
1094   event =
1095       gst_event_new_seek (1.0, GST_FORMAT_BYTES,
1096       GST_SEEK_FLAG_FLUSH | GST_SEEK_FLAG_ACCURATE, GST_SEEK_TYPE_SET, offset,
1097       GST_SEEK_TYPE_NONE, -1);
1098
1099   res = gst_pad_push_event (queue->sinkpad, event);
1100   GST_QUEUE2_MUTEX_LOCK (queue);
1101
1102   if (res)
1103     queue->current = add_range (queue, offset);
1104
1105   return res;
1106 }
1107
1108 /* see if there is enough data in the file to read a full buffer */
1109 static gboolean
1110 gst_queue2_have_data (GstQueue2 * queue, guint64 offset, guint length)
1111 {
1112   GstQueue2Range *range;
1113
1114   GST_DEBUG_OBJECT (queue, "looking for offset %" G_GUINT64_FORMAT ", len %u",
1115       offset, length);
1116
1117   if ((range = find_range (queue, offset))) {
1118     if (queue->current != range) {
1119       GST_DEBUG_OBJECT (queue, "switching ranges, do seek to range position");
1120       perform_seek_to_offset (queue, range->writing_pos);
1121     }
1122
1123     GST_INFO_OBJECT (queue, "cur_level.bytes %u (max %" G_GUINT64_FORMAT ")",
1124         queue->cur_level.bytes, QUEUE_MAX_BYTES (queue));
1125
1126     /* we have a range for offset */
1127     GST_DEBUG_OBJECT (queue,
1128         "we have a range %p, offset %" G_GUINT64_FORMAT ", writing_pos %"
1129         G_GUINT64_FORMAT, range, range->offset, range->writing_pos);
1130
1131     if (!QUEUE_IS_USING_RING_BUFFER (queue) && queue->is_eos)
1132       return TRUE;
1133
1134     if (offset + length <= range->writing_pos)
1135       return TRUE;
1136     else
1137       GST_DEBUG_OBJECT (queue,
1138           "Need more data (%" G_GUINT64_FORMAT " bytes more)",
1139           (offset + length) - range->writing_pos);
1140
1141   } else {
1142     GST_INFO_OBJECT (queue, "not found in any range off %" G_GUINT64_FORMAT
1143         " len %u", offset, length);
1144     /* we don't have the range, see how far away we are */
1145     if (!queue->is_eos && queue->current) {
1146       /* FIXME, find a good threshold based on the incoming rate. */
1147       guint64 threshold = 1024 * 512;
1148
1149       if (QUEUE_IS_USING_RING_BUFFER (queue)) {
1150         guint64 distance;
1151
1152         distance = QUEUE_MAX_BYTES (queue) - queue->cur_level.bytes;
1153         /* don't wait for the complete buffer to fill */
1154         distance = MIN (distance, threshold);
1155
1156         if (offset >= queue->current->offset && offset <=
1157             queue->current->writing_pos + distance) {
1158           GST_INFO_OBJECT (queue,
1159               "requested data is within range, wait for data");
1160           return FALSE;
1161         }
1162       } else if (offset < queue->current->writing_pos + threshold) {
1163         update_cur_pos (queue, queue->current, offset + length);
1164         GST_INFO_OBJECT (queue, "wait for data");
1165         return FALSE;
1166       }
1167     }
1168
1169     /* too far away, do a seek */
1170     perform_seek_to_offset (queue, offset);
1171   }
1172
1173   return FALSE;
1174 }
1175
1176 #ifdef HAVE_FSEEKO
1177 #define FSEEK_FILE(file,offset)  (fseeko (file, (off_t) offset, SEEK_SET) != 0)
1178 #elif defined (G_OS_UNIX) || defined (G_OS_WIN32)
1179 #define FSEEK_FILE(file,offset)  (lseek (fileno (file), (off_t) offset, SEEK_SET) == (off_t) -1)
1180 #else
1181 #define FSEEK_FILE(file,offset)  (fseek (file, offset, SEEK_SET) != 0)
1182 #endif
1183
1184 static gint64
1185 gst_queue2_read_data_at_offset (GstQueue2 * queue, guint64 offset, guint length,
1186     guint8 * dst)
1187 {
1188   guint8 *ring_buffer;
1189   size_t res;
1190
1191   ring_buffer = queue->ring_buffer;
1192
1193   if (QUEUE_IS_USING_TEMP_FILE (queue) && FSEEK_FILE (queue->temp_file, offset))
1194     goto seek_failed;
1195
1196   /* this should not block */
1197   GST_LOG_OBJECT (queue, "Reading %d bytes from offset %" G_GUINT64_FORMAT,
1198       length, offset);
1199   if (QUEUE_IS_USING_TEMP_FILE (queue)) {
1200     res = fread (dst, 1, length, queue->temp_file);
1201   } else {
1202     memcpy (dst, ring_buffer + offset, length);
1203     res = length;
1204   }
1205
1206   GST_LOG_OBJECT (queue, "read %" G_GSIZE_FORMAT " bytes", res);
1207
1208   if (G_UNLIKELY (res < length)) {
1209     if (!QUEUE_IS_USING_TEMP_FILE (queue))
1210       goto could_not_read;
1211     /* check for errors or EOF */
1212     if (ferror (queue->temp_file))
1213       goto could_not_read;
1214     if (feof (queue->temp_file) && length > 0)
1215       goto eos;
1216   }
1217
1218   return res;
1219
1220 seek_failed:
1221   {
1222     GST_ELEMENT_ERROR (queue, RESOURCE, SEEK, (NULL), GST_ERROR_SYSTEM);
1223     return GST_FLOW_ERROR;
1224   }
1225 could_not_read:
1226   {
1227     GST_ELEMENT_ERROR (queue, RESOURCE, READ, (NULL), GST_ERROR_SYSTEM);
1228     return GST_FLOW_ERROR;
1229   }
1230 eos:
1231   {
1232     GST_DEBUG ("non-regular file hits EOS");
1233     return GST_FLOW_UNEXPECTED;
1234   }
1235 }
1236
1237 static GstFlowReturn
1238 gst_queue2_create_read (GstQueue2 * queue, guint64 offset, guint length,
1239     GstBuffer ** buffer)
1240 {
1241   GstBuffer *buf;
1242   guint8 *data;
1243   guint64 file_offset;
1244   guint block_length, remaining, read_length;
1245   gint64 read_return;
1246   guint64 rb_size;
1247   guint64 max_size;
1248   guint64 rpos;
1249
1250   /* allocate the output buffer of the requested size */
1251   buf = gst_buffer_new_and_alloc (length);
1252   data = GST_BUFFER_DATA (buf);
1253
1254   GST_DEBUG_OBJECT (queue, "Reading %u bytes from %" G_GUINT64_FORMAT, length,
1255       offset);
1256
1257   rpos = offset;
1258   rb_size = queue->ring_buffer_max_size;
1259   max_size = QUEUE_MAX_BYTES (queue);
1260
1261   remaining = length;
1262   while (remaining > 0) {
1263     /* configure how much/whether to read */
1264     if (!gst_queue2_have_data (queue, rpos, remaining)) {
1265       read_length = 0;
1266
1267       if (QUEUE_IS_USING_RING_BUFFER (queue)) {
1268         guint64 level;
1269
1270         /* calculate how far away the offset is */
1271         if (queue->current->writing_pos > rpos)
1272           level = queue->current->writing_pos - rpos;
1273         else
1274           level = 0;
1275
1276         GST_DEBUG_OBJECT (queue,
1277             "reading %" G_GUINT64_FORMAT ", writing %" G_GUINT64_FORMAT
1278             ", level %" G_GUINT64_FORMAT ", max %" G_GUINT64_FORMAT,
1279             rpos, queue->current->writing_pos, level, max_size);
1280
1281         if (level >= max_size) {
1282           /* we don't have the data but if we have a ring buffer that is full, we
1283            * need to read */
1284           GST_DEBUG_OBJECT (queue,
1285               "ring buffer full, reading QUEUE_MAX_BYTES %"
1286               G_GUINT64_FORMAT " bytes", max_size);
1287           read_length = max_size;
1288         } else if (queue->is_eos) {
1289           /* won't get any more data so read any data we have */
1290           if (level) {
1291             GST_DEBUG_OBJECT (queue,
1292                 "EOS hit but read %" G_GUINT64_FORMAT " bytes that we have",
1293                 level);
1294             read_length = level;
1295             remaining = level;
1296             length = level;
1297           } else {
1298             GST_DEBUG_OBJECT (queue,
1299                 "EOS hit and we don't have any requested data");
1300             gst_buffer_unref (buf);
1301             return GST_FLOW_UNEXPECTED;
1302           }
1303         }
1304       }
1305
1306       if (read_length == 0) {
1307         if (QUEUE_IS_USING_RING_BUFFER (queue)
1308             && queue->current->max_reading_pos > rpos) {
1309           /* protect cached data (data between offset and max_reading_pos)
1310            * and update current level */
1311           GST_DEBUG_OBJECT (queue,
1312               "protecting cached data [%" G_GUINT64_FORMAT "-%" G_GUINT64_FORMAT
1313               "]", rpos, queue->current->max_reading_pos);
1314           queue->current->max_reading_pos = rpos;
1315           update_cur_level (queue, queue->current);
1316         }
1317         GST_DEBUG_OBJECT (queue, "waiting for add");
1318         GST_QUEUE2_WAIT_ADD_CHECK (queue, queue->srcresult, out_flushing);
1319         continue;
1320       }
1321     } else {
1322       /* we have the requested data so read it */
1323       read_length = remaining;
1324     }
1325
1326     /* set range reading_pos to actual reading position for this read */
1327     queue->current->reading_pos = rpos;
1328
1329     /* configure how much and from where to read */
1330     if (QUEUE_IS_USING_RING_BUFFER (queue)) {
1331       file_offset =
1332           (queue->current->rb_offset + (rpos -
1333               queue->current->offset)) % rb_size;
1334       if (file_offset + read_length > rb_size) {
1335         block_length = rb_size - file_offset;
1336       } else {
1337         block_length = read_length;
1338       }
1339     } else {
1340       file_offset = rpos;
1341       block_length = read_length;
1342     }
1343
1344     /* while we still have data to read, we loop */
1345     while (read_length > 0) {
1346       read_return =
1347           gst_queue2_read_data_at_offset (queue, file_offset, block_length,
1348           data);
1349       if (read_return < 0)
1350         goto read_error;
1351
1352       file_offset += read_return;
1353       if (QUEUE_IS_USING_RING_BUFFER (queue))
1354         file_offset %= rb_size;
1355
1356       data += read_return;
1357       read_length -= read_return;
1358       block_length = read_length;
1359       remaining -= read_return;
1360
1361       rpos = (queue->current->reading_pos += read_return);
1362       update_cur_pos (queue, queue->current, queue->current->reading_pos);
1363     }
1364     GST_QUEUE2_SIGNAL_DEL (queue);
1365     GST_DEBUG_OBJECT (queue, "%u bytes left to read", remaining);
1366   }
1367
1368   GST_BUFFER_SIZE (buf) = length;
1369   GST_BUFFER_OFFSET (buf) = offset;
1370   GST_BUFFER_OFFSET_END (buf) = offset + length;
1371
1372   *buffer = buf;
1373
1374   return GST_FLOW_OK;
1375
1376   /* ERRORS */
1377 out_flushing:
1378   {
1379     GST_DEBUG_OBJECT (queue, "we are flushing");
1380     gst_buffer_unref (buf);
1381     return GST_FLOW_WRONG_STATE;
1382   }
1383 read_error:
1384   {
1385     GST_DEBUG_OBJECT (queue, "we have a read error");
1386     gst_buffer_unref (buf);
1387     return read_return;
1388   }
1389 }
1390
1391 /* should be called with QUEUE_LOCK */
1392 static GstMiniObject *
1393 gst_queue2_read_item_from_file (GstQueue2 * queue)
1394 {
1395   GstMiniObject *item;
1396
1397   if (queue->starting_segment != NULL) {
1398     item = GST_MINI_OBJECT_CAST (queue->starting_segment);
1399     queue->starting_segment = NULL;
1400   } else {
1401     GstFlowReturn ret;
1402     GstBuffer *buffer;
1403     guint64 reading_pos;
1404
1405     reading_pos = queue->current->reading_pos;
1406
1407     ret =
1408         gst_queue2_create_read (queue, reading_pos, DEFAULT_BUFFER_SIZE,
1409         &buffer);
1410
1411     switch (ret) {
1412       case GST_FLOW_OK:
1413         item = GST_MINI_OBJECT_CAST (buffer);
1414         break;
1415       case GST_FLOW_UNEXPECTED:
1416         item = GST_MINI_OBJECT_CAST (gst_event_new_eos ());
1417         break;
1418       default:
1419         item = NULL;
1420         break;
1421     }
1422   }
1423   return item;
1424 }
1425
1426 /* must be called with MUTEX_LOCK. Will briefly release the lock when notifying
1427  * the temp filename. */
1428 static gboolean
1429 gst_queue2_open_temp_location_file (GstQueue2 * queue)
1430 {
1431   gint fd = -1;
1432   gchar *name = NULL;
1433
1434   if (queue->temp_file)
1435     goto already_opened;
1436
1437   GST_DEBUG_OBJECT (queue, "opening temp file %s", queue->temp_template);
1438
1439   /* we have two cases:
1440    * - temp_location was set to something !NULL (Deprecated). in this case we
1441    *   open the specified filename.
1442    * - temp_template was set, allocate a filename and open that filename
1443    */
1444   if (!queue->temp_location_set) {
1445     /* nothing to do */
1446     if (queue->temp_template == NULL)
1447       goto no_directory;
1448
1449     /* make copy of the template, we don't want to change this */
1450     name = g_strdup (queue->temp_template);
1451     fd = g_mkstemp (name);
1452     if (fd == -1)
1453       goto mkstemp_failed;
1454
1455     /* open the file for update/writing */
1456     queue->temp_file = fdopen (fd, "wb+");
1457     /* error creating file */
1458     if (queue->temp_file == NULL)
1459       goto open_failed;
1460
1461     g_free (queue->temp_location);
1462     queue->temp_location = name;
1463
1464     GST_QUEUE2_MUTEX_UNLOCK (queue);
1465
1466     /* we can't emit the notify with the lock */
1467     g_object_notify (G_OBJECT (queue), "temp-location");
1468
1469     GST_QUEUE2_MUTEX_LOCK (queue);
1470   } else {
1471     /* open the file for update/writing, this is deprecated but we still need to
1472      * support it for API/ABI compatibility */
1473     queue->temp_file = g_fopen (queue->temp_location, "wb+");
1474     /* error creating file */
1475     if (queue->temp_file == NULL)
1476       goto open_failed;
1477   }
1478   GST_DEBUG_OBJECT (queue, "opened temp file %s", queue->temp_template);
1479
1480   return TRUE;
1481
1482   /* ERRORS */
1483 already_opened:
1484   {
1485     GST_DEBUG_OBJECT (queue, "temp file was already open");
1486     return TRUE;
1487   }
1488 no_directory:
1489   {
1490     GST_ELEMENT_ERROR (queue, RESOURCE, NOT_FOUND,
1491         (_("No Temp directory specified.")), (NULL));
1492     return FALSE;
1493   }
1494 mkstemp_failed:
1495   {
1496     GST_ELEMENT_ERROR (queue, RESOURCE, OPEN_READ,
1497         (_("Could not create temp file \"%s\"."), queue->temp_template),
1498         GST_ERROR_SYSTEM);
1499     g_free (name);
1500     return FALSE;
1501   }
1502 open_failed:
1503   {
1504     GST_ELEMENT_ERROR (queue, RESOURCE, OPEN_READ,
1505         (_("Could not open file \"%s\" for reading."), name), GST_ERROR_SYSTEM);
1506     g_free (name);
1507     if (fd != -1)
1508       close (fd);
1509     return FALSE;
1510   }
1511 }
1512
1513 static void
1514 gst_queue2_close_temp_location_file (GstQueue2 * queue)
1515 {
1516   /* nothing to do */
1517   if (queue->temp_file == NULL)
1518     return;
1519
1520   GST_DEBUG_OBJECT (queue, "closing temp file");
1521
1522   fflush (queue->temp_file);
1523   fclose (queue->temp_file);
1524
1525   if (queue->temp_remove)
1526     remove (queue->temp_location);
1527
1528   queue->temp_file = NULL;
1529   clean_ranges (queue);
1530 }
1531
1532 static void
1533 gst_queue2_flush_temp_file (GstQueue2 * queue)
1534 {
1535   if (queue->temp_file == NULL)
1536     return;
1537
1538   GST_DEBUG_OBJECT (queue, "flushing temp file");
1539
1540   queue->temp_file = g_freopen (queue->temp_location, "wb+", queue->temp_file);
1541 }
1542
1543 static void
1544 gst_queue2_locked_flush (GstQueue2 * queue)
1545 {
1546   if (!QUEUE_IS_USING_QUEUE (queue)) {
1547     if (QUEUE_IS_USING_TEMP_FILE (queue))
1548       gst_queue2_flush_temp_file (queue);
1549     init_ranges (queue);
1550   } else {
1551     while (!g_queue_is_empty (&queue->queue)) {
1552       GstMiniObject *data = g_queue_pop_head (&queue->queue);
1553
1554       /* Then lose another reference because we are supposed to destroy that
1555          data when flushing */
1556       gst_mini_object_unref (data);
1557     }
1558   }
1559   GST_QUEUE2_CLEAR_LEVEL (queue->cur_level);
1560   gst_segment_init (&queue->sink_segment, GST_FORMAT_TIME);
1561   gst_segment_init (&queue->src_segment, GST_FORMAT_TIME);
1562   queue->sinktime = queue->srctime = GST_CLOCK_TIME_NONE;
1563   queue->sink_tainted = queue->src_tainted = TRUE;
1564   if (queue->starting_segment != NULL)
1565     gst_event_unref (queue->starting_segment);
1566   queue->starting_segment = NULL;
1567   queue->segment_event_received = FALSE;
1568
1569   /* we deleted a lot of something */
1570   GST_QUEUE2_SIGNAL_DEL (queue);
1571 }
1572
1573 static gboolean
1574 gst_queue2_wait_free_space (GstQueue2 * queue)
1575 {
1576   /* We make space available if we're "full" according to whatever
1577    * the user defined as "full". */
1578   if (gst_queue2_is_filled (queue)) {
1579     gboolean started;
1580
1581     /* pause the timer while we wait. The fact that we are waiting does not mean
1582      * the byterate on the input pad is lower */
1583     if ((started = queue->in_timer_started))
1584       g_timer_stop (queue->in_timer);
1585
1586     GST_CAT_DEBUG_OBJECT (queue_dataflow, queue,
1587         "queue is full, waiting for free space");
1588     do {
1589       /* Wait for space to be available, we could be unlocked because of a flush. */
1590       GST_QUEUE2_WAIT_DEL_CHECK (queue, queue->sinkresult, out_flushing);
1591     }
1592     while (gst_queue2_is_filled (queue));
1593
1594     /* and continue if we were running before */
1595     if (started)
1596       g_timer_continue (queue->in_timer);
1597   }
1598   return TRUE;
1599
1600   /* ERRORS */
1601 out_flushing:
1602   {
1603     GST_CAT_DEBUG_OBJECT (queue_dataflow, queue, "queue is flushing");
1604     return FALSE;
1605   }
1606 }
1607
1608 static gboolean
1609 gst_queue2_create_write (GstQueue2 * queue, GstBuffer * buffer)
1610 {
1611   guint8 *data, *ring_buffer;
1612   guint size, rb_size;
1613   guint64 writing_pos, new_writing_pos;
1614   GstQueue2Range *range, *prev, *next;
1615
1616   if (QUEUE_IS_USING_RING_BUFFER (queue))
1617     writing_pos = queue->current->rb_writing_pos;
1618   else
1619     writing_pos = queue->current->writing_pos;
1620   ring_buffer = queue->ring_buffer;
1621   rb_size = queue->ring_buffer_max_size;
1622
1623   size = GST_BUFFER_SIZE (buffer);
1624   data = GST_BUFFER_DATA (buffer);
1625
1626   GST_DEBUG_OBJECT (queue, "Writing %u bytes to %" G_GUINT64_FORMAT, size,
1627       GST_BUFFER_OFFSET (buffer));
1628
1629   while (size > 0) {
1630     guint to_write;
1631
1632     if (QUEUE_IS_USING_RING_BUFFER (queue)) {
1633       gint64 space;
1634
1635       /* calculate the space in the ring buffer not used by data from
1636        * the current range */
1637       while (QUEUE_MAX_BYTES (queue) <= queue->cur_level.bytes) {
1638         /* wait until there is some free space */
1639         GST_QUEUE2_WAIT_DEL_CHECK (queue, queue->sinkresult, out_flushing);
1640       }
1641       /* get the amount of space we have */
1642       space = QUEUE_MAX_BYTES (queue) - queue->cur_level.bytes;
1643
1644       /* calculate if we need to split or if we can write the entire
1645        * buffer now */
1646       to_write = MIN (size, space);
1647
1648       /* the writing position in the ring buffer after writing (part
1649        * or all of) the buffer */
1650       new_writing_pos = (writing_pos + to_write) % rb_size;
1651
1652       prev = NULL;
1653       range = queue->ranges;
1654
1655       /* if we need to overwrite data in the ring buffer, we need to
1656        * update the ranges
1657        *
1658        * warning: this code is complicated and includes some
1659        * simplifications - pen, paper and diagrams for the cases
1660        * recommended! */
1661       while (range) {
1662         guint64 range_data_start, range_data_end;
1663         GstQueue2Range *range_to_destroy = NULL;
1664
1665         range_data_start = range->rb_offset;
1666         range_data_end = range->rb_writing_pos;
1667
1668         /* handle the special case where the range has no data in it */
1669         if (range->writing_pos == range->offset) {
1670           if (range != queue->current) {
1671             GST_DEBUG_OBJECT (queue,
1672                 "Removing range: offset %" G_GUINT64_FORMAT ", wpos %"
1673                 G_GUINT64_FORMAT, range->offset, range->writing_pos);
1674             /* remove range */
1675             range_to_destroy = range;
1676             if (prev)
1677               prev->next = range->next;
1678           }
1679           goto next_range;
1680         }
1681
1682         if (range_data_end > range_data_start) {
1683           if (writing_pos >= range_data_end && new_writing_pos >= writing_pos)
1684             goto next_range;
1685
1686           if (new_writing_pos > range_data_start) {
1687             if (new_writing_pos >= range_data_end) {
1688               GST_DEBUG_OBJECT (queue,
1689                   "Removing range: offset %" G_GUINT64_FORMAT ", wpos %"
1690                   G_GUINT64_FORMAT, range->offset, range->writing_pos);
1691               /* remove range */
1692               range_to_destroy = range;
1693               if (prev)
1694                 prev->next = range->next;
1695             } else {
1696               GST_DEBUG_OBJECT (queue,
1697                   "advancing offsets from %" G_GUINT64_FORMAT " (%"
1698                   G_GUINT64_FORMAT ") to %" G_GUINT64_FORMAT " (%"
1699                   G_GUINT64_FORMAT ")", range->offset, range->rb_offset,
1700                   range->offset + new_writing_pos - range_data_start,
1701                   new_writing_pos);
1702               range->offset += (new_writing_pos - range_data_start);
1703               range->rb_offset = new_writing_pos;
1704             }
1705           }
1706         } else {
1707           guint64 new_wpos_virt = writing_pos + to_write;
1708
1709           if (new_wpos_virt <= range_data_start)
1710             goto next_range;
1711
1712           if (new_wpos_virt > rb_size && new_writing_pos >= range_data_end) {
1713             GST_DEBUG_OBJECT (queue,
1714                 "Removing range: offset %" G_GUINT64_FORMAT ", wpos %"
1715                 G_GUINT64_FORMAT, range->offset, range->writing_pos);
1716             /* remove range */
1717             range_to_destroy = range;
1718             if (prev)
1719               prev->next = range->next;
1720           } else {
1721             GST_DEBUG_OBJECT (queue,
1722                 "advancing offsets from %" G_GUINT64_FORMAT " (%"
1723                 G_GUINT64_FORMAT ") to %" G_GUINT64_FORMAT " (%"
1724                 G_GUINT64_FORMAT ")", range->offset, range->rb_offset,
1725                 range->offset + new_writing_pos - range_data_start,
1726                 new_writing_pos);
1727             range->offset += (new_wpos_virt - range_data_start);
1728             range->rb_offset = new_writing_pos;
1729           }
1730         }
1731
1732       next_range:
1733         if (!range_to_destroy)
1734           prev = range;
1735
1736         range = range->next;
1737         if (range_to_destroy) {
1738           if (range_to_destroy == queue->ranges)
1739             queue->ranges = range;
1740           g_slice_free (GstQueue2Range, range_to_destroy);
1741           range_to_destroy = NULL;
1742         }
1743       }
1744     } else {
1745       to_write = size;
1746       new_writing_pos = writing_pos + to_write;
1747     }
1748
1749     if (QUEUE_IS_USING_TEMP_FILE (queue)
1750         && FSEEK_FILE (queue->temp_file, writing_pos))
1751       goto seek_failed;
1752
1753     if (new_writing_pos > writing_pos) {
1754       GST_INFO_OBJECT (queue,
1755           "writing %u bytes to range [%" G_GUINT64_FORMAT "-%" G_GUINT64_FORMAT
1756           "] (rb wpos %" G_GUINT64_FORMAT ")", to_write, queue->current->offset,
1757           queue->current->writing_pos, queue->current->rb_writing_pos);
1758       /* either not using ring buffer or no wrapping, just write */
1759       if (QUEUE_IS_USING_TEMP_FILE (queue)) {
1760         if (fwrite (data, to_write, 1, queue->temp_file) != 1)
1761           goto handle_error;
1762       } else {
1763         memcpy (ring_buffer + writing_pos, data, to_write);
1764       }
1765
1766       if (!QUEUE_IS_USING_RING_BUFFER (queue)) {
1767         /* try to merge with next range */
1768         while ((next = queue->current->next)) {
1769           GST_INFO_OBJECT (queue,
1770               "checking merge with next range %" G_GUINT64_FORMAT " < %"
1771               G_GUINT64_FORMAT, new_writing_pos, next->offset);
1772           if (new_writing_pos < next->offset)
1773             break;
1774
1775           GST_DEBUG_OBJECT (queue, "merging ranges %" G_GUINT64_FORMAT,
1776               next->writing_pos);
1777
1778           /* remove the group, we could choose to not read the data in this range
1779            * again. This would involve us doing a seek to the current writing position
1780            * in the range. FIXME, It would probably make sense to do a seek when there
1781            * is a lot of data in the range we merged with to avoid reading it all
1782            * again. */
1783           queue->current->next = next->next;
1784           g_slice_free (GstQueue2Range, next);
1785
1786           debug_ranges (queue);
1787         }
1788         goto update_and_signal;
1789       }
1790     } else {
1791       /* wrapping */
1792       guint block_one, block_two;
1793
1794       block_one = rb_size - writing_pos;
1795       block_two = to_write - block_one;
1796
1797       if (block_one > 0) {
1798         GST_INFO_OBJECT (queue, "writing %u bytes", block_one);
1799         /* write data to end of ring buffer */
1800         if (QUEUE_IS_USING_TEMP_FILE (queue)) {
1801           if (fwrite (data, block_one, 1, queue->temp_file) != 1)
1802             goto handle_error;
1803         } else {
1804           memcpy (ring_buffer + writing_pos, data, block_one);
1805         }
1806       }
1807
1808       if (QUEUE_IS_USING_TEMP_FILE (queue) && FSEEK_FILE (queue->temp_file, 0))
1809         goto seek_failed;
1810
1811       if (block_two > 0) {
1812         GST_INFO_OBJECT (queue, "writing %u bytes", block_two);
1813         if (QUEUE_IS_USING_TEMP_FILE (queue)) {
1814           if (fwrite (data + block_one, block_two, 1, queue->temp_file) != 1)
1815             goto handle_error;
1816         } else {
1817           memcpy (ring_buffer, data + block_one, block_two);
1818         }
1819       }
1820     }
1821
1822   update_and_signal:
1823     /* update the writing positions */
1824     size -= to_write;
1825     GST_INFO_OBJECT (queue,
1826         "wrote %u bytes to %" G_GUINT64_FORMAT " (%u bytes remaining to write)",
1827         to_write, writing_pos, size);
1828
1829     if (QUEUE_IS_USING_RING_BUFFER (queue)) {
1830       data += to_write;
1831       queue->current->writing_pos += to_write;
1832       queue->current->rb_writing_pos = writing_pos = new_writing_pos;
1833     } else {
1834       queue->current->writing_pos = writing_pos = new_writing_pos;
1835     }
1836     update_cur_level (queue, queue->current);
1837
1838     /* update the buffering status */
1839     if (queue->use_buffering)
1840       update_buffering (queue);
1841
1842     GST_INFO_OBJECT (queue, "cur_level.bytes %u (max %" G_GUINT64_FORMAT ")",
1843         queue->cur_level.bytes, QUEUE_MAX_BYTES (queue));
1844
1845     GST_QUEUE2_SIGNAL_ADD (queue);
1846   };
1847
1848   return TRUE;
1849
1850   /* ERRORS */
1851 out_flushing:
1852   {
1853     GST_DEBUG_OBJECT (queue, "we are flushing");
1854     /* FIXME - GST_FLOW_UNEXPECTED ? */
1855     return FALSE;
1856   }
1857 seek_failed:
1858   {
1859     GST_ELEMENT_ERROR (queue, RESOURCE, SEEK, (NULL), GST_ERROR_SYSTEM);
1860     return FALSE;
1861   }
1862 handle_error:
1863   {
1864     switch (errno) {
1865       case ENOSPC:{
1866         GST_ELEMENT_ERROR (queue, RESOURCE, NO_SPACE_LEFT, (NULL), (NULL));
1867         break;
1868       }
1869       default:{
1870         GST_ELEMENT_ERROR (queue, RESOURCE, WRITE,
1871             (_("Error while writing to download file.")),
1872             ("%s", g_strerror (errno)));
1873       }
1874     }
1875     return FALSE;
1876   }
1877 }
1878
1879 static GstBufferListItem
1880 buffer_list_create_write (GstBuffer ** buf, guint group, guint idx, gpointer q)
1881 {
1882   GstQueue2 *queue = q;
1883
1884   GST_TRACE_OBJECT (queue, "writing buffer %u in group %u of size %u bytes",
1885       idx, group, GST_BUFFER_SIZE (*buf));
1886
1887   if (!gst_queue2_create_write (queue, *buf)) {
1888     GST_INFO_OBJECT (queue, "create_write() returned FALSE, bailing out");
1889     return GST_BUFFER_LIST_END;
1890   }
1891
1892   return GST_BUFFER_LIST_CONTINUE;
1893 }
1894
1895 static GstBufferListItem
1896 buffer_list_calc_size (GstBuffer ** buf, guint group, guint idx, gpointer data)
1897 {
1898   guint *p_size = data;
1899   guint buf_size;
1900
1901   buf_size = GST_BUFFER_SIZE (*buf);
1902   GST_TRACE ("buffer %u in group %u has size %u", idx, group, buf_size);
1903   *p_size += buf_size;
1904
1905   return GST_BUFFER_LIST_CONTINUE;
1906 }
1907
1908 /* enqueue an item an update the level stats */
1909 static void
1910 gst_queue2_locked_enqueue (GstQueue2 * queue, gpointer item,
1911     GstQueue2ItemType item_type)
1912 {
1913   if (item_type == GST_QUEUE2_ITEM_TYPE_BUFFER) {
1914     GstBuffer *buffer;
1915     guint size;
1916
1917     buffer = GST_BUFFER_CAST (item);
1918     size = GST_BUFFER_SIZE (buffer);
1919
1920     /* add buffer to the statistics */
1921     if (QUEUE_IS_USING_QUEUE (queue)) {
1922       queue->cur_level.buffers++;
1923       queue->cur_level.bytes += size;
1924     }
1925     queue->bytes_in += size;
1926
1927     /* apply new buffer to segment stats */
1928     apply_buffer (queue, buffer, &queue->sink_segment, TRUE);
1929     /* update the byterate stats */
1930     update_in_rates (queue);
1931
1932     if (!QUEUE_IS_USING_QUEUE (queue)) {
1933       /* FIXME - check return value? */
1934       gst_queue2_create_write (queue, buffer);
1935     }
1936   } else if (item_type == GST_QUEUE2_ITEM_TYPE_BUFFER_LIST) {
1937     GstBufferList *buffer_list;
1938     guint size = 0;
1939
1940     buffer_list = GST_BUFFER_LIST_CAST (item);
1941
1942     gst_buffer_list_foreach (buffer_list, buffer_list_calc_size, &size);
1943     GST_LOG_OBJECT (queue, "total size of buffer list: %u bytes", size);
1944
1945     /* add buffer to the statistics */
1946     if (QUEUE_IS_USING_QUEUE (queue)) {
1947       queue->cur_level.buffers++;
1948       queue->cur_level.bytes += size;
1949     }
1950     queue->bytes_in += size;
1951
1952     /* apply new buffer to segment stats */
1953     apply_buffer_list (queue, buffer_list, &queue->sink_segment, TRUE);
1954
1955     /* update the byterate stats */
1956     update_in_rates (queue);
1957
1958     if (!QUEUE_IS_USING_QUEUE (queue)) {
1959       gst_buffer_list_foreach (buffer_list, buffer_list_create_write, queue);
1960     }
1961   } else if (item_type == GST_QUEUE2_ITEM_TYPE_EVENT) {
1962     GstEvent *event;
1963
1964     event = GST_EVENT_CAST (item);
1965
1966     switch (GST_EVENT_TYPE (event)) {
1967       case GST_EVENT_EOS:
1968         /* Zero the thresholds, this makes sure the queue is completely
1969          * filled and we can read all data from the queue. */
1970         GST_DEBUG_OBJECT (queue, "we have EOS");
1971         queue->is_eos = TRUE;
1972         break;
1973       case GST_EVENT_NEWSEGMENT:
1974         apply_segment (queue, event, &queue->sink_segment, TRUE);
1975         /* This is our first new segment, we hold it
1976          * as we can't save it on the temp file */
1977         if (!QUEUE_IS_USING_QUEUE (queue)) {
1978           if (queue->segment_event_received)
1979             goto unexpected_event;
1980
1981           queue->segment_event_received = TRUE;
1982           if (queue->starting_segment != NULL)
1983             gst_event_unref (queue->starting_segment);
1984           queue->starting_segment = event;
1985           item = NULL;
1986         }
1987         /* a new segment allows us to accept more buffers if we got UNEXPECTED
1988          * from downstream */
1989         queue->unexpected = FALSE;
1990         break;
1991       default:
1992         if (!QUEUE_IS_USING_QUEUE (queue))
1993           goto unexpected_event;
1994         break;
1995     }
1996   } else {
1997     g_warning ("Unexpected item %p added in queue %s (refcounting problem?)",
1998         item, GST_OBJECT_NAME (queue));
1999     /* we can't really unref since we don't know what it is */
2000     item = NULL;
2001   }
2002
2003   if (item) {
2004     /* update the buffering status */
2005     if (queue->use_buffering)
2006       update_buffering (queue);
2007
2008     if (QUEUE_IS_USING_QUEUE (queue)) {
2009       g_queue_push_tail (&queue->queue, item);
2010     } else {
2011       gst_mini_object_unref (GST_MINI_OBJECT_CAST (item));
2012     }
2013
2014     GST_QUEUE2_SIGNAL_ADD (queue);
2015   }
2016
2017   return;
2018
2019   /* ERRORS */
2020 unexpected_event:
2021   {
2022     g_warning
2023         ("Unexpected event of kind %s can't be added in temp file of queue %s ",
2024         gst_event_type_get_name (GST_EVENT_TYPE (item)),
2025         GST_OBJECT_NAME (queue));
2026     gst_event_unref (GST_EVENT_CAST (item));
2027     return;
2028   }
2029 }
2030
2031 /* dequeue an item from the queue and update level stats */
2032 static GstMiniObject *
2033 gst_queue2_locked_dequeue (GstQueue2 * queue, GstQueue2ItemType * item_type)
2034 {
2035   GstMiniObject *item;
2036
2037   if (!QUEUE_IS_USING_QUEUE (queue))
2038     item = gst_queue2_read_item_from_file (queue);
2039   else
2040     item = g_queue_pop_head (&queue->queue);
2041
2042   if (item == NULL)
2043     goto no_item;
2044
2045   if (GST_IS_BUFFER (item)) {
2046     GstBuffer *buffer;
2047     guint size;
2048
2049     buffer = GST_BUFFER_CAST (item);
2050     size = GST_BUFFER_SIZE (buffer);
2051     *item_type = GST_QUEUE2_ITEM_TYPE_BUFFER;
2052
2053     GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2054         "retrieved buffer %p from queue", buffer);
2055
2056     if (QUEUE_IS_USING_QUEUE (queue)) {
2057       queue->cur_level.buffers--;
2058       queue->cur_level.bytes -= size;
2059     }
2060     queue->bytes_out += size;
2061
2062     apply_buffer (queue, buffer, &queue->src_segment, FALSE);
2063     /* update the byterate stats */
2064     update_out_rates (queue);
2065     /* update the buffering */
2066     if (queue->use_buffering)
2067       update_buffering (queue);
2068
2069   } else if (GST_IS_EVENT (item)) {
2070     GstEvent *event = GST_EVENT_CAST (item);
2071
2072     *item_type = GST_QUEUE2_ITEM_TYPE_EVENT;
2073
2074     GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2075         "retrieved event %p from queue", event);
2076
2077     switch (GST_EVENT_TYPE (event)) {
2078       case GST_EVENT_EOS:
2079         /* queue is empty now that we dequeued the EOS */
2080         GST_QUEUE2_CLEAR_LEVEL (queue->cur_level);
2081         break;
2082       case GST_EVENT_NEWSEGMENT:
2083         apply_segment (queue, event, &queue->src_segment, FALSE);
2084         break;
2085       default:
2086         break;
2087     }
2088   } else if (GST_IS_BUFFER_LIST (item)) {
2089     GstBufferList *buffer_list;
2090     guint size = 0;
2091
2092     buffer_list = GST_BUFFER_LIST_CAST (item);
2093     gst_buffer_list_foreach (buffer_list, buffer_list_calc_size, &size);
2094     *item_type = GST_QUEUE2_ITEM_TYPE_BUFFER_LIST;
2095
2096     GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2097         "retrieved buffer list %p from queue", buffer_list);
2098
2099     if (QUEUE_IS_USING_QUEUE (queue)) {
2100       queue->cur_level.buffers--;
2101       queue->cur_level.bytes -= size;
2102     }
2103     queue->bytes_out += size;
2104
2105     apply_buffer_list (queue, buffer_list, &queue->src_segment, FALSE);
2106     /* update the byterate stats */
2107     update_out_rates (queue);
2108     /* update the buffering */
2109     if (queue->use_buffering)
2110       update_buffering (queue);
2111
2112   } else {
2113     g_warning
2114         ("Unexpected item %p dequeued from queue %s (refcounting problem?)",
2115         item, GST_OBJECT_NAME (queue));
2116     item = NULL;
2117     *item_type = GST_QUEUE2_ITEM_TYPE_UNKNOWN;
2118   }
2119   GST_QUEUE2_SIGNAL_DEL (queue);
2120
2121   return item;
2122
2123   /* ERRORS */
2124 no_item:
2125   {
2126     GST_CAT_LOG_OBJECT (queue_dataflow, queue, "the queue is empty");
2127     return NULL;
2128   }
2129 }
2130
2131 static gboolean
2132 gst_queue2_handle_sink_event (GstPad * pad, GstEvent * event)
2133 {
2134   GstQueue2 *queue;
2135
2136   queue = GST_QUEUE2 (GST_OBJECT_PARENT (pad));
2137
2138   switch (GST_EVENT_TYPE (event)) {
2139     case GST_EVENT_FLUSH_START:
2140     {
2141       GST_CAT_LOG_OBJECT (queue_dataflow, queue, "received flush start event");
2142       if (queue->srcpad->mode == GST_ACTIVATE_PUSH) {
2143         /* forward event */
2144         gst_pad_push_event (queue->srcpad, event);
2145
2146         /* now unblock the chain function */
2147         GST_QUEUE2_MUTEX_LOCK (queue);
2148         queue->srcresult = GST_FLOW_WRONG_STATE;
2149         queue->sinkresult = GST_FLOW_WRONG_STATE;
2150         /* unblock the loop and chain functions */
2151         GST_QUEUE2_SIGNAL_ADD (queue);
2152         GST_QUEUE2_SIGNAL_DEL (queue);
2153         GST_QUEUE2_MUTEX_UNLOCK (queue);
2154
2155         /* make sure it pauses, this should happen since we sent
2156          * flush_start downstream. */
2157         gst_pad_pause_task (queue->srcpad);
2158         GST_CAT_LOG_OBJECT (queue_dataflow, queue, "loop stopped");
2159       } else {
2160         GST_QUEUE2_MUTEX_LOCK (queue);
2161         /* flush the sink pad */
2162         queue->sinkresult = GST_FLOW_WRONG_STATE;
2163         GST_QUEUE2_SIGNAL_DEL (queue);
2164         GST_QUEUE2_MUTEX_UNLOCK (queue);
2165
2166         gst_event_unref (event);
2167       }
2168       goto done;
2169     }
2170     case GST_EVENT_FLUSH_STOP:
2171     {
2172       GST_CAT_LOG_OBJECT (queue_dataflow, queue, "received flush stop event");
2173
2174       if (queue->srcpad->mode == GST_ACTIVATE_PUSH) {
2175         /* forward event */
2176         gst_pad_push_event (queue->srcpad, event);
2177
2178         GST_QUEUE2_MUTEX_LOCK (queue);
2179         gst_queue2_locked_flush (queue);
2180         queue->srcresult = GST_FLOW_OK;
2181         queue->sinkresult = GST_FLOW_OK;
2182         queue->is_eos = FALSE;
2183         queue->unexpected = FALSE;
2184         queue->seeking = FALSE;
2185         /* reset rate counters */
2186         reset_rate_timer (queue);
2187         gst_pad_start_task (queue->srcpad, (GstTaskFunction) gst_queue2_loop,
2188             queue->srcpad);
2189         GST_QUEUE2_MUTEX_UNLOCK (queue);
2190       } else {
2191         GST_QUEUE2_MUTEX_LOCK (queue);
2192         queue->segment_event_received = FALSE;
2193         queue->is_eos = FALSE;
2194         queue->unexpected = FALSE;
2195         queue->sinkresult = GST_FLOW_OK;
2196         queue->seeking = FALSE;
2197         GST_QUEUE2_MUTEX_UNLOCK (queue);
2198
2199         gst_event_unref (event);
2200       }
2201       goto done;
2202     }
2203     default:
2204       if (GST_EVENT_IS_SERIALIZED (event)) {
2205         /* serialized events go in the queue */
2206         GST_QUEUE2_MUTEX_LOCK_CHECK (queue, queue->sinkresult, out_flushing);
2207         /* refuse more events on EOS */
2208         if (queue->is_eos)
2209           goto out_eos;
2210         gst_queue2_locked_enqueue (queue, event, GST_QUEUE2_ITEM_TYPE_EVENT);
2211         GST_QUEUE2_MUTEX_UNLOCK (queue);
2212       } else {
2213         /* non-serialized events are passed upstream. */
2214         gst_pad_push_event (queue->srcpad, event);
2215       }
2216       break;
2217   }
2218 done:
2219   return TRUE;
2220
2221   /* ERRORS */
2222 out_flushing:
2223   {
2224     GST_DEBUG_OBJECT (queue, "refusing event, we are flushing");
2225     GST_QUEUE2_MUTEX_UNLOCK (queue);
2226     gst_event_unref (event);
2227     return FALSE;
2228   }
2229 out_eos:
2230   {
2231     GST_DEBUG_OBJECT (queue, "refusing event, we are EOS");
2232     GST_QUEUE2_MUTEX_UNLOCK (queue);
2233     gst_event_unref (event);
2234     return FALSE;
2235   }
2236 }
2237
2238 static gboolean
2239 gst_queue2_is_empty (GstQueue2 * queue)
2240 {
2241   /* never empty on EOS */
2242   if (queue->is_eos)
2243     return FALSE;
2244
2245   if (!QUEUE_IS_USING_QUEUE (queue) && queue->current) {
2246     return queue->current->writing_pos <= queue->current->max_reading_pos;
2247   } else {
2248     if (queue->queue.length == 0)
2249       return TRUE;
2250   }
2251
2252   return FALSE;
2253 }
2254
2255 static gboolean
2256 gst_queue2_is_filled (GstQueue2 * queue)
2257 {
2258   gboolean res;
2259
2260   /* always filled on EOS */
2261   if (queue->is_eos)
2262     return TRUE;
2263
2264 #define CHECK_FILLED(format,alt_max) ((queue->max_level.format) > 0 && \
2265     (queue->cur_level.format) >= ((alt_max) ? \
2266       MIN ((queue->max_level.format), (alt_max)) : (queue->max_level.format)))
2267
2268   /* if using a ring buffer we're filled if all ring buffer space is used
2269    * _by the current range_ */
2270   if (QUEUE_IS_USING_RING_BUFFER (queue)) {
2271     guint64 rb_size = queue->ring_buffer_max_size;
2272     GST_DEBUG_OBJECT (queue,
2273         "max bytes %u, rb size %" G_GUINT64_FORMAT ", cur bytes %u",
2274         queue->max_level.bytes, rb_size, queue->cur_level.bytes);
2275     return CHECK_FILLED (bytes, rb_size);
2276   }
2277
2278   /* if using file, we're never filled if we don't have EOS */
2279   if (QUEUE_IS_USING_TEMP_FILE (queue))
2280     return FALSE;
2281
2282   /* we are never filled when we have no buffers at all */
2283   if (queue->cur_level.buffers == 0)
2284     return FALSE;
2285
2286   /* we are filled if one of the current levels exceeds the max */
2287   res = CHECK_FILLED (buffers, 0) || CHECK_FILLED (bytes, 0)
2288       || CHECK_FILLED (time, 0);
2289
2290   /* if we need to, use the rate estimate to check against the max time we are
2291    * allowed to queue */
2292   if (queue->use_rate_estimate)
2293     res |= CHECK_FILLED (rate_time, 0);
2294
2295 #undef CHECK_FILLED
2296   return res;
2297 }
2298
2299 static GstFlowReturn
2300 gst_queue2_chain_buffer_or_buffer_list (GstQueue2 * queue,
2301     GstMiniObject * item, GstQueue2ItemType item_type)
2302 {
2303   /* we have to lock the queue since we span threads */
2304   GST_QUEUE2_MUTEX_LOCK_CHECK (queue, queue->sinkresult, out_flushing);
2305   /* when we received EOS, we refuse more data */
2306   if (queue->is_eos)
2307     goto out_eos;
2308   /* when we received unexpected from downstream, refuse more buffers */
2309   if (queue->unexpected)
2310     goto out_unexpected;
2311
2312   /* while we didn't receive the newsegment, we're seeking and we skip data */
2313   if (queue->seeking)
2314     goto out_seeking;
2315
2316   if (!gst_queue2_wait_free_space (queue))
2317     goto out_flushing;
2318
2319   /* put buffer in queue now */
2320   gst_queue2_locked_enqueue (queue, item, item_type);
2321   GST_QUEUE2_MUTEX_UNLOCK (queue);
2322
2323   return GST_FLOW_OK;
2324
2325   /* special conditions */
2326 out_flushing:
2327   {
2328     GstFlowReturn ret = queue->sinkresult;
2329
2330     GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2331         "exit because task paused, reason: %s", gst_flow_get_name (ret));
2332     GST_QUEUE2_MUTEX_UNLOCK (queue);
2333     gst_mini_object_unref (item);
2334
2335     return ret;
2336   }
2337 out_eos:
2338   {
2339     GST_CAT_LOG_OBJECT (queue_dataflow, queue, "exit because we received EOS");
2340     GST_QUEUE2_MUTEX_UNLOCK (queue);
2341     gst_mini_object_unref (item);
2342
2343     return GST_FLOW_UNEXPECTED;
2344   }
2345 out_seeking:
2346   {
2347     GST_CAT_LOG_OBJECT (queue_dataflow, queue, "exit because we are seeking");
2348     GST_QUEUE2_MUTEX_UNLOCK (queue);
2349     gst_mini_object_unref (item);
2350
2351     return GST_FLOW_OK;
2352   }
2353 out_unexpected:
2354   {
2355     GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2356         "exit because we received UNEXPECTED");
2357     GST_QUEUE2_MUTEX_UNLOCK (queue);
2358     gst_mini_object_unref (item);
2359
2360     return GST_FLOW_UNEXPECTED;
2361   }
2362 }
2363
2364 static GstFlowReturn
2365 gst_queue2_chain (GstPad * pad, GstBuffer * buffer)
2366 {
2367   GstQueue2 *queue;
2368
2369   queue = GST_QUEUE2 (GST_OBJECT_PARENT (pad));
2370
2371   GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2372       "received buffer %p of size %d, time %" GST_TIME_FORMAT ", duration %"
2373       GST_TIME_FORMAT, buffer, GST_BUFFER_SIZE (buffer),
2374       GST_TIME_ARGS (GST_BUFFER_TIMESTAMP (buffer)),
2375       GST_TIME_ARGS (GST_BUFFER_DURATION (buffer)));
2376
2377   return gst_queue2_chain_buffer_or_buffer_list (queue,
2378       GST_MINI_OBJECT_CAST (buffer), GST_QUEUE2_ITEM_TYPE_BUFFER);
2379 }
2380
2381 static GstFlowReturn
2382 gst_queue2_chain_list (GstPad * pad, GstBufferList * buffer_list)
2383 {
2384   GstQueue2 *queue;
2385
2386   queue = GST_QUEUE2 (GST_OBJECT_PARENT (pad));
2387
2388   GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2389       "received buffer list %p", buffer_list);
2390
2391   return gst_queue2_chain_buffer_or_buffer_list (queue,
2392       GST_MINI_OBJECT_CAST (buffer_list), GST_QUEUE2_ITEM_TYPE_BUFFER_LIST);
2393 }
2394
2395 static GstMiniObject *
2396 gst_queue2_dequeue_on_unexpected (GstQueue2 * queue,
2397     GstQueue2ItemType * item_type)
2398 {
2399   GstMiniObject *data;
2400
2401   GST_CAT_LOG_OBJECT (queue_dataflow, queue, "got UNEXPECTED from downstream");
2402
2403   /* stop pushing buffers, we dequeue all items until we see an item that we
2404    * can push again, which is EOS or NEWSEGMENT. If there is nothing in the
2405    * queue we can push, we set a flag to make the sinkpad refuse more
2406    * buffers with an UNEXPECTED return value until we receive something
2407    * pushable again or we get flushed. */
2408   while ((data = gst_queue2_locked_dequeue (queue, item_type))) {
2409     if (*item_type == GST_QUEUE2_ITEM_TYPE_BUFFER) {
2410       GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2411           "dropping UNEXPECTED buffer %p", data);
2412       gst_buffer_unref (GST_BUFFER_CAST (data));
2413     } else if (*item_type == GST_QUEUE2_ITEM_TYPE_EVENT) {
2414       GstEvent *event = GST_EVENT_CAST (data);
2415       GstEventType type = GST_EVENT_TYPE (event);
2416
2417       if (type == GST_EVENT_EOS || type == GST_EVENT_NEWSEGMENT) {
2418         /* we found a pushable item in the queue, push it out */
2419         GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2420             "pushing pushable event %s after UNEXPECTED",
2421             GST_EVENT_TYPE_NAME (event));
2422         return data;
2423       }
2424       GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2425           "dropping UNEXPECTED event %p", event);
2426       gst_event_unref (event);
2427     } else if (*item_type == GST_QUEUE2_ITEM_TYPE_BUFFER_LIST) {
2428       GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2429           "dropping UNEXPECTED buffer list %p", data);
2430       gst_buffer_list_unref (GST_BUFFER_LIST_CAST (data));
2431     }
2432   }
2433   /* no more items in the queue. Set the unexpected flag so that upstream
2434    * make us refuse any more buffers on the sinkpad. Since we will still
2435    * accept EOS and NEWSEGMENT we return _FLOW_OK to the caller so that the
2436    * task function does not shut down. */
2437   queue->unexpected = TRUE;
2438   return NULL;
2439 }
2440
2441 /* dequeue an item from the queue an push it downstream. This functions returns
2442  * the result of the push. */
2443 static GstFlowReturn
2444 gst_queue2_push_one (GstQueue2 * queue)
2445 {
2446   GstFlowReturn result = GST_FLOW_OK;
2447   GstMiniObject *data;
2448   GstQueue2ItemType item_type;
2449
2450   data = gst_queue2_locked_dequeue (queue, &item_type);
2451   if (data == NULL)
2452     goto no_item;
2453
2454 next:
2455   GST_QUEUE2_MUTEX_UNLOCK (queue);
2456
2457   if (item_type == GST_QUEUE2_ITEM_TYPE_BUFFER) {
2458     GstBuffer *buffer;
2459     GstCaps *caps;
2460
2461     buffer = GST_BUFFER_CAST (data);
2462     caps = GST_BUFFER_CAPS (buffer);
2463
2464     /* set caps before pushing the buffer so that core does not try to do
2465      * something fancy to check if this is possible. */
2466     if (caps && caps != GST_PAD_CAPS (queue->srcpad))
2467       gst_pad_set_caps (queue->srcpad, caps);
2468
2469     result = gst_pad_push (queue->srcpad, buffer);
2470
2471     /* need to check for srcresult here as well */
2472     GST_QUEUE2_MUTEX_LOCK_CHECK (queue, queue->srcresult, out_flushing);
2473     if (result == GST_FLOW_UNEXPECTED) {
2474       data = gst_queue2_dequeue_on_unexpected (queue, &item_type);
2475       if (data != NULL)
2476         goto next;
2477       /* Since we will still accept EOS and NEWSEGMENT we return _FLOW_OK
2478        * to the caller so that the task function does not shut down */
2479       result = GST_FLOW_OK;
2480     }
2481   } else if (item_type == GST_QUEUE2_ITEM_TYPE_EVENT) {
2482     GstEvent *event = GST_EVENT_CAST (data);
2483     GstEventType type = GST_EVENT_TYPE (event);
2484
2485     gst_pad_push_event (queue->srcpad, event);
2486
2487     /* if we're EOS, return UNEXPECTED so that the task pauses. */
2488     if (type == GST_EVENT_EOS) {
2489       GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2490           "pushed EOS event %p, return UNEXPECTED", event);
2491       result = GST_FLOW_UNEXPECTED;
2492     }
2493
2494     GST_QUEUE2_MUTEX_LOCK_CHECK (queue, queue->srcresult, out_flushing);
2495   } else if (item_type == GST_QUEUE2_ITEM_TYPE_BUFFER_LIST) {
2496     GstBufferList *buffer_list;
2497     GstBuffer *first_buf;
2498     GstCaps *caps;
2499
2500     buffer_list = GST_BUFFER_LIST_CAST (data);
2501
2502     first_buf = gst_buffer_list_get (buffer_list, 0, 0);
2503     caps = (first_buf != NULL) ? GST_BUFFER_CAPS (first_buf) : NULL;
2504
2505     /* set caps before pushing the buffer so that core does not try to do
2506      * something fancy to check if this is possible. */
2507     if (caps && caps != GST_PAD_CAPS (queue->srcpad))
2508       gst_pad_set_caps (queue->srcpad, caps);
2509
2510     result = gst_pad_push_list (queue->srcpad, buffer_list);
2511
2512     /* need to check for srcresult here as well */
2513     GST_QUEUE2_MUTEX_LOCK_CHECK (queue, queue->srcresult, out_flushing);
2514     if (result == GST_FLOW_UNEXPECTED) {
2515       data = gst_queue2_dequeue_on_unexpected (queue, &item_type);
2516       if (data != NULL)
2517         goto next;
2518       /* Since we will still accept EOS and NEWSEGMENT we return _FLOW_OK
2519        * to the caller so that the task function does not shut down */
2520       result = GST_FLOW_OK;
2521     }
2522   }
2523   return result;
2524
2525   /* ERRORS */
2526 no_item:
2527   {
2528     GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2529         "exit because we have no item in the queue");
2530     return GST_FLOW_ERROR;
2531   }
2532 out_flushing:
2533   {
2534     GST_CAT_LOG_OBJECT (queue_dataflow, queue, "exit because we are flushing");
2535     return GST_FLOW_WRONG_STATE;
2536   }
2537 }
2538
2539 /* called repeatedly with @pad as the source pad. This function should push out
2540  * data to the peer element. */
2541 static void
2542 gst_queue2_loop (GstPad * pad)
2543 {
2544   GstQueue2 *queue;
2545   GstFlowReturn ret;
2546
2547   queue = GST_QUEUE2 (GST_PAD_PARENT (pad));
2548
2549   /* have to lock for thread-safety */
2550   GST_QUEUE2_MUTEX_LOCK_CHECK (queue, queue->srcresult, out_flushing);
2551
2552   if (gst_queue2_is_empty (queue)) {
2553     gboolean started;
2554
2555     /* pause the timer while we wait. The fact that we are waiting does not mean
2556      * the byterate on the output pad is lower */
2557     if ((started = queue->out_timer_started))
2558       g_timer_stop (queue->out_timer);
2559
2560     GST_CAT_DEBUG_OBJECT (queue_dataflow, queue,
2561         "queue is empty, waiting for new data");
2562     do {
2563       /* Wait for data to be available, we could be unlocked because of a flush. */
2564       GST_QUEUE2_WAIT_ADD_CHECK (queue, queue->srcresult, out_flushing);
2565     }
2566     while (gst_queue2_is_empty (queue));
2567
2568     /* and continue if we were running before */
2569     if (started)
2570       g_timer_continue (queue->out_timer);
2571   }
2572   ret = gst_queue2_push_one (queue);
2573   queue->srcresult = ret;
2574   queue->sinkresult = ret;
2575   if (ret != GST_FLOW_OK)
2576     goto out_flushing;
2577
2578   GST_QUEUE2_MUTEX_UNLOCK (queue);
2579
2580   return;
2581
2582   /* ERRORS */
2583 out_flushing:
2584   {
2585     gboolean eos = queue->is_eos;
2586     GstFlowReturn ret = queue->srcresult;
2587
2588     gst_pad_pause_task (queue->srcpad);
2589     GST_QUEUE2_MUTEX_UNLOCK (queue);
2590     GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2591         "pause task, reason:  %s", gst_flow_get_name (queue->srcresult));
2592     /* let app know about us giving up if upstream is not expected to do so */
2593     /* UNEXPECTED is already taken care of elsewhere */
2594     if (eos && (ret == GST_FLOW_NOT_LINKED || ret < GST_FLOW_UNEXPECTED)) {
2595       GST_ELEMENT_ERROR (queue, STREAM, FAILED,
2596           (_("Internal data flow error.")),
2597           ("streaming task paused, reason %s (%d)",
2598               gst_flow_get_name (ret), ret));
2599       gst_pad_push_event (queue->srcpad, gst_event_new_eos ());
2600     }
2601     return;
2602   }
2603 }
2604
2605 static gboolean
2606 gst_queue2_handle_src_event (GstPad * pad, GstEvent * event)
2607 {
2608   gboolean res = TRUE;
2609   GstQueue2 *queue = GST_QUEUE2 (gst_pad_get_parent (pad));
2610
2611   if (G_UNLIKELY (queue == NULL)) {
2612     gst_event_unref (event);
2613     return FALSE;
2614   }
2615 #ifndef GST_DISABLE_GST_DEBUG
2616   GST_CAT_DEBUG_OBJECT (queue_dataflow, queue, "got event %p (%s)",
2617       event, GST_EVENT_TYPE_NAME (event));
2618 #endif
2619
2620   switch (GST_EVENT_TYPE (event)) {
2621     case GST_EVENT_FLUSH_START:
2622       if (QUEUE_IS_USING_QUEUE (queue)) {
2623         /* just forward upstream */
2624         res = gst_pad_push_event (queue->sinkpad, event);
2625       } else {
2626         /* now unblock the getrange function */
2627         GST_QUEUE2_MUTEX_LOCK (queue);
2628         GST_DEBUG_OBJECT (queue, "flushing");
2629         queue->srcresult = GST_FLOW_WRONG_STATE;
2630         GST_QUEUE2_SIGNAL_ADD (queue);
2631         GST_QUEUE2_MUTEX_UNLOCK (queue);
2632
2633         /* when using a temp file, we eat the event */
2634         res = TRUE;
2635         gst_event_unref (event);
2636       }
2637       break;
2638     case GST_EVENT_FLUSH_STOP:
2639       if (QUEUE_IS_USING_QUEUE (queue)) {
2640         /* just forward upstream */
2641         res = gst_pad_push_event (queue->sinkpad, event);
2642       } else {
2643         /* now unblock the getrange function */
2644         GST_QUEUE2_MUTEX_LOCK (queue);
2645         queue->srcresult = GST_FLOW_OK;
2646         GST_QUEUE2_MUTEX_UNLOCK (queue);
2647
2648         /* when using a temp file, we eat the event */
2649         res = TRUE;
2650         gst_event_unref (event);
2651       }
2652       break;
2653     default:
2654       res = gst_pad_push_event (queue->sinkpad, event);
2655       break;
2656   }
2657
2658   gst_object_unref (queue);
2659   return res;
2660 }
2661
2662 static gboolean
2663 gst_queue2_peer_query (GstQueue2 * queue, GstPad * pad, GstQuery * query)
2664 {
2665   gboolean ret = FALSE;
2666   GstPad *peer;
2667
2668   if ((peer = gst_pad_get_peer (pad))) {
2669     ret = gst_pad_query (peer, query);
2670     gst_object_unref (peer);
2671   }
2672   return ret;
2673 }
2674
2675 static gboolean
2676 gst_queue2_handle_src_query (GstPad * pad, GstQuery * query)
2677 {
2678   GstQueue2 *queue;
2679
2680   queue = GST_QUEUE2 (gst_pad_get_parent (pad));
2681   if (G_UNLIKELY (queue == NULL))
2682     return FALSE;
2683
2684   switch (GST_QUERY_TYPE (query)) {
2685     case GST_QUERY_POSITION:
2686     {
2687       gint64 peer_pos;
2688       GstFormat format;
2689
2690       if (!gst_queue2_peer_query (queue, queue->sinkpad, query))
2691         goto peer_failed;
2692
2693       /* get peer position */
2694       gst_query_parse_position (query, &format, &peer_pos);
2695
2696       /* FIXME: this code assumes that there's no discont in the queue */
2697       switch (format) {
2698         case GST_FORMAT_BYTES:
2699           peer_pos -= queue->cur_level.bytes;
2700           break;
2701         case GST_FORMAT_TIME:
2702           peer_pos -= queue->cur_level.time;
2703           break;
2704         default:
2705           GST_WARNING_OBJECT (queue, "dropping query in %s format, don't "
2706               "know how to adjust value", gst_format_get_name (format));
2707           return FALSE;
2708       }
2709       /* set updated position */
2710       gst_query_set_position (query, format, peer_pos);
2711       break;
2712     }
2713     case GST_QUERY_DURATION:
2714     {
2715       GST_DEBUG_OBJECT (queue, "doing peer query");
2716
2717       if (!gst_queue2_peer_query (queue, queue->sinkpad, query))
2718         goto peer_failed;
2719
2720       GST_DEBUG_OBJECT (queue, "peer query success");
2721       break;
2722     }
2723     case GST_QUERY_BUFFERING:
2724     {
2725       GstFormat format;
2726
2727       GST_DEBUG_OBJECT (queue, "query buffering");
2728
2729       /* FIXME - is this condition correct? what should ring buffer do? */
2730       if (QUEUE_IS_USING_QUEUE (queue)) {
2731         /* no temp file, just forward to the peer */
2732         if (!gst_queue2_peer_query (queue, queue->sinkpad, query))
2733           goto peer_failed;
2734         GST_DEBUG_OBJECT (queue, "buffering forwarded to peer");
2735       } else {
2736         gint64 start, stop, range_start, range_stop;
2737         guint64 writing_pos;
2738         gint percent;
2739         gint64 estimated_total, buffering_left;
2740         GstFormat peer_fmt;
2741         gint64 duration;
2742         gboolean peer_res, is_buffering, is_eos;
2743         gdouble byte_in_rate, byte_out_rate;
2744         GstQueue2Range *queued_ranges;
2745
2746         /* we need a current download region */
2747         if (queue->current == NULL)
2748           return FALSE;
2749
2750         writing_pos = queue->current->writing_pos;
2751         byte_in_rate = queue->byte_in_rate;
2752         byte_out_rate = queue->byte_out_rate;
2753         is_buffering = queue->is_buffering;
2754         is_eos = queue->is_eos;
2755         percent = queue->buffering_percent;
2756
2757         if (is_eos) {
2758           /* we're EOS, we know the duration in bytes now */
2759           peer_res = TRUE;
2760           duration = writing_pos;
2761         } else {
2762           /* get duration of upstream in bytes */
2763           peer_fmt = GST_FORMAT_BYTES;
2764           peer_res = gst_pad_query_peer_duration (queue->sinkpad, &peer_fmt,
2765               &duration);
2766         }
2767
2768         /* calculate remaining and total download time */
2769         if (peer_res && byte_in_rate > 0.0) {
2770           estimated_total = (duration * 1000) / byte_in_rate;
2771           buffering_left = ((duration - writing_pos) * 1000) / byte_in_rate;
2772         } else {
2773           estimated_total = -1;
2774           buffering_left = -1;
2775         }
2776         GST_DEBUG_OBJECT (queue, "estimated %" G_GINT64_FORMAT ", left %"
2777             G_GINT64_FORMAT, estimated_total, buffering_left);
2778
2779         gst_query_parse_buffering_range (query, &format, NULL, NULL, NULL);
2780
2781         switch (format) {
2782           case GST_FORMAT_PERCENT:
2783             /* we need duration */
2784             if (!peer_res)
2785               goto peer_failed;
2786
2787             GST_DEBUG_OBJECT (queue,
2788                 "duration %" G_GINT64_FORMAT ", writing %" G_GINT64_FORMAT,
2789                 duration, writing_pos);
2790
2791             start = 0;
2792             /* get our available data relative to the duration */
2793             if (duration != -1)
2794               stop = GST_FORMAT_PERCENT_MAX * writing_pos / duration;
2795             else
2796               stop = -1;
2797             break;
2798           case GST_FORMAT_BYTES:
2799             start = 0;
2800             stop = writing_pos;
2801             break;
2802           default:
2803             start = -1;
2804             stop = -1;
2805             break;
2806         }
2807
2808         /* fill out the buffered ranges */
2809         for (queued_ranges = queue->ranges; queued_ranges;
2810             queued_ranges = queued_ranges->next) {
2811           switch (format) {
2812             case GST_FORMAT_PERCENT:
2813               if (duration == -1) {
2814                 range_start = 0;
2815                 range_stop = 0;
2816                 break;
2817               }
2818               range_start = 100 * queued_ranges->offset / duration;
2819               range_stop = 100 * queued_ranges->writing_pos / duration;
2820               break;
2821             case GST_FORMAT_BYTES:
2822               range_start = queued_ranges->offset;
2823               range_stop = queued_ranges->writing_pos;
2824               break;
2825             default:
2826               range_start = -1;
2827               range_stop = -1;
2828               break;
2829           }
2830           if (range_start == range_stop)
2831             continue;
2832           GST_DEBUG_OBJECT (queue,
2833               "range starting at %" G_GINT64_FORMAT " and finishing at %"
2834               G_GINT64_FORMAT, range_start, range_stop);
2835           gst_query_add_buffering_range (query, range_start, range_stop);
2836         }
2837
2838         gst_query_set_buffering_percent (query, is_buffering, percent);
2839         gst_query_set_buffering_range (query, format, start, stop,
2840             estimated_total);
2841         gst_query_set_buffering_stats (query, GST_BUFFERING_DOWNLOAD,
2842             byte_in_rate, byte_out_rate, buffering_left);
2843       }
2844       break;
2845     }
2846     default:
2847       /* peer handled other queries */
2848       if (!gst_queue2_peer_query (queue, queue->sinkpad, query))
2849         goto peer_failed;
2850       break;
2851   }
2852
2853   gst_object_unref (queue);
2854   return TRUE;
2855
2856   /* ERRORS */
2857 peer_failed:
2858   {
2859     GST_DEBUG_OBJECT (queue, "failed peer query");
2860     gst_object_unref (queue);
2861     return FALSE;
2862   }
2863 }
2864
2865 static gboolean
2866 gst_queue2_handle_query (GstElement * element, GstQuery * query)
2867 {
2868   /* simply forward to the srcpad query function */
2869   return gst_queue2_handle_src_query (GST_QUEUE2_CAST (element)->srcpad, query);
2870 }
2871
2872 static void
2873 gst_queue2_update_upstream_size (GstQueue2 * queue)
2874 {
2875   GstFormat fmt = GST_FORMAT_BYTES;
2876   gint64 upstream_size = -1;
2877
2878   if (gst_pad_query_peer_duration (queue->sinkpad, &fmt, &upstream_size)) {
2879     GST_INFO_OBJECT (queue, "upstream size: %" G_GINT64_FORMAT, upstream_size);
2880     queue->upstream_size = upstream_size;
2881   }
2882 }
2883
2884 static GstFlowReturn
2885 gst_queue2_get_range (GstPad * pad, guint64 offset, guint length,
2886     GstBuffer ** buffer)
2887 {
2888   GstQueue2 *queue;
2889   GstFlowReturn ret;
2890
2891   queue = GST_QUEUE2_CAST (gst_pad_get_parent (pad));
2892
2893   length = (length == -1) ? DEFAULT_BUFFER_SIZE : length;
2894   GST_QUEUE2_MUTEX_LOCK_CHECK (queue, queue->srcresult, out_flushing);
2895   offset = (offset == -1) ? queue->current->reading_pos : offset;
2896
2897   GST_DEBUG_OBJECT (queue,
2898       "Getting range: offset %" G_GUINT64_FORMAT ", length %u", offset, length);
2899
2900   /* catch any reads beyond the size of the file here to make sure queue2
2901    * doesn't send seek events beyond the size of the file upstream, since
2902    * that would confuse elements such as souphttpsrc and/or http servers.
2903    * Demuxers often just loop until EOS at the end of the file to figure out
2904    * when they've read all the end-headers or index chunks. */
2905   if (G_UNLIKELY (offset >= queue->upstream_size)) {
2906     gst_queue2_update_upstream_size (queue);
2907     if (queue->upstream_size > 0 && offset >= queue->upstream_size)
2908       goto out_unexpected;
2909   }
2910
2911   if (G_UNLIKELY (offset + length > queue->upstream_size)) {
2912     gst_queue2_update_upstream_size (queue);
2913     if (queue->upstream_size > 0 && offset + length >= queue->upstream_size) {
2914       length = queue->upstream_size - offset;
2915       GST_DEBUG_OBJECT (queue, "adjusting length downto %d", length);
2916     }
2917   }
2918
2919   /* FIXME - function will block when the range is not yet available */
2920   ret = gst_queue2_create_read (queue, offset, length, buffer);
2921   GST_QUEUE2_MUTEX_UNLOCK (queue);
2922
2923   gst_object_unref (queue);
2924
2925   return ret;
2926
2927   /* ERRORS */
2928 out_flushing:
2929   {
2930     ret = queue->srcresult;
2931
2932     GST_DEBUG_OBJECT (queue, "we are flushing");
2933     GST_QUEUE2_MUTEX_UNLOCK (queue);
2934     gst_object_unref (queue);
2935     return ret;
2936   }
2937 out_unexpected:
2938   {
2939     GST_DEBUG_OBJECT (queue, "read beyond end of file");
2940     GST_QUEUE2_MUTEX_UNLOCK (queue);
2941     gst_object_unref (queue);
2942     return GST_FLOW_UNEXPECTED;
2943   }
2944 }
2945
2946 static gboolean
2947 gst_queue2_src_checkgetrange_function (GstPad * pad)
2948 {
2949   GstQueue2 *queue;
2950   gboolean ret;
2951
2952   queue = GST_QUEUE2 (gst_pad_get_parent (pad));
2953
2954   /* we can operate in pull mode when we are using a tempfile */
2955   ret = !QUEUE_IS_USING_QUEUE (queue);
2956
2957   gst_object_unref (GST_OBJECT (queue));
2958
2959   return ret;
2960 }
2961
2962 /* sink currently only operates in push mode */
2963 static gboolean
2964 gst_queue2_sink_activate_push (GstPad * pad, gboolean active)
2965 {
2966   gboolean result = TRUE;
2967   GstQueue2 *queue;
2968
2969   queue = GST_QUEUE2 (gst_pad_get_parent (pad));
2970
2971   if (active) {
2972     GST_QUEUE2_MUTEX_LOCK (queue);
2973     GST_DEBUG_OBJECT (queue, "activating push mode");
2974     queue->srcresult = GST_FLOW_OK;
2975     queue->sinkresult = GST_FLOW_OK;
2976     queue->is_eos = FALSE;
2977     queue->unexpected = FALSE;
2978     reset_rate_timer (queue);
2979     GST_QUEUE2_MUTEX_UNLOCK (queue);
2980   } else {
2981     /* unblock chain function */
2982     GST_QUEUE2_MUTEX_LOCK (queue);
2983     GST_DEBUG_OBJECT (queue, "deactivating push mode");
2984     queue->srcresult = GST_FLOW_WRONG_STATE;
2985     queue->sinkresult = GST_FLOW_WRONG_STATE;
2986     gst_queue2_locked_flush (queue);
2987     GST_QUEUE2_MUTEX_UNLOCK (queue);
2988   }
2989
2990   gst_object_unref (queue);
2991
2992   return result;
2993 }
2994
2995 /* src operating in push mode, we start a task on the source pad that pushes out
2996  * buffers from the queue */
2997 static gboolean
2998 gst_queue2_src_activate_push (GstPad * pad, gboolean active)
2999 {
3000   gboolean result = FALSE;
3001   GstQueue2 *queue;
3002
3003   queue = GST_QUEUE2 (gst_pad_get_parent (pad));
3004
3005   if (active) {
3006     GST_QUEUE2_MUTEX_LOCK (queue);
3007     GST_DEBUG_OBJECT (queue, "activating push mode");
3008     queue->srcresult = GST_FLOW_OK;
3009     queue->sinkresult = GST_FLOW_OK;
3010     queue->is_eos = FALSE;
3011     queue->unexpected = FALSE;
3012     result = gst_pad_start_task (pad, (GstTaskFunction) gst_queue2_loop, pad);
3013     GST_QUEUE2_MUTEX_UNLOCK (queue);
3014   } else {
3015     /* unblock loop function */
3016     GST_QUEUE2_MUTEX_LOCK (queue);
3017     GST_DEBUG_OBJECT (queue, "deactivating push mode");
3018     queue->srcresult = GST_FLOW_WRONG_STATE;
3019     queue->sinkresult = GST_FLOW_WRONG_STATE;
3020     /* the item add signal will unblock */
3021     GST_QUEUE2_SIGNAL_ADD (queue);
3022     GST_QUEUE2_MUTEX_UNLOCK (queue);
3023
3024     /* step 2, make sure streaming finishes */
3025     result = gst_pad_stop_task (pad);
3026   }
3027
3028   gst_object_unref (queue);
3029
3030   return result;
3031 }
3032
3033 /* pull mode, downstream will call our getrange function */
3034 static gboolean
3035 gst_queue2_src_activate_pull (GstPad * pad, gboolean active)
3036 {
3037   gboolean result;
3038   GstQueue2 *queue;
3039
3040   queue = GST_QUEUE2 (gst_pad_get_parent (pad));
3041
3042   if (active) {
3043     GST_QUEUE2_MUTEX_LOCK (queue);
3044     if (!QUEUE_IS_USING_QUEUE (queue)) {
3045       if (QUEUE_IS_USING_TEMP_FILE (queue)) {
3046         /* open the temp file now */
3047         result = gst_queue2_open_temp_location_file (queue);
3048       } else if (!queue->ring_buffer) {
3049         queue->ring_buffer = g_malloc (queue->ring_buffer_max_size);
3050         result = ! !queue->ring_buffer;
3051       } else {
3052         result = TRUE;
3053       }
3054
3055       GST_DEBUG_OBJECT (queue, "activating pull mode");
3056       init_ranges (queue);
3057       queue->srcresult = GST_FLOW_OK;
3058       queue->sinkresult = GST_FLOW_OK;
3059       queue->is_eos = FALSE;
3060       queue->unexpected = FALSE;
3061       queue->upstream_size = 0;
3062     } else {
3063       GST_DEBUG_OBJECT (queue, "no temp file, cannot activate pull mode");
3064       /* this is not allowed, we cannot operate in pull mode without a temp
3065        * file. */
3066       queue->srcresult = GST_FLOW_WRONG_STATE;
3067       queue->sinkresult = GST_FLOW_WRONG_STATE;
3068       result = FALSE;
3069     }
3070     GST_QUEUE2_MUTEX_UNLOCK (queue);
3071   } else {
3072     GST_QUEUE2_MUTEX_LOCK (queue);
3073     GST_DEBUG_OBJECT (queue, "deactivating pull mode");
3074     queue->srcresult = GST_FLOW_WRONG_STATE;
3075     queue->sinkresult = GST_FLOW_WRONG_STATE;
3076     /* this will unlock getrange */
3077     GST_QUEUE2_SIGNAL_ADD (queue);
3078     result = TRUE;
3079     GST_QUEUE2_MUTEX_UNLOCK (queue);
3080   }
3081   gst_object_unref (queue);
3082
3083   return result;
3084 }
3085
3086 static GstStateChangeReturn
3087 gst_queue2_change_state (GstElement * element, GstStateChange transition)
3088 {
3089   GstQueue2 *queue;
3090   GstStateChangeReturn ret = GST_STATE_CHANGE_SUCCESS;
3091
3092   queue = GST_QUEUE2 (element);
3093
3094   switch (transition) {
3095     case GST_STATE_CHANGE_NULL_TO_READY:
3096       break;
3097     case GST_STATE_CHANGE_READY_TO_PAUSED:
3098       GST_QUEUE2_MUTEX_LOCK (queue);
3099       if (!QUEUE_IS_USING_QUEUE (queue)) {
3100         if (QUEUE_IS_USING_TEMP_FILE (queue)) {
3101           if (!gst_queue2_open_temp_location_file (queue))
3102             ret = GST_STATE_CHANGE_FAILURE;
3103         } else {
3104           if (queue->ring_buffer) {
3105             g_free (queue->ring_buffer);
3106             queue->ring_buffer = NULL;
3107           }
3108           if (!(queue->ring_buffer = g_malloc (queue->ring_buffer_max_size)))
3109             ret = GST_STATE_CHANGE_FAILURE;
3110         }
3111         init_ranges (queue);
3112       }
3113       queue->segment_event_received = FALSE;
3114       queue->starting_segment = NULL;
3115       GST_QUEUE2_MUTEX_UNLOCK (queue);
3116       break;
3117     case GST_STATE_CHANGE_PAUSED_TO_PLAYING:
3118       break;
3119     default:
3120       break;
3121   }
3122
3123   if (ret == GST_STATE_CHANGE_FAILURE)
3124     return ret;
3125
3126   ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
3127
3128   if (ret == GST_STATE_CHANGE_FAILURE)
3129     return ret;
3130
3131   switch (transition) {
3132     case GST_STATE_CHANGE_PLAYING_TO_PAUSED:
3133       break;
3134     case GST_STATE_CHANGE_PAUSED_TO_READY:
3135       GST_QUEUE2_MUTEX_LOCK (queue);
3136       if (!QUEUE_IS_USING_QUEUE (queue)) {
3137         if (QUEUE_IS_USING_TEMP_FILE (queue)) {
3138           gst_queue2_close_temp_location_file (queue);
3139         } else if (queue->ring_buffer) {
3140           g_free (queue->ring_buffer);
3141           queue->ring_buffer = NULL;
3142         }
3143         clean_ranges (queue);
3144       }
3145       if (queue->starting_segment != NULL) {
3146         gst_event_unref (queue->starting_segment);
3147         queue->starting_segment = NULL;
3148       }
3149       GST_QUEUE2_MUTEX_UNLOCK (queue);
3150       break;
3151     case GST_STATE_CHANGE_READY_TO_NULL:
3152       break;
3153     default:
3154       break;
3155   }
3156
3157   return ret;
3158 }
3159
3160 /* changing the capacity of the queue must wake up
3161  * the _chain function, it might have more room now
3162  * to store the buffer/event in the queue */
3163 #define QUEUE_CAPACITY_CHANGE(q)\
3164   GST_QUEUE2_SIGNAL_DEL (queue);
3165
3166 /* Changing the minimum required fill level must
3167  * wake up the _loop function as it might now
3168  * be able to preceed.
3169  */
3170 #define QUEUE_THRESHOLD_CHANGE(q)\
3171   GST_QUEUE2_SIGNAL_ADD (queue);
3172
3173 static void
3174 gst_queue2_set_temp_template (GstQueue2 * queue, const gchar * template)
3175 {
3176   GstState state;
3177
3178   /* the element must be stopped in order to do this */
3179   GST_OBJECT_LOCK (queue);
3180   state = GST_STATE (queue);
3181   if (state != GST_STATE_READY && state != GST_STATE_NULL)
3182     goto wrong_state;
3183   GST_OBJECT_UNLOCK (queue);
3184
3185   /* set new location */
3186   g_free (queue->temp_template);
3187   queue->temp_template = g_strdup (template);
3188
3189   return;
3190
3191 /* ERROR */
3192 wrong_state:
3193   {
3194     GST_WARNING_OBJECT (queue, "setting temp-template property in wrong state");
3195     GST_OBJECT_UNLOCK (queue);
3196   }
3197 }
3198
3199 static void
3200 gst_queue2_set_property (GObject * object,
3201     guint prop_id, const GValue * value, GParamSpec * pspec)
3202 {
3203   GstQueue2 *queue = GST_QUEUE2 (object);
3204
3205   /* someone could change levels here, and since this
3206    * affects the get/put funcs, we need to lock for safety. */
3207   GST_QUEUE2_MUTEX_LOCK (queue);
3208
3209   switch (prop_id) {
3210     case PROP_MAX_SIZE_BYTES:
3211       queue->max_level.bytes = g_value_get_uint (value);
3212       QUEUE_CAPACITY_CHANGE (queue);
3213       break;
3214     case PROP_MAX_SIZE_BUFFERS:
3215       queue->max_level.buffers = g_value_get_uint (value);
3216       QUEUE_CAPACITY_CHANGE (queue);
3217       break;
3218     case PROP_MAX_SIZE_TIME:
3219       queue->max_level.time = g_value_get_uint64 (value);
3220       /* set rate_time to the same value. We use an extra field in the level
3221        * structure so that we can easily access and compare it */
3222       queue->max_level.rate_time = queue->max_level.time;
3223       QUEUE_CAPACITY_CHANGE (queue);
3224       break;
3225     case PROP_USE_BUFFERING:
3226       queue->use_buffering = g_value_get_boolean (value);
3227       break;
3228     case PROP_USE_RATE_ESTIMATE:
3229       queue->use_rate_estimate = g_value_get_boolean (value);
3230       break;
3231     case PROP_LOW_PERCENT:
3232       queue->low_percent = g_value_get_int (value);
3233       break;
3234     case PROP_HIGH_PERCENT:
3235       queue->high_percent = g_value_get_int (value);
3236       break;
3237     case PROP_TEMP_TEMPLATE:
3238       gst_queue2_set_temp_template (queue, g_value_get_string (value));
3239       break;
3240     case PROP_TEMP_LOCATION:
3241       g_free (queue->temp_location);
3242       queue->temp_location = g_value_dup_string (value);
3243       /* you can set the property back to NULL to make it use the temp-tmpl
3244        * property. */
3245       queue->temp_location_set = queue->temp_location != NULL;
3246       break;
3247     case PROP_TEMP_REMOVE:
3248       queue->temp_remove = g_value_get_boolean (value);
3249       break;
3250     case PROP_RING_BUFFER_MAX_SIZE:
3251       queue->ring_buffer_max_size = g_value_get_uint64 (value);
3252       break;
3253     default:
3254       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
3255       break;
3256   }
3257
3258   GST_QUEUE2_MUTEX_UNLOCK (queue);
3259 }
3260
3261 static void
3262 gst_queue2_get_property (GObject * object,
3263     guint prop_id, GValue * value, GParamSpec * pspec)
3264 {
3265   GstQueue2 *queue = GST_QUEUE2 (object);
3266
3267   GST_QUEUE2_MUTEX_LOCK (queue);
3268
3269   switch (prop_id) {
3270     case PROP_CUR_LEVEL_BYTES:
3271       g_value_set_uint (value, queue->cur_level.bytes);
3272       break;
3273     case PROP_CUR_LEVEL_BUFFERS:
3274       g_value_set_uint (value, queue->cur_level.buffers);
3275       break;
3276     case PROP_CUR_LEVEL_TIME:
3277       g_value_set_uint64 (value, queue->cur_level.time);
3278       break;
3279     case PROP_MAX_SIZE_BYTES:
3280       g_value_set_uint (value, queue->max_level.bytes);
3281       break;
3282     case PROP_MAX_SIZE_BUFFERS:
3283       g_value_set_uint (value, queue->max_level.buffers);
3284       break;
3285     case PROP_MAX_SIZE_TIME:
3286       g_value_set_uint64 (value, queue->max_level.time);
3287       break;
3288     case PROP_USE_BUFFERING:
3289       g_value_set_boolean (value, queue->use_buffering);
3290       break;
3291     case PROP_USE_RATE_ESTIMATE:
3292       g_value_set_boolean (value, queue->use_rate_estimate);
3293       break;
3294     case PROP_LOW_PERCENT:
3295       g_value_set_int (value, queue->low_percent);
3296       break;
3297     case PROP_HIGH_PERCENT:
3298       g_value_set_int (value, queue->high_percent);
3299       break;
3300     case PROP_TEMP_TEMPLATE:
3301       g_value_set_string (value, queue->temp_template);
3302       break;
3303     case PROP_TEMP_LOCATION:
3304       g_value_set_string (value, queue->temp_location);
3305       break;
3306     case PROP_TEMP_REMOVE:
3307       g_value_set_boolean (value, queue->temp_remove);
3308       break;
3309     case PROP_RING_BUFFER_MAX_SIZE:
3310       g_value_set_uint64 (value, queue->ring_buffer_max_size);
3311       break;
3312     default:
3313       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
3314       break;
3315   }
3316
3317   GST_QUEUE2_MUTEX_UNLOCK (queue);
3318 }