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 *
generate_caps(void)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 *
generate_test_buffer_full(GstClockTime dts,guint seq_num,guint32 rtp_ts,guint ssrc)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 *
generate_test_buffer(guint seq_num,guint ssrc)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 *
_pt_map_requested(GstElement * element,guint pt,gpointer data)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 *
session_harness_new(void)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
session_harness_free(SessionHarness * h)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
session_harness_send_rtp(SessionHarness * h,GstBuffer * buf)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
session_harness_recv_rtp(SessionHarness * h,GstBuffer * buf)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
session_harness_recv_rtcp(SessionHarness * h,GstBuffer * buf)168 session_harness_recv_rtcp (SessionHarness * h, GstBuffer * buf)
169 {
170 return gst_harness_push (h->rtcp_h, buf);
171 }
172
173 static GstBuffer *
session_harness_pull_rtcp(SessionHarness * h)174 session_harness_pull_rtcp (SessionHarness * h)
175 {
176 return gst_harness_pull (h->rtcp_h);
177 }
178
179 static void
session_harness_crank_clock(SessionHarness * h)180 session_harness_crank_clock (SessionHarness * h)
181 {
182 gst_test_clock_crank (h->testclock);
183 }
184
185 static gboolean
session_harness_advance_and_crank(SessionHarness * h,GstClockTime delta)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
session_harness_produce_rtcp(SessionHarness * h,gint num_rtcp_packets)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
session_harness_force_key_unit(SessionHarness * h,guint count,guint ssrc,guint payload,gint * reqid,guint64 * sfr)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
session_harness_rtp_retransmission_request(SessionHarness * h,guint ssrc,guint seqnum,guint delay,guint deadline,guint avg_rtt)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
GST_START_TEST(test_multiple_ssrc_rr)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 */
GST_START_TEST(test_multiple_senders_roundrobin_rbs)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
GST_START_TEST(test_no_rbs_for_internal_senders)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
GST_START_TEST(test_internal_sources_timeout)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
on_app_rtcp_cb(GObject * session,guint subtype,guint ssrc,const gchar * name,GstBuffer * data,RTCPAppResult * result)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
GST_START_TEST(test_receive_rtcp_app_packet)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
stats_test_cb(GObject * object,GParamSpec * spec,gpointer data)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
GST_START_TEST(test_dont_lock_on_stats)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
suspicious_bye_cb(GObject * object,GParamSpec * spec,gpointer data)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 *
create_bye_rtcp(guint32 ssrc)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
GST_START_TEST(test_ignore_suspicious_bye)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 *
create_buffer(guint8 * data,gsize size)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
GST_START_TEST(test_receive_regular_pli)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
GST_START_TEST(test_receive_pli_no_sender_ssrc)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
add_rtcp_sdes_packet(GstBuffer * gstbuf,guint32 ssrc,const char * cname)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
GST_START_TEST(test_ssrc_collision_when_sending)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
GST_START_TEST(test_request_fir)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
GST_START_TEST(test_request_pli)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
GST_START_TEST(test_illegal_rtcp_fb_packet)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
feedback_rtcp_cb(GstElement * element,guint fbtype,guint fmt,guint sender_ssrc,guint media_ssrc,GstBuffer * fci,FeedbackRTCPCallbackData * cb_data)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 *
send_feedback_rtcp(SessionHarness * h)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
GST_START_TEST(test_feedback_rtcp_race)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
GST_START_TEST(test_dont_send_rtcp_while_idle)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
GST_START_TEST(test_send_rtcp_when_signalled)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
validate_sdes_priv(GstBuffer * buf,const char * name_ref,const char * value)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
GST_START_TEST(test_change_sent_sdes)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
GST_START_TEST(test_request_nack)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
on_rtcp_pad_blocked(GstPad * pad,GstPadProbeInfo * info,gpointer user_data)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
session_harness_block_rtcp(SessionHarness * h,BlockingProbeData * probe)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
session_harness_unblock_rtcp(SessionHarness * h,BlockingProbeData * probe)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
GST_START_TEST(test_request_nack_surplus)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
GST_START_TEST(test_request_nack_packing)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
GST_START_TEST(test_disable_sr_timestamp)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
on_sending_nacks(GObject * internal_session,guint sender_ssrc,guint media_ssrc,GArray * nacks,GstBuffer * buffer)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
GST_START_TEST(test_on_sending_nacks)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
disable_probation_on_new_ssrc(GObject * session,GObject * source)1867 disable_probation_on_new_ssrc (GObject * session, GObject * source)
1868 {
1869 g_object_set (source, "probation", 0, NULL);
1870 }
1871
GST_START_TEST(test_disable_probation)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
GST_START_TEST(test_request_late_nack)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
_push_caps_events(gpointer user_data)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
GST_START_TEST(test_clear_pt_map_stress)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 *
rtpsession_suite(void)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);
2039