subprojects/gst-plugins-bad/ext/webrtc/webrtcdatachannel.c

   1 /* GStreamer
   2  * Copyright (C) 2018 Matthew Waters <matthew@centricular.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 /**
  21  * SECTION:gstwebrtc-datachannel
  22  * @short_description: RTCDataChannel object
  23  * @title: GstWebRTCDataChannel
  24  * @see_also: #GstWebRTCRTPTransceiver
  25  *
  26  * <http://w3c.github.io/webrtc-pc/#dom-rtcsctptransport>
  27  */
  28
  29 #ifdef HAVE_CONFIG_H
  30 # include "config.h"
  31 #endif
  32
  33 #include "webrtcdatachannel.h"
  34 #include <gst/app/gstappsink.h>
  35 #include <gst/app/gstappsrc.h>
  36 #include <gst/base/gstbytereader.h>
  37 #include <gst/base/gstbytewriter.h>
  38 #include <gst/sctp/sctpreceivemeta.h>
  39 #include <gst/sctp/sctpsendmeta.h>
  40
  41 #include "gstwebrtcbin.h"
  42 #include "utils.h"
  43
  44 #define GST_CAT_DEFAULT webrtc_data_channel_debug
  45 GST_DEBUG_CATEGORY_STATIC (GST_CAT_DEFAULT);
  46
  47 #define webrtc_data_channel_parent_class parent_class
  48 G_DEFINE_TYPE_WITH_CODE (WebRTCDataChannel, webrtc_data_channel,
  49     GST_TYPE_WEBRTC_DATA_CHANNEL,
  50     GST_DEBUG_CATEGORY_INIT (webrtc_data_channel_debug, "webrtcdatachannel", 0,
  51         "webrtcdatachannel"););
  52
  53 G_LOCK_DEFINE_STATIC (outstanding_channels_lock);
  54 static GList *outstanding_channels;
  55
  56 typedef enum
  57 {
  58   DATA_CHANNEL_PPID_WEBRTC_CONTROL = 50,
  59   DATA_CHANNEL_PPID_WEBRTC_STRING = 51,
  60   DATA_CHANNEL_PPID_WEBRTC_BINARY_PARTIAL = 52, /* deprecated */
  61   DATA_CHANNEL_PPID_WEBRTC_BINARY = 53,
  62   DATA_CHANNEL_PPID_WEBRTC_STRING_PARTIAL = 54, /* deprecated */
  63   DATA_CHANNEL_PPID_WEBRTC_BINARY_EMPTY = 56,
  64   DATA_CHANNEL_PPID_WEBRTC_STRING_EMPTY = 57,
  65 } DataChannelPPID;
  66
  67 typedef enum
  68 {
  69   CHANNEL_TYPE_RELIABLE = 0x00,
  70   CHANNEL_TYPE_RELIABLE_UNORDERED = 0x80,
  71   CHANNEL_TYPE_PARTIAL_RELIABLE_REXMIT = 0x01,
  72   CHANNEL_TYPE_PARTIAL_RELIABLE_REXMIT_UNORDERED = 0x81,
  73   CHANNEL_TYPE_PARTIAL_RELIABLE_TIMED = 0x02,
  74   CHANNEL_TYPE_PARTIAL_RELIABLE_TIMED_UNORDERED = 0x82,
  75 } DataChannelReliabilityType;
  76
  77 typedef enum
  78 {
  79   CHANNEL_MESSAGE_ACK = 0x02,
  80   CHANNEL_MESSAGE_OPEN = 0x03,
  81 } DataChannelMessage;
  82
  83 static guint16
  84 priority_type_to_uint (GstWebRTCPriorityType pri)
  85 {
  86   switch (pri) {
  87     case GST_WEBRTC_PRIORITY_TYPE_VERY_LOW:
  88       return 64;
  89     case GST_WEBRTC_PRIORITY_TYPE_LOW:
  90       return 192;
  91     case GST_WEBRTC_PRIORITY_TYPE_MEDIUM:
  92       return 384;
  93     case GST_WEBRTC_PRIORITY_TYPE_HIGH:
  94       return 768;
  95   }
  96   g_assert_not_reached ();
  97   return 0;
  98 }
  99
 100 static GstWebRTCPriorityType
 101 priority_uint_to_type (guint16 val)
 102 {
 103   if (val <= 128)
 104     return GST_WEBRTC_PRIORITY_TYPE_VERY_LOW;
 105   if (val <= 256)
 106     return GST_WEBRTC_PRIORITY_TYPE_LOW;
 107   if (val <= 512)
 108     return GST_WEBRTC_PRIORITY_TYPE_MEDIUM;
 109   return GST_WEBRTC_PRIORITY_TYPE_HIGH;
 110 }
 111
 112 static GstBuffer *
 113 construct_open_packet (WebRTCDataChannel * channel)
 114 {
 115   GstByteWriter w;
 116   gsize label_len = strlen (channel->parent.label);
 117   gsize proto_len = strlen (channel->parent.protocol);
 118   gsize size = 12 + label_len + proto_len;
 119   DataChannelReliabilityType reliability = 0;
 120   guint32 reliability_param = 0;
 121   guint16 priority;
 122   GstBuffer *buf;
 123
 124 /*
 125  *    0                   1                   2                   3
 126  *    0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
 127  *   +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 128  *   |  Message Type |  Channel Type |            Priority           |
 129  *   +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 130  *   |                    Reliability Parameter                      |
 131  *   +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 132  *   |         Label Length          |       Protocol Length         |
 133  *   +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 134  *   \                                                               /
 135  *   |                             Label                             |
 136  *   /                                                               \
 137  *   +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 138  *   \                                                               /
 139  *   |                            Protocol                           |
 140  *   /                                                               \
 141  *   +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 142  */
 143
 144   gst_byte_writer_init_with_size (&w, size, FALSE);
 145
 146   if (!gst_byte_writer_put_uint8 (&w, (guint8) CHANNEL_MESSAGE_OPEN))
 147     g_return_val_if_reached (NULL);
 148
 149   if (!channel->parent.ordered)
 150     reliability |= 0x80;
 151   if (channel->parent.max_retransmits != -1) {
 152     reliability |= 0x01;
 153     reliability_param = channel->parent.max_retransmits;
 154   }
 155   if (channel->parent.max_packet_lifetime != -1) {
 156     reliability |= 0x02;
 157     reliability_param = channel->parent.max_packet_lifetime;
 158   }
 159
 160   priority = priority_type_to_uint (channel->parent.priority);
 161
 162   if (!gst_byte_writer_put_uint8 (&w, (guint8) reliability))
 163     g_return_val_if_reached (NULL);
 164   if (!gst_byte_writer_put_uint16_be (&w, (guint16) priority))
 165     g_return_val_if_reached (NULL);
 166   if (!gst_byte_writer_put_uint32_be (&w, (guint32) reliability_param))
 167     g_return_val_if_reached (NULL);
 168   if (!gst_byte_writer_put_uint16_be (&w, (guint16) label_len))
 169     g_return_val_if_reached (NULL);
 170   if (!gst_byte_writer_put_uint16_be (&w, (guint16) proto_len))
 171     g_return_val_if_reached (NULL);
 172   if (!gst_byte_writer_put_data (&w, (guint8 *) channel->parent.label,
 173           label_len))
 174     g_return_val_if_reached (NULL);
 175   if (!gst_byte_writer_put_data (&w, (guint8 *) channel->parent.protocol,
 176           proto_len))
 177     g_return_val_if_reached (NULL);
 178
 179   buf = gst_byte_writer_reset_and_get_buffer (&w);
 180
 181   /* send reliable and ordered */
 182   gst_sctp_buffer_add_send_meta (buf, DATA_CHANNEL_PPID_WEBRTC_CONTROL, TRUE,
 183       GST_SCTP_SEND_META_PARTIAL_RELIABILITY_NONE, 0);
 184
 185   return buf;
 186 }
 187
 188 static GstBuffer *
 189 construct_ack_packet (WebRTCDataChannel * channel)
 190 {
 191   GstByteWriter w;
 192   GstBuffer *buf;
 193
 194 /*
 195  *   0                   1                   2                   3
 196  *   0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
 197  *   +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 198  *   |  Message Type |
 199  *   +-+-+-+-+-+-+-+-+
 200  */
 201
 202   gst_byte_writer_init_with_size (&w, 1, FALSE);
 203
 204   if (!gst_byte_writer_put_uint8 (&w, (guint8) CHANNEL_MESSAGE_ACK))
 205     g_return_val_if_reached (NULL);
 206
 207   buf = gst_byte_writer_reset_and_get_buffer (&w);
 208
 209   /* send reliable and ordered */
 210   gst_sctp_buffer_add_send_meta (buf, DATA_CHANNEL_PPID_WEBRTC_CONTROL, TRUE,
 211       GST_SCTP_SEND_META_PARTIAL_RELIABILITY_NONE, 0);
 212
 213   return buf;
 214 }
 215
 216 typedef void (*ChannelTask) (GstWebRTCDataChannel * channel,
 217     gpointer user_data);
 218
 219 struct task
 220 {
 221   GstWebRTCDataChannel *channel;
 222   ChannelTask func;
 223   gpointer user_data;
 224   GDestroyNotify notify;
 225 };
 226
 227 static GstStructure *
 228 _execute_task (GstWebRTCBin * webrtc, struct task *task)
 229 {
 230   if (task->func)
 231     task->func (task->channel, task->user_data);
 232
 233   return NULL;
 234 }
 235
 236 static void
 237 _free_task (struct task *task)
 238 {
 239   gst_object_unref (task->channel);
 240
 241   if (task->notify)
 242     task->notify (task->user_data);
 243   g_free (task);
 244 }
 245
 246 static void
 247 _channel_enqueue_task (WebRTCDataChannel * channel, ChannelTask func,
 248     gpointer user_data, GDestroyNotify notify)
 249 {
 250   struct task *task = g_new0 (struct task, 1);
 251
 252   task->channel = gst_object_ref (channel);
 253   task->func = func;
 254   task->user_data = user_data;
 255   task->notify = notify;
 256
 257   gst_webrtc_bin_enqueue_task (channel->webrtcbin,
 258       (GstWebRTCBinFunc) _execute_task, task, (GDestroyNotify) _free_task,
 259       NULL);
 260 }
 261
 262 static void
 263 _channel_store_error (WebRTCDataChannel * channel, GError * error)
 264 {
 265   GST_WEBRTC_DATA_CHANNEL_LOCK (channel);
 266   if (error) {
 267     GST_WARNING_OBJECT (channel, "Error: %s",
 268         error ? error->message : "Unknown");
 269     if (!channel->stored_error)
 270       channel->stored_error = error;
 271     else
 272       g_clear_error (&error);
 273   }
 274   GST_WEBRTC_DATA_CHANNEL_UNLOCK (channel);
 275 }
 276
 277 static void
 278 _emit_on_open (WebRTCDataChannel * channel, gpointer user_data)
 279 {
 280   gst_webrtc_data_channel_on_open (GST_WEBRTC_DATA_CHANNEL (channel));
 281 }
 282
 283 static void
 284 _transport_closed (WebRTCDataChannel * channel)
 285 {
 286   GError *error;
 287   gboolean both_sides_closed;
 288
 289   GST_WEBRTC_DATA_CHANNEL_LOCK (channel);
 290   error = channel->stored_error;
 291   channel->stored_error = NULL;
 292
 293   both_sides_closed =
 294       channel->peer_closed && channel->parent.buffered_amount <= 0;
 295   if (both_sides_closed || error) {
 296     channel->peer_closed = FALSE;
 297   }
 298   GST_WEBRTC_DATA_CHANNEL_UNLOCK (channel);
 299
 300   if (error) {
 301     gst_webrtc_data_channel_on_error (GST_WEBRTC_DATA_CHANNEL (channel), error);
 302     g_clear_error (&error);
 303   }
 304   if (both_sides_closed || error) {
 305     gst_webrtc_data_channel_on_close (GST_WEBRTC_DATA_CHANNEL (channel));
 306   }
 307 }
 308
 309 static void
 310 _close_sctp_stream (WebRTCDataChannel * channel, gpointer user_data)
 311 {
 312   GstPad *pad, *peer;
 313
 314   GST_INFO_OBJECT (channel, "Closing outgoing SCTP stream %i label \"%s\"",
 315       channel->parent.id, channel->parent.label);
 316
 317   pad = gst_element_get_static_pad (channel->appsrc, "src");
 318   peer = gst_pad_get_peer (pad);
 319   gst_object_unref (pad);
 320
 321   if (peer) {
 322     GstElement *sctpenc = gst_pad_get_parent_element (peer);
 323
 324     if (sctpenc) {
 325       gst_element_release_request_pad (sctpenc, peer);
 326       gst_object_unref (sctpenc);
 327     }
 328     gst_object_unref (peer);
 329   }
 330
 331   _transport_closed (channel);
 332 }
 333
 334 static void
 335 _close_procedure (WebRTCDataChannel * channel, gpointer user_data)
 336 {
 337   /* https://www.w3.org/TR/webrtc/#data-transport-closing-procedure */
 338   GST_WEBRTC_DATA_CHANNEL_LOCK (channel);
 339   if (channel->parent.ready_state == GST_WEBRTC_DATA_CHANNEL_STATE_CLOSED) {
 340     GST_WEBRTC_DATA_CHANNEL_UNLOCK (channel);
 341     return;
 342   } else if (channel->parent.ready_state ==
 343       GST_WEBRTC_DATA_CHANNEL_STATE_CLOSING) {
 344     _channel_enqueue_task (channel, (ChannelTask) _transport_closed, NULL,
 345         NULL);
 346   } else if (channel->parent.ready_state == GST_WEBRTC_DATA_CHANNEL_STATE_OPEN) {
 347     channel->parent.ready_state = GST_WEBRTC_DATA_CHANNEL_STATE_CLOSING;
 348     GST_WEBRTC_DATA_CHANNEL_UNLOCK (channel);
 349     g_object_notify (G_OBJECT (channel), "ready-state");
 350
 351     GST_WEBRTC_DATA_CHANNEL_LOCK (channel);
 352     if (channel->parent.buffered_amount <= 0) {
 353       _channel_enqueue_task (channel, (ChannelTask) _close_sctp_stream,
 354           NULL, NULL);
 355     }
 356   }
 357
 358   GST_WEBRTC_DATA_CHANNEL_UNLOCK (channel);
 359 }
 360
 361 static void
 362 _on_sctp_stream_reset (WebRTCSCTPTransport * sctp, guint stream_id,
 363     WebRTCDataChannel * channel)
 364 {
 365   if (channel->parent.id == stream_id) {
 366     GST_INFO_OBJECT (channel,
 367         "Received channel close for SCTP stream %i label \"%s\"",
 368         channel->parent.id, channel->parent.label);
 369
 370     GST_WEBRTC_DATA_CHANNEL_LOCK (channel);
 371     channel->peer_closed = TRUE;
 372     GST_WEBRTC_DATA_CHANNEL_UNLOCK (channel);
 373
 374     _channel_enqueue_task (channel, (ChannelTask) _close_procedure,
 375         GUINT_TO_POINTER (stream_id), NULL);
 376   }
 377 }
 378
 379 static void
 380 webrtc_data_channel_close (GstWebRTCDataChannel * channel)
 381 {
 382   _close_procedure (WEBRTC_DATA_CHANNEL (channel), NULL);
 383 }
 384
 385 static GstFlowReturn
 386 _parse_control_packet (WebRTCDataChannel * channel, guint8 * data,
 387     gsize size, GError ** error)
 388 {
 389   GstByteReader r;
 390   guint8 message_type;
 391   gchar *label = NULL;
 392   gchar *proto = NULL;
 393
 394   if (!data)
 395     g_return_val_if_reached (GST_FLOW_ERROR);
 396   if (size < 1)
 397     g_return_val_if_reached (GST_FLOW_ERROR);
 398
 399   gst_byte_reader_init (&r, data, size);
 400
 401   if (!gst_byte_reader_get_uint8 (&r, &message_type))
 402     g_return_val_if_reached (GST_FLOW_ERROR);
 403
 404   if (message_type == CHANNEL_MESSAGE_ACK) {
 405     /* all good */
 406     GST_INFO_OBJECT (channel, "Received channel ack");
 407     return GST_FLOW_OK;
 408   } else if (message_type == CHANNEL_MESSAGE_OPEN) {
 409     guint8 reliability;
 410     guint32 reliability_param;
 411     guint16 priority, label_len, proto_len;
 412     const guint8 *src;
 413     GstBuffer *buffer;
 414     GstFlowReturn ret;
 415
 416     GST_INFO_OBJECT (channel, "Received channel open");
 417
 418     if (channel->parent.negotiated) {
 419       g_set_error (error, GST_WEBRTC_ERROR,
 420           GST_WEBRTC_ERROR_DATA_CHANNEL_FAILURE,
 421           "Data channel was signalled as negotiated already");
 422       g_return_val_if_reached (GST_FLOW_ERROR);
 423     }
 424
 425     if (channel->opened)
 426       return GST_FLOW_OK;
 427
 428     if (!gst_byte_reader_get_uint8 (&r, &reliability))
 429       goto parse_error;
 430     if (!gst_byte_reader_get_uint16_be (&r, &priority))
 431       goto parse_error;
 432     if (!gst_byte_reader_get_uint32_be (&r, &reliability_param))
 433       goto parse_error;
 434     if (!gst_byte_reader_get_uint16_be (&r, &label_len))
 435       goto parse_error;
 436     if (!gst_byte_reader_get_uint16_be (&r, &proto_len))
 437       goto parse_error;
 438
 439     label = g_new0 (gchar, (gsize) label_len + 1);
 440     proto = g_new0 (gchar, (gsize) proto_len + 1);
 441
 442     if (!gst_byte_reader_get_data (&r, label_len, &src))
 443       goto parse_error;
 444     memcpy (label, src, label_len);
 445     label[label_len] = '\0';
 446     if (!gst_byte_reader_get_data (&r, proto_len, &src))
 447       goto parse_error;
 448     memcpy (proto, src, proto_len);
 449     proto[proto_len] = '\0';
 450
 451     g_free (channel->parent.label);
 452     channel->parent.label = label;
 453     g_free (channel->parent.protocol);
 454     channel->parent.protocol = proto;
 455     channel->parent.priority = priority_uint_to_type (priority);
 456     channel->parent.ordered = !(reliability & 0x80);
 457     if (reliability & 0x01) {
 458       channel->parent.max_retransmits = reliability_param;
 459       channel->parent.max_packet_lifetime = -1;
 460     } else if (reliability & 0x02) {
 461       channel->parent.max_retransmits = -1;
 462       channel->parent.max_packet_lifetime = reliability_param;
 463     } else {
 464       channel->parent.max_retransmits = -1;
 465       channel->parent.max_packet_lifetime = -1;
 466     }
 467     channel->opened = TRUE;
 468
 469     GST_INFO_OBJECT (channel, "Received channel open for SCTP stream %i "
 470         "label \"%s\" protocol %s ordered %s", channel->parent.id,
 471         channel->parent.label, channel->parent.protocol,
 472         channel->parent.ordered ? "true" : "false");
 473
 474     _channel_enqueue_task (channel, (ChannelTask) _emit_on_open, NULL, NULL);
 475
 476     GST_INFO_OBJECT (channel, "Sending channel ack");
 477     buffer = construct_ack_packet (channel);
 478
 479     GST_WEBRTC_DATA_CHANNEL_LOCK (channel);
 480     channel->parent.buffered_amount += gst_buffer_get_size (buffer);
 481     GST_WEBRTC_DATA_CHANNEL_UNLOCK (channel);
 482
 483     ret = gst_app_src_push_buffer (GST_APP_SRC (channel->appsrc), buffer);
 484     if (ret != GST_FLOW_OK) {
 485       g_set_error (error, GST_WEBRTC_ERROR,
 486           GST_WEBRTC_ERROR_DATA_CHANNEL_FAILURE, "Could not send ack packet");
 487       return ret;
 488     }
 489
 490     return ret;
 491   } else {
 492     g_set_error (error, GST_WEBRTC_ERROR,
 493         GST_WEBRTC_ERROR_DATA_CHANNEL_FAILURE,
 494         "Unknown message type in control protocol");
 495     return GST_FLOW_ERROR;
 496   }
 497
 498 parse_error:
 499   {
 500     g_free (label);
 501     g_free (proto);
 502     g_set_error (error, GST_WEBRTC_ERROR,
 503         GST_WEBRTC_ERROR_DATA_CHANNEL_FAILURE, "Failed to parse packet");
 504     g_return_val_if_reached (GST_FLOW_ERROR);
 505   }
 506 }
 507
 508 static void
 509 on_sink_eos (GstAppSink * sink, gpointer user_data)
 510 {
 511 }
 512
 513 struct map_info
 514 {
 515   GstBuffer *buffer;
 516   GstMapInfo map_info;
 517 };
 518
 519 static void
 520 buffer_unmap_and_unref (struct map_info *info)
 521 {
 522   gst_buffer_unmap (info->buffer, &info->map_info);
 523   gst_buffer_unref (info->buffer);
 524   g_free (info);
 525 }
 526
 527 static void
 528 _emit_have_data (WebRTCDataChannel * channel, GBytes * data)
 529 {
 530   gst_webrtc_data_channel_on_message_data (GST_WEBRTC_DATA_CHANNEL (channel),
 531       data);
 532 }
 533
 534 static void
 535 _emit_have_string (GstWebRTCDataChannel * channel, gchar * str)
 536 {
 537   gst_webrtc_data_channel_on_message_string (GST_WEBRTC_DATA_CHANNEL (channel),
 538       str);
 539 }
 540
 541 static GstFlowReturn
 542 _data_channel_have_sample (WebRTCDataChannel * channel, GstSample * sample,
 543     GError ** error)
 544 {
 545   GstSctpReceiveMeta *receive;
 546   GstBuffer *buffer;
 547   GstFlowReturn ret = GST_FLOW_OK;
 548
 549   GST_LOG_OBJECT (channel, "Received sample %" GST_PTR_FORMAT, sample);
 550
 551   g_return_val_if_fail (channel->sctp_transport != NULL, GST_FLOW_ERROR);
 552
 553   buffer = gst_sample_get_buffer (sample);
 554   if (!buffer) {
 555     g_set_error (error, GST_WEBRTC_ERROR,
 556         GST_WEBRTC_ERROR_DATA_CHANNEL_FAILURE, "No buffer to handle");
 557     return GST_FLOW_ERROR;
 558   }
 559   receive = gst_sctp_buffer_get_receive_meta (buffer);
 560   if (!receive) {
 561     g_set_error (error, GST_WEBRTC_ERROR,
 562         GST_WEBRTC_ERROR_DATA_CHANNEL_FAILURE,
 563         "No SCTP Receive meta on the buffer");
 564     return GST_FLOW_ERROR;
 565   }
 566
 567   switch (receive->ppid) {
 568     case DATA_CHANNEL_PPID_WEBRTC_CONTROL:{
 569       GstMapInfo info = GST_MAP_INFO_INIT;
 570       if (!gst_buffer_map (buffer, &info, GST_MAP_READ)) {
 571         g_set_error (error, GST_WEBRTC_ERROR,
 572             GST_WEBRTC_ERROR_DATA_CHANNEL_FAILURE,
 573             "Failed to map received buffer");
 574         ret = GST_FLOW_ERROR;
 575       } else {
 576         ret = _parse_control_packet (channel, info.data, info.size, error);
 577         gst_buffer_unmap (buffer, &info);
 578       }
 579       break;
 580     }
 581     case DATA_CHANNEL_PPID_WEBRTC_STRING:
 582     case DATA_CHANNEL_PPID_WEBRTC_STRING_PARTIAL:{
 583       GstMapInfo info = GST_MAP_INFO_INIT;
 584       if (!gst_buffer_map (buffer, &info, GST_MAP_READ)) {
 585         g_set_error (error, GST_WEBRTC_ERROR,
 586             GST_WEBRTC_ERROR_DATA_CHANNEL_FAILURE,
 587             "Failed to map received buffer");
 588         ret = GST_FLOW_ERROR;
 589       } else {
 590         gchar *str = g_strndup ((gchar *) info.data, info.size);
 591         _channel_enqueue_task (channel, (ChannelTask) _emit_have_string, str,
 592             g_free);
 593         gst_buffer_unmap (buffer, &info);
 594       }
 595       break;
 596     }
 597     case DATA_CHANNEL_PPID_WEBRTC_BINARY:
 598     case DATA_CHANNEL_PPID_WEBRTC_BINARY_PARTIAL:{
 599       struct map_info *info = g_new0 (struct map_info, 1);
 600       if (!gst_buffer_map (buffer, &info->map_info, GST_MAP_READ)) {
 601         g_set_error (error, GST_WEBRTC_ERROR,
 602             GST_WEBRTC_ERROR_DATA_CHANNEL_FAILURE,
 603             "Failed to map received buffer");
 604         ret = GST_FLOW_ERROR;
 605       } else {
 606         GBytes *data = g_bytes_new_with_free_func (info->map_info.data,
 607             info->map_info.size, (GDestroyNotify) buffer_unmap_and_unref, info);
 608         info->buffer = gst_buffer_ref (buffer);
 609         _channel_enqueue_task (channel, (ChannelTask) _emit_have_data, data,
 610             (GDestroyNotify) g_bytes_unref);
 611       }
 612       break;
 613     }
 614     case DATA_CHANNEL_PPID_WEBRTC_BINARY_EMPTY:
 615       _channel_enqueue_task (channel, (ChannelTask) _emit_have_data, NULL,
 616           NULL);
 617       break;
 618     case DATA_CHANNEL_PPID_WEBRTC_STRING_EMPTY:
 619       _channel_enqueue_task (channel, (ChannelTask) _emit_have_string, NULL,
 620           NULL);
 621       break;
 622     default:
 623       g_set_error (error, GST_WEBRTC_ERROR,
 624           GST_WEBRTC_ERROR_DATA_CHANNEL_FAILURE,
 625           "Unknown SCTP PPID %u received", receive->ppid);
 626       ret = GST_FLOW_ERROR;
 627       break;
 628   }
 629
 630   return ret;
 631 }
 632
 633 static GstFlowReturn
 634 on_sink_preroll (GstAppSink * sink, gpointer user_data)
 635 {
 636   WebRTCDataChannel *channel = user_data;
 637   GstSample *sample = gst_app_sink_pull_preroll (sink);
 638   GstFlowReturn ret;
 639
 640   if (sample) {
 641     /* This sample also seems to be provided by the sample callback
 642        ret = _data_channel_have_sample (channel, sample); */
 643     ret = GST_FLOW_OK;
 644     gst_sample_unref (sample);
 645   } else if (gst_app_sink_is_eos (sink)) {
 646     ret = GST_FLOW_EOS;
 647   } else {
 648     ret = GST_FLOW_ERROR;
 649   }
 650
 651   if (ret != GST_FLOW_OK) {
 652     _channel_enqueue_task (channel, (ChannelTask) _close_procedure, NULL, NULL);
 653   }
 654
 655   return ret;
 656 }
 657
 658 static GstFlowReturn
 659 on_sink_sample (GstAppSink * sink, gpointer user_data)
 660 {
 661   WebRTCDataChannel *channel = user_data;
 662   GstSample *sample = gst_app_sink_pull_sample (sink);
 663   GstFlowReturn ret;
 664   GError *error = NULL;
 665
 666   if (sample) {
 667     ret = _data_channel_have_sample (channel, sample, &error);
 668     gst_sample_unref (sample);
 669   } else if (gst_app_sink_is_eos (sink)) {
 670     ret = GST_FLOW_EOS;
 671   } else {
 672     ret = GST_FLOW_ERROR;
 673   }
 674
 675   if (error)
 676     _channel_store_error (channel, error);
 677
 678   if (ret != GST_FLOW_OK) {
 679     _channel_enqueue_task (channel, (ChannelTask) _close_procedure, NULL, NULL);
 680   }
 681
 682   return ret;
 683 }
 684
 685 static GstAppSinkCallbacks sink_callbacks = {
 686   on_sink_eos,
 687   on_sink_preroll,
 688   on_sink_sample,
 689 };
 690
 691 void
 692 webrtc_data_channel_start_negotiation (WebRTCDataChannel * channel)
 693 {
 694   GstBuffer *buffer;
 695
 696   g_return_if_fail (!channel->parent.negotiated);
 697   g_return_if_fail (channel->parent.id != -1);
 698   g_return_if_fail (channel->sctp_transport != NULL);
 699
 700   buffer = construct_open_packet (channel);
 701
 702   GST_INFO_OBJECT (channel, "Sending channel open for SCTP stream %i "
 703       "label \"%s\" protocol %s ordered %s", channel->parent.id,
 704       channel->parent.label, channel->parent.protocol,
 705       channel->parent.ordered ? "true" : "false");
 706
 707   GST_WEBRTC_DATA_CHANNEL_LOCK (channel);
 708   channel->parent.buffered_amount += gst_buffer_get_size (buffer);
 709   GST_WEBRTC_DATA_CHANNEL_UNLOCK (channel);
 710   g_object_notify (G_OBJECT (&channel->parent), "buffered-amount");
 711
 712   if (gst_app_src_push_buffer (GST_APP_SRC (channel->appsrc),
 713           buffer) == GST_FLOW_OK) {
 714     channel->opened = TRUE;
 715     _channel_enqueue_task (channel, (ChannelTask) _emit_on_open, NULL, NULL);
 716   } else {
 717     GError *error = NULL;
 718     g_set_error (&error, GST_WEBRTC_ERROR,
 719         GST_WEBRTC_ERROR_DATA_CHANNEL_FAILURE,
 720         "Failed to send DCEP open packet");
 721     _channel_store_error (channel, error);
 722     _channel_enqueue_task (channel, (ChannelTask) _close_procedure, NULL, NULL);
 723   }
 724 }
 725
 726 static void
 727 _get_sctp_reliability (WebRTCDataChannel * channel,
 728     GstSctpSendMetaPartiallyReliability * reliability, guint * rel_param)
 729 {
 730   if (channel->parent.max_retransmits != -1) {
 731     *reliability = GST_SCTP_SEND_META_PARTIAL_RELIABILITY_RTX;
 732     *rel_param = channel->parent.max_retransmits;
 733   } else if (channel->parent.max_packet_lifetime != -1) {
 734     *reliability = GST_SCTP_SEND_META_PARTIAL_RELIABILITY_TTL;
 735     *rel_param = channel->parent.max_packet_lifetime;
 736   } else {
 737     *reliability = GST_SCTP_SEND_META_PARTIAL_RELIABILITY_NONE;
 738     *rel_param = 0;
 739   }
 740 }
 741
 742 static gboolean
 743 _is_within_max_message_size (WebRTCDataChannel * channel, gsize size)
 744 {
 745   return size <= channel->sctp_transport->max_message_size;
 746 }
 747
 748 static void
 749 webrtc_data_channel_send_data (GstWebRTCDataChannel * base_channel,
 750     GBytes * bytes)
 751 {
 752   WebRTCDataChannel *channel = WEBRTC_DATA_CHANNEL (base_channel);
 753   GstSctpSendMetaPartiallyReliability reliability;
 754   guint rel_param;
 755   guint32 ppid;
 756   GstBuffer *buffer;
 757   GstFlowReturn ret;
 758
 759   if (!bytes) {
 760     buffer = gst_buffer_new ();
 761     ppid = DATA_CHANNEL_PPID_WEBRTC_BINARY_EMPTY;
 762   } else {
 763     gsize size;
 764     guint8 *data;
 765
 766     data = (guint8 *) g_bytes_get_data (bytes, &size);
 767     g_return_if_fail (data != NULL);
 768     if (!_is_within_max_message_size (channel, size)) {
 769       GError *error = NULL;
 770       g_set_error (&error, GST_WEBRTC_ERROR,
 771           GST_WEBRTC_ERROR_DATA_CHANNEL_FAILURE,
 772           "Requested to send data that is too large");
 773       _channel_store_error (channel, error);
 774       _channel_enqueue_task (channel, (ChannelTask) _close_procedure, NULL,
 775           NULL);
 776       return;
 777     }
 778
 779     buffer = gst_buffer_new_wrapped_full (GST_MEMORY_FLAG_READONLY, data, size,
 780         0, size, g_bytes_ref (bytes), (GDestroyNotify) g_bytes_unref);
 781     ppid = DATA_CHANNEL_PPID_WEBRTC_BINARY;
 782   }
 783
 784   _get_sctp_reliability (channel, &reliability, &rel_param);
 785   gst_sctp_buffer_add_send_meta (buffer, ppid, channel->parent.ordered,
 786       reliability, rel_param);
 787
 788   GST_LOG_OBJECT (channel, "Sending data using buffer %" GST_PTR_FORMAT,
 789       buffer);
 790
 791   GST_WEBRTC_DATA_CHANNEL_LOCK (channel);
 792   channel->parent.buffered_amount += gst_buffer_get_size (buffer);
 793   GST_WEBRTC_DATA_CHANNEL_UNLOCK (channel);
 794   g_object_notify (G_OBJECT (&channel->parent), "buffered-amount");
 795
 796   ret = gst_app_src_push_buffer (GST_APP_SRC (channel->appsrc), buffer);
 797
 798   if (ret != GST_FLOW_OK) {
 799     GError *error = NULL;
 800     g_set_error (&error, GST_WEBRTC_ERROR,
 801         GST_WEBRTC_ERROR_DATA_CHANNEL_FAILURE, "Failed to send data");
 802     _channel_store_error (channel, error);
 803     _channel_enqueue_task (channel, (ChannelTask) _close_procedure, NULL, NULL);
 804   }
 805 }
 806
 807 static void
 808 webrtc_data_channel_send_string (GstWebRTCDataChannel * base_channel,
 809     const gchar * str)
 810 {
 811   WebRTCDataChannel *channel = WEBRTC_DATA_CHANNEL (base_channel);
 812   GstSctpSendMetaPartiallyReliability reliability;
 813   guint rel_param;
 814   guint32 ppid;
 815   GstBuffer *buffer;
 816   GstFlowReturn ret;
 817
 818   if (!channel->parent.negotiated)
 819     g_return_if_fail (channel->opened);
 820   g_return_if_fail (channel->sctp_transport != NULL);
 821
 822   if (!str) {
 823     buffer = gst_buffer_new ();
 824     ppid = DATA_CHANNEL_PPID_WEBRTC_STRING_EMPTY;
 825   } else {
 826     gsize size = strlen (str);
 827     gchar *str_copy;
 828
 829     if (!_is_within_max_message_size (channel, size)) {
 830       GError *error = NULL;
 831       g_set_error (&error, GST_WEBRTC_ERROR,
 832           GST_WEBRTC_ERROR_DATA_CHANNEL_FAILURE,
 833           "Requested to send a string that is too large");
 834       _channel_store_error (channel, error);
 835       _channel_enqueue_task (channel, (ChannelTask) _close_procedure, NULL,
 836           NULL);
 837       return;
 838     }
 839
 840     str_copy = g_strdup (str);
 841     buffer =
 842         gst_buffer_new_wrapped_full (GST_MEMORY_FLAG_READONLY, str_copy,
 843         size, 0, size, str_copy, g_free);
 844     ppid = DATA_CHANNEL_PPID_WEBRTC_STRING;
 845   }
 846
 847   _get_sctp_reliability (channel, &reliability, &rel_param);
 848   gst_sctp_buffer_add_send_meta (buffer, ppid, channel->parent.ordered,
 849       reliability, rel_param);
 850
 851   GST_TRACE_OBJECT (channel, "Sending string using buffer %" GST_PTR_FORMAT,
 852       buffer);
 853
 854   GST_WEBRTC_DATA_CHANNEL_LOCK (channel);
 855   channel->parent.buffered_amount += gst_buffer_get_size (buffer);
 856   GST_WEBRTC_DATA_CHANNEL_UNLOCK (channel);
 857   g_object_notify (G_OBJECT (&channel->parent), "buffered-amount");
 858
 859   ret = gst_app_src_push_buffer (GST_APP_SRC (channel->appsrc), buffer);
 860
 861   if (ret != GST_FLOW_OK) {
 862     GError *error = NULL;
 863     g_set_error (&error, GST_WEBRTC_ERROR,
 864         GST_WEBRTC_ERROR_DATA_CHANNEL_FAILURE, "Failed to send string");
 865     _channel_store_error (channel, error);
 866     _channel_enqueue_task (channel, (ChannelTask) _close_procedure, NULL, NULL);
 867   }
 868 }
 869
 870 static void
 871 _on_sctp_notify_state_unlocked (GObject * sctp_transport,
 872     WebRTCDataChannel * channel)
 873 {
 874   GstWebRTCSCTPTransportState state;
 875
 876   g_object_get (sctp_transport, "state", &state, NULL);
 877   if (state == GST_WEBRTC_SCTP_TRANSPORT_STATE_CONNECTED) {
 878     if (channel->parent.negotiated)
 879       _channel_enqueue_task (channel, (ChannelTask) _emit_on_open, NULL, NULL);
 880   }
 881 }
 882
 883 static WebRTCDataChannel *
 884 ensure_channel_alive (WebRTCDataChannel * channel)
 885 {
 886   /* ghetto impl of, does the channel still exist?.
 887    * Needed because g_signal_handler_disconnect*() will not disconnect any
 888    * running functions and _finalize() implementation can complete and
 889    * invalidate channel */
 890   G_LOCK (outstanding_channels_lock);
 891   if (g_list_find (outstanding_channels, channel)) {
 892     g_object_ref (channel);
 893   } else {
 894     G_UNLOCK (outstanding_channels_lock);
 895     return NULL;
 896   }
 897   G_UNLOCK (outstanding_channels_lock);
 898
 899   return channel;
 900 }
 901
 902 static void
 903 _on_sctp_notify_state (GObject * sctp_transport, GParamSpec * pspec,
 904     WebRTCDataChannel * channel)
 905 {
 906   if (!(channel = ensure_channel_alive (channel)))
 907     return;
 908
 909   GST_WEBRTC_DATA_CHANNEL_LOCK (channel);
 910   _on_sctp_notify_state_unlocked (sctp_transport, channel);
 911   GST_WEBRTC_DATA_CHANNEL_UNLOCK (channel);
 912
 913   g_object_unref (channel);
 914 }
 915
 916 static void
 917 _emit_low_threshold (WebRTCDataChannel * channel, gpointer user_data)
 918 {
 919   gst_webrtc_data_channel_on_buffered_amount_low (GST_WEBRTC_DATA_CHANNEL
 920       (channel));
 921 }
 922
 923 static GstPadProbeReturn
 924 on_appsrc_data (GstPad * pad, GstPadProbeInfo * info, gpointer user_data)
 925 {
 926   WebRTCDataChannel *channel = user_data;
 927   guint64 prev_amount;
 928   guint64 size = 0;
 929
 930   if (GST_PAD_PROBE_INFO_TYPE (info) & (GST_PAD_PROBE_TYPE_BUFFER)) {
 931     GstBuffer *buffer = GST_PAD_PROBE_INFO_BUFFER (info);
 932     size = gst_buffer_get_size (buffer);
 933   } else if (GST_PAD_PROBE_INFO_TYPE (info) & GST_PAD_PROBE_TYPE_BUFFER_LIST) {
 934     GstBufferList *list = GST_PAD_PROBE_INFO_BUFFER_LIST (info);
 935     size = gst_buffer_list_calculate_size (list);
 936   }
 937
 938   if (size > 0) {
 939     GST_WEBRTC_DATA_CHANNEL_LOCK (channel);
 940     prev_amount = channel->parent.buffered_amount;
 941     channel->parent.buffered_amount -= size;
 942     GST_TRACE_OBJECT (channel, "checking low-threshold: prev %"
 943         G_GUINT64_FORMAT " low-threshold %" G_GUINT64_FORMAT " buffered %"
 944         G_GUINT64_FORMAT, prev_amount,
 945         channel->parent.buffered_amount_low_threshold,
 946         channel->parent.buffered_amount);
 947     if (prev_amount >= channel->parent.buffered_amount_low_threshold
 948         && channel->parent.buffered_amount <
 949         channel->parent.buffered_amount_low_threshold) {
 950       _channel_enqueue_task (channel, (ChannelTask) _emit_low_threshold, NULL,
 951           NULL);
 952     }
 953
 954     if (channel->parent.ready_state == GST_WEBRTC_DATA_CHANNEL_STATE_CLOSING
 955         && channel->parent.buffered_amount <= 0) {
 956       _channel_enqueue_task (channel, (ChannelTask) _close_sctp_stream, NULL,
 957           NULL);
 958     }
 959     GST_WEBRTC_DATA_CHANNEL_UNLOCK (channel);
 960     g_object_notify (G_OBJECT (&channel->parent), "buffered-amount");
 961   }
 962
 963   return GST_PAD_PROBE_OK;
 964 }
 965
 966 static void
 967 gst_webrtc_data_channel_constructed (GObject * object)
 968 {
 969   WebRTCDataChannel *channel = WEBRTC_DATA_CHANNEL (object);
 970   GstPad *pad;
 971   GstCaps *caps;
 972
 973   caps = gst_caps_new_any ();
 974
 975   channel->appsrc = gst_element_factory_make ("appsrc", NULL);
 976   gst_object_ref_sink (channel->appsrc);
 977   pad = gst_element_get_static_pad (channel->appsrc, "src");
 978
 979   channel->src_probe = gst_pad_add_probe (pad, GST_PAD_PROBE_TYPE_DATA_BOTH,
 980       (GstPadProbeCallback) on_appsrc_data, channel, NULL);
 981
 982   channel->appsink = gst_element_factory_make ("appsink", NULL);
 983   gst_object_ref_sink (channel->appsink);
 984   g_object_set (channel->appsink, "sync", FALSE, "async", FALSE, "caps", caps,
 985       NULL);
 986   gst_app_sink_set_callbacks (GST_APP_SINK (channel->appsink), &sink_callbacks,
 987       channel, NULL);
 988
 989   gst_object_unref (pad);
 990   gst_caps_unref (caps);
 991 }
 992
 993 static void
 994 gst_webrtc_data_channel_dispose (GObject * object)
 995 {
 996   G_LOCK (outstanding_channels_lock);
 997   outstanding_channels = g_list_remove (outstanding_channels, object);
 998   G_UNLOCK (outstanding_channels_lock);
 999
1000   G_OBJECT_CLASS (parent_class)->dispose (object);
1001 }
1002
1003 static void
1004 gst_webrtc_data_channel_finalize (GObject * object)
1005 {
1006   WebRTCDataChannel *channel = WEBRTC_DATA_CHANNEL (object);
1007
1008   if (channel->src_probe) {
1009     GstPad *pad = gst_element_get_static_pad (channel->appsrc, "src");
1010     gst_pad_remove_probe (pad, channel->src_probe);
1011     gst_object_unref (pad);
1012     channel->src_probe = 0;
1013   }
1014
1015   if (channel->sctp_transport)
1016     g_signal_handlers_disconnect_by_data (channel->sctp_transport, channel);
1017   g_clear_object (&channel->sctp_transport);
1018
1019   g_clear_object (&channel->appsrc);
1020   g_clear_object (&channel->appsink);
1021
1022   G_OBJECT_CLASS (parent_class)->finalize (object);
1023 }
1024
1025 static void
1026 webrtc_data_channel_class_init (WebRTCDataChannelClass * klass)
1027 {
1028   GObjectClass *gobject_class = (GObjectClass *) klass;
1029   GstWebRTCDataChannelClass *channel_class =
1030       (GstWebRTCDataChannelClass *) klass;
1031
1032   gobject_class->constructed = gst_webrtc_data_channel_constructed;
1033   gobject_class->dispose = gst_webrtc_data_channel_dispose;
1034   gobject_class->finalize = gst_webrtc_data_channel_finalize;
1035
1036   channel_class->send_data = webrtc_data_channel_send_data;
1037   channel_class->send_string = webrtc_data_channel_send_string;
1038   channel_class->close = webrtc_data_channel_close;
1039 }
1040
1041 static void
1042 webrtc_data_channel_init (WebRTCDataChannel * channel)
1043 {
1044   G_LOCK (outstanding_channels_lock);
1045   outstanding_channels = g_list_prepend (outstanding_channels, channel);
1046   G_UNLOCK (outstanding_channels_lock);
1047 }
1048
1049 static void
1050 _data_channel_set_sctp_transport (WebRTCDataChannel * channel,
1051     WebRTCSCTPTransport * sctp)
1052 {
1053   g_return_if_fail (GST_IS_WEBRTC_DATA_CHANNEL (channel));
1054   g_return_if_fail (GST_IS_WEBRTC_SCTP_TRANSPORT (sctp));
1055
1056   GST_WEBRTC_DATA_CHANNEL_LOCK (channel);
1057   if (channel->sctp_transport)
1058     g_signal_handlers_disconnect_by_data (channel->sctp_transport, channel);
1059   GST_TRACE ("%p set sctp %p", channel, sctp);
1060
1061   gst_object_replace ((GstObject **) & channel->sctp_transport,
1062       GST_OBJECT (sctp));
1063
1064   if (sctp) {
1065     g_signal_connect (sctp, "stream-reset", G_CALLBACK (_on_sctp_stream_reset),
1066         channel);
1067     g_signal_connect (sctp, "notify::state", G_CALLBACK (_on_sctp_notify_state),
1068         channel);
1069   }
1070   GST_WEBRTC_DATA_CHANNEL_UNLOCK (channel);
1071 }
1072
1073 void
1074 webrtc_data_channel_link_to_sctp (WebRTCDataChannel * channel,
1075     WebRTCSCTPTransport * sctp_transport)
1076 {
1077   if (sctp_transport && !channel->sctp_transport) {
1078     gint id;
1079
1080     g_object_get (channel, "id", &id, NULL);
1081
1082     if (sctp_transport->association_established && id != -1) {
1083       gchar *pad_name;
1084
1085       _data_channel_set_sctp_transport (channel, sctp_transport);
1086       pad_name = g_strdup_printf ("sink_%u", id);
1087       if (!gst_element_link_pads (channel->appsrc, "src",
1088               channel->sctp_transport->sctpenc, pad_name))
1089         g_warn_if_reached ();
1090       g_free (pad_name);
1091
1092       _on_sctp_notify_state_unlocked (G_OBJECT (sctp_transport), channel);
1093     }
1094   }
1095 }