tests/check/elements/rtpsession.c

   1 /* GStreamer
   2  *
   3  * unit test for gstrtpsession
   4  *
   5  * Copyright (C) <2009> Wim Taymans <wim.taymans@gmail.com>
   6  * Copyright (C) 2013 Collabora Ltd.
   7  *
   8  * This library is free software; you can redistribute it and/or
   9  * modify it under the terms of the GNU Library General Public
  10  * License as published by the Free Software Foundation; either
  11  * version 2 of the License, or (at your option) any later version.
  12  *
  13  * This library is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16  * Library General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU Library General Public
  19  * License along with this library; if not, write to the
  20  * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
  21  * Boston, MA 02110-1301, USA.
  22  */
  23 #define GLIB_DISABLE_DEPRECATION_WARNINGS
  24
  25 #include <gst/check/gstharness.h>
  26 #include <gst/check/gstcheck.h>
  27 #include <gst/check/gsttestclock.h>
  28 #include <gst/check/gstharness.h>
  29
  30 #include <gst/rtp/gstrtpbuffer.h>
  31 #include <gst/rtp/gstrtcpbuffer.h>
  32 #include <gst/net/gstnetaddressmeta.h>
  33 #include <gst/video/video.h>
  34
  35 #define TEST_BUF_CLOCK_RATE 8000
  36 #define TEST_BUF_PT 0
  37 #define TEST_BUF_SSRC 0x01BADBAD
  38 #define TEST_BUF_MS  20
  39 #define TEST_BUF_DURATION (TEST_BUF_MS * GST_MSECOND)
  40 #define TEST_BUF_SIZE (64000 * TEST_BUF_MS / 1000)
  41 #define TEST_RTP_TS_DURATION (TEST_BUF_CLOCK_RATE * TEST_BUF_MS / 1000)
  42
  43 static GstCaps *
  44 generate_caps (void)
  45 {
  46   return gst_caps_new_simple ("application/x-rtp",
  47       "clock-rate", G_TYPE_INT, TEST_BUF_CLOCK_RATE,
  48       "payload", G_TYPE_INT, TEST_BUF_PT, NULL);
  49 }
  50
  51 static GstBuffer *
  52 generate_test_buffer_full (GstClockTime dts,
  53     guint seq_num, guint32 rtp_ts, guint ssrc)
  54 {
  55   GstBuffer *buf;
  56   guint8 *payload;
  57   guint i;
  58   GstRTPBuffer rtp = GST_RTP_BUFFER_INIT;
  59
  60   buf = gst_rtp_buffer_new_allocate (TEST_BUF_SIZE, 0, 0);
  61   GST_BUFFER_DTS (buf) = dts;
  62
  63   gst_rtp_buffer_map (buf, GST_MAP_READWRITE, &rtp);
  64   gst_rtp_buffer_set_payload_type (&rtp, TEST_BUF_PT);
  65   gst_rtp_buffer_set_seq (&rtp, seq_num);
  66   gst_rtp_buffer_set_timestamp (&rtp, rtp_ts);
  67   gst_rtp_buffer_set_ssrc (&rtp, ssrc);
  68
  69   payload = gst_rtp_buffer_get_payload (&rtp);
  70   for (i = 0; i < TEST_BUF_SIZE; i++)
  71     payload[i] = 0xff;
  72
  73   gst_rtp_buffer_unmap (&rtp);
  74
  75   return buf;
  76 }
  77
  78 static GstBuffer *
  79 generate_test_buffer (guint seq_num, guint ssrc)
  80 {
  81   return generate_test_buffer_full (seq_num * TEST_BUF_DURATION,
  82       seq_num, seq_num * TEST_RTP_TS_DURATION, ssrc);
  83 }
  84
  85 typedef struct
  86 {
  87   GstHarness *send_rtp_h;
  88   GstHarness *recv_rtp_h;
  89   GstHarness *rtcp_h;
  90
  91   GstElement *session;
  92   GObject *internal_session;
  93   GstTestClock *testclock;
  94   GstCaps *caps;
  95
  96   gboolean running;
  97 } SessionHarness;
  98
  99 static GstCaps *
 100 _pt_map_requested (GstElement * element, guint pt, gpointer data)
 101 {
 102   SessionHarness *h = data;
 103   return gst_caps_copy (h->caps);
 104 }
 105
 106 static SessionHarness *
 107 session_harness_new (void)
 108 {
 109   SessionHarness *h = g_new0 (SessionHarness, 1);
 110   h->caps = generate_caps ();
 111
 112   h->testclock = GST_TEST_CLOCK_CAST (gst_test_clock_new ());
 113   gst_system_clock_set_default (GST_CLOCK_CAST (h->testclock));
 114
 115   h->session = gst_element_factory_make ("rtpsession", NULL);
 116   gst_element_set_clock (h->session, GST_CLOCK_CAST (h->testclock));
 117
 118   h->send_rtp_h = gst_harness_new_with_element (h->session,
 119       "send_rtp_sink", "send_rtp_src");
 120   gst_harness_set_src_caps (h->send_rtp_h, gst_caps_copy (h->caps));
 121
 122   h->recv_rtp_h = gst_harness_new_with_element (h->session,
 123       "recv_rtp_sink", "recv_rtp_src");
 124   gst_harness_set_src_caps (h->recv_rtp_h, gst_caps_copy (h->caps));
 125
 126   h->rtcp_h = gst_harness_new_with_element (h->session,
 127       "recv_rtcp_sink", "send_rtcp_src");
 128   gst_harness_set_src_caps_str (h->rtcp_h, "application/x-rtcp");
 129
 130   g_signal_connect (h->session, "request-pt-map",
 131       (GCallback) _pt_map_requested, h);
 132
 133   g_object_get (h->session, "internal-session", &h->internal_session, NULL);
 134
 135   return h;
 136 }
 137
 138 static void
 139 session_harness_free (SessionHarness * h)
 140 {
 141   gst_system_clock_set_default (NULL);
 142
 143   gst_caps_unref (h->caps);
 144   gst_object_unref (h->testclock);
 145
 146   gst_harness_teardown (h->rtcp_h);
 147   gst_harness_teardown (h->recv_rtp_h);
 148   gst_harness_teardown (h->send_rtp_h);
 149
 150   g_object_unref (h->internal_session);
 151   gst_object_unref (h->session);
 152   g_free (h);
 153 }
 154
 155 static GstFlowReturn
 156 session_harness_send_rtp (SessionHarness * h, GstBuffer * buf)
 157 {
 158   return gst_harness_push (h->send_rtp_h, buf);
 159 }
 160
 161 static GstFlowReturn
 162 session_harness_recv_rtp (SessionHarness * h, GstBuffer * buf)
 163 {
 164   return gst_harness_push (h->recv_rtp_h, buf);
 165 }
 166
 167 static GstFlowReturn
 168 session_harness_recv_rtcp (SessionHarness * h, GstBuffer * buf)
 169 {
 170   return gst_harness_push (h->rtcp_h, buf);
 171 }
 172
 173 static GstBuffer *
 174 session_harness_pull_rtcp (SessionHarness * h)
 175 {
 176   return gst_harness_pull (h->rtcp_h);
 177 }
 178
 179 static void
 180 session_harness_crank_clock (SessionHarness * h)
 181 {
 182   gst_test_clock_crank (h->testclock);
 183 }
 184
 185 static gboolean
 186 session_harness_advance_and_crank (SessionHarness * h, GstClockTime delta)
 187 {
 188   GstClockID res, pending;
 189   gboolean result;
 190   gst_test_clock_wait_for_next_pending_id (h->testclock, &pending);
 191   gst_test_clock_advance_time (h->testclock, delta);
 192   res = gst_test_clock_process_next_clock_id (h->testclock);
 193   if (res == pending)
 194     result = TRUE;
 195   else
 196     result = FALSE;
 197   if (res)
 198     gst_clock_id_unref (res);
 199   gst_clock_id_unref (pending);
 200   return result;
 201 }
 202
 203 static void
 204 session_harness_produce_rtcp (SessionHarness * h, gint num_rtcp_packets)
 205 {
 206   /* due to randomness in rescheduling of RTCP timeout, we need to
 207      keep cranking until we have the desired amount of packets */
 208   while (gst_harness_buffers_in_queue (h->rtcp_h) < num_rtcp_packets) {
 209     session_harness_crank_clock (h);
 210     /* allow the rtcp-thread to settle before checking the queue */
 211     gst_test_clock_wait_for_next_pending_id (h->testclock, NULL);
 212   }
 213 }
 214
 215 static void
 216 session_harness_force_key_unit (SessionHarness * h,
 217     guint count, guint ssrc, guint payload, gint * reqid, guint64 * sfr)
 218 {
 219   GstClockTime running_time = GST_CLOCK_TIME_NONE;
 220   gboolean all_headers = TRUE;
 221
 222   GstStructure *s = gst_structure_new ("GstForceKeyUnit",
 223       "running-time", GST_TYPE_CLOCK_TIME, running_time,
 224       "all-headers", G_TYPE_BOOLEAN, all_headers,
 225       "count", G_TYPE_UINT, count,
 226       "ssrc", G_TYPE_UINT, ssrc,
 227       "payload", G_TYPE_UINT, payload,
 228       NULL);
 229
 230   if (reqid)
 231     gst_structure_set (s, "reqid", G_TYPE_INT, *reqid, NULL);
 232   if (sfr)
 233     gst_structure_set (s, "sfr", G_TYPE_UINT64, *sfr, NULL);
 234
 235   gst_harness_push_upstream_event (h->recv_rtp_h,
 236       gst_event_new_custom (GST_EVENT_CUSTOM_UPSTREAM, s));
 237 }
 238
 239 static void
 240 session_harness_rtp_retransmission_request (SessionHarness * h,
 241     guint ssrc, guint seqnum, guint delay, guint deadline, guint avg_rtt)
 242 {
 243   GstClockTime running_time = GST_CLOCK_TIME_NONE;
 244
 245   GstStructure *s = gst_structure_new ("GstRTPRetransmissionRequest",
 246       "running-time", GST_TYPE_CLOCK_TIME, running_time,
 247       "ssrc", G_TYPE_UINT, ssrc,
 248       "seqnum", G_TYPE_UINT, seqnum,
 249       "delay", G_TYPE_UINT, delay,
 250       "deadline", G_TYPE_UINT, deadline,
 251       "avg-rtt", G_TYPE_UINT, avg_rtt,
 252       NULL);
 253   gst_harness_push_upstream_event (h->recv_rtp_h,
 254       gst_event_new_custom (GST_EVENT_CUSTOM_UPSTREAM, s));
 255 }
 256
 257 GST_START_TEST (test_multiple_ssrc_rr)
 258 {
 259   SessionHarness *h = session_harness_new ();
 260   GstFlowReturn res;
 261   GstBuffer *in_buf, *out_buf;
 262   GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT;
 263   GstRTCPPacket rtcp_packet;
 264   gint i, j;
 265   guint ssrc_match;
 266
 267   guint ssrcs[] = {
 268     0x01BADBAD,
 269     0xDEADBEEF,
 270   };
 271
 272   /* receive buffers with multiple ssrcs */
 273   for (i = 0; i < 2; i++) {
 274     for (j = 0; j < G_N_ELEMENTS (ssrcs); j++) {
 275       in_buf = generate_test_buffer (i, ssrcs[j]);
 276       res = session_harness_recv_rtp (h, in_buf);
 277       fail_unless_equals_int (GST_FLOW_OK, res);
 278     }
 279   }
 280
 281   /* crank the rtcp-thread and pull out the rtcp-packet we have generated */
 282   session_harness_crank_clock (h);
 283   out_buf = session_harness_pull_rtcp (h);
 284
 285   /* verify we have report blocks for both ssrcs */
 286   g_assert (out_buf != NULL);
 287   fail_unless (gst_rtcp_buffer_validate (out_buf));
 288   gst_rtcp_buffer_map (out_buf, GST_MAP_READ, &rtcp);
 289   g_assert (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet));
 290   fail_unless_equals_int (GST_RTCP_TYPE_RR,
 291       gst_rtcp_packet_get_type (&rtcp_packet));
 292
 293   fail_unless_equals_int (G_N_ELEMENTS (ssrcs),
 294       gst_rtcp_packet_get_rb_count (&rtcp_packet));
 295
 296   ssrc_match = 0;
 297   for (i = 0; i < G_N_ELEMENTS (ssrcs); i++) {
 298     guint32 ssrc;
 299     gst_rtcp_packet_get_rb (&rtcp_packet, i, &ssrc,
 300         NULL, NULL, NULL, NULL, NULL, NULL);
 301     for (j = 0; j < G_N_ELEMENTS (ssrcs); j++) {
 302       if (ssrcs[j] == ssrc)
 303         ssrc_match++;
 304     }
 305   }
 306   fail_unless_equals_int (G_N_ELEMENTS (ssrcs), ssrc_match);
 307
 308   gst_rtcp_buffer_unmap (&rtcp);
 309   gst_buffer_unref (out_buf);
 310
 311   session_harness_free (h);
 312 }
 313
 314 GST_END_TEST;
 315
 316 /* This verifies that rtpsession will correctly place RBs round-robin
 317  * across multiple RRs when there are too many senders that their RBs
 318  * do not fit in one RR */
 319 GST_START_TEST (test_multiple_senders_roundrobin_rbs)
 320 {
 321   SessionHarness *h = session_harness_new ();
 322   GstFlowReturn res;
 323   GstBuffer *buf;
 324   GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT;
 325   GstRTCPPacket rtcp_packet;
 326   gint i, j, k;
 327   guint32 ssrc;
 328   GHashTable *rb_ssrcs, *tmp_set;
 329
 330   g_object_set (h->internal_session, "internal-ssrc", 0xDEADBEEF, NULL);
 331
 332   for (i = 0; i < 2; i++) {     /* cycles between RR reports */
 333     for (j = 0; j < 5; j++) {   /* packets per ssrc */
 334       gint seq = (i * 5) + j;
 335       for (k = 0; k < 35; k++) {        /* number of ssrcs */
 336         buf = generate_test_buffer (seq, 10000 + k);
 337         res = session_harness_recv_rtp (h, buf);
 338         fail_unless_equals_int (GST_FLOW_OK, res);
 339       }
 340     }
 341   }
 342
 343   rb_ssrcs = g_hash_table_new_full (g_direct_hash, g_direct_equal, NULL,
 344       (GDestroyNotify) g_hash_table_unref);
 345
 346   /* verify the rtcp packets */
 347   for (i = 0; i < 2; i++) {
 348     guint expected_rb_count = (i < 1) ? GST_RTCP_MAX_RB_COUNT :
 349         (35 - GST_RTCP_MAX_RB_COUNT);
 350
 351     session_harness_produce_rtcp (h, 1);
 352     buf = session_harness_pull_rtcp (h);
 353     g_assert (buf != NULL);
 354     fail_unless (gst_rtcp_buffer_validate (buf));
 355
 356     gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp);
 357     fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet));
 358     fail_unless_equals_int (GST_RTCP_TYPE_RR,
 359         gst_rtcp_packet_get_type (&rtcp_packet));
 360
 361     ssrc = gst_rtcp_packet_rr_get_ssrc (&rtcp_packet);
 362     fail_unless_equals_int (0xDEADBEEF, ssrc);
 363
 364     /* inspect the RBs */
 365     fail_unless_equals_int (expected_rb_count,
 366         gst_rtcp_packet_get_rb_count (&rtcp_packet));
 367
 368     if (i == 0) {
 369       tmp_set = g_hash_table_new (g_direct_hash, g_direct_equal);
 370       g_hash_table_insert (rb_ssrcs, GUINT_TO_POINTER (ssrc), tmp_set);
 371     } else {
 372       tmp_set = g_hash_table_lookup (rb_ssrcs, GUINT_TO_POINTER (ssrc));
 373       g_assert (tmp_set);
 374     }
 375
 376     for (j = 0; j < expected_rb_count; j++) {
 377       gst_rtcp_packet_get_rb (&rtcp_packet, j, &ssrc, NULL, NULL,
 378           NULL, NULL, NULL, NULL);
 379       g_assert_cmpint (ssrc, >=, 10000);
 380       g_assert_cmpint (ssrc, <=, 10035);
 381       g_hash_table_add (tmp_set, GUINT_TO_POINTER (ssrc));
 382     }
 383
 384     gst_rtcp_buffer_unmap (&rtcp);
 385     gst_buffer_unref (buf);
 386   }
 387
 388   /* now verify all received ssrcs have been reported */
 389   fail_unless_equals_int (1, g_hash_table_size (rb_ssrcs));
 390   tmp_set = g_hash_table_lookup (rb_ssrcs, GUINT_TO_POINTER (0xDEADBEEF));
 391   g_assert (tmp_set);
 392   fail_unless_equals_int (35, g_hash_table_size (tmp_set));
 393
 394   g_hash_table_unref (rb_ssrcs);
 395   session_harness_free (h);
 396 }
 397
 398 GST_END_TEST;
 399
 400 GST_START_TEST (test_no_rbs_for_internal_senders)
 401 {
 402   SessionHarness *h = session_harness_new ();
 403   GstFlowReturn res;
 404   GstBuffer *buf;
 405   GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT;
 406   GstRTCPPacket rtcp_packet;
 407   gint i, j, k;
 408   guint32 ssrc;
 409   GHashTable *sr_ssrcs;
 410   GHashTable *rb_ssrcs, *tmp_set;
 411
 412   /* Push RTP from our send SSRCs */
 413   for (j = 0; j < 5; j++) {     /* packets per ssrc */
 414     for (k = 0; k < 2; k++) {   /* number of ssrcs */
 415       buf = generate_test_buffer (j, 10000 + k);
 416       res = session_harness_send_rtp (h, buf);
 417       fail_unless_equals_int (GST_FLOW_OK, res);
 418     }
 419   }
 420
 421   /* crank the RTCP pad thread */
 422   session_harness_crank_clock (h);
 423
 424   sr_ssrcs = g_hash_table_new (g_direct_hash, g_direct_equal);
 425
 426   /* verify the rtcp packets */
 427   for (i = 0; i < 2; i++) {
 428     buf = session_harness_pull_rtcp (h);
 429     g_assert (buf != NULL);
 430     g_assert (gst_rtcp_buffer_validate (buf));
 431
 432     gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp);
 433     g_assert (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet));
 434     fail_unless_equals_int (GST_RTCP_TYPE_SR,
 435         gst_rtcp_packet_get_type (&rtcp_packet));
 436
 437     gst_rtcp_packet_sr_get_sender_info (&rtcp_packet, &ssrc, NULL, NULL,
 438         NULL, NULL);
 439     g_assert_cmpint (ssrc, >=, 10000);
 440     g_assert_cmpint (ssrc, <=, 10001);
 441     g_hash_table_add (sr_ssrcs, GUINT_TO_POINTER (ssrc));
 442
 443     /* There should be no RBs as there are no remote senders */
 444     fail_unless_equals_int (0, gst_rtcp_packet_get_rb_count (&rtcp_packet));
 445
 446     gst_rtcp_buffer_unmap (&rtcp);
 447     gst_buffer_unref (buf);
 448   }
 449
 450   /* Ensure both internal senders generated RTCP */
 451   fail_unless_equals_int (2, g_hash_table_size (sr_ssrcs));
 452   g_hash_table_unref (sr_ssrcs);
 453
 454   /* Generate RTP from remote side */
 455   for (j = 0; j < 5; j++) {     /* packets per ssrc */
 456     for (k = 0; k < 2; k++) {   /* number of ssrcs */
 457       buf = generate_test_buffer (j, 20000 + k);
 458       res = session_harness_recv_rtp (h, buf);
 459       fail_unless_equals_int (GST_FLOW_OK, res);
 460     }
 461   }
 462
 463   sr_ssrcs = g_hash_table_new (g_direct_hash, g_direct_equal);
 464   rb_ssrcs = g_hash_table_new_full (g_direct_hash, g_direct_equal, NULL,
 465       (GDestroyNotify) g_hash_table_unref);
 466
 467   /* verify the rtcp packets */
 468   for (i = 0; i < 2; i++) {
 469     session_harness_produce_rtcp (h, 1);
 470     buf = session_harness_pull_rtcp (h);
 471     g_assert (buf != NULL);
 472     g_assert (gst_rtcp_buffer_validate (buf));
 473
 474     gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp);
 475     g_assert (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet));
 476     fail_unless_equals_int (GST_RTCP_TYPE_SR,
 477         gst_rtcp_packet_get_type (&rtcp_packet));
 478
 479     gst_rtcp_packet_sr_get_sender_info (&rtcp_packet, &ssrc, NULL, NULL,
 480         NULL, NULL);
 481     g_assert_cmpint (ssrc, >=, 10000);
 482     g_assert_cmpint (ssrc, <=, 10001);
 483     g_hash_table_add (sr_ssrcs, GUINT_TO_POINTER (ssrc));
 484
 485     /* There should be 2 RBs: one for each remote sender */
 486     fail_unless_equals_int (2, gst_rtcp_packet_get_rb_count (&rtcp_packet));
 487
 488     tmp_set = g_hash_table_new (g_direct_hash, g_direct_equal);
 489     g_hash_table_insert (rb_ssrcs, GUINT_TO_POINTER (ssrc), tmp_set);
 490
 491     for (j = 0; j < 2; j++) {
 492       gst_rtcp_packet_get_rb (&rtcp_packet, j, &ssrc, NULL, NULL,
 493           NULL, NULL, NULL, NULL);
 494       g_assert_cmpint (ssrc, >=, 20000);
 495       g_assert_cmpint (ssrc, <=, 20001);
 496       g_hash_table_add (tmp_set, GUINT_TO_POINTER (ssrc));
 497     }
 498
 499     gst_rtcp_buffer_unmap (&rtcp);
 500     gst_buffer_unref (buf);
 501   }
 502
 503   /* now verify all received ssrcs have been reported */
 504   fail_unless_equals_int (2, g_hash_table_size (sr_ssrcs));
 505   fail_unless_equals_int (2, g_hash_table_size (rb_ssrcs));
 506   for (i = 10000; i < 10002; i++) {
 507     tmp_set = g_hash_table_lookup (rb_ssrcs, GUINT_TO_POINTER (i));
 508     g_assert (tmp_set);
 509     fail_unless_equals_int (2, g_hash_table_size (tmp_set));
 510   }
 511
 512   g_hash_table_unref (rb_ssrcs);
 513   g_hash_table_unref (sr_ssrcs);
 514
 515   session_harness_free (h);
 516 }
 517
 518 GST_END_TEST;
 519
 520 GST_START_TEST (test_internal_sources_timeout)
 521 {
 522   SessionHarness *h = session_harness_new ();
 523   guint internal_ssrc;
 524   guint32 ssrc;
 525   GstBuffer *buf;
 526   GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT;
 527   GstRTCPPacket rtcp_packet;
 528   GstRTCPType rtcp_type;
 529   GstFlowReturn res;
 530   gint i, j;
 531   GstCaps *caps;
 532   gboolean seen_bye;
 533
 534   g_object_set (h->internal_session, "internal-ssrc", 0xDEADBEEF, NULL);
 535   g_object_get (h->internal_session, "internal-ssrc", &internal_ssrc, NULL);
 536   fail_unless_equals_int (0xDEADBEEF, internal_ssrc);
 537
 538   for (i = 1; i < 4; i++) {
 539     buf = generate_test_buffer (i, 0xBEEFDEAD);
 540     res = session_harness_recv_rtp (h, buf);
 541     fail_unless_equals_int (GST_FLOW_OK, res);
 542   }
 543
 544   /* verify that rtpsession has sent RR for an internally-created
 545    * RTPSource that is using the internal-ssrc */
 546   session_harness_produce_rtcp (h, 1);
 547   buf = session_harness_pull_rtcp (h);
 548
 549   fail_unless (buf != NULL);
 550   fail_unless (gst_rtcp_buffer_validate (buf));
 551   gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp);
 552   fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet));
 553   fail_unless_equals_int (GST_RTCP_TYPE_RR,
 554       gst_rtcp_packet_get_type (&rtcp_packet));
 555   ssrc = gst_rtcp_packet_rr_get_ssrc (&rtcp_packet);
 556   fail_unless_equals_int (ssrc, internal_ssrc);
 557   gst_rtcp_buffer_unmap (&rtcp);
 558   gst_buffer_unref (buf);
 559
 560   /* ok, now let's push some RTP packets */
 561   caps = gst_caps_new_simple ("application/x-rtp",
 562       "ssrc", G_TYPE_UINT, 0x01BADBAD, NULL);
 563   gst_harness_set_src_caps (h->send_rtp_h, caps);
 564
 565   for (i = 1; i < 4; i++) {
 566     buf = generate_test_buffer (i, 0x01BADBAD);
 567     res = session_harness_send_rtp (h, buf);
 568     fail_unless_equals_int (GST_FLOW_OK, res);
 569   }
 570
 571   /* internal ssrc must have changed already */
 572   g_object_get (h->internal_session, "internal-ssrc", &internal_ssrc, NULL);
 573   fail_unless (internal_ssrc != ssrc);
 574   fail_unless_equals_int (0x01BADBAD, internal_ssrc);
 575
 576   /* verify SR and RR */
 577   j = 0;
 578   for (i = 0; i < 5; i++) {
 579     session_harness_produce_rtcp (h, 1);
 580     buf = session_harness_pull_rtcp (h);
 581     g_assert (buf != NULL);
 582     fail_unless (gst_rtcp_buffer_validate (buf));
 583     gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp);
 584     fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet));
 585     rtcp_type = gst_rtcp_packet_get_type (&rtcp_packet);
 586
 587     if (rtcp_type == GST_RTCP_TYPE_SR) {
 588       gst_rtcp_packet_sr_get_sender_info (&rtcp_packet, &ssrc, NULL, NULL, NULL,
 589           NULL);
 590       fail_unless_equals_int (internal_ssrc, ssrc);
 591       fail_unless_equals_int (0x01BADBAD, ssrc);
 592       j |= 0x1;
 593     } else if (rtcp_type == GST_RTCP_TYPE_RR) {
 594       ssrc = gst_rtcp_packet_rr_get_ssrc (&rtcp_packet);
 595       if (internal_ssrc != ssrc)
 596         j |= 0x2;
 597     }
 598     gst_rtcp_buffer_unmap (&rtcp);
 599     gst_buffer_unref (buf);
 600   }
 601   fail_unless_equals_int (0x3, j);      /* verify we got both SR and RR */
 602
 603   /* go 30 seconds in the future and observe both sources timing out:
 604    * 0xDEADBEEF -> BYE, 0x01BADBAD -> becomes receiver only */
 605   fail_unless (session_harness_advance_and_crank (h, 30 * GST_SECOND));
 606
 607   /* verify BYE and RR */
 608   j = 0;
 609   seen_bye = FALSE;
 610   while (!seen_bye) {
 611     session_harness_produce_rtcp (h, 1);
 612     buf = session_harness_pull_rtcp (h);
 613     fail_unless (buf != NULL);
 614     fail_unless (gst_rtcp_buffer_validate (buf));
 615     gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp);
 616     fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet));
 617     rtcp_type = gst_rtcp_packet_get_type (&rtcp_packet);
 618
 619     if (rtcp_type == GST_RTCP_TYPE_RR) {
 620       ssrc = gst_rtcp_packet_rr_get_ssrc (&rtcp_packet);
 621       if (ssrc == 0x01BADBAD) {
 622         j |= 0x1;
 623         fail_unless_equals_int (internal_ssrc, ssrc);
 624         /* 2 => RR, SDES. There is no BYE here */
 625         fail_unless_equals_int (2, gst_rtcp_buffer_get_packet_count (&rtcp));
 626       } else if (ssrc == 0xDEADBEEF) {
 627         j |= 0x2;
 628         g_assert_cmpint (ssrc, !=, internal_ssrc);
 629         /* 3 => RR, SDES, BYE */
 630         if (gst_rtcp_buffer_get_packet_count (&rtcp) == 3) {
 631           fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet));
 632           fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet));
 633           fail_unless_equals_int (GST_RTCP_TYPE_BYE,
 634               gst_rtcp_packet_get_type (&rtcp_packet));
 635           seen_bye = TRUE;
 636         }
 637       }
 638     }
 639     gst_rtcp_buffer_unmap (&rtcp);
 640     gst_buffer_unref (buf);
 641   }
 642   fail_unless_equals_int (0x3, j);      /* verify we got both BYE and RR */
 643
 644   session_harness_free (h);
 645 }
 646
 647 GST_END_TEST;
 648
 649 typedef struct
 650 {
 651   guint8 subtype;
 652   guint32 ssrc;
 653   gchar *name;
 654   GstBuffer *data;
 655 } RTCPAppResult;
 656
 657 static void
 658 on_app_rtcp_cb (GObject * session, guint subtype, guint ssrc,
 659     const gchar * name, GstBuffer * data, RTCPAppResult * result)
 660 {
 661   result->subtype = subtype;
 662   result->ssrc = ssrc;
 663   result->name = g_strdup (name);
 664   result->data = data ? gst_buffer_ref (data) : NULL;
 665 }
 666
 667 GST_START_TEST (test_receive_rtcp_app_packet)
 668 {
 669   SessionHarness *h = session_harness_new ();
 670   GstBuffer *buf;
 671   GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT;
 672   GstRTCPPacket packet;
 673   RTCPAppResult result = { 0 };
 674   guint8 data[] = { 0x11, 0x22, 0x33, 0x44 };
 675
 676   g_signal_connect (h->internal_session, "on-app-rtcp",
 677       G_CALLBACK (on_app_rtcp_cb), &result);
 678
 679   /* Push APP buffer with no data */
 680   buf = gst_rtcp_buffer_new (1000);
 681   fail_unless (gst_rtcp_buffer_map (buf, GST_MAP_READWRITE, &rtcp));
 682   fail_unless (gst_rtcp_buffer_add_packet (&rtcp, GST_RTCP_TYPE_APP, &packet));
 683   gst_rtcp_packet_app_set_subtype (&packet, 21);
 684   gst_rtcp_packet_app_set_ssrc (&packet, 0x11111111);
 685   gst_rtcp_packet_app_set_name (&packet, "Test");
 686   gst_rtcp_buffer_unmap (&rtcp);
 687
 688   fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtcp (h, buf));
 689
 690   fail_unless_equals_int (21, result.subtype);
 691   fail_unless_equals_int (0x11111111, result.ssrc);
 692   fail_unless_equals_string ("Test", result.name);
 693   fail_unless_equals_pointer (NULL, result.data);
 694
 695   g_free (result.name);
 696
 697   /* Push APP buffer with data */
 698   memset (&result, 0, sizeof (result));
 699   buf = gst_rtcp_buffer_new (1000);
 700   fail_unless (gst_rtcp_buffer_map (buf, GST_MAP_READWRITE, &rtcp));
 701   fail_unless (gst_rtcp_buffer_add_packet (&rtcp, GST_RTCP_TYPE_APP, &packet));
 702   gst_rtcp_packet_app_set_subtype (&packet, 22);
 703   gst_rtcp_packet_app_set_ssrc (&packet, 0x22222222);
 704   gst_rtcp_packet_app_set_name (&packet, "Test");
 705   gst_rtcp_packet_app_set_data_length (&packet, sizeof (data) / 4);
 706   memcpy (gst_rtcp_packet_app_get_data (&packet), data, sizeof (data));
 707   gst_rtcp_buffer_unmap (&rtcp);
 708
 709   fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtcp (h, buf));
 710
 711   fail_unless_equals_int (22, result.subtype);
 712   fail_unless_equals_int (0x22222222, result.ssrc);
 713   fail_unless_equals_string ("Test", result.name);
 714   fail_unless (gst_buffer_memcmp (result.data, 0, data, sizeof (data)) == 0);
 715
 716   g_free (result.name);
 717   gst_buffer_unref (result.data);
 718
 719   session_harness_free (h);
 720 }
 721
 722 GST_END_TEST;
 723
 724 static void
 725 stats_test_cb (GObject * object, GParamSpec * spec, gpointer data)
 726 {
 727   guint num_sources = 0;
 728   gboolean *cb_called = data;
 729   g_assert (*cb_called == FALSE);
 730
 731   /* We should be able to get a rtpsession property
 732      without introducing the deadlock */
 733   g_object_get (object, "num-sources", &num_sources, NULL);
 734
 735   *cb_called = TRUE;
 736 }
 737
 738 GST_START_TEST (test_dont_lock_on_stats)
 739 {
 740   SessionHarness *h = session_harness_new ();
 741   gboolean cb_called = FALSE;
 742
 743   /* connect to the stats-reporting */
 744   g_signal_connect (h->session, "notify::stats",
 745       G_CALLBACK (stats_test_cb), &cb_called);
 746
 747   /* Push RTP buffer to make sure RTCP-thread have started */
 748   fail_unless_equals_int (GST_FLOW_OK,
 749       session_harness_send_rtp (h, generate_test_buffer (0, 0xDEADBEEF)));
 750
 751   /* crank the RTCP-thread and pull out rtcp, generating a stats-callback */
 752   session_harness_crank_clock (h);
 753   gst_buffer_unref (session_harness_pull_rtcp (h));
 754   fail_unless (cb_called);
 755
 756   session_harness_free (h);
 757 }
 758
 759 GST_END_TEST;
 760
 761 static void
 762 suspicious_bye_cb (GObject * object, GParamSpec * spec, gpointer data)
 763 {
 764   GValueArray *stats_arr;
 765   GstStructure *stats, *internal_stats;
 766   gboolean *cb_called = data;
 767   gboolean internal = FALSE, sent_bye = TRUE;
 768   guint ssrc = 0;
 769   guint i;
 770
 771   g_assert (*cb_called == FALSE);
 772   *cb_called = TRUE;
 773
 774   g_object_get (object, "stats", &stats, NULL);
 775   stats_arr =
 776       g_value_get_boxed (gst_structure_get_value (stats, "source-stats"));
 777   g_assert (stats_arr != NULL);
 778   fail_unless (stats_arr->n_values >= 1);
 779
 780   for (i = 0; i < stats_arr->n_values; i++) {
 781     internal_stats = g_value_get_boxed (g_value_array_get_nth (stats_arr, i));
 782     g_assert (internal_stats != NULL);
 783
 784     gst_structure_get (internal_stats,
 785         "ssrc", G_TYPE_UINT, &ssrc,
 786         "internal", G_TYPE_BOOLEAN, &internal,
 787         "received-bye", G_TYPE_BOOLEAN, &sent_bye, NULL);
 788
 789     if (ssrc == 0xDEADBEEF) {
 790       fail_unless (internal);
 791       fail_unless (!sent_bye);
 792       break;
 793     }
 794   }
 795   fail_unless_equals_int (ssrc, 0xDEADBEEF);
 796
 797   gst_structure_free (stats);
 798 }
 799
 800 static GstBuffer *
 801 create_bye_rtcp (guint32 ssrc)
 802 {
 803   GstRTCPPacket packet;
 804   GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT;
 805   GSocketAddress *saddr;
 806   GstBuffer *buffer = gst_rtcp_buffer_new (1000);
 807
 808   fail_unless (gst_rtcp_buffer_map (buffer, GST_MAP_READWRITE, &rtcp));
 809   fail_unless (gst_rtcp_buffer_add_packet (&rtcp, GST_RTCP_TYPE_BYE, &packet));
 810   gst_rtcp_packet_bye_add_ssrc (&packet, ssrc);
 811   gst_rtcp_buffer_unmap (&rtcp);
 812
 813   /* Need to add meta to trigger collision detection */
 814   saddr = g_inet_socket_address_new_from_string ("127.0.0.1", 3490);
 815   gst_buffer_add_net_address_meta (buffer, saddr);
 816   g_object_unref (saddr);
 817   return buffer;
 818 }
 819
 820 GST_START_TEST (test_ignore_suspicious_bye)
 821 {
 822   SessionHarness *h = session_harness_new ();
 823   gboolean cb_called = FALSE;
 824
 825   /* connect to the stats-reporting */
 826   g_signal_connect (h->session, "notify::stats",
 827       G_CALLBACK (suspicious_bye_cb), &cb_called);
 828
 829   /* Push RTP buffer making our internal SSRC=0xDEADBEEF */
 830   fail_unless_equals_int (GST_FLOW_OK,
 831       session_harness_send_rtp (h, generate_test_buffer (0, 0xDEADBEEF)));
 832
 833   /* Receive BYE RTCP referencing our internal SSRC(!?!) (0xDEADBEEF) */
 834   fail_unless_equals_int (GST_FLOW_OK,
 835       session_harness_recv_rtcp (h, create_bye_rtcp (0xDEADBEEF)));
 836
 837   /* "crank" and check the stats */
 838   session_harness_crank_clock (h);
 839   gst_buffer_unref (session_harness_pull_rtcp (h));
 840   fail_unless (cb_called);
 841
 842   session_harness_free (h);
 843 }
 844
 845 GST_END_TEST;
 846
 847 static GstBuffer *
 848 create_buffer (guint8 * data, gsize size)
 849 {
 850   return gst_buffer_new_wrapped_full (GST_MEMORY_FLAG_READONLY,
 851       data, size, 0, size, NULL, NULL);
 852 }
 853
 854 GST_START_TEST (test_receive_regular_pli)
 855 {
 856   SessionHarness *h = session_harness_new ();
 857   GstEvent *ev;
 858
 859   /* PLI packet */
 860   guint8 rtcp_pkt[] = {
 861     0x81,                       /* PLI */
 862     0xce,                       /* Type 206 Application layer feedback */
 863     0x00, 0x02,                 /* Length */
 864     0x37, 0x56, 0x93, 0xed,     /* Sender SSRC */
 865     0x37, 0x56, 0x93, 0xed      /* Media SSRC */
 866   };
 867
 868   fail_unless_equals_int (GST_FLOW_OK,
 869       session_harness_send_rtp (h, generate_test_buffer (0, 928420845)));
 870
 871   session_harness_recv_rtcp (h, create_buffer (rtcp_pkt, sizeof (rtcp_pkt)));
 872   fail_unless_equals_int (3,
 873       gst_harness_upstream_events_received (h->send_rtp_h));
 874
 875   /* Remove the first 2 reconfigure events */
 876   fail_unless ((ev = gst_harness_pull_upstream_event (h->send_rtp_h)) != NULL);
 877   fail_unless_equals_int (GST_EVENT_RECONFIGURE, GST_EVENT_TYPE (ev));
 878   gst_event_unref (ev);
 879   fail_unless ((ev = gst_harness_pull_upstream_event (h->send_rtp_h)) != NULL);
 880   fail_unless_equals_int (GST_EVENT_RECONFIGURE, GST_EVENT_TYPE (ev));
 881   gst_event_unref (ev);
 882
 883   /* Then pull and check the force key-unit event */
 884   fail_unless ((ev = gst_harness_pull_upstream_event (h->send_rtp_h)) != NULL);
 885   fail_unless_equals_int (GST_EVENT_CUSTOM_UPSTREAM, GST_EVENT_TYPE (ev));
 886   fail_unless (gst_video_event_is_force_key_unit (ev));
 887   gst_event_unref (ev);
 888
 889   session_harness_free (h);
 890 }
 891
 892 GST_END_TEST;
 893
 894 GST_START_TEST (test_receive_pli_no_sender_ssrc)
 895 {
 896   SessionHarness *h = session_harness_new ();
 897   GstEvent *ev;
 898
 899   /* PLI packet */
 900   guint8 rtcp_pkt[] = {
 901     0x81,                       /* PLI */
 902     0xce,                       /* Type 206 Application layer feedback */
 903     0x00, 0x02,                 /* Length */
 904     0x00, 0x00, 0x00, 0x00,     /* Sender SSRC */
 905     0x37, 0x56, 0x93, 0xed      /* Media SSRC */
 906   };
 907
 908   fail_unless_equals_int (GST_FLOW_OK,
 909       session_harness_send_rtp (h, generate_test_buffer (0, 928420845)));
 910
 911   session_harness_recv_rtcp (h, create_buffer (rtcp_pkt, sizeof (rtcp_pkt)));
 912   fail_unless_equals_int (3,
 913       gst_harness_upstream_events_received (h->send_rtp_h));
 914
 915   /* Remove the first 2 reconfigure events */
 916   fail_unless ((ev = gst_harness_pull_upstream_event (h->send_rtp_h)) != NULL);
 917   fail_unless_equals_int (GST_EVENT_RECONFIGURE, GST_EVENT_TYPE (ev));
 918   gst_event_unref (ev);
 919   fail_unless ((ev = gst_harness_pull_upstream_event (h->send_rtp_h)) != NULL);
 920   fail_unless_equals_int (GST_EVENT_RECONFIGURE, GST_EVENT_TYPE (ev));
 921   gst_event_unref (ev);
 922
 923   /* Then pull and check the force key-unit event */
 924   fail_unless ((ev = gst_harness_pull_upstream_event (h->send_rtp_h)) != NULL);
 925   fail_unless_equals_int (GST_EVENT_CUSTOM_UPSTREAM, GST_EVENT_TYPE (ev));
 926   fail_unless (gst_video_event_is_force_key_unit (ev));
 927   gst_event_unref (ev);
 928
 929   session_harness_free (h);
 930 }
 931
 932 GST_END_TEST;
 933
 934 static void
 935 add_rtcp_sdes_packet (GstBuffer * gstbuf, guint32 ssrc, const char *cname)
 936 {
 937   GstRTCPPacket packet;
 938   GstRTCPBuffer buffer = GST_RTCP_BUFFER_INIT;
 939
 940   gst_rtcp_buffer_map (gstbuf, GST_MAP_READWRITE, &buffer);
 941
 942   fail_unless (gst_rtcp_buffer_add_packet (&buffer, GST_RTCP_TYPE_SDES,
 943           &packet) == TRUE);
 944   fail_unless (gst_rtcp_packet_sdes_add_item (&packet, ssrc) == TRUE);
 945   fail_unless (gst_rtcp_packet_sdes_add_entry (&packet, GST_RTCP_SDES_CNAME,
 946           strlen (cname), (const guint8 *) cname));
 947
 948   gst_rtcp_buffer_unmap (&buffer);
 949 }
 950
 951 GST_START_TEST (test_ssrc_collision_when_sending)
 952 {
 953   SessionHarness *h = session_harness_new ();
 954   GstBuffer *buf = gst_rtcp_buffer_new (1400);
 955
 956 /* Push SDES with identical SSRC as what we will use for sending RTP,
 957    establishing this as a non-internal SSRC */
 958   add_rtcp_sdes_packet (buf, 0x12345678, "test@foo.bar");
 959   session_harness_recv_rtcp (h, buf);
 960
 961   /* Push RTP buffer making our internal SSRC=0x12345678 */
 962   fail_unless_equals_int (GST_FLOW_OK,
 963       session_harness_send_rtp (h, generate_test_buffer (0, 0x12345678)));
 964
 965   /* Verify the packet we just sent is not being boomeranged back to us
 966      as a received packet! */
 967   fail_unless_equals_int (0, gst_harness_buffers_in_queue (h->recv_rtp_h));
 968
 969   /* FIXME: verify a Collision event coming upstream! */
 970
 971   session_harness_free (h);
 972 }
 973
 974 GST_END_TEST;
 975
 976 GST_START_TEST (test_request_fir)
 977 {
 978   SessionHarness *h = session_harness_new ();
 979   GstBuffer *buf;
 980   GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT;
 981   GstRTCPPacket rtcp_packet;
 982   guint8 *fci_data;
 983
 984   /* add FIR-capabilites to our caps */
 985   gst_caps_set_simple (h->caps, "rtcp-fb-ccm-fir", G_TYPE_BOOLEAN, TRUE, NULL);
 986   /* clear pt-map to removed the cached caps without fir */
 987   g_signal_emit_by_name (h->session, "clear-pt-map");
 988
 989   g_object_set (h->internal_session, "internal-ssrc", 0xDEADBEEF, NULL);
 990
 991   /* Receive a RTP buffer from the wire from 2 different ssrcs */
 992   fail_unless_equals_int (GST_FLOW_OK,
 993       session_harness_recv_rtp (h, generate_test_buffer (0, 0x12345678)));
 994   fail_unless_equals_int (GST_FLOW_OK,
 995       session_harness_recv_rtp (h, generate_test_buffer (0, 0x87654321)));
 996
 997   /* fix to make the test deterministic: We need to wait for the RTCP-thread
 998      to have settled to ensure the key-unit will considered once released */
 999   gst_test_clock_wait_for_next_pending_id (h->testclock, NULL);
1000
1001   /* request FIR for both SSRCs */
1002   session_harness_force_key_unit (h, 0, 0x12345678, TEST_BUF_PT, NULL, NULL);
1003   session_harness_force_key_unit (h, 0, 0x87654321, TEST_BUF_PT, NULL, NULL);
1004
1005   session_harness_produce_rtcp (h, 1);
1006   buf = session_harness_pull_rtcp (h);
1007
1008   fail_unless (gst_rtcp_buffer_validate (buf));
1009   gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp);
1010   fail_unless_equals_int (3, gst_rtcp_buffer_get_packet_count (&rtcp));
1011   fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet));
1012
1013   /* first a Receiver Report */
1014   fail_unless_equals_int (GST_RTCP_TYPE_RR,
1015       gst_rtcp_packet_get_type (&rtcp_packet));
1016   fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet));
1017
1018   /* then a SDES */
1019   fail_unless_equals_int (GST_RTCP_TYPE_SDES,
1020       gst_rtcp_packet_get_type (&rtcp_packet));
1021   fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet));
1022
1023   /* and then our FIR */
1024   fail_unless_equals_int (GST_RTCP_TYPE_PSFB,
1025       gst_rtcp_packet_get_type (&rtcp_packet));
1026   fail_unless_equals_int (GST_RTCP_PSFB_TYPE_FIR,
1027       gst_rtcp_packet_fb_get_type (&rtcp_packet));
1028
1029   /* FIR has sender-ssrc as normal, but media-ssrc set to 0, because
1030      it can have multiple media-ssrcs in its fci-data */
1031   fail_unless_equals_int (0xDEADBEEF,
1032       gst_rtcp_packet_fb_get_sender_ssrc (&rtcp_packet));
1033   fail_unless_equals_int (0, gst_rtcp_packet_fb_get_media_ssrc (&rtcp_packet));
1034   fci_data = gst_rtcp_packet_fb_get_fci (&rtcp_packet);
1035
1036   fail_unless_equals_int (16,
1037       gst_rtcp_packet_fb_get_fci_length (&rtcp_packet) * sizeof (guint32));
1038
1039   /* verify the FIR contains both SSRCs */
1040   fail_unless_equals_int (0x87654321, GST_READ_UINT32_BE (fci_data));
1041   fail_unless_equals_int (1, fci_data[4]);
1042   fail_unless_equals_int (0, fci_data[5]);
1043   fail_unless_equals_int (0, fci_data[6]);
1044   fail_unless_equals_int (0, fci_data[7]);
1045   fci_data += 8;
1046
1047   fail_unless_equals_int (0x12345678, GST_READ_UINT32_BE (fci_data));
1048   fail_unless_equals_int (1, fci_data[4]);
1049   fail_unless_equals_int (0, fci_data[5]);
1050   fail_unless_equals_int (0, fci_data[6]);
1051   fail_unless_equals_int (0, fci_data[7]);
1052
1053   gst_rtcp_buffer_unmap (&rtcp);
1054   gst_buffer_unref (buf);
1055   session_harness_free (h);
1056 }
1057
1058 GST_END_TEST;
1059
1060 GST_START_TEST (test_request_pli)
1061 {
1062   SessionHarness *h = session_harness_new ();
1063   GstBuffer *buf;
1064   GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT;
1065   GstRTCPPacket rtcp_packet;
1066
1067   /* add PLI-capabilites to our caps */
1068   gst_caps_set_simple (h->caps, "rtcp-fb-nack-pli", G_TYPE_BOOLEAN, TRUE, NULL);
1069   /* clear pt-map to removed the cached caps without PLI */
1070   g_signal_emit_by_name (h->session, "clear-pt-map");
1071
1072   g_object_set (h->internal_session, "internal-ssrc", 0xDEADBEEF, NULL);
1073
1074   /* Receive a RTP buffer from the wire */
1075   fail_unless_equals_int (GST_FLOW_OK,
1076       session_harness_recv_rtp (h, generate_test_buffer (0, 0x12345678)));
1077
1078   /* Wait for first regular RTCP to be sent so that we are clear to send early RTCP */
1079   session_harness_produce_rtcp (h, 1);
1080   gst_buffer_unref (session_harness_pull_rtcp (h));
1081
1082   /* request PLI */
1083   session_harness_force_key_unit (h, 0, 0x12345678, TEST_BUF_PT, NULL, NULL);
1084
1085   /* PLI should be produced immediately as early RTCP is allowed. Pull buffer
1086      without advancing the clock to ensure this is the case */
1087   buf = session_harness_pull_rtcp (h);
1088   fail_unless (gst_rtcp_buffer_validate (buf));
1089   gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp);
1090   fail_unless_equals_int (3, gst_rtcp_buffer_get_packet_count (&rtcp));
1091   fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet));
1092
1093   /* first a Receiver Report */
1094   fail_unless_equals_int (GST_RTCP_TYPE_RR,
1095       gst_rtcp_packet_get_type (&rtcp_packet));
1096   fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet));
1097
1098   /* then a SDES */
1099   fail_unless_equals_int (GST_RTCP_TYPE_SDES,
1100       gst_rtcp_packet_get_type (&rtcp_packet));
1101   fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet));
1102
1103   /* and then our PLI */
1104   fail_unless_equals_int (GST_RTCP_TYPE_PSFB,
1105       gst_rtcp_packet_get_type (&rtcp_packet));
1106   fail_unless_equals_int (GST_RTCP_PSFB_TYPE_PLI,
1107       gst_rtcp_packet_fb_get_type (&rtcp_packet));
1108
1109   fail_unless_equals_int (0xDEADBEEF,
1110       gst_rtcp_packet_fb_get_sender_ssrc (&rtcp_packet));
1111   fail_unless_equals_int (0x12345678,
1112       gst_rtcp_packet_fb_get_media_ssrc (&rtcp_packet));
1113   fail_unless_equals_int (0, gst_rtcp_packet_fb_get_fci_length (&rtcp_packet));
1114
1115   gst_rtcp_buffer_unmap (&rtcp);
1116   gst_buffer_unref (buf);
1117   session_harness_free (h);
1118 }
1119
1120 GST_END_TEST;
1121
1122 GST_START_TEST (test_illegal_rtcp_fb_packet)
1123 {
1124   SessionHarness *h = session_harness_new ();
1125   GstBuffer *buf;
1126   /* Zero length RTCP feedback packet (reduced size) */
1127   const guint8 rtcp_zero_fb_pkt[] = { 0x8f, 0xce, 0x00, 0x00 };
1128
1129   g_object_set (h->internal_session, "internal-ssrc", 0xDEADBEEF, NULL);
1130
1131   buf = gst_buffer_new_and_alloc (sizeof (rtcp_zero_fb_pkt));
1132   gst_buffer_fill (buf, 0, rtcp_zero_fb_pkt, sizeof (rtcp_zero_fb_pkt));
1133   GST_BUFFER_DTS (buf) = GST_BUFFER_PTS (buf) = G_GUINT64_CONSTANT (0);
1134
1135   /* Push the packet, this did previously crash because length of packet was
1136    * never validated. */
1137   fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtcp (h, buf));
1138
1139   session_harness_free (h);
1140 }
1141
1142 GST_END_TEST;
1143
1144 typedef struct
1145 {
1146   GCond *cond;
1147   GMutex *mutex;
1148   gboolean fired;
1149 } FeedbackRTCPCallbackData;
1150
1151 static void
1152 feedback_rtcp_cb (GstElement * element, guint fbtype, guint fmt,
1153     guint sender_ssrc, guint media_ssrc, GstBuffer * fci,
1154     FeedbackRTCPCallbackData * cb_data)
1155 {
1156   g_mutex_lock (cb_data->mutex);
1157   cb_data->fired = TRUE;
1158   g_cond_wait (cb_data->cond, cb_data->mutex);
1159   g_mutex_unlock (cb_data->mutex);
1160 }
1161
1162 static void *
1163 send_feedback_rtcp (SessionHarness * h)
1164 {
1165   GstRTCPPacket packet;
1166   GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT;
1167   GstBuffer *buffer = gst_rtcp_buffer_new (1000);
1168
1169   fail_unless (gst_rtcp_buffer_map (buffer, GST_MAP_READWRITE, &rtcp));
1170   fail_unless (gst_rtcp_buffer_add_packet (&rtcp, GST_RTCP_TYPE_PSFB, &packet));
1171   gst_rtcp_packet_fb_set_type (&packet, GST_RTCP_PSFB_TYPE_PLI);
1172   gst_rtcp_packet_fb_set_fci_length (&packet, 0);
1173   gst_rtcp_packet_fb_set_media_ssrc (&packet, 0xABE2B0B);
1174   gst_rtcp_packet_fb_set_media_ssrc (&packet, 0xDEADBEEF);
1175   gst_rtcp_buffer_unmap (&rtcp);
1176   fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtcp (h, buffer));
1177
1178   return NULL;
1179 }
1180
1181 GST_START_TEST (test_feedback_rtcp_race)
1182 {
1183   SessionHarness *h = session_harness_new ();
1184
1185   GCond cond;
1186   GMutex mutex;
1187   FeedbackRTCPCallbackData cb_data;
1188   GThread *send_rtcp_thread;
1189
1190   g_cond_init (&cond);
1191   g_mutex_init (&mutex);
1192   cb_data.cond = &cond;
1193   cb_data.mutex = &mutex;
1194   cb_data.fired = FALSE;
1195   g_signal_connect (h->internal_session, "on-feedback-rtcp",
1196       G_CALLBACK (feedback_rtcp_cb), &cb_data);
1197
1198   /* Push RTP buffer making external source with SSRC=0xDEADBEEF */
1199   fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtp (h,
1200           generate_test_buffer (0, 0xDEADBEEF)));
1201
1202   /* Push feedback RTCP with media SSRC=0xDEADBEEF */
1203   send_rtcp_thread = g_thread_new (NULL, (GThreadFunc) send_feedback_rtcp, h);
1204
1205   /* Waiting for feedback RTCP callback to fire */
1206   while (!cb_data.fired)
1207     g_usleep (G_USEC_PER_SEC / 100);
1208
1209   /* While send_rtcp_thread thread is waiting for our signal
1210      advance the clock by 30sec triggering removal of 0xDEADBEEF,
1211      as if the source was inactive for too long */
1212   session_harness_advance_and_crank (h, GST_SECOND * 30);
1213   gst_buffer_unref (session_harness_pull_rtcp (h));
1214
1215   /* Let send_rtcp_thread finish */
1216   g_mutex_lock (&mutex);
1217   g_cond_signal (&cond);
1218   g_mutex_unlock (&mutex);
1219   g_thread_join (send_rtcp_thread);
1220
1221   session_harness_free (h);
1222 }
1223
1224 GST_END_TEST;
1225
1226 GST_START_TEST (test_dont_send_rtcp_while_idle)
1227 {
1228   SessionHarness *h = session_harness_new ();
1229
1230   /* verify the RTCP thread has not started */
1231   fail_unless_equals_int (0, gst_test_clock_peek_id_count (h->testclock));
1232   /* and that no RTCP has been pushed */
1233   fail_unless_equals_int (0, gst_harness_buffers_in_queue (h->rtcp_h));
1234
1235   session_harness_free (h);
1236 }
1237
1238 GST_END_TEST;
1239
1240 GST_START_TEST (test_send_rtcp_when_signalled)
1241 {
1242   SessionHarness *h = session_harness_new ();
1243   gboolean ret;
1244
1245   /* verify the RTCP thread has not started */
1246   fail_unless_equals_int (0, gst_test_clock_peek_id_count (h->testclock));
1247   /* and that no RTCP has been pushed */
1248   fail_unless_equals_int (0, gst_harness_buffers_in_queue (h->rtcp_h));
1249
1250   /* then ask explicitly to send RTCP */
1251   g_signal_emit_by_name (h->internal_session,
1252       "send-rtcp-full", GST_SECOND, &ret);
1253   /* this is FALSE due to no next RTCP check time */
1254   fail_unless (ret == FALSE);
1255
1256   /* "crank" and verify RTCP now was sent */
1257   session_harness_crank_clock (h);
1258   gst_buffer_unref (session_harness_pull_rtcp (h));
1259
1260   session_harness_free (h);
1261 }
1262
1263 GST_END_TEST;
1264
1265 static void
1266 validate_sdes_priv (GstBuffer * buf, const char *name_ref, const char *value)
1267 {
1268   GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT;
1269   GstRTCPPacket pkt;
1270
1271   fail_unless (gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp));
1272
1273   fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &pkt));
1274
1275   do {
1276     if (gst_rtcp_packet_get_type (&pkt) == GST_RTCP_TYPE_SDES) {
1277       fail_unless (gst_rtcp_packet_sdes_first_entry (&pkt));
1278
1279       do {
1280         GstRTCPSDESType type;
1281         guint8 len;
1282         guint8 *data;
1283
1284         fail_unless (gst_rtcp_packet_sdes_get_entry (&pkt, &type, &len, &data));
1285
1286         if (type == GST_RTCP_SDES_PRIV) {
1287           char *name = g_strndup ((const gchar *) &data[1], data[0]);
1288           len -= data[0] + 1;
1289           data += data[0] + 1;
1290
1291           fail_unless_equals_int (len, strlen (value));
1292           fail_unless (!strncmp (value, (char *) data, len));
1293           fail_unless_equals_string (name, name_ref);
1294           g_free (name);
1295           goto sdes_done;
1296         }
1297       } while (gst_rtcp_packet_sdes_next_entry (&pkt));
1298
1299       g_assert_not_reached ();
1300     }
1301   } while (gst_rtcp_packet_move_to_next (&pkt));
1302
1303   g_assert_not_reached ();
1304
1305 sdes_done:
1306
1307   fail_unless (gst_rtcp_buffer_unmap (&rtcp));
1308
1309 }
1310
1311 GST_START_TEST (test_change_sent_sdes)
1312 {
1313   SessionHarness *h = session_harness_new ();
1314   GstStructure *s;
1315   GstBuffer *buf;
1316   gboolean ret;
1317   GstFlowReturn res;
1318
1319   /* verify the RTCP thread has not started */
1320   fail_unless_equals_int (0, gst_test_clock_peek_id_count (h->testclock));
1321   /* and that no RTCP has been pushed */
1322   fail_unless_equals_int (0, gst_harness_buffers_in_queue (h->rtcp_h));
1323
1324   s = gst_structure_new ("application/x-rtp-source-sdes",
1325       "other", G_TYPE_STRING, "first", NULL);
1326   g_object_set (h->internal_session, "sdes", s, NULL);
1327   gst_structure_free (s);
1328
1329   /* then ask explicitly to send RTCP */
1330   g_signal_emit_by_name (h->internal_session,
1331       "send-rtcp-full", GST_SECOND, &ret);
1332   /* this is FALSE due to no next RTCP check time */
1333   fail_unless (ret == FALSE);
1334
1335   /* "crank" and verify RTCP now was sent */
1336   session_harness_crank_clock (h);
1337   buf = session_harness_pull_rtcp (h);
1338   fail_unless (buf);
1339   validate_sdes_priv (buf, "other", "first");
1340   gst_buffer_unref (buf);
1341
1342   /* Change the SDES */
1343   s = gst_structure_new ("application/x-rtp-source-sdes",
1344       "other", G_TYPE_STRING, "second", NULL);
1345   g_object_set (h->internal_session, "sdes", s, NULL);
1346   gst_structure_free (s);
1347
1348   /* Send an RTP packet */
1349   buf = generate_test_buffer (22, 10000);
1350   res = session_harness_send_rtp (h, buf);
1351   fail_unless_equals_int (GST_FLOW_OK, res);
1352
1353   /* "crank" enough to ensure a RTCP packet has been produced ! */
1354   session_harness_crank_clock (h);
1355   session_harness_crank_clock (h);
1356   session_harness_crank_clock (h);
1357   session_harness_crank_clock (h);
1358   session_harness_crank_clock (h);
1359   session_harness_crank_clock (h);
1360   session_harness_crank_clock (h);
1361   session_harness_crank_clock (h);
1362   session_harness_crank_clock (h);
1363   session_harness_crank_clock (h);
1364
1365   /* and verify RTCP now was sent with new SDES */
1366   buf = session_harness_pull_rtcp (h);
1367   validate_sdes_priv (buf, "other", "second");
1368   gst_buffer_unref (buf);
1369
1370   session_harness_free (h);
1371 }
1372
1373 GST_END_TEST;
1374
1375 GST_START_TEST (test_request_nack)
1376 {
1377   SessionHarness *h = session_harness_new ();
1378   GstBuffer *buf;
1379   GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT;
1380   GstRTCPPacket rtcp_packet;
1381   guint8 *fci_data;
1382   guint32 fci_length;
1383
1384   g_object_set (h->internal_session, "internal-ssrc", 0xDEADBEEF, NULL);
1385
1386   /* Receive a RTP buffer from the wire */
1387   fail_unless_equals_int (GST_FLOW_OK,
1388       session_harness_recv_rtp (h, generate_test_buffer (0, 0x12345678)));
1389
1390   /* Wait for first regular RTCP to be sent so that we are clear to send early RTCP */
1391   session_harness_produce_rtcp (h, 1);
1392   gst_buffer_unref (session_harness_pull_rtcp (h));
1393
1394   /* request NACK immediately */
1395   session_harness_rtp_retransmission_request (h, 0x12345678, 1234, 0, 0, 0);
1396
1397   /* NACK should be produced immediately as early RTCP is allowed. Pull buffer
1398      without advancing the clock to ensure this is the case */
1399   buf = session_harness_pull_rtcp (h);
1400
1401   fail_unless (gst_rtcp_buffer_validate (buf));
1402   gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp);
1403   fail_unless_equals_int (3, gst_rtcp_buffer_get_packet_count (&rtcp));
1404   fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet));
1405
1406   /* first a Receiver Report */
1407   fail_unless_equals_int (GST_RTCP_TYPE_RR,
1408       gst_rtcp_packet_get_type (&rtcp_packet));
1409   fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet));
1410
1411   /* then a SDES */
1412   fail_unless_equals_int (GST_RTCP_TYPE_SDES,
1413       gst_rtcp_packet_get_type (&rtcp_packet));
1414   fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet));
1415
1416   /* and then our NACK */
1417   fail_unless_equals_int (GST_RTCP_TYPE_RTPFB,
1418       gst_rtcp_packet_get_type (&rtcp_packet));
1419   fail_unless_equals_int (GST_RTCP_RTPFB_TYPE_NACK,
1420       gst_rtcp_packet_fb_get_type (&rtcp_packet));
1421
1422   fail_unless_equals_int (0xDEADBEEF,
1423       gst_rtcp_packet_fb_get_sender_ssrc (&rtcp_packet));
1424   fail_unless_equals_int (0x12345678,
1425       gst_rtcp_packet_fb_get_media_ssrc (&rtcp_packet));
1426
1427   fci_data = gst_rtcp_packet_fb_get_fci (&rtcp_packet);
1428   fci_length =
1429       gst_rtcp_packet_fb_get_fci_length (&rtcp_packet) * sizeof (guint32);
1430   fail_unless_equals_int (4, fci_length);
1431   fail_unless_equals_int (GST_READ_UINT32_BE (fci_data), 1234L << 16);
1432
1433   gst_rtcp_buffer_unmap (&rtcp);
1434   gst_buffer_unref (buf);
1435
1436   session_harness_free (h);
1437 }
1438
1439 GST_END_TEST;
1440
1441 typedef struct
1442 {
1443   gulong id;
1444   GstPad *pad;
1445   GMutex mutex;
1446   GCond cond;
1447   gboolean blocked;
1448 } BlockingProbeData;
1449
1450 static GstPadProbeReturn
1451 on_rtcp_pad_blocked (GstPad * pad, GstPadProbeInfo * info, gpointer user_data)
1452 {
1453   BlockingProbeData *probe = user_data;
1454
1455   g_mutex_lock (&probe->mutex);
1456   probe->blocked = TRUE;
1457   g_cond_signal (&probe->cond);
1458   g_mutex_unlock (&probe->mutex);
1459
1460   return GST_PAD_PROBE_OK;
1461 }
1462
1463 static void
1464 session_harness_block_rtcp (SessionHarness * h, BlockingProbeData * probe)
1465 {
1466   probe->pad = gst_element_get_static_pad (h->session, "send_rtcp_src");
1467   fail_unless (probe->pad);
1468
1469   g_mutex_init (&probe->mutex);
1470   g_cond_init (&probe->cond);
1471   probe->blocked = FALSE;
1472   probe->id = gst_pad_add_probe (probe->pad,
1473       GST_PAD_PROBE_TYPE_BLOCK | GST_PAD_PROBE_TYPE_BUFFER |
1474       GST_PAD_PROBE_TYPE_BUFFER_LIST, on_rtcp_pad_blocked, probe, NULL);
1475
1476   g_mutex_lock (&probe->mutex);
1477   while (!probe->blocked) {
1478     session_harness_crank_clock (h);
1479     g_cond_wait (&probe->cond, &probe->mutex);
1480   }
1481   g_mutex_unlock (&probe->mutex);
1482 }
1483
1484 static void
1485 session_harness_unblock_rtcp (SessionHarness * h, BlockingProbeData * probe)
1486 {
1487   gst_pad_remove_probe (probe->pad, probe->id);
1488   gst_object_unref (probe->pad);
1489   g_mutex_clear (&probe->mutex);
1490 }
1491
1492 GST_START_TEST (test_request_nack_surplus)
1493 {
1494   SessionHarness *h = session_harness_new ();
1495   GstRTCPPacket rtcp_packet;
1496   BlockingProbeData probe;
1497   GstBuffer *buf;
1498   GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT;
1499   guint8 *fci_data;
1500   gint i;
1501   GstStructure *sdes;
1502
1503   g_object_set (h->internal_session, "internal-ssrc", 0xDEADBEEF, NULL);
1504
1505   /* sdes cname has variable size, fix it */
1506   g_object_get (h->internal_session, "sdes", &sdes, NULL);
1507   gst_structure_set (sdes, "cname", G_TYPE_STRING, "user@test", NULL);
1508   g_object_set (h->internal_session, "sdes", sdes, NULL);
1509   gst_structure_free (sdes);
1510
1511   /* Receive a RTP buffer from the wire */
1512   fail_unless_equals_int (GST_FLOW_OK,
1513       session_harness_recv_rtp (h, generate_test_buffer (0, 0x12345678)));
1514
1515   /* Block on first regular RTCP so we can fill the nack list */
1516   session_harness_block_rtcp (h, &probe);
1517
1518   /* request 400 NACK with 17 seqnum distance to optain the worst possible
1519    * packing  */
1520   for (i = 0; i < 350; i++)
1521     session_harness_rtp_retransmission_request (h, 0x12345678, 1234 + i * 17,
1522         0, 0, 0);
1523   /* and the last 50 with a 2s deadline */
1524   for (i = 350; i < 400; i++)
1525     session_harness_rtp_retransmission_request (h, 0x12345678, 1234 + i * 17,
1526         0, 2000, 0);
1527
1528   /* Unblock and wait for the regular and first early packet */
1529   session_harness_unblock_rtcp (h, &probe);
1530   session_harness_produce_rtcp (h, 2);
1531
1532   /* Move time forward, so that only the remaining 50 are still up to date */
1533   session_harness_advance_and_crank (h, GST_SECOND);
1534   session_harness_produce_rtcp (h, 3);
1535
1536   /* ignore the regular RTCP packet */
1537   buf = session_harness_pull_rtcp (h);
1538   gst_buffer_unref (buf);
1539
1540   /* validate the first early RTCP which should hold 335 Nack */
1541   buf = session_harness_pull_rtcp (h);
1542
1543   fail_unless (gst_rtcp_buffer_validate (buf));
1544   gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp);
1545   fail_unless_equals_int (3, gst_rtcp_buffer_get_packet_count (&rtcp));
1546   fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet));
1547
1548   /* first a Receiver Report */
1549   fail_unless_equals_int (GST_RTCP_TYPE_RR,
1550       gst_rtcp_packet_get_type (&rtcp_packet));
1551   fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet));
1552
1553   /* then a SDES */
1554   fail_unless_equals_int (GST_RTCP_TYPE_SDES,
1555       gst_rtcp_packet_get_type (&rtcp_packet));
1556   fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet));
1557
1558   /* and then our NACK */
1559   fail_unless_equals_int (GST_RTCP_TYPE_RTPFB,
1560       gst_rtcp_packet_get_type (&rtcp_packet));
1561   fail_unless_equals_int (GST_RTCP_RTPFB_TYPE_NACK,
1562       gst_rtcp_packet_fb_get_type (&rtcp_packet));
1563
1564   fail_unless_equals_int (0xDEADBEEF,
1565       gst_rtcp_packet_fb_get_sender_ssrc (&rtcp_packet));
1566   fail_unless_equals_int (0x12345678,
1567       gst_rtcp_packet_fb_get_media_ssrc (&rtcp_packet));
1568
1569   fail_unless_equals_int (340,
1570       gst_rtcp_packet_fb_get_fci_length (&rtcp_packet));
1571   fci_data = gst_rtcp_packet_fb_get_fci (&rtcp_packet);
1572   fail_unless_equals_int (GST_READ_UINT32_BE (fci_data), 1234L << 16);
1573
1574   gst_rtcp_buffer_unmap (&rtcp);
1575   gst_buffer_unref (buf);
1576
1577   /* validate the second early RTCP which should hold 50 Nack */
1578   buf = session_harness_pull_rtcp (h);
1579
1580   fail_unless (gst_rtcp_buffer_validate (buf));
1581   gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp);
1582   fail_unless_equals_int (3, gst_rtcp_buffer_get_packet_count (&rtcp));
1583   fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet));
1584
1585   /* first a Receiver Report */
1586   fail_unless_equals_int (GST_RTCP_TYPE_RR,
1587       gst_rtcp_packet_get_type (&rtcp_packet));
1588   fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet));
1589
1590   /* then a SDES */
1591   fail_unless_equals_int (GST_RTCP_TYPE_SDES,
1592       gst_rtcp_packet_get_type (&rtcp_packet));
1593   fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet));
1594
1595   /* and then our NACK */
1596   fail_unless_equals_int (GST_RTCP_TYPE_RTPFB,
1597       gst_rtcp_packet_get_type (&rtcp_packet));
1598   fail_unless_equals_int (GST_RTCP_RTPFB_TYPE_NACK,
1599       gst_rtcp_packet_fb_get_type (&rtcp_packet));
1600
1601   fail_unless_equals_int (0xDEADBEEF,
1602       gst_rtcp_packet_fb_get_sender_ssrc (&rtcp_packet));
1603   fail_unless_equals_int (0x12345678,
1604       gst_rtcp_packet_fb_get_media_ssrc (&rtcp_packet));
1605
1606   fail_unless_equals_int (50, gst_rtcp_packet_fb_get_fci_length (&rtcp_packet));
1607   fci_data = gst_rtcp_packet_fb_get_fci (&rtcp_packet);
1608   fail_unless_equals_int (GST_READ_UINT32_BE (fci_data),
1609       (guint16) (1234 + 350 * 17) << 16);
1610
1611   gst_rtcp_buffer_unmap (&rtcp);
1612   gst_buffer_unref (buf);
1613
1614   session_harness_free (h);
1615 }
1616
1617 GST_END_TEST;
1618
1619 GST_START_TEST (test_request_nack_packing)
1620 {
1621   SessionHarness *h = session_harness_new ();
1622   GstRTCPPacket rtcp_packet;
1623   BlockingProbeData probe;
1624   GstBuffer *buf;
1625   GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT;
1626   guint8 *fci_data;
1627   gint i;
1628
1629   g_object_set (h->internal_session, "internal-ssrc", 0xDEADBEEF, NULL);
1630
1631   /* Receive a RTP buffer from the wire */
1632   fail_unless_equals_int (GST_FLOW_OK,
1633       session_harness_recv_rtp (h, generate_test_buffer (0, 0x12345678)));
1634
1635   /* Block on first regular RTCP so we can fill the nack list */
1636   session_harness_block_rtcp (h, &probe);
1637
1638   /* append 16 consecutive seqnum */
1639   for (i = 1; i < 17; i++)
1640     session_harness_rtp_retransmission_request (h, 0x12345678, 1234 + i,
1641         0, 0, 0);
1642   /* prepend one, still consecutive */
1643   session_harness_rtp_retransmission_request (h, 0x12345678, 1234, 0, 0, 0);
1644   /* update it */
1645   session_harness_rtp_retransmission_request (h, 0x12345678, 1234, 0, 0, 0);
1646
1647   /* Unblock and wait for the regular and first early packet */
1648   session_harness_unblock_rtcp (h, &probe);
1649   session_harness_produce_rtcp (h, 2);
1650
1651   /* ignore the regular RTCP packet */
1652   buf = session_harness_pull_rtcp (h);
1653   gst_buffer_unref (buf);
1654
1655   /* validate the early RTCP which should hold 1 Nack */
1656   buf = session_harness_pull_rtcp (h);
1657
1658   fail_unless (gst_rtcp_buffer_validate (buf));
1659   gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp);
1660   fail_unless_equals_int (3, gst_rtcp_buffer_get_packet_count (&rtcp));
1661   fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet));
1662
1663   /* first a Receiver Report */
1664   fail_unless_equals_int (GST_RTCP_TYPE_RR,
1665       gst_rtcp_packet_get_type (&rtcp_packet));
1666   fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet));
1667
1668   /* then a SDES */
1669   fail_unless_equals_int (GST_RTCP_TYPE_SDES,
1670       gst_rtcp_packet_get_type (&rtcp_packet));
1671   fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet));
1672
1673   /* and then our NACK */
1674   fail_unless_equals_int (GST_RTCP_TYPE_RTPFB,
1675       gst_rtcp_packet_get_type (&rtcp_packet));
1676   fail_unless_equals_int (GST_RTCP_RTPFB_TYPE_NACK,
1677       gst_rtcp_packet_fb_get_type (&rtcp_packet));
1678
1679   fail_unless_equals_int (0xDEADBEEF,
1680       gst_rtcp_packet_fb_get_sender_ssrc (&rtcp_packet));
1681   fail_unless_equals_int (0x12345678,
1682       gst_rtcp_packet_fb_get_media_ssrc (&rtcp_packet));
1683
1684   fail_unless_equals_int (1, gst_rtcp_packet_fb_get_fci_length (&rtcp_packet));
1685   fci_data = gst_rtcp_packet_fb_get_fci (&rtcp_packet);
1686   fail_unless_equals_int (GST_READ_UINT32_BE (fci_data), 1234L << 16 | 0xFFFF);
1687
1688   gst_rtcp_buffer_unmap (&rtcp);
1689   gst_buffer_unref (buf);
1690
1691   session_harness_free (h);
1692 }
1693
1694 GST_END_TEST;
1695
1696 GST_START_TEST (test_disable_sr_timestamp)
1697 {
1698   SessionHarness *h = session_harness_new ();
1699   GstBuffer *buf;
1700   GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT;
1701   GstRTCPPacket rtcp_packet;
1702   guint64 ntptime;
1703   guint32 rtptime;
1704
1705   g_object_set (h->internal_session, "disable-sr-timestamp", TRUE, NULL);
1706
1707   /* Push RTP buffer to make sure RTCP-thread have started */
1708   fail_unless_equals_int (GST_FLOW_OK,
1709       session_harness_send_rtp (h, generate_test_buffer (0, 0xDEADBEEF)));
1710
1711   /* crank the RTCP-thread and pull out rtcp, generating a stats-callback */
1712   session_harness_crank_clock (h);
1713   buf = session_harness_pull_rtcp (h);
1714
1715   gst_rtcp_buffer_map (buf, GST_MAP_READWRITE, &rtcp);
1716
1717   fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet));
1718
1719   fail_unless_equals_int (GST_RTCP_TYPE_SR,
1720       gst_rtcp_packet_get_type (&rtcp_packet));
1721
1722   gst_rtcp_packet_sr_get_sender_info (&rtcp_packet, NULL, &ntptime, &rtptime,
1723       NULL, NULL);
1724
1725   fail_unless_equals_uint64 (ntptime, 0);
1726   fail_unless (rtptime == 0);
1727
1728   gst_rtcp_buffer_unmap (&rtcp);
1729   gst_buffer_unref (buf);
1730
1731   session_harness_free (h);
1732 }
1733
1734 GST_END_TEST;
1735
1736 static guint
1737 on_sending_nacks (GObject * internal_session, guint sender_ssrc,
1738     guint media_ssrc, GArray * nacks, GstBuffer * buffer)
1739 {
1740   GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT;
1741   GstRTCPPacket packet;
1742   guint16 seqnum = g_array_index (nacks, guint16, 0);
1743   guint8 *data;
1744
1745   if (seqnum == 1235)
1746     return 0;
1747
1748   fail_unless (gst_rtcp_buffer_map (buffer, GST_MAP_READWRITE, &rtcp));
1749   fail_unless (gst_rtcp_buffer_add_packet (&rtcp, GST_RTCP_TYPE_APP, &packet));
1750
1751   gst_rtcp_packet_app_set_ssrc (&packet, media_ssrc);
1752   gst_rtcp_packet_app_set_name (&packet, "TEST");
1753
1754   fail_unless (gst_rtcp_packet_app_set_data_length (&packet, 1));
1755   data = gst_rtcp_packet_app_get_data (&packet);
1756   GST_WRITE_UINT32_BE (data, seqnum);
1757
1758   gst_rtcp_buffer_unmap (&rtcp);
1759   return 1;
1760 }
1761
1762 GST_START_TEST (test_on_sending_nacks)
1763 {
1764   SessionHarness *h = session_harness_new ();
1765   BlockingProbeData probe;
1766   GstBuffer *buf;
1767   GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT;
1768   GstRTCPPacket rtcp_packet;
1769   guint8 *data;
1770
1771   g_object_set (h->internal_session, "internal-ssrc", 0xDEADBEEF, NULL);
1772
1773   /* Receive a RTP buffer from the wire */
1774   fail_unless_equals_int (GST_FLOW_OK,
1775       session_harness_recv_rtp (h, generate_test_buffer (0, 0x12345678)));
1776
1777   /* Block on first regular RTCP so we can fill the nack list */
1778   session_harness_block_rtcp (h, &probe);
1779   g_signal_connect (h->internal_session, "on-sending-nacks",
1780       G_CALLBACK (on_sending_nacks), NULL);
1781
1782   /* request NACK immediately */
1783   session_harness_rtp_retransmission_request (h, 0x12345678, 1234, 0, 0, 0);
1784   session_harness_rtp_retransmission_request (h, 0x12345678, 1235, 0, 0, 0);
1785
1786   session_harness_unblock_rtcp (h, &probe);
1787   gst_buffer_unref (session_harness_pull_rtcp (h));
1788   session_harness_produce_rtcp (h, 2);
1789
1790   /* first packet only includes seqnum 1234 in an APP FB */
1791   buf = session_harness_pull_rtcp (h);
1792
1793   fail_unless (gst_rtcp_buffer_validate (buf));
1794   gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp);
1795   fail_unless_equals_int (3, gst_rtcp_buffer_get_packet_count (&rtcp));
1796   fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet));
1797
1798   /* first a Receiver Report */
1799   fail_unless_equals_int (GST_RTCP_TYPE_RR,
1800       gst_rtcp_packet_get_type (&rtcp_packet));
1801   fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet));
1802
1803   /* then a SDES */
1804   fail_unless_equals_int (GST_RTCP_TYPE_SDES,
1805       gst_rtcp_packet_get_type (&rtcp_packet));
1806   fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet));
1807
1808   /* and then our NACK */
1809   fail_unless_equals_int (GST_RTCP_TYPE_APP,
1810       gst_rtcp_packet_get_type (&rtcp_packet));
1811   fail_unless_equals_string ("TEST",
1812       gst_rtcp_packet_app_get_name (&rtcp_packet));
1813
1814   fail_unless_equals_int (0x12345678,
1815       gst_rtcp_packet_app_get_ssrc (&rtcp_packet));
1816
1817   fail_unless_equals_int (1,
1818       gst_rtcp_packet_app_get_data_length (&rtcp_packet));
1819   data = gst_rtcp_packet_app_get_data (&rtcp_packet);
1820   fail_unless_equals_int (GST_READ_UINT32_BE (data), 1234L);
1821
1822   gst_rtcp_buffer_unmap (&rtcp);
1823   gst_buffer_unref (buf);
1824
1825   /* second will contain seqnum 1235 in a generic nack packet */
1826   buf = session_harness_pull_rtcp (h);
1827
1828   fail_unless (gst_rtcp_buffer_validate (buf));
1829   gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp);
1830   fail_unless_equals_int (3, gst_rtcp_buffer_get_packet_count (&rtcp));
1831   fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet));
1832
1833   /* first a Receiver Report */
1834   fail_unless_equals_int (GST_RTCP_TYPE_RR,
1835       gst_rtcp_packet_get_type (&rtcp_packet));
1836   fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet));
1837
1838   /* then a SDES */
1839   fail_unless_equals_int (GST_RTCP_TYPE_SDES,
1840       gst_rtcp_packet_get_type (&rtcp_packet));
1841   fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet));
1842
1843   /* and then our NACK */
1844   fail_unless_equals_int (GST_RTCP_TYPE_RTPFB,
1845       gst_rtcp_packet_get_type (&rtcp_packet));
1846   fail_unless_equals_int (GST_RTCP_RTPFB_TYPE_NACK,
1847       gst_rtcp_packet_fb_get_type (&rtcp_packet));
1848
1849   fail_unless_equals_int (0xDEADBEEF,
1850       gst_rtcp_packet_fb_get_sender_ssrc (&rtcp_packet));
1851   fail_unless_equals_int (0x12345678,
1852       gst_rtcp_packet_fb_get_media_ssrc (&rtcp_packet));
1853
1854   fail_unless_equals_int (1, gst_rtcp_packet_fb_get_fci_length (&rtcp_packet));
1855   data = gst_rtcp_packet_fb_get_fci (&rtcp_packet);
1856   fail_unless_equals_int (GST_READ_UINT32_BE (data), 1235L << 16);
1857
1858   gst_rtcp_buffer_unmap (&rtcp);
1859   gst_buffer_unref (buf);
1860
1861   session_harness_free (h);
1862 }
1863
1864 GST_END_TEST;
1865
1866 static void
1867 disable_probation_on_new_ssrc (GObject * session, GObject * source)
1868 {
1869   g_object_set (source, "probation", 0, NULL);
1870 }
1871
1872 GST_START_TEST (test_disable_probation)
1873 {
1874   SessionHarness *h = session_harness_new ();
1875
1876   g_object_set (h->internal_session, "internal-ssrc", 0xDEADBEEF, NULL);
1877   g_signal_connect (h->internal_session, "on-new-ssrc",
1878       G_CALLBACK (disable_probation_on_new_ssrc), NULL);
1879
1880   /* Receive a RTP buffer from the wire */
1881   fail_unless_equals_int (GST_FLOW_OK,
1882       session_harness_recv_rtp (h, generate_test_buffer (0, 0x12345678)));
1883
1884   /* When probation is disable, the packet should be produced immediatly */
1885   fail_unless_equals_int (1, gst_harness_buffers_in_queue (h->recv_rtp_h));
1886
1887   session_harness_free (h);
1888 }
1889
1890 GST_END_TEST;
1891
1892 GST_START_TEST (test_request_late_nack)
1893 {
1894   SessionHarness *h = session_harness_new ();
1895   GstBuffer *buf;
1896   GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT;
1897   GstRTCPPacket rtcp_packet;
1898   guint8 *fci_data;
1899   guint32 fci_length;
1900
1901   g_object_set (h->internal_session, "internal-ssrc", 0xDEADBEEF, NULL);
1902
1903   /* Receive a RTP buffer from the wire */
1904   fail_unless_equals_int (GST_FLOW_OK,
1905       session_harness_recv_rtp (h, generate_test_buffer (0, 0x12345678)));
1906
1907   /* Wait for first regular RTCP to be sent so that we are clear to send early RTCP */
1908   session_harness_produce_rtcp (h, 1);
1909   gst_buffer_unref (session_harness_pull_rtcp (h));
1910
1911   /* request NACK immediately, but also advance the clock, so the request is
1912    * now late, but it should be kept to avoid sendign an early rtcp without
1913    * NACK. This would otherwise lead to a stall if the late packet was cause
1914    * by high RTT, we need to send some RTX in order to update that statistic. */
1915   session_harness_rtp_retransmission_request (h, 0x12345678, 1234, 0, 0, 0);
1916   gst_test_clock_advance_time (h->testclock, 100 * GST_USECOND);
1917
1918   /* NACK should be produced immediately as early RTCP is allowed. Pull buffer
1919      without advancing the clock to ensure this is the case */
1920   buf = session_harness_pull_rtcp (h);
1921
1922   fail_unless (gst_rtcp_buffer_validate (buf));
1923   gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp);
1924   fail_unless_equals_int (3, gst_rtcp_buffer_get_packet_count (&rtcp));
1925   fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet));
1926
1927   /* first a Receiver Report */
1928   fail_unless_equals_int (GST_RTCP_TYPE_RR,
1929       gst_rtcp_packet_get_type (&rtcp_packet));
1930   fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet));
1931
1932   /* then a SDES */
1933   fail_unless_equals_int (GST_RTCP_TYPE_SDES,
1934       gst_rtcp_packet_get_type (&rtcp_packet));
1935   fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet));
1936
1937   /* and then our NACK */
1938   fail_unless_equals_int (GST_RTCP_TYPE_RTPFB,
1939       gst_rtcp_packet_get_type (&rtcp_packet));
1940   fail_unless_equals_int (GST_RTCP_RTPFB_TYPE_NACK,
1941       gst_rtcp_packet_fb_get_type (&rtcp_packet));
1942
1943   fail_unless_equals_int (0xDEADBEEF,
1944       gst_rtcp_packet_fb_get_sender_ssrc (&rtcp_packet));
1945   fail_unless_equals_int (0x12345678,
1946       gst_rtcp_packet_fb_get_media_ssrc (&rtcp_packet));
1947
1948   fci_data = gst_rtcp_packet_fb_get_fci (&rtcp_packet);
1949   fci_length =
1950       gst_rtcp_packet_fb_get_fci_length (&rtcp_packet) * sizeof (guint32);
1951   fail_unless_equals_int (4, fci_length);
1952   fail_unless_equals_int (GST_READ_UINT32_BE (fci_data), 1234L << 16);
1953
1954   gst_rtcp_buffer_unmap (&rtcp);
1955   gst_buffer_unref (buf);
1956
1957   session_harness_free (h);
1958 }
1959
1960 GST_END_TEST;
1961
1962 static gpointer
1963 _push_caps_events (gpointer user_data)
1964 {
1965   SessionHarness *h = user_data;
1966   gint payload = 0;
1967   while (h->running) {
1968
1969     GstCaps *caps = gst_caps_new_simple ("application/x-rtp",
1970         "payload", G_TYPE_INT, payload,
1971         NULL);
1972     gst_harness_set_src_caps (h->recv_rtp_h, caps);
1973     g_thread_yield ();
1974     payload++;
1975   }
1976
1977   return NULL;
1978 }
1979
1980 GST_START_TEST (test_clear_pt_map_stress)
1981 {
1982   SessionHarness *h = session_harness_new ();
1983   GThread *thread;
1984   guint i;
1985
1986   h->running = TRUE;
1987   thread = g_thread_new (NULL, _push_caps_events, h);
1988
1989   for (i = 0; i < 1000; i++) {
1990     g_signal_emit_by_name (h->session, "clear-pt-map");
1991     g_thread_yield ();
1992   }
1993
1994   h->running = FALSE;
1995   g_thread_join (thread);
1996
1997   session_harness_free (h);
1998 }
1999
2000 GST_END_TEST;
2001
2002 static Suite *
2003 rtpsession_suite (void)
2004 {
2005   Suite *s = suite_create ("rtpsession");
2006   TCase *tc_chain = tcase_create ("general");
2007
2008   suite_add_tcase (s, tc_chain);
2009   tcase_add_test (tc_chain, test_multiple_ssrc_rr);
2010   tcase_add_test (tc_chain, test_multiple_senders_roundrobin_rbs);
2011   tcase_add_test (tc_chain, test_no_rbs_for_internal_senders);
2012   tcase_add_test (tc_chain, test_internal_sources_timeout);
2013   tcase_add_test (tc_chain, test_receive_rtcp_app_packet);
2014   tcase_add_test (tc_chain, test_dont_lock_on_stats);
2015   tcase_add_test (tc_chain, test_ignore_suspicious_bye);
2016   tcase_add_test (tc_chain, test_ssrc_collision_when_sending);
2017   tcase_add_test (tc_chain, test_request_fir);
2018   tcase_add_test (tc_chain, test_request_pli);
2019   tcase_add_test (tc_chain, test_request_nack);
2020   tcase_add_test (tc_chain, test_request_nack_surplus);
2021   tcase_add_test (tc_chain, test_request_nack_packing);
2022   tcase_add_test (tc_chain, test_illegal_rtcp_fb_packet);
2023   tcase_add_test (tc_chain, test_feedback_rtcp_race);
2024   tcase_add_test (tc_chain, test_receive_regular_pli);
2025   tcase_add_test (tc_chain, test_receive_pli_no_sender_ssrc);
2026   tcase_add_test (tc_chain, test_dont_send_rtcp_while_idle);
2027   tcase_add_test (tc_chain, test_send_rtcp_when_signalled);
2028   tcase_add_test (tc_chain, test_change_sent_sdes);
2029   tcase_add_test (tc_chain, test_disable_sr_timestamp);
2030   tcase_add_test (tc_chain, test_on_sending_nacks);
2031   tcase_add_test (tc_chain, test_disable_probation);
2032   tcase_add_test (tc_chain, test_request_late_nack);
2033   tcase_add_test (tc_chain, test_clear_pt_map_stress);
2034
2035   return s;
2036 }
2037
2038 GST_CHECK_MAIN (rtpsession);