gst/rtsp-server/rtsp-stream.c

   1 /* GStreamer
   2  * Copyright (C) 2008 Wim Taymans <wim.taymans at gmail.com>
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Library General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Library General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Library General Public
  15  * License along with this library; if not, write to the
  16  * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
  17  * Boston, MA 02110-1301, USA.
  18  */
  19
  20 #include <stdlib.h>
  21 #include <stdio.h>
  22 #include <string.h>
  23
  24 #include <gio/gio.h>
  25
  26 #include <gst/app/gstappsrc.h>
  27 #include <gst/app/gstappsink.h>
  28
  29 #include "rtsp-stream.h"
  30
  31 #define GST_RTSP_STREAM_GET_PRIVATE(obj)  \
  32      (G_TYPE_INSTANCE_GET_PRIVATE ((obj), GST_TYPE_RTSP_STREAM, GstRTSPStreamPrivate))
  33
  34 struct _GstRTSPStreamPrivate
  35 {
  36   GMutex lock;
  37   guint idx;
  38   GstPad *srcpad;
  39   GstElement *payloader;
  40   guint buffer_size;
  41   gboolean is_joined;
  42   gchar *control;
  43
  44   /* pads on the rtpbin */
  45   GstPad *send_rtp_sink;
  46   GstPad *recv_sink[2];
  47   GstPad *send_src[2];
  48
  49   /* the RTPSession object */
  50   GObject *session;
  51
  52   /* sinks used for sending and receiving RTP and RTCP over ipv4, they share
  53    * sockets */
  54   GstElement *udpsrc_v4[2];
  55
  56   /* sinks used for sending and receiving RTP and RTCP over ipv6, they share
  57    * sockets */
  58   GstElement *udpsrc_v6[2];
  59
  60   GstElement *udpsink[2];
  61
  62   /* for TCP transport */
  63   GstElement *appsrc[2];
  64   GstElement *appqueue[2];
  65   GstElement *appsink[2];
  66
  67   GstElement *tee[2];
  68   GstElement *funnel[2];
  69
  70   /* server ports for sending/receiving over ipv4 */
  71   GstRTSPRange server_port_v4;
  72   GstRTSPAddress *server_addr_v4;
  73   gboolean have_ipv4;
  74
  75   /* server ports for sending/receiving over ipv6 */
  76   GstRTSPRange server_port_v6;
  77   GstRTSPAddress *server_addr_v6;
  78   gboolean have_ipv6;
  79
  80   /* multicast addresses */
  81   GstRTSPAddressPool *pool;
  82   GstRTSPAddress *addr_v4;
  83   GstRTSPAddress *addr_v6;
  84
  85   /* the caps of the stream */
  86   gulong caps_sig;
  87   GstCaps *caps;
  88
  89   /* transports we stream to */
  90   guint n_active;
  91   GList *transports;
  92
  93   gint dscp_qos;
  94 };
  95
  96 #define DEFAULT_CONTROL NULL
  97
  98 enum
  99 {
 100   PROP_0,
 101   PROP_CONTROL,
 102   PROP_LAST
 103 };
 104
 105 GST_DEBUG_CATEGORY_STATIC (rtsp_stream_debug);
 106 #define GST_CAT_DEFAULT rtsp_stream_debug
 107
 108 static GQuark ssrc_stream_map_key;
 109
 110 static void gst_rtsp_stream_get_property (GObject * object, guint propid,
 111     GValue * value, GParamSpec * pspec);
 112 static void gst_rtsp_stream_set_property (GObject * object, guint propid,
 113     const GValue * value, GParamSpec * pspec);
 114
 115 static void gst_rtsp_stream_finalize (GObject * obj);
 116
 117 G_DEFINE_TYPE (GstRTSPStream, gst_rtsp_stream, G_TYPE_OBJECT);
 118
 119 static void
 120 gst_rtsp_stream_class_init (GstRTSPStreamClass * klass)
 121 {
 122   GObjectClass *gobject_class;
 123
 124   g_type_class_add_private (klass, sizeof (GstRTSPStreamPrivate));
 125
 126   gobject_class = G_OBJECT_CLASS (klass);
 127
 128   gobject_class->get_property = gst_rtsp_stream_get_property;
 129   gobject_class->set_property = gst_rtsp_stream_set_property;
 130   gobject_class->finalize = gst_rtsp_stream_finalize;
 131
 132   g_object_class_install_property (gobject_class, PROP_CONTROL,
 133       g_param_spec_string ("control", "Control",
 134           "The control string for this stream", DEFAULT_CONTROL,
 135           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 136
 137   GST_DEBUG_CATEGORY_INIT (rtsp_stream_debug, "rtspstream", 0, "GstRTSPStream");
 138
 139   ssrc_stream_map_key = g_quark_from_static_string ("GstRTSPServer.stream");
 140 }
 141
 142 static void
 143 gst_rtsp_stream_init (GstRTSPStream * stream)
 144 {
 145   GstRTSPStreamPrivate *priv = GST_RTSP_STREAM_GET_PRIVATE (stream);
 146
 147   GST_DEBUG ("new stream %p", stream);
 148
 149   stream->priv = priv;
 150
 151   priv->dscp_qos = -1;
 152   priv->control = g_strdup (DEFAULT_CONTROL);
 153
 154   g_mutex_init (&priv->lock);
 155 }
 156
 157 static void
 158 gst_rtsp_stream_finalize (GObject * obj)
 159 {
 160   GstRTSPStream *stream;
 161   GstRTSPStreamPrivate *priv;
 162
 163   stream = GST_RTSP_STREAM (obj);
 164   priv = stream->priv;
 165
 166   GST_DEBUG ("finalize stream %p", stream);
 167
 168   /* we really need to be unjoined now */
 169   g_return_if_fail (!priv->is_joined);
 170
 171   if (priv->addr_v4)
 172     gst_rtsp_address_free (priv->addr_v4);
 173   if (priv->addr_v6)
 174     gst_rtsp_address_free (priv->addr_v6);
 175   if (priv->server_addr_v4)
 176     gst_rtsp_address_free (priv->server_addr_v4);
 177   if (priv->server_addr_v6)
 178     gst_rtsp_address_free (priv->server_addr_v6);
 179   if (priv->pool)
 180     g_object_unref (priv->pool);
 181   gst_object_unref (priv->payloader);
 182   gst_object_unref (priv->srcpad);
 183   g_free (priv->control);
 184   g_mutex_clear (&priv->lock);
 185
 186   G_OBJECT_CLASS (gst_rtsp_stream_parent_class)->finalize (obj);
 187 }
 188
 189 static void
 190 gst_rtsp_stream_get_property (GObject * object, guint propid,
 191     GValue * value, GParamSpec * pspec)
 192 {
 193   GstRTSPStream *stream = GST_RTSP_STREAM (object);
 194
 195   switch (propid) {
 196     case PROP_CONTROL:
 197       g_value_take_string (value, gst_rtsp_stream_get_control (stream));
 198       break;
 199     default:
 200       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, propid, pspec);
 201   }
 202 }
 203
 204 static void
 205 gst_rtsp_stream_set_property (GObject * object, guint propid,
 206     const GValue * value, GParamSpec * pspec)
 207 {
 208   GstRTSPStream *stream = GST_RTSP_STREAM (object);
 209
 210   switch (propid) {
 211     case PROP_CONTROL:
 212       gst_rtsp_stream_set_control (stream, g_value_get_string (value));
 213       break;
 214     default:
 215       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, propid, pspec);
 216   }
 217 }
 218
 219 /**
 220  * gst_rtsp_stream_new:
 221  * @idx: an index
 222  * @srcpad: a #GstPad
 223  * @payloader: a #GstElement
 224  *
 225  * Create a new media stream with index @idx that handles RTP data on
 226  * @srcpad and has a payloader element @payloader.
 227  *
 228  * Returns: a new #GstRTSPStream
 229  */
 230 GstRTSPStream *
 231 gst_rtsp_stream_new (guint idx, GstElement * payloader, GstPad * srcpad)
 232 {
 233   GstRTSPStreamPrivate *priv;
 234   GstRTSPStream *stream;
 235
 236   g_return_val_if_fail (GST_IS_ELEMENT (payloader), NULL);
 237   g_return_val_if_fail (GST_IS_PAD (srcpad), NULL);
 238   g_return_val_if_fail (GST_PAD_IS_SRC (srcpad), NULL);
 239
 240   stream = g_object_new (GST_TYPE_RTSP_STREAM, NULL);
 241   priv = stream->priv;
 242   priv->idx = idx;
 243   priv->payloader = gst_object_ref (payloader);
 244   priv->srcpad = gst_object_ref (srcpad);
 245
 246   return stream;
 247 }
 248
 249 /**
 250  * gst_rtsp_stream_get_index:
 251  * @stream: a #GstRTSPStream
 252  *
 253  * Get the stream index.
 254  *
 255  * Return: the stream index.
 256  */
 257 guint
 258 gst_rtsp_stream_get_index (GstRTSPStream * stream)
 259 {
 260   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), -1);
 261
 262   return stream->priv->idx;
 263 }
 264
 265 /**
 266  * gst_rtsp_stream_get_srcpad:
 267  * @stream: a #GstRTSPStream
 268  *
 269  * Get the srcpad associated with @stream.
 270  *
 271  * Return: the srcpad. Unref after usage.
 272  */
 273 GstPad *
 274 gst_rtsp_stream_get_srcpad (GstRTSPStream * stream)
 275 {
 276   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), NULL);
 277
 278   return gst_object_ref (stream->priv->srcpad);
 279 }
 280
 281 /**
 282  * gst_rtsp_stream_get_control:
 283  * @stream: a #GstRTSPStream
 284  *
 285  * Get the control string to identify this stream.
 286  *
 287  * Return: the control string. free after usage.
 288  */
 289 gchar *
 290 gst_rtsp_stream_get_control (GstRTSPStream * stream)
 291 {
 292   GstRTSPStreamPrivate *priv;
 293   gchar *result;
 294
 295   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), NULL);
 296
 297   priv = stream->priv;
 298
 299   g_mutex_lock (&priv->lock);
 300   if ((result = g_strdup (priv->control)) == NULL)
 301     result = g_strdup_printf ("stream=%u", priv->idx);
 302   g_mutex_unlock (&priv->lock);
 303
 304   return result;
 305 }
 306
 307 /**
 308  * gst_rtsp_stream_set_control:
 309  * @stream: a #GstRTSPStream
 310  * @control: a control string
 311  *
 312  * Set the control string in @stream.
 313  */
 314 void
 315 gst_rtsp_stream_set_control (GstRTSPStream * stream, const gchar * control)
 316 {
 317   GstRTSPStreamPrivate *priv;
 318
 319   g_return_if_fail (GST_IS_RTSP_STREAM (stream));
 320
 321   priv = stream->priv;
 322
 323   g_mutex_lock (&priv->lock);
 324   g_free (priv->control);
 325   priv->control = g_strdup (control);
 326   g_mutex_unlock (&priv->lock);
 327 }
 328
 329 /**
 330  * gst_rtsp_stream_has_control:
 331  * @stream: a #GstRTSPStream
 332  * @control: a control string
 333  *
 334  * Check if @stream has the control string @control.
 335  *
 336  * Returns: %TRUE is @stream has @control as the control string
 337  */
 338 gboolean
 339 gst_rtsp_stream_has_control (GstRTSPStream * stream, const gchar * control)
 340 {
 341   GstRTSPStreamPrivate *priv;
 342   gboolean res;
 343
 344   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), FALSE);
 345
 346   priv = stream->priv;
 347
 348   g_mutex_lock (&priv->lock);
 349   if (priv->control)
 350     res = g_strcmp0 (priv->control, control);
 351   else {
 352     guint streamid;
 353     sscanf (control, "stream=%u", &streamid);
 354     res = (streamid == priv->idx);
 355   }
 356   g_mutex_unlock (&priv->lock);
 357
 358   return res;
 359 }
 360
 361 /**
 362  * gst_rtsp_stream_set_mtu:
 363  * @stream: a #GstRTSPStream
 364  * @mtu: a new MTU
 365  *
 366  * Configure the mtu in the payloader of @stream to @mtu.
 367  */
 368 void
 369 gst_rtsp_stream_set_mtu (GstRTSPStream * stream, guint mtu)
 370 {
 371   GstRTSPStreamPrivate *priv;
 372
 373   g_return_if_fail (GST_IS_RTSP_STREAM (stream));
 374
 375   priv = stream->priv;
 376
 377   GST_LOG_OBJECT (stream, "set MTU %u", mtu);
 378
 379   g_object_set (G_OBJECT (priv->payloader), "mtu", mtu, NULL);
 380 }
 381
 382 /**
 383  * gst_rtsp_stream_get_mtu:
 384  * @stream: a #GstRTSPStream
 385  *
 386  * Get the configured MTU in the payloader of @stream.
 387  *
 388  * Returns: the MTU of the payloader.
 389  */
 390 guint
 391 gst_rtsp_stream_get_mtu (GstRTSPStream * stream)
 392 {
 393   GstRTSPStreamPrivate *priv;
 394   guint mtu;
 395
 396   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), 0);
 397
 398   priv = stream->priv;
 399
 400   g_object_get (G_OBJECT (priv->payloader), "mtu", &mtu, NULL);
 401
 402   return mtu;
 403 }
 404
 405 /* Update the dscp qos property on the udp sinks */
 406 static void
 407 update_dscp_qos (GstRTSPStream * stream)
 408 {
 409   GstRTSPStreamPrivate *priv;
 410
 411   g_return_if_fail (GST_IS_RTSP_STREAM (stream));
 412
 413   priv = stream->priv;
 414
 415   if (priv->udpsink[0]) {
 416     g_object_set (G_OBJECT (priv->udpsink[0]), "qos-dscp", priv->dscp_qos,
 417         NULL);
 418   }
 419
 420   if (priv->udpsink[1]) {
 421     g_object_set (G_OBJECT (priv->udpsink[1]), "qos-dscp", priv->dscp_qos,
 422         NULL);
 423   }
 424 }
 425
 426 /**
 427  * gst_rtsp_stream_set_dscp_qos:
 428  * @stream: a #GstRTSPStream
 429  * @dscp_qos: a new dscp qos value (0-63, or -1 to disable)
 430  *
 431  * Configure the dscp qos of the outgoing sockets to @dscp_qos.
 432  */
 433 void
 434 gst_rtsp_stream_set_dscp_qos (GstRTSPStream * stream, gint dscp_qos)
 435 {
 436   GstRTSPStreamPrivate *priv;
 437
 438   g_return_if_fail (GST_IS_RTSP_STREAM (stream));
 439
 440   priv = stream->priv;
 441
 442   GST_LOG_OBJECT (stream, "set DSCP QoS %d", dscp_qos);
 443
 444   if (dscp_qos < -1 || dscp_qos > 63) {
 445     GST_WARNING_OBJECT (stream, "trying to set illegal dscp qos %d", dscp_qos);
 446     return;
 447   }
 448
 449   priv->dscp_qos = dscp_qos;
 450
 451   update_dscp_qos (stream);
 452 }
 453
 454 /**
 455  * gst_rtsp_stream_get_dscp_qos:
 456  * @stream: a #GstRTSPStream
 457  *
 458  * Get the configured DSCP QoS in of the outgoing sockets.
 459  *
 460  * Returns: the DSCP QoS value of the outgoing sockets, or -1 if disbled.
 461  */
 462 gint
 463 gst_rtsp_stream_get_dscp_qos (GstRTSPStream * stream)
 464 {
 465   GstRTSPStreamPrivate *priv;
 466
 467   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), -1);
 468
 469   priv = stream->priv;
 470
 471   return priv->dscp_qos;
 472 }
 473
 474
 475 /**
 476  * gst_rtsp_stream_set_address_pool:
 477  * @stream: a #GstRTSPStream
 478  * @pool: a #GstRTSPAddressPool
 479  *
 480  * configure @pool to be used as the address pool of @stream.
 481  */
 482 void
 483 gst_rtsp_stream_set_address_pool (GstRTSPStream * stream,
 484     GstRTSPAddressPool * pool)
 485 {
 486   GstRTSPStreamPrivate *priv;
 487   GstRTSPAddressPool *old;
 488
 489   g_return_if_fail (GST_IS_RTSP_STREAM (stream));
 490
 491   priv = stream->priv;
 492
 493   GST_LOG_OBJECT (stream, "set address pool %p", pool);
 494
 495   g_mutex_lock (&priv->lock);
 496   if ((old = priv->pool) != pool)
 497     priv->pool = pool ? g_object_ref (pool) : NULL;
 498   else
 499     old = NULL;
 500   g_mutex_unlock (&priv->lock);
 501
 502   if (old)
 503     g_object_unref (old);
 504 }
 505
 506 /**
 507  * gst_rtsp_stream_get_address_pool:
 508  * @stream: a #GstRTSPStream
 509  *
 510  * Get the #GstRTSPAddressPool used as the address pool of @stream.
 511  *
 512  * Returns: (transfer full): the #GstRTSPAddressPool of @stream. g_object_unref() after
 513  * usage.
 514  */
 515 GstRTSPAddressPool *
 516 gst_rtsp_stream_get_address_pool (GstRTSPStream * stream)
 517 {
 518   GstRTSPStreamPrivate *priv;
 519   GstRTSPAddressPool *result;
 520
 521   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), NULL);
 522
 523   priv = stream->priv;
 524
 525   g_mutex_lock (&priv->lock);
 526   if ((result = priv->pool))
 527     g_object_ref (result);
 528   g_mutex_unlock (&priv->lock);
 529
 530   return result;
 531 }
 532
 533 /**
 534  * gst_rtsp_stream_get_multicast_address:
 535  * @stream: a #GstRTSPStream
 536  * @family: the #GSocketFamily
 537  *
 538  * Get the multicast address of @stream for @family.
 539  *
 540  * Returns: the #GstRTSPAddress of @stream or %NULL when no address could be
 541  * allocated. gst_rtsp_address_free() after usage.
 542  */
 543 GstRTSPAddress *
 544 gst_rtsp_stream_get_multicast_address (GstRTSPStream * stream,
 545     GSocketFamily family)
 546 {
 547   GstRTSPStreamPrivate *priv;
 548   GstRTSPAddress *result;
 549   GstRTSPAddress **addrp;
 550   GstRTSPAddressFlags flags;
 551
 552   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), NULL);
 553
 554   priv = stream->priv;
 555
 556   if (family == G_SOCKET_FAMILY_IPV6) {
 557     flags = GST_RTSP_ADDRESS_FLAG_IPV6;
 558     addrp = &priv->addr_v4;
 559   } else {
 560     flags = GST_RTSP_ADDRESS_FLAG_IPV4;
 561     addrp = &priv->addr_v6;
 562   }
 563
 564   g_mutex_lock (&priv->lock);
 565   if (*addrp == NULL) {
 566     if (priv->pool == NULL)
 567       goto no_pool;
 568
 569     flags |= GST_RTSP_ADDRESS_FLAG_EVEN_PORT | GST_RTSP_ADDRESS_FLAG_MULTICAST;
 570
 571     *addrp = gst_rtsp_address_pool_acquire_address (priv->pool, flags, 2);
 572     if (*addrp == NULL)
 573       goto no_address;
 574   }
 575   result = gst_rtsp_address_copy (*addrp);
 576   g_mutex_unlock (&priv->lock);
 577
 578   return result;
 579
 580   /* ERRORS */
 581 no_pool:
 582   {
 583     GST_ERROR_OBJECT (stream, "no address pool specified");
 584     g_mutex_unlock (&priv->lock);
 585     return NULL;
 586   }
 587 no_address:
 588   {
 589     GST_ERROR_OBJECT (stream, "failed to acquire address from pool");
 590     g_mutex_unlock (&priv->lock);
 591     return NULL;
 592   }
 593 }
 594
 595 /**
 596  * gst_rtsp_stream_reserve_address:
 597  * @stream: a #GstRTSPStream
 598  * @address: an address
 599  * @port: a port
 600  * @n_ports: n_ports
 601  * @ttl: a TTL
 602  *
 603  * Reserve @address and @port as the address and port of @stream.
 604  *
 605  * Returns: the #GstRTSPAddress of @stream or %NULL when the address could be
 606  * reserved. gst_rtsp_address_free() after usage.
 607  */
 608 GstRTSPAddress *
 609 gst_rtsp_stream_reserve_address (GstRTSPStream * stream,
 610     const gchar * address, guint port, guint n_ports, guint ttl)
 611 {
 612   GstRTSPStreamPrivate *priv;
 613   GstRTSPAddress *result;
 614   GInetAddress *addr;
 615   GSocketFamily family;
 616   GstRTSPAddress **addrp;
 617
 618   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), NULL);
 619   g_return_val_if_fail (address != NULL, NULL);
 620   g_return_val_if_fail (port > 0, NULL);
 621   g_return_val_if_fail (n_ports > 0, NULL);
 622   g_return_val_if_fail (ttl > 0, NULL);
 623
 624   priv = stream->priv;
 625
 626   addr = g_inet_address_new_from_string (address);
 627   if (!addr) {
 628     GST_ERROR ("failed to get inet addr from %s", address);
 629     family = G_SOCKET_FAMILY_IPV4;
 630   } else {
 631     family = g_inet_address_get_family (addr);
 632     g_object_unref (addr);
 633   }
 634
 635   if (family == G_SOCKET_FAMILY_IPV6)
 636     addrp = &priv->addr_v4;
 637   else
 638     addrp = &priv->addr_v6;
 639
 640   g_mutex_lock (&priv->lock);
 641   if (*addrp == NULL) {
 642     if (priv->pool == NULL)
 643       goto no_pool;
 644
 645     *addrp = gst_rtsp_address_pool_reserve_address (priv->pool, address,
 646         port, n_ports, ttl);
 647     if (*addrp == NULL)
 648       goto no_address;
 649   } else {
 650     if (strcmp ((*addrp)->address, address) ||
 651         (*addrp)->port != port || (*addrp)->n_ports != n_ports ||
 652         (*addrp)->ttl != ttl)
 653       goto different_address;
 654   }
 655   result = gst_rtsp_address_copy (*addrp);
 656   g_mutex_unlock (&priv->lock);
 657
 658   return result;
 659
 660   /* ERRORS */
 661 no_pool:
 662   {
 663     GST_ERROR_OBJECT (stream, "no address pool specified");
 664     g_mutex_unlock (&priv->lock);
 665     return NULL;
 666   }
 667 no_address:
 668   {
 669     GST_ERROR_OBJECT (stream, "failed to acquire address %s from pool",
 670         address);
 671     g_mutex_unlock (&priv->lock);
 672     return NULL;
 673   }
 674 different_address:
 675   {
 676     GST_ERROR_OBJECT (stream, "address %s is not the same that was already"
 677         " reserved", address);
 678     g_mutex_unlock (&priv->lock);
 679     return NULL;
 680   }
 681 }
 682
 683 static gboolean
 684 alloc_ports_one_family (GstRTSPAddressPool * pool, gint buffer_size,
 685     GSocketFamily family, GstElement * udpsrc_out[2],
 686     GstElement * udpsink_out[2], GstRTSPRange * server_port_out,
 687     GstRTSPAddress ** server_addr_out)
 688 {
 689   GstStateChangeReturn ret;
 690   GstElement *udpsrc0, *udpsrc1;
 691   GstElement *udpsink0, *udpsink1;
 692   GSocket *rtp_socket = NULL;
 693   GSocket *rtcp_socket;
 694   gint tmp_rtp, tmp_rtcp;
 695   guint count;
 696   gint rtpport, rtcpport;
 697   GList *rejected_addresses = NULL;
 698   GstRTSPAddress *addr = NULL;
 699   GInetAddress *inetaddr = NULL;
 700   GSocketAddress *rtp_sockaddr = NULL;
 701   GSocketAddress *rtcp_sockaddr = NULL;
 702   const gchar *multisink_socket;
 703
 704   if (family == G_SOCKET_FAMILY_IPV6)
 705     multisink_socket = "socket-v6";
 706   else
 707     multisink_socket = "socket";
 708
 709   udpsrc0 = NULL;
 710   udpsrc1 = NULL;
 711   udpsink0 = NULL;
 712   udpsink1 = NULL;
 713   count = 0;
 714
 715   /* Start with random port */
 716   tmp_rtp = 0;
 717
 718   rtcp_socket = g_socket_new (family, G_SOCKET_TYPE_DATAGRAM,
 719       G_SOCKET_PROTOCOL_UDP, NULL);
 720   if (!rtcp_socket)
 721     goto no_udp_protocol;
 722
 723   if (*server_addr_out)
 724     gst_rtsp_address_free (*server_addr_out);
 725
 726   /* try to allocate 2 UDP ports, the RTP port should be an even
 727    * number and the RTCP port should be the next (uneven) port */
 728 again:
 729
 730   if (rtp_socket == NULL) {
 731     rtp_socket = g_socket_new (family, G_SOCKET_TYPE_DATAGRAM,
 732         G_SOCKET_PROTOCOL_UDP, NULL);
 733     if (!rtp_socket)
 734       goto no_udp_protocol;
 735   }
 736
 737   if (pool && gst_rtsp_address_pool_has_unicast_addresses (pool)) {
 738     GstRTSPAddressFlags flags;
 739
 740     if (addr)
 741       rejected_addresses = g_list_prepend (rejected_addresses, addr);
 742
 743     flags = GST_RTSP_ADDRESS_FLAG_EVEN_PORT | GST_RTSP_ADDRESS_FLAG_UNICAST;
 744     if (family == G_SOCKET_FAMILY_IPV6)
 745       flags |= GST_RTSP_ADDRESS_FLAG_IPV6;
 746     else
 747       flags |= GST_RTSP_ADDRESS_FLAG_IPV4;
 748
 749     addr = gst_rtsp_address_pool_acquire_address (pool, flags, 2);
 750
 751     if (addr == NULL)
 752       goto no_ports;
 753
 754     tmp_rtp = addr->port;
 755
 756     g_clear_object (&inetaddr);
 757     inetaddr = g_inet_address_new_from_string (addr->address);
 758   } else {
 759     if (tmp_rtp != 0) {
 760       tmp_rtp += 2;
 761       if (++count > 20)
 762         goto no_ports;
 763     }
 764
 765     if (inetaddr == NULL)
 766       inetaddr = g_inet_address_new_any (family);
 767   }
 768
 769   rtp_sockaddr = g_inet_socket_address_new (inetaddr, tmp_rtp);
 770   if (!g_socket_bind (rtp_socket, rtp_sockaddr, FALSE, NULL)) {
 771     g_object_unref (rtp_sockaddr);
 772     goto again;
 773   }
 774   g_object_unref (rtp_sockaddr);
 775
 776   rtp_sockaddr = g_socket_get_local_address (rtp_socket, NULL);
 777   if (rtp_sockaddr == NULL || !G_IS_INET_SOCKET_ADDRESS (rtp_sockaddr)) {
 778     g_clear_object (&rtp_sockaddr);
 779     goto socket_error;
 780   }
 781
 782   tmp_rtp =
 783       g_inet_socket_address_get_port (G_INET_SOCKET_ADDRESS (rtp_sockaddr));
 784   g_object_unref (rtp_sockaddr);
 785
 786   /* check if port is even */
 787   if ((tmp_rtp & 1) != 0) {
 788     /* port not even, close and allocate another */
 789     tmp_rtp++;
 790     g_clear_object (&rtp_socket);
 791     goto again;
 792   }
 793
 794   /* set port */
 795   tmp_rtcp = tmp_rtp + 1;
 796
 797   rtcp_sockaddr = g_inet_socket_address_new (inetaddr, tmp_rtcp);
 798   if (!g_socket_bind (rtcp_socket, rtcp_sockaddr, FALSE, NULL)) {
 799     g_object_unref (rtcp_sockaddr);
 800     g_clear_object (&rtp_socket);
 801     goto again;
 802   }
 803   g_object_unref (rtcp_sockaddr);
 804
 805   g_clear_object (&inetaddr);
 806
 807   udpsrc0 = gst_element_factory_make ("udpsrc", NULL);
 808   udpsrc1 = gst_element_factory_make ("udpsrc", NULL);
 809
 810   if (udpsrc0 == NULL || udpsrc1 == NULL)
 811     goto no_udp_protocol;
 812
 813   g_object_set (G_OBJECT (udpsrc0), "socket", rtp_socket, NULL);
 814   g_object_set (G_OBJECT (udpsrc1), "socket", rtcp_socket, NULL);
 815
 816   ret = gst_element_set_state (udpsrc0, GST_STATE_PAUSED);
 817   if (ret == GST_STATE_CHANGE_FAILURE)
 818     goto element_error;
 819   ret = gst_element_set_state (udpsrc1, GST_STATE_PAUSED);
 820   if (ret == GST_STATE_CHANGE_FAILURE)
 821     goto element_error;
 822
 823   /* all fine, do port check */
 824   g_object_get (G_OBJECT (udpsrc0), "port", &rtpport, NULL);
 825   g_object_get (G_OBJECT (udpsrc1), "port", &rtcpport, NULL);
 826
 827   /* this should not happen... */
 828   if (rtpport != tmp_rtp || rtcpport != tmp_rtcp)
 829     goto port_error;
 830
 831   if (udpsink_out[0])
 832     udpsink0 = udpsink_out[0];
 833   else
 834     udpsink0 = gst_element_factory_make ("multiudpsink", NULL);
 835
 836   if (!udpsink0)
 837     goto no_udp_protocol;
 838
 839   g_object_set (G_OBJECT (udpsink0), "close-socket", FALSE, NULL);
 840   g_object_set (G_OBJECT (udpsink0), multisink_socket, rtp_socket, NULL);
 841
 842   if (udpsink_out[1])
 843     udpsink1 = udpsink_out[1];
 844   else
 845     udpsink1 = gst_element_factory_make ("multiudpsink", NULL);
 846
 847   if (!udpsink1)
 848     goto no_udp_protocol;
 849
 850   g_object_set (G_OBJECT (udpsink0), "send-duplicates", FALSE, NULL);
 851   g_object_set (G_OBJECT (udpsink1), "send-duplicates", FALSE, NULL);
 852   g_object_set (G_OBJECT (udpsink0), "buffer-size", buffer_size, NULL);
 853
 854   g_object_set (G_OBJECT (udpsink1), "close-socket", FALSE, NULL);
 855   g_object_set (G_OBJECT (udpsink1), multisink_socket, rtcp_socket, NULL);
 856   g_object_set (G_OBJECT (udpsink1), "sync", FALSE, NULL);
 857   g_object_set (G_OBJECT (udpsink1), "async", FALSE, NULL);
 858   g_object_set (G_OBJECT (udpsink0), "auto-multicast", FALSE, NULL);
 859   g_object_set (G_OBJECT (udpsink0), "loop", FALSE, NULL);
 860   g_object_set (G_OBJECT (udpsink1), "auto-multicast", FALSE, NULL);
 861   g_object_set (G_OBJECT (udpsink1), "loop", FALSE, NULL);
 862
 863   /* we keep these elements, we will further configure them when the
 864    * client told us to really use the UDP ports. */
 865   udpsrc_out[0] = udpsrc0;
 866   udpsrc_out[1] = udpsrc1;
 867   udpsink_out[0] = udpsink0;
 868   udpsink_out[1] = udpsink1;
 869   server_port_out->min = rtpport;
 870   server_port_out->max = rtcpport;
 871
 872   *server_addr_out = addr;
 873   g_list_free_full (rejected_addresses, (GDestroyNotify) gst_rtsp_address_free);
 874
 875   g_object_unref (rtp_socket);
 876   g_object_unref (rtcp_socket);
 877
 878   return TRUE;
 879
 880   /* ERRORS */
 881 no_udp_protocol:
 882   {
 883     goto cleanup;
 884   }
 885 no_ports:
 886   {
 887     goto cleanup;
 888   }
 889 port_error:
 890   {
 891     goto cleanup;
 892   }
 893 socket_error:
 894   {
 895     goto cleanup;
 896   }
 897 element_error:
 898   {
 899     goto cleanup;
 900   }
 901 cleanup:
 902   {
 903     if (udpsrc0) {
 904       gst_element_set_state (udpsrc0, GST_STATE_NULL);
 905       gst_object_unref (udpsrc0);
 906     }
 907     if (udpsrc1) {
 908       gst_element_set_state (udpsrc1, GST_STATE_NULL);
 909       gst_object_unref (udpsrc1);
 910     }
 911     if (udpsink0) {
 912       gst_element_set_state (udpsink0, GST_STATE_NULL);
 913       gst_object_unref (udpsink0);
 914     }
 915     if (udpsink1) {
 916       gst_element_set_state (udpsink1, GST_STATE_NULL);
 917       gst_object_unref (udpsink1);
 918     }
 919     if (inetaddr)
 920       g_object_unref (inetaddr);
 921     g_list_free_full (rejected_addresses,
 922         (GDestroyNotify) gst_rtsp_address_free);
 923     if (addr)
 924       gst_rtsp_address_free (addr);
 925     if (rtp_socket)
 926       g_object_unref (rtp_socket);
 927     if (rtcp_socket)
 928       g_object_unref (rtcp_socket);
 929     return FALSE;
 930   }
 931 }
 932
 933 /* must be called with lock */
 934 static gboolean
 935 alloc_ports (GstRTSPStream * stream)
 936 {
 937   GstRTSPStreamPrivate *priv = stream->priv;
 938
 939   priv->have_ipv4 = alloc_ports_one_family (priv->pool, priv->buffer_size,
 940       G_SOCKET_FAMILY_IPV4, priv->udpsrc_v4, priv->udpsink,
 941       &priv->server_port_v4, &priv->server_addr_v4);
 942
 943   priv->have_ipv6 = alloc_ports_one_family (priv->pool, priv->buffer_size,
 944       G_SOCKET_FAMILY_IPV6, priv->udpsrc_v6, priv->udpsink,
 945       &priv->server_port_v6, &priv->server_addr_v6);
 946
 947   return priv->have_ipv4 || priv->have_ipv6;
 948 }
 949
 950 /**
 951  * gst_rtsp_stream_get_server_port:
 952  * @stream: a #GstRTSPStream
 953  * @server_port: (out): result server port
 954  * @family: the port family to get
 955  *
 956  * Fill @server_port with the port pair used by the server. This function can
 957  * only be called when @stream has been joined.
 958  */
 959 void
 960 gst_rtsp_stream_get_server_port (GstRTSPStream * stream,
 961     GstRTSPRange * server_port, GSocketFamily family)
 962 {
 963   GstRTSPStreamPrivate *priv;
 964
 965   g_return_if_fail (GST_IS_RTSP_STREAM (stream));
 966   priv = stream->priv;
 967   g_return_if_fail (priv->is_joined);
 968
 969   g_mutex_lock (&priv->lock);
 970   if (family == G_SOCKET_FAMILY_IPV4) {
 971     if (server_port)
 972       *server_port = priv->server_port_v4;
 973   } else {
 974     if (server_port)
 975       *server_port = priv->server_port_v6;
 976   }
 977   g_mutex_unlock (&priv->lock);
 978 }
 979
 980 /**
 981  * gst_rtsp_stream_get_rtpsession:
 982  * @stream: a #GstRTSPStream
 983  *
 984  * Get the RTP session of this stream.
 985  *
 986  * Returns: The RTP session of this stream. Unref after usage.
 987  */
 988 GObject *
 989 gst_rtsp_stream_get_rtpsession (GstRTSPStream * stream)
 990 {
 991   GstRTSPStreamPrivate *priv;
 992   GObject *session;
 993
 994   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), NULL);
 995
 996   priv = stream->priv;
 997
 998   g_mutex_lock (&priv->lock);
 999   if ((session = priv->session))
1000     g_object_ref (session);
1001   g_mutex_unlock (&priv->lock);
1002
1003   return session;
1004 }
1005
1006 /**
1007  * gst_rtsp_stream_get_ssrc:
1008  * @stream: a #GstRTSPStream
1009  * @ssrc: (out): result ssrc
1010  *
1011  * Get the SSRC used by the RTP session of this stream. This function can only
1012  * be called when @stream has been joined.
1013  */
1014 void
1015 gst_rtsp_stream_get_ssrc (GstRTSPStream * stream, guint * ssrc)
1016 {
1017   GstRTSPStreamPrivate *priv;
1018
1019   g_return_if_fail (GST_IS_RTSP_STREAM (stream));
1020   priv = stream->priv;
1021   g_return_if_fail (priv->is_joined);
1022
1023   g_mutex_lock (&priv->lock);
1024   if (ssrc && priv->session)
1025     g_object_get (priv->session, "internal-ssrc", ssrc, NULL);
1026   g_mutex_unlock (&priv->lock);
1027 }
1028
1029 /* executed from streaming thread */
1030 static void
1031 caps_notify (GstPad * pad, GParamSpec * unused, GstRTSPStream * stream)
1032 {
1033   GstRTSPStreamPrivate *priv = stream->priv;
1034   GstCaps *newcaps, *oldcaps;
1035
1036   newcaps = gst_pad_get_current_caps (pad);
1037
1038   GST_INFO ("stream %p received caps %p, %" GST_PTR_FORMAT, stream, newcaps,
1039       newcaps);
1040
1041   g_mutex_lock (&priv->lock);
1042   oldcaps = priv->caps;
1043   priv->caps = newcaps;
1044   g_mutex_unlock (&priv->lock);
1045
1046   if (oldcaps)
1047     gst_caps_unref (oldcaps);
1048 }
1049
1050 static void
1051 dump_structure (const GstStructure * s)
1052 {
1053   gchar *sstr;
1054
1055   sstr = gst_structure_to_string (s);
1056   GST_INFO ("structure: %s", sstr);
1057   g_free (sstr);
1058 }
1059
1060 static GstRTSPStreamTransport *
1061 find_transport (GstRTSPStream * stream, const gchar * rtcp_from)
1062 {
1063   GstRTSPStreamPrivate *priv = stream->priv;
1064   GList *walk;
1065   GstRTSPStreamTransport *result = NULL;
1066   const gchar *tmp;
1067   gchar *dest;
1068   guint port;
1069
1070   if (rtcp_from == NULL)
1071     return NULL;
1072
1073   tmp = g_strrstr (rtcp_from, ":");
1074   if (tmp == NULL)
1075     return NULL;
1076
1077   port = atoi (tmp + 1);
1078   dest = g_strndup (rtcp_from, tmp - rtcp_from);
1079
1080   g_mutex_lock (&priv->lock);
1081   GST_INFO ("finding %s:%d in %d transports", dest, port,
1082       g_list_length (priv->transports));
1083
1084   for (walk = priv->transports; walk; walk = g_list_next (walk)) {
1085     GstRTSPStreamTransport *trans = walk->data;
1086     const GstRTSPTransport *tr;
1087     gint min, max;
1088
1089     tr = gst_rtsp_stream_transport_get_transport (trans);
1090
1091     min = tr->client_port.min;
1092     max = tr->client_port.max;
1093
1094     if ((strcmp (tr->destination, dest) == 0) && (min == port || max == port)) {
1095       result = trans;
1096       break;
1097     }
1098   }
1099   if (result)
1100     g_object_ref (result);
1101   g_mutex_unlock (&priv->lock);
1102
1103   g_free (dest);
1104
1105   return result;
1106 }
1107
1108 static GstRTSPStreamTransport *
1109 check_transport (GObject * source, GstRTSPStream * stream)
1110 {
1111   GstStructure *stats;
1112   GstRTSPStreamTransport *trans;
1113
1114   /* see if we have a stream to match with the origin of the RTCP packet */
1115   trans = g_object_get_qdata (source, ssrc_stream_map_key);
1116   if (trans == NULL) {
1117     g_object_get (source, "stats", &stats, NULL);
1118     if (stats) {
1119       const gchar *rtcp_from;
1120
1121       dump_structure (stats);
1122
1123       rtcp_from = gst_structure_get_string (stats, "rtcp-from");
1124       if ((trans = find_transport (stream, rtcp_from))) {
1125         GST_INFO ("%p: found transport %p for source  %p", stream, trans,
1126             source);
1127         g_object_set_qdata_full (source, ssrc_stream_map_key, trans,
1128             g_object_unref);
1129       }
1130       gst_structure_free (stats);
1131     }
1132   }
1133   return trans;
1134 }
1135
1136
1137 static void
1138 on_new_ssrc (GObject * session, GObject * source, GstRTSPStream * stream)
1139 {
1140   GstRTSPStreamTransport *trans;
1141
1142   GST_INFO ("%p: new source %p", stream, source);
1143
1144   trans = check_transport (source, stream);
1145
1146   if (trans)
1147     GST_INFO ("%p: source %p for transport %p", stream, source, trans);
1148 }
1149
1150 static void
1151 on_ssrc_sdes (GObject * session, GObject * source, GstRTSPStream * stream)
1152 {
1153   GST_INFO ("%p: new SDES %p", stream, source);
1154 }
1155
1156 static void
1157 on_ssrc_active (GObject * session, GObject * source, GstRTSPStream * stream)
1158 {
1159   GstRTSPStreamTransport *trans;
1160
1161   trans = check_transport (source, stream);
1162
1163   if (trans) {
1164     GST_INFO ("%p: source %p in transport %p is active", stream, source, trans);
1165     gst_rtsp_stream_transport_keep_alive (trans);
1166   }
1167 #ifdef DUMP_STATS
1168   {
1169     GstStructure *stats;
1170     g_object_get (source, "stats", &stats, NULL);
1171     if (stats) {
1172       dump_structure (stats);
1173       gst_structure_free (stats);
1174     }
1175   }
1176 #endif
1177 }
1178
1179 static void
1180 on_bye_ssrc (GObject * session, GObject * source, GstRTSPStream * stream)
1181 {
1182   GST_INFO ("%p: source %p bye", stream, source);
1183 }
1184
1185 static void
1186 on_bye_timeout (GObject * session, GObject * source, GstRTSPStream * stream)
1187 {
1188   GstRTSPStreamTransport *trans;
1189
1190   GST_INFO ("%p: source %p bye timeout", stream, source);
1191
1192   if ((trans = g_object_get_qdata (source, ssrc_stream_map_key))) {
1193     gst_rtsp_stream_transport_set_timed_out (trans, TRUE);
1194     g_object_set_qdata (source, ssrc_stream_map_key, NULL);
1195   }
1196 }
1197
1198 static void
1199 on_timeout (GObject * session, GObject * source, GstRTSPStream * stream)
1200 {
1201   GstRTSPStreamTransport *trans;
1202
1203   GST_INFO ("%p: source %p timeout", stream, source);
1204
1205   if ((trans = g_object_get_qdata (source, ssrc_stream_map_key))) {
1206     gst_rtsp_stream_transport_set_timed_out (trans, TRUE);
1207     g_object_set_qdata (source, ssrc_stream_map_key, NULL);
1208   }
1209 }
1210
1211 static GstFlowReturn
1212 handle_new_sample (GstAppSink * sink, gpointer user_data)
1213 {
1214   GstRTSPStreamPrivate *priv;
1215   GList *walk;
1216   GstSample *sample;
1217   GstBuffer *buffer;
1218   GstRTSPStream *stream;
1219
1220   sample = gst_app_sink_pull_sample (sink);
1221   if (!sample)
1222     return GST_FLOW_OK;
1223
1224   stream = (GstRTSPStream *) user_data;
1225   priv = stream->priv;
1226   buffer = gst_sample_get_buffer (sample);
1227
1228   g_mutex_lock (&priv->lock);
1229   for (walk = priv->transports; walk; walk = g_list_next (walk)) {
1230     GstRTSPStreamTransport *tr = (GstRTSPStreamTransport *) walk->data;
1231
1232     if (GST_ELEMENT_CAST (sink) == priv->appsink[0]) {
1233       gst_rtsp_stream_transport_send_rtp (tr, buffer);
1234     } else {
1235       gst_rtsp_stream_transport_send_rtcp (tr, buffer);
1236     }
1237   }
1238   g_mutex_unlock (&priv->lock);
1239
1240   gst_sample_unref (sample);
1241
1242   return GST_FLOW_OK;
1243 }
1244
1245 static GstAppSinkCallbacks sink_cb = {
1246   NULL,                         /* not interested in EOS */
1247   NULL,                         /* not interested in preroll samples */
1248   handle_new_sample,
1249 };
1250
1251 /**
1252  * gst_rtsp_stream_join_bin:
1253  * @stream: a #GstRTSPStream
1254  * @bin: a #GstBin to join
1255  * @rtpbin: a rtpbin element in @bin
1256  * @state: the target state of the new elements
1257  *
1258  * Join the #Gstbin @bin that contains the element @rtpbin.
1259  *
1260  * @stream will link to @rtpbin, which must be inside @bin. The elements
1261  * added to @bin will be set to the state given in @state.
1262  *
1263  * Returns: %TRUE on success.
1264  */
1265 gboolean
1266 gst_rtsp_stream_join_bin (GstRTSPStream * stream, GstBin * bin,
1267     GstElement * rtpbin, GstState state)
1268 {
1269   GstRTSPStreamPrivate *priv;
1270   gint i;
1271   guint idx;
1272   gchar *name;
1273   GstPad *pad, *teepad, *queuepad, *selpad;
1274   GstPadLinkReturn ret;
1275
1276   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), FALSE);
1277   g_return_val_if_fail (GST_IS_BIN (bin), FALSE);
1278   g_return_val_if_fail (GST_IS_ELEMENT (rtpbin), FALSE);
1279
1280   priv = stream->priv;
1281
1282   g_mutex_lock (&priv->lock);
1283   if (priv->is_joined)
1284     goto was_joined;
1285
1286   /* create a session with the same index as the stream */
1287   idx = priv->idx;
1288
1289   GST_INFO ("stream %p joining bin as session %u", stream, idx);
1290
1291   if (!alloc_ports (stream))
1292     goto no_ports;
1293
1294   /* update the dscp qos field in the sinks */
1295   update_dscp_qos (stream);
1296
1297   /* get a pad for sending RTP */
1298   name = g_strdup_printf ("send_rtp_sink_%u", idx);
1299   priv->send_rtp_sink = gst_element_get_request_pad (rtpbin, name);
1300   g_free (name);
1301   /* link the RTP pad to the session manager, it should not really fail unless
1302    * this is not really an RTP pad */
1303   ret = gst_pad_link (priv->srcpad, priv->send_rtp_sink);
1304   if (ret != GST_PAD_LINK_OK)
1305     goto link_failed;
1306
1307   /* get pads from the RTP session element for sending and receiving
1308    * RTP/RTCP*/
1309   name = g_strdup_printf ("send_rtp_src_%u", idx);
1310   priv->send_src[0] = gst_element_get_static_pad (rtpbin, name);
1311   g_free (name);
1312   name = g_strdup_printf ("send_rtcp_src_%u", idx);
1313   priv->send_src[1] = gst_element_get_request_pad (rtpbin, name);
1314   g_free (name);
1315   name = g_strdup_printf ("recv_rtp_sink_%u", idx);
1316   priv->recv_sink[0] = gst_element_get_request_pad (rtpbin, name);
1317   g_free (name);
1318   name = g_strdup_printf ("recv_rtcp_sink_%u", idx);
1319   priv->recv_sink[1] = gst_element_get_request_pad (rtpbin, name);
1320   g_free (name);
1321
1322   /* get the session */
1323   g_signal_emit_by_name (rtpbin, "get-internal-session", idx, &priv->session);
1324
1325   g_signal_connect (priv->session, "on-new-ssrc", (GCallback) on_new_ssrc,
1326       stream);
1327   g_signal_connect (priv->session, "on-ssrc-sdes", (GCallback) on_ssrc_sdes,
1328       stream);
1329   g_signal_connect (priv->session, "on-ssrc-active",
1330       (GCallback) on_ssrc_active, stream);
1331   g_signal_connect (priv->session, "on-bye-ssrc", (GCallback) on_bye_ssrc,
1332       stream);
1333   g_signal_connect (priv->session, "on-bye-timeout",
1334       (GCallback) on_bye_timeout, stream);
1335   g_signal_connect (priv->session, "on-timeout", (GCallback) on_timeout,
1336       stream);
1337
1338   for (i = 0; i < 2; i++) {
1339     /* For the sender we create this bit of pipeline for both
1340      * RTP and RTCP. Sync and preroll are enabled on udpsink so
1341      * we need to add a queue before appsink to make the pipeline
1342      * not block. For the TCP case, we want to pump data to the
1343      * client as fast as possible anyway.
1344      *
1345      * .--------.      .-----.    .---------.
1346      * | rtpbin |      | tee |    | udpsink |
1347      * |       send->sink   src->sink       |
1348      * '--------'      |     |    '---------'
1349      *                 |     |    .---------.    .---------.
1350      *                 |     |    |  queue  |    | appsink |
1351      *                 |    src->sink      src->sink       |
1352      *                 '-----'    '---------'    '---------'
1353      */
1354     /* make tee for RTP/RTCP */
1355     priv->tee[i] = gst_element_factory_make ("tee", NULL);
1356     gst_bin_add (bin, priv->tee[i]);
1357
1358     /* and link to rtpbin send pad */
1359     pad = gst_element_get_static_pad (priv->tee[i], "sink");
1360     gst_pad_link (priv->send_src[i], pad);
1361     gst_object_unref (pad);
1362
1363     /* add udpsink */
1364     gst_bin_add (bin, priv->udpsink[i]);
1365
1366     /* link tee to udpsink */
1367     teepad = gst_element_get_request_pad (priv->tee[i], "src_%u");
1368     pad = gst_element_get_static_pad (priv->udpsink[i], "sink");
1369     gst_pad_link (teepad, pad);
1370     gst_object_unref (pad);
1371     gst_object_unref (teepad);
1372
1373     /* make queue */
1374     priv->appqueue[i] = gst_element_factory_make ("queue", NULL);
1375     gst_bin_add (bin, priv->appqueue[i]);
1376     /* and link to tee */
1377     teepad = gst_element_get_request_pad (priv->tee[i], "src_%u");
1378     pad = gst_element_get_static_pad (priv->appqueue[i], "sink");
1379     gst_pad_link (teepad, pad);
1380     gst_object_unref (pad);
1381     gst_object_unref (teepad);
1382
1383     /* make appsink */
1384     priv->appsink[i] = gst_element_factory_make ("appsink", NULL);
1385     g_object_set (priv->appsink[i], "async", FALSE, "sync", FALSE, NULL);
1386     g_object_set (priv->appsink[i], "emit-signals", FALSE, NULL);
1387     gst_bin_add (bin, priv->appsink[i]);
1388     gst_app_sink_set_callbacks (GST_APP_SINK_CAST (priv->appsink[i]),
1389         &sink_cb, stream, NULL);
1390     /* and link to queue */
1391     queuepad = gst_element_get_static_pad (priv->appqueue[i], "src");
1392     pad = gst_element_get_static_pad (priv->appsink[i], "sink");
1393     gst_pad_link (queuepad, pad);
1394     gst_object_unref (pad);
1395     gst_object_unref (queuepad);
1396
1397     /* For the receiver we create this bit of pipeline for both
1398      * RTP and RTCP. We receive RTP/RTCP on appsrc and udpsrc
1399      * and it is all funneled into the rtpbin receive pad.
1400      *
1401      * .--------.     .--------.    .--------.
1402      * | udpsrc |     | funnel |    | rtpbin |
1403      * |       src->sink      src->sink      |
1404      * '--------'     |        |    '--------'
1405      * .--------.     |        |
1406      * | appsrc |     |        |
1407      * |       src->sink       |
1408      * '--------'     '--------'
1409      */
1410     /* make funnel for the RTP/RTCP receivers */
1411     priv->funnel[i] = gst_element_factory_make ("funnel", NULL);
1412     gst_bin_add (bin, priv->funnel[i]);
1413
1414     pad = gst_element_get_static_pad (priv->funnel[i], "src");
1415     gst_pad_link (pad, priv->recv_sink[i]);
1416     gst_object_unref (pad);
1417
1418     if (priv->udpsrc_v4[i]) {
1419       /* we set and keep these to playing so that they don't cause NO_PREROLL return
1420        * values */
1421       gst_element_set_state (priv->udpsrc_v4[i], GST_STATE_PLAYING);
1422       gst_element_set_locked_state (priv->udpsrc_v4[i], TRUE);
1423       /* add udpsrc */
1424       gst_bin_add (bin, priv->udpsrc_v4[i]);
1425
1426       /* and link to the funnel v4 */
1427       selpad = gst_element_get_request_pad (priv->funnel[i], "sink_%u");
1428       pad = gst_element_get_static_pad (priv->udpsrc_v4[i], "src");
1429       gst_pad_link (pad, selpad);
1430       gst_object_unref (pad);
1431       gst_object_unref (selpad);
1432     }
1433
1434     if (priv->udpsrc_v6[i]) {
1435       gst_element_set_state (priv->udpsrc_v6[i], GST_STATE_PLAYING);
1436       gst_element_set_locked_state (priv->udpsrc_v6[i], TRUE);
1437       gst_bin_add (bin, priv->udpsrc_v6[i]);
1438
1439       /* and link to the funnel v6 */
1440       selpad = gst_element_get_request_pad (priv->funnel[i], "sink_%u");
1441       pad = gst_element_get_static_pad (priv->udpsrc_v6[i], "src");
1442       gst_pad_link (pad, selpad);
1443       gst_object_unref (pad);
1444       gst_object_unref (selpad);
1445     }
1446
1447     /* make and add appsrc */
1448     priv->appsrc[i] = gst_element_factory_make ("appsrc", NULL);
1449     gst_bin_add (bin, priv->appsrc[i]);
1450     /* and link to the funnel */
1451     selpad = gst_element_get_request_pad (priv->funnel[i], "sink_%u");
1452     pad = gst_element_get_static_pad (priv->appsrc[i], "src");
1453     gst_pad_link (pad, selpad);
1454     gst_object_unref (pad);
1455     gst_object_unref (selpad);
1456
1457     /* check if we need to set to a special state */
1458     if (state != GST_STATE_NULL) {
1459       gst_element_set_state (priv->udpsink[i], state);
1460       gst_element_set_state (priv->appsink[i], state);
1461       gst_element_set_state (priv->appqueue[i], state);
1462       gst_element_set_state (priv->tee[i], state);
1463       gst_element_set_state (priv->funnel[i], state);
1464       gst_element_set_state (priv->appsrc[i], state);
1465     }
1466   }
1467
1468   /* be notified of caps changes */
1469   priv->caps_sig = g_signal_connect (priv->send_rtp_sink, "notify::caps",
1470       (GCallback) caps_notify, stream);
1471
1472   priv->is_joined = TRUE;
1473   g_mutex_unlock (&priv->lock);
1474
1475   return TRUE;
1476
1477   /* ERRORS */
1478 was_joined:
1479   {
1480     g_mutex_unlock (&priv->lock);
1481     return TRUE;
1482   }
1483 no_ports:
1484   {
1485     g_mutex_unlock (&priv->lock);
1486     GST_WARNING ("failed to allocate ports %u", idx);
1487     return FALSE;
1488   }
1489 link_failed:
1490   {
1491     GST_WARNING ("failed to link stream %u", idx);
1492     gst_object_unref (priv->send_rtp_sink);
1493     priv->send_rtp_sink = NULL;
1494     g_mutex_unlock (&priv->lock);
1495     return FALSE;
1496   }
1497 }
1498
1499 /**
1500  * gst_rtsp_stream_leave_bin:
1501  * @stream: a #GstRTSPStream
1502  * @bin: a #GstBin
1503  * @rtpbin: a rtpbin #GstElement
1504  *
1505  * Remove the elements of @stream from @bin.
1506  *
1507  * Return: %TRUE on success.
1508  */
1509 gboolean
1510 gst_rtsp_stream_leave_bin (GstRTSPStream * stream, GstBin * bin,
1511     GstElement * rtpbin)
1512 {
1513   GstRTSPStreamPrivate *priv;
1514   gint i;
1515
1516   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), FALSE);
1517   g_return_val_if_fail (GST_IS_BIN (bin), FALSE);
1518   g_return_val_if_fail (GST_IS_ELEMENT (rtpbin), FALSE);
1519
1520   priv = stream->priv;
1521
1522   g_mutex_lock (&priv->lock);
1523   if (!priv->is_joined)
1524     goto was_not_joined;
1525
1526   /* all transports must be removed by now */
1527   g_return_val_if_fail (priv->transports == NULL, FALSE);
1528
1529   GST_INFO ("stream %p leaving bin", stream);
1530
1531   gst_pad_unlink (priv->srcpad, priv->send_rtp_sink);
1532   g_signal_handler_disconnect (priv->send_rtp_sink, priv->caps_sig);
1533   gst_element_release_request_pad (rtpbin, priv->send_rtp_sink);
1534   gst_object_unref (priv->send_rtp_sink);
1535   priv->send_rtp_sink = NULL;
1536
1537   for (i = 0; i < 2; i++) {
1538     gst_element_set_state (priv->udpsink[i], GST_STATE_NULL);
1539     gst_element_set_state (priv->appsink[i], GST_STATE_NULL);
1540     gst_element_set_state (priv->appqueue[i], GST_STATE_NULL);
1541     gst_element_set_state (priv->tee[i], GST_STATE_NULL);
1542     gst_element_set_state (priv->funnel[i], GST_STATE_NULL);
1543     gst_element_set_state (priv->appsrc[i], GST_STATE_NULL);
1544     if (priv->udpsrc_v4[i]) {
1545       /* and set udpsrc to NULL now before removing */
1546       gst_element_set_locked_state (priv->udpsrc_v4[i], FALSE);
1547       gst_element_set_state (priv->udpsrc_v4[i], GST_STATE_NULL);
1548       /* removing them should also nicely release the request
1549        * pads when they finalize */
1550       gst_bin_remove (bin, priv->udpsrc_v4[i]);
1551     }
1552     if (priv->udpsrc_v6[i]) {
1553       gst_element_set_locked_state (priv->udpsrc_v6[i], FALSE);
1554       gst_element_set_state (priv->udpsrc_v6[i], GST_STATE_NULL);
1555       gst_bin_remove (bin, priv->udpsrc_v6[i]);
1556     }
1557     gst_bin_remove (bin, priv->udpsink[i]);
1558     gst_bin_remove (bin, priv->appsrc[i]);
1559     gst_bin_remove (bin, priv->appsink[i]);
1560     gst_bin_remove (bin, priv->appqueue[i]);
1561     gst_bin_remove (bin, priv->tee[i]);
1562     gst_bin_remove (bin, priv->funnel[i]);
1563
1564     gst_element_release_request_pad (rtpbin, priv->recv_sink[i]);
1565     gst_object_unref (priv->recv_sink[i]);
1566     priv->recv_sink[i] = NULL;
1567
1568     priv->udpsrc_v4[i] = NULL;
1569     priv->udpsrc_v6[i] = NULL;
1570     priv->udpsink[i] = NULL;
1571     priv->appsrc[i] = NULL;
1572     priv->appsink[i] = NULL;
1573     priv->appqueue[i] = NULL;
1574     priv->tee[i] = NULL;
1575     priv->funnel[i] = NULL;
1576   }
1577   gst_object_unref (priv->send_src[0]);
1578   priv->send_src[0] = NULL;
1579
1580   gst_element_release_request_pad (rtpbin, priv->send_src[1]);
1581   gst_object_unref (priv->send_src[1]);
1582   priv->send_src[1] = NULL;
1583
1584   g_object_unref (priv->session);
1585   priv->session = NULL;
1586   if (priv->caps)
1587     gst_caps_unref (priv->caps);
1588   priv->caps = NULL;
1589
1590   priv->is_joined = FALSE;
1591   g_mutex_unlock (&priv->lock);
1592
1593   return TRUE;
1594
1595 was_not_joined:
1596   {
1597     return TRUE;
1598   }
1599 }
1600
1601 /**
1602  * gst_rtsp_stream_get_rtpinfo:
1603  * @stream: a #GstRTSPStream
1604  * @rtptime: result RTP timestamp
1605  * @seq: result RTP seqnum
1606  *
1607  * Retrieve the current rtptime and seq. This is used to
1608  * construct a RTPInfo reply header.
1609  *
1610  * Returns: %TRUE when rtptime and seq could be determined.
1611  */
1612 gboolean
1613 gst_rtsp_stream_get_rtpinfo (GstRTSPStream * stream,
1614     guint * rtptime, guint * seq)
1615 {
1616   GstRTSPStreamPrivate *priv;
1617   GObjectClass *payobjclass;
1618
1619   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), FALSE);
1620   g_return_val_if_fail (rtptime != NULL, FALSE);
1621   g_return_val_if_fail (seq != NULL, FALSE);
1622
1623   priv = stream->priv;
1624
1625   payobjclass = G_OBJECT_GET_CLASS (priv->payloader);
1626
1627   if (!g_object_class_find_property (payobjclass, "seqnum") ||
1628       !g_object_class_find_property (payobjclass, "timestamp"))
1629     return FALSE;
1630
1631   g_object_get (priv->payloader, "seqnum", seq, "timestamp", rtptime, NULL);
1632
1633   return TRUE;
1634 }
1635
1636 /**
1637  * gst_rtsp_stream_get_caps:
1638  * @stream: a #GstRTSPStream
1639  *
1640  * Retrieve the current caps of @stream.
1641  *
1642  * Returns: (transfer full): the #GstCaps of @stream. use gst_caps_unref()
1643  *    after usage.
1644  */
1645 GstCaps *
1646 gst_rtsp_stream_get_caps (GstRTSPStream * stream)
1647 {
1648   GstRTSPStreamPrivate *priv;
1649   GstCaps *result;
1650
1651   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), NULL);
1652
1653   priv = stream->priv;
1654
1655   g_mutex_lock (&priv->lock);
1656   if ((result = priv->caps))
1657     gst_caps_ref (result);
1658   g_mutex_unlock (&priv->lock);
1659
1660   return result;
1661 }
1662
1663 /**
1664  * gst_rtsp_stream_recv_rtp:
1665  * @stream: a #GstRTSPStream
1666  * @buffer: (transfer full): a #GstBuffer
1667  *
1668  * Handle an RTP buffer for the stream. This method is usually called when a
1669  * message has been received from a client using the TCP transport.
1670  *
1671  * This function takes ownership of @buffer.
1672  *
1673  * Returns: a GstFlowReturn.
1674  */
1675 GstFlowReturn
1676 gst_rtsp_stream_recv_rtp (GstRTSPStream * stream, GstBuffer * buffer)
1677 {
1678   GstRTSPStreamPrivate *priv;
1679   GstFlowReturn ret;
1680   GstElement *element;
1681
1682   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), GST_FLOW_ERROR);
1683   priv = stream->priv;
1684   g_return_val_if_fail (GST_IS_BUFFER (buffer), GST_FLOW_ERROR);
1685   g_return_val_if_fail (priv->is_joined, FALSE);
1686
1687   g_mutex_lock (&priv->lock);
1688   element = gst_object_ref (priv->appsrc[0]);
1689   g_mutex_unlock (&priv->lock);
1690
1691   ret = gst_app_src_push_buffer (GST_APP_SRC_CAST (element), buffer);
1692
1693   gst_object_unref (element);
1694
1695   return ret;
1696 }
1697
1698 /**
1699  * gst_rtsp_stream_recv_rtcp:
1700  * @stream: a #GstRTSPStream
1701  * @buffer: (transfer full): a #GstBuffer
1702  *
1703  * Handle an RTCP buffer for the stream. This method is usually called when a
1704  * message has been received from a client using the TCP transport.
1705  *
1706  * This function takes ownership of @buffer.
1707  *
1708  * Returns: a GstFlowReturn.
1709  */
1710 GstFlowReturn
1711 gst_rtsp_stream_recv_rtcp (GstRTSPStream * stream, GstBuffer * buffer)
1712 {
1713   GstRTSPStreamPrivate *priv;
1714   GstFlowReturn ret;
1715   GstElement *element;
1716
1717   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), GST_FLOW_ERROR);
1718   priv = stream->priv;
1719   g_return_val_if_fail (GST_IS_BUFFER (buffer), GST_FLOW_ERROR);
1720   g_return_val_if_fail (priv->is_joined, FALSE);
1721
1722   g_mutex_lock (&priv->lock);
1723   element = gst_object_ref (priv->appsrc[1]);
1724   g_mutex_unlock (&priv->lock);
1725
1726   ret = gst_app_src_push_buffer (GST_APP_SRC_CAST (element), buffer);
1727
1728   gst_object_unref (element);
1729
1730   return ret;
1731 }
1732
1733 /* must be called with lock */
1734 static gboolean
1735 update_transport (GstRTSPStream * stream, GstRTSPStreamTransport * trans,
1736     gboolean add)
1737 {
1738   GstRTSPStreamPrivate *priv = stream->priv;
1739   const GstRTSPTransport *tr;
1740
1741   tr = gst_rtsp_stream_transport_get_transport (trans);
1742
1743   switch (tr->lower_transport) {
1744     case GST_RTSP_LOWER_TRANS_UDP:
1745     case GST_RTSP_LOWER_TRANS_UDP_MCAST:
1746     {
1747       gchar *dest;
1748       gint min, max;
1749       guint ttl = 0;
1750
1751       dest = tr->destination;
1752       if (tr->lower_transport == GST_RTSP_LOWER_TRANS_UDP_MCAST) {
1753         min = tr->port.min;
1754         max = tr->port.max;
1755         ttl = tr->ttl;
1756       } else {
1757         min = tr->client_port.min;
1758         max = tr->client_port.max;
1759       }
1760
1761       if (add) {
1762         GST_INFO ("adding %s:%d-%d", dest, min, max);
1763         g_signal_emit_by_name (priv->udpsink[0], "add", dest, min, NULL);
1764         g_signal_emit_by_name (priv->udpsink[1], "add", dest, max, NULL);
1765         if (ttl > 0) {
1766           GST_INFO ("setting ttl-mc %d", ttl);
1767           g_object_set (G_OBJECT (priv->udpsink[0]), "ttl-mc", ttl, NULL);
1768           g_object_set (G_OBJECT (priv->udpsink[1]), "ttl-mc", ttl, NULL);
1769         }
1770         priv->transports = g_list_prepend (priv->transports, trans);
1771       } else {
1772         GST_INFO ("removing %s:%d-%d", dest, min, max);
1773         g_signal_emit_by_name (priv->udpsink[0], "remove", dest, min, NULL);
1774         g_signal_emit_by_name (priv->udpsink[1], "remove", dest, max, NULL);
1775         priv->transports = g_list_remove (priv->transports, trans);
1776       }
1777       break;
1778     }
1779     case GST_RTSP_LOWER_TRANS_TCP:
1780       if (add) {
1781         GST_INFO ("adding TCP %s", tr->destination);
1782         priv->transports = g_list_prepend (priv->transports, trans);
1783       } else {
1784         GST_INFO ("removing TCP %s", tr->destination);
1785         priv->transports = g_list_remove (priv->transports, trans);
1786       }
1787       break;
1788     default:
1789       goto unknown_transport;
1790   }
1791   return TRUE;
1792
1793   /* ERRORS */
1794 unknown_transport:
1795   {
1796     GST_INFO ("Unknown transport %d", tr->lower_transport);
1797     return FALSE;
1798   }
1799 }
1800
1801
1802 /**
1803  * gst_rtsp_stream_add_transport:
1804  * @stream: a #GstRTSPStream
1805  * @trans: a #GstRTSPStreamTransport
1806  *
1807  * Add the transport in @trans to @stream. The media of @stream will
1808  * then also be send to the values configured in @trans.
1809  *
1810  * @stream must be joined to a bin.
1811  *
1812  * @trans must contain a valid #GstRTSPTransport.
1813  *
1814  * Returns: %TRUE if @trans was added
1815  */
1816 gboolean
1817 gst_rtsp_stream_add_transport (GstRTSPStream * stream,
1818     GstRTSPStreamTransport * trans)
1819 {
1820   GstRTSPStreamPrivate *priv;
1821   gboolean res;
1822
1823   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), FALSE);
1824   priv = stream->priv;
1825   g_return_val_if_fail (GST_IS_RTSP_STREAM_TRANSPORT (trans), FALSE);
1826   g_return_val_if_fail (priv->is_joined, FALSE);
1827
1828   g_mutex_lock (&priv->lock);
1829   res = update_transport (stream, trans, TRUE);
1830   g_mutex_unlock (&priv->lock);
1831
1832   return res;
1833 }
1834
1835 /**
1836  * gst_rtsp_stream_remove_transport:
1837  * @stream: a #GstRTSPStream
1838  * @trans: a #GstRTSPStreamTransport
1839  *
1840  * Remove the transport in @trans from @stream. The media of @stream will
1841  * not be sent to the values configured in @trans.
1842  *
1843  * @stream must be joined to a bin.
1844  *
1845  * @trans must contain a valid #GstRTSPTransport.
1846  *
1847  * Returns: %TRUE if @trans was removed
1848  */
1849 gboolean
1850 gst_rtsp_stream_remove_transport (GstRTSPStream * stream,
1851     GstRTSPStreamTransport * trans)
1852 {
1853   GstRTSPStreamPrivate *priv;
1854   gboolean res;
1855
1856   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), FALSE);
1857   priv = stream->priv;
1858   g_return_val_if_fail (GST_IS_RTSP_STREAM_TRANSPORT (trans), FALSE);
1859   g_return_val_if_fail (priv->is_joined, FALSE);
1860
1861   g_mutex_lock (&priv->lock);
1862   res = update_transport (stream, trans, FALSE);
1863   g_mutex_unlock (&priv->lock);
1864
1865   return res;
1866 }