/* GStreamer * * unit test for gstrtpsession * * Copyright (C) <2009> Wim Taymans * Copyright (C) 2013 Collabora Ltd. * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Library General Public * License as published by the Free Software Foundation; either * version 2 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Library General Public License for more details. * * You should have received a copy of the GNU Library General Public * License along with this library; if not, write to the * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor, * Boston, MA 02110-1301, USA. */ #define GLIB_DISABLE_DEPRECATION_WARNINGS #include #include #include #include #include #include #include #include #define TEST_BUF_CLOCK_RATE 8000 #define TEST_BUF_PT 0 #define TEST_BUF_SSRC 0x01BADBAD #define TEST_BUF_MS 20 #define TEST_BUF_DURATION (TEST_BUF_MS * GST_MSECOND) #define TEST_BUF_BPS 512000 #define TEST_BUF_SIZE (TEST_BUF_BPS * TEST_BUF_MS / (1000 * 8)) #define TEST_RTP_TS_DURATION (TEST_BUF_CLOCK_RATE * TEST_BUF_MS / 1000) #define TEST_TWCC_EXT_ID 5 #define TWCC_EXTMAP_STR "http://www.ietf.org/id/draft-holmer-rmcat-transport-wide-cc-extensions-01" static GstCaps * generate_caps (void) { return gst_caps_new_simple ("application/x-rtp", "clock-rate", G_TYPE_INT, TEST_BUF_CLOCK_RATE, "payload", G_TYPE_INT, TEST_BUF_PT, NULL); } static GstBuffer * generate_test_buffer_full (GstClockTime ts, guint seqnum, guint32 rtp_ts, guint ssrc, gboolean marker_bit, guint8 twcc_ext_id, guint16 twcc_seqnum) { GstBuffer *buf; guint8 *payload; guint i; GstRTPBuffer rtp = GST_RTP_BUFFER_INIT; buf = gst_rtp_buffer_new_allocate (TEST_BUF_SIZE, 0, 0); GST_BUFFER_PTS (buf) = ts; GST_BUFFER_DTS (buf) = ts; gst_rtp_buffer_map (buf, GST_MAP_READWRITE, &rtp); gst_rtp_buffer_set_payload_type (&rtp, TEST_BUF_PT); gst_rtp_buffer_set_seq (&rtp, seqnum); gst_rtp_buffer_set_timestamp (&rtp, rtp_ts); gst_rtp_buffer_set_ssrc (&rtp, ssrc); gst_rtp_buffer_set_marker (&rtp, marker_bit); payload = gst_rtp_buffer_get_payload (&rtp); for (i = 0; i < TEST_BUF_SIZE; i++) payload[i] = 0xff; if (twcc_ext_id > 0) { guint8 twcc_seqnum_be[2]; GST_WRITE_UINT16_BE (twcc_seqnum_be, twcc_seqnum); gst_rtp_buffer_add_extension_onebyte_header (&rtp, twcc_ext_id, twcc_seqnum_be, sizeof (twcc_seqnum_be)); } gst_rtp_buffer_unmap (&rtp); return buf; } static GstBuffer * generate_test_buffer (guint seqnum, guint ssrc) { return generate_test_buffer_full (seqnum * TEST_BUF_DURATION, seqnum, seqnum * TEST_RTP_TS_DURATION, ssrc, FALSE, 0, 0); } static GstBuffer * generate_twcc_recv_buffer (guint seqnum, GstClockTime arrival_time, gboolean marker_bit) { return generate_test_buffer_full (arrival_time, seqnum, seqnum * TEST_RTP_TS_DURATION, TEST_BUF_SSRC, marker_bit, TEST_TWCC_EXT_ID, seqnum); } static GstBuffer * generate_twcc_send_buffer (guint seqnum, gboolean marker_bit) { return generate_test_buffer_full (seqnum * TEST_BUF_DURATION, seqnum, seqnum * TEST_RTP_TS_DURATION, TEST_BUF_SSRC, marker_bit, TEST_TWCC_EXT_ID, seqnum); } typedef struct { GstHarness *send_rtp_h; GstHarness *recv_rtp_h; GstHarness *rtcp_h; GstElement *session; GObject *internal_session; GstTestClock *testclock; GstCaps *caps; gboolean running; GMutex lock; GstStructure *last_twcc_stats; } SessionHarness; static GstCaps * _pt_map_requested (GstElement * element, guint pt, gpointer data) { SessionHarness *h = data; return gst_caps_copy (h->caps); } static void _notify_twcc_stats (GParamSpec * spec G_GNUC_UNUSED, GObject * object G_GNUC_UNUSED, gpointer data) { SessionHarness *h = data; GstStructure *stats; g_object_get (h->session, "twcc-stats", &stats, NULL); g_mutex_lock (&h->lock); if (h->last_twcc_stats) gst_structure_free (h->last_twcc_stats); h->last_twcc_stats = stats; g_mutex_unlock (&h->lock); } static GstStructure * session_harness_get_last_twcc_stats (SessionHarness * h) { GstStructure *ret = NULL; g_mutex_lock (&h->lock); if (h->last_twcc_stats) ret = gst_structure_copy (h->last_twcc_stats); g_mutex_unlock (&h->lock); return ret; } static SessionHarness * session_harness_new (void) { SessionHarness *h = g_new0 (SessionHarness, 1); h->caps = generate_caps (); g_mutex_init (&h->lock); h->testclock = GST_TEST_CLOCK_CAST (gst_test_clock_new ()); gst_system_clock_set_default (GST_CLOCK_CAST (h->testclock)); h->session = gst_element_factory_make ("rtpsession", NULL); gst_element_set_clock (h->session, GST_CLOCK_CAST (h->testclock)); h->send_rtp_h = gst_harness_new_with_element (h->session, "send_rtp_sink", "send_rtp_src"); gst_harness_set_src_caps (h->send_rtp_h, gst_caps_copy (h->caps)); h->recv_rtp_h = gst_harness_new_with_element (h->session, "recv_rtp_sink", "recv_rtp_src"); gst_harness_set_src_caps (h->recv_rtp_h, gst_caps_copy (h->caps)); h->rtcp_h = gst_harness_new_with_element (h->session, "recv_rtcp_sink", "send_rtcp_src"); gst_harness_set_src_caps_str (h->rtcp_h, "application/x-rtcp"); g_signal_connect (h->session, "request-pt-map", (GCallback) _pt_map_requested, h); g_signal_connect (h->session, "notify::twcc-stats", (GCallback) _notify_twcc_stats, h); g_object_get (h->session, "internal-session", &h->internal_session, NULL); return h; } static void session_harness_free (SessionHarness * h) { gst_system_clock_set_default (NULL); gst_caps_unref (h->caps); gst_object_unref (h->testclock); gst_harness_teardown (h->rtcp_h); gst_harness_teardown (h->recv_rtp_h); gst_harness_teardown (h->send_rtp_h); g_mutex_clear (&h->lock); if (h->last_twcc_stats) gst_structure_free (h->last_twcc_stats); g_object_unref (h->internal_session); gst_object_unref (h->session); g_free (h); } static GstFlowReturn session_harness_send_rtp (SessionHarness * h, GstBuffer * buf) { return gst_harness_push (h->send_rtp_h, buf); } static GstBuffer * session_harness_pull_send_rtp (SessionHarness * h) { return gst_harness_pull (h->send_rtp_h); } static GstFlowReturn session_harness_recv_rtp (SessionHarness * h, GstBuffer * buf) { return gst_harness_push (h->recv_rtp_h, buf); } static GstFlowReturn session_harness_recv_rtcp (SessionHarness * h, GstBuffer * buf) { return gst_harness_push (h->rtcp_h, buf); } static GstBuffer * session_harness_pull_rtcp (SessionHarness * h) { return gst_harness_pull (h->rtcp_h); } static void session_harness_crank_clock (SessionHarness * h) { gst_test_clock_crank (h->testclock); } static gboolean session_harness_advance_and_crank (SessionHarness * h, GstClockTime delta) { GstClockID res, pending; gboolean result; gst_test_clock_wait_for_next_pending_id (h->testclock, &pending); gst_test_clock_advance_time (h->testclock, delta); res = gst_test_clock_process_next_clock_id (h->testclock); if (res == pending) result = TRUE; else result = FALSE; if (res) gst_clock_id_unref (res); gst_clock_id_unref (pending); return result; } static void session_harness_produce_rtcp (SessionHarness * h, gint num_rtcp_packets) { /* due to randomness in rescheduling of RTCP timeout, we need to keep cranking until we have the desired amount of packets */ while (gst_harness_buffers_in_queue (h->rtcp_h) < num_rtcp_packets) { session_harness_crank_clock (h); /* allow the rtcp-thread to settle before checking the queue */ gst_test_clock_wait_for_next_pending_id (h->testclock, NULL); } } static void session_harness_force_key_unit (SessionHarness * h, guint count, guint ssrc, guint payload, gint * reqid, guint64 * sfr) { GstClockTime running_time = GST_CLOCK_TIME_NONE; gboolean all_headers = TRUE; GstStructure *s = gst_structure_new ("GstForceKeyUnit", "running-time", GST_TYPE_CLOCK_TIME, running_time, "all-headers", G_TYPE_BOOLEAN, all_headers, "count", G_TYPE_UINT, count, "ssrc", G_TYPE_UINT, ssrc, "payload", G_TYPE_UINT, payload, NULL); if (reqid) gst_structure_set (s, "reqid", G_TYPE_INT, *reqid, NULL); if (sfr) gst_structure_set (s, "sfr", G_TYPE_UINT64, *sfr, NULL); gst_harness_push_upstream_event (h->recv_rtp_h, gst_event_new_custom (GST_EVENT_CUSTOM_UPSTREAM, s)); } static void session_harness_rtp_retransmission_request (SessionHarness * h, guint ssrc, guint seqnum, guint delay, guint deadline, guint avg_rtt) { GstClockTime running_time = GST_CLOCK_TIME_NONE; GstStructure *s = gst_structure_new ("GstRTPRetransmissionRequest", "running-time", GST_TYPE_CLOCK_TIME, running_time, "ssrc", G_TYPE_UINT, ssrc, "seqnum", G_TYPE_UINT, seqnum, "delay", G_TYPE_UINT, delay, "deadline", G_TYPE_UINT, deadline, "avg-rtt", G_TYPE_UINT, avg_rtt, NULL); gst_harness_push_upstream_event (h->recv_rtp_h, gst_event_new_custom (GST_EVENT_CUSTOM_UPSTREAM, s)); } static void _add_twcc_field_to_caps (GstCaps * caps, guint8 ext_id) { gchar *name = g_strdup_printf ("extmap-%u", ext_id); gst_caps_set_simple (caps, name, G_TYPE_STRING, TWCC_EXTMAP_STR, NULL); g_free (name); } static void session_harness_set_twcc_recv_ext_id (SessionHarness * h, guint8 ext_id) { _add_twcc_field_to_caps (h->caps, ext_id); g_signal_emit_by_name (h->session, "clear-pt-map"); } static void session_harness_set_twcc_send_ext_id (SessionHarness * h, guint8 ext_id) { GstCaps *caps = gst_caps_copy (h->caps); _add_twcc_field_to_caps (caps, ext_id); gst_harness_set_src_caps (h->send_rtp_h, caps); } GST_START_TEST (test_multiple_ssrc_rr) { SessionHarness *h = session_harness_new (); GstFlowReturn res; GstBuffer *in_buf, *out_buf; GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT; GstRTCPPacket rtcp_packet; gint i, j; guint ssrc_match; guint ssrcs[] = { 0x01BADBAD, 0xDEADBEEF, }; /* receive buffers with multiple ssrcs */ for (i = 0; i < 2; i++) { for (j = 0; j < G_N_ELEMENTS (ssrcs); j++) { in_buf = generate_test_buffer (i, ssrcs[j]); res = session_harness_recv_rtp (h, in_buf); fail_unless_equals_int (GST_FLOW_OK, res); } } /* crank the rtcp-thread and pull out the rtcp-packet we have generated */ session_harness_crank_clock (h); out_buf = session_harness_pull_rtcp (h); /* verify we have report blocks for both ssrcs */ g_assert (out_buf != NULL); fail_unless (gst_rtcp_buffer_validate (out_buf)); gst_rtcp_buffer_map (out_buf, GST_MAP_READ, &rtcp); g_assert (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet)); fail_unless_equals_int (GST_RTCP_TYPE_RR, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless_equals_int (G_N_ELEMENTS (ssrcs), gst_rtcp_packet_get_rb_count (&rtcp_packet)); ssrc_match = 0; for (i = 0; i < G_N_ELEMENTS (ssrcs); i++) { guint32 ssrc; gst_rtcp_packet_get_rb (&rtcp_packet, i, &ssrc, NULL, NULL, NULL, NULL, NULL, NULL); for (j = 0; j < G_N_ELEMENTS (ssrcs); j++) { if (ssrcs[j] == ssrc) ssrc_match++; } } fail_unless_equals_int (G_N_ELEMENTS (ssrcs), ssrc_match); gst_rtcp_buffer_unmap (&rtcp); gst_buffer_unref (out_buf); session_harness_free (h); } GST_END_TEST; /* This verifies that rtpsession will correctly place RBs round-robin * across multiple RRs when there are too many senders that their RBs * do not fit in one RR */ GST_START_TEST (test_multiple_senders_roundrobin_rbs) { SessionHarness *h = session_harness_new (); GstFlowReturn res; GstBuffer *buf; GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT; GstRTCPPacket rtcp_packet; gint i, j, k; guint32 ssrc; GHashTable *rb_ssrcs, *tmp_set; g_object_set (h->internal_session, "internal-ssrc", 0xDEADBEEF, NULL); for (i = 0; i < 2; i++) { /* cycles between RR reports */ for (j = 0; j < 5; j++) { /* packets per ssrc */ gint seq = (i * 5) + j; for (k = 0; k < 35; k++) { /* number of ssrcs */ buf = generate_test_buffer (seq, 10000 + k); res = session_harness_recv_rtp (h, buf); fail_unless_equals_int (GST_FLOW_OK, res); } } } rb_ssrcs = g_hash_table_new_full (g_direct_hash, g_direct_equal, NULL, (GDestroyNotify) g_hash_table_unref); /* verify the rtcp packets */ for (i = 0; i < 2; i++) { guint expected_rb_count = (i < 1) ? GST_RTCP_MAX_RB_COUNT : (35 - GST_RTCP_MAX_RB_COUNT); session_harness_produce_rtcp (h, 1); buf = session_harness_pull_rtcp (h); g_assert (buf != NULL); fail_unless (gst_rtcp_buffer_validate (buf)); gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp); fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet)); fail_unless_equals_int (GST_RTCP_TYPE_RR, gst_rtcp_packet_get_type (&rtcp_packet)); ssrc = gst_rtcp_packet_rr_get_ssrc (&rtcp_packet); fail_unless_equals_int (0xDEADBEEF, ssrc); /* inspect the RBs */ fail_unless_equals_int (expected_rb_count, gst_rtcp_packet_get_rb_count (&rtcp_packet)); if (i == 0) { tmp_set = g_hash_table_new (g_direct_hash, g_direct_equal); g_hash_table_insert (rb_ssrcs, GUINT_TO_POINTER (ssrc), tmp_set); } else { tmp_set = g_hash_table_lookup (rb_ssrcs, GUINT_TO_POINTER (ssrc)); g_assert (tmp_set); } for (j = 0; j < expected_rb_count; j++) { gst_rtcp_packet_get_rb (&rtcp_packet, j, &ssrc, NULL, NULL, NULL, NULL, NULL, NULL); g_assert_cmpint (ssrc, >=, 10000); g_assert_cmpint (ssrc, <=, 10035); g_hash_table_add (tmp_set, GUINT_TO_POINTER (ssrc)); } gst_rtcp_buffer_unmap (&rtcp); gst_buffer_unref (buf); } /* now verify all received ssrcs have been reported */ fail_unless_equals_int (1, g_hash_table_size (rb_ssrcs)); tmp_set = g_hash_table_lookup (rb_ssrcs, GUINT_TO_POINTER (0xDEADBEEF)); g_assert (tmp_set); fail_unless_equals_int (35, g_hash_table_size (tmp_set)); g_hash_table_unref (rb_ssrcs); session_harness_free (h); } GST_END_TEST; GST_START_TEST (test_no_rbs_for_internal_senders) { SessionHarness *h = session_harness_new (); GstFlowReturn res; GstBuffer *buf; GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT; GstRTCPPacket rtcp_packet; gint i, j, k; guint32 ssrc; GHashTable *sr_ssrcs; GHashTable *rb_ssrcs, *tmp_set; /* Push RTP from our send SSRCs */ for (j = 0; j < 5; j++) { /* packets per ssrc */ for (k = 0; k < 2; k++) { /* number of ssrcs */ buf = generate_test_buffer (j, 10000 + k); res = session_harness_send_rtp (h, buf); fail_unless_equals_int (GST_FLOW_OK, res); } } /* crank the RTCP pad thread */ session_harness_crank_clock (h); sr_ssrcs = g_hash_table_new (g_direct_hash, g_direct_equal); /* verify the rtcp packets */ for (i = 0; i < 2; i++) { buf = session_harness_pull_rtcp (h); g_assert (buf != NULL); g_assert (gst_rtcp_buffer_validate (buf)); gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp); g_assert (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet)); fail_unless_equals_int (GST_RTCP_TYPE_SR, gst_rtcp_packet_get_type (&rtcp_packet)); gst_rtcp_packet_sr_get_sender_info (&rtcp_packet, &ssrc, NULL, NULL, NULL, NULL); g_assert_cmpint (ssrc, >=, 10000); g_assert_cmpint (ssrc, <=, 10001); g_hash_table_add (sr_ssrcs, GUINT_TO_POINTER (ssrc)); /* There should be no RBs as there are no remote senders */ fail_unless_equals_int (0, gst_rtcp_packet_get_rb_count (&rtcp_packet)); gst_rtcp_buffer_unmap (&rtcp); gst_buffer_unref (buf); } /* Ensure both internal senders generated RTCP */ fail_unless_equals_int (2, g_hash_table_size (sr_ssrcs)); g_hash_table_unref (sr_ssrcs); /* Generate RTP from remote side */ for (j = 0; j < 5; j++) { /* packets per ssrc */ for (k = 0; k < 2; k++) { /* number of ssrcs */ buf = generate_test_buffer (j, 20000 + k); res = session_harness_recv_rtp (h, buf); fail_unless_equals_int (GST_FLOW_OK, res); } } sr_ssrcs = g_hash_table_new (g_direct_hash, g_direct_equal); rb_ssrcs = g_hash_table_new_full (g_direct_hash, g_direct_equal, NULL, (GDestroyNotify) g_hash_table_unref); /* verify the rtcp packets */ for (i = 0; i < 2; i++) { session_harness_produce_rtcp (h, 1); buf = session_harness_pull_rtcp (h); g_assert (buf != NULL); g_assert (gst_rtcp_buffer_validate (buf)); gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp); g_assert (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet)); fail_unless_equals_int (GST_RTCP_TYPE_SR, gst_rtcp_packet_get_type (&rtcp_packet)); gst_rtcp_packet_sr_get_sender_info (&rtcp_packet, &ssrc, NULL, NULL, NULL, NULL); g_assert_cmpint (ssrc, >=, 10000); g_assert_cmpint (ssrc, <=, 10001); g_hash_table_add (sr_ssrcs, GUINT_TO_POINTER (ssrc)); /* There should be 2 RBs: one for each remote sender */ fail_unless_equals_int (2, gst_rtcp_packet_get_rb_count (&rtcp_packet)); tmp_set = g_hash_table_new (g_direct_hash, g_direct_equal); g_hash_table_insert (rb_ssrcs, GUINT_TO_POINTER (ssrc), tmp_set); for (j = 0; j < 2; j++) { gst_rtcp_packet_get_rb (&rtcp_packet, j, &ssrc, NULL, NULL, NULL, NULL, NULL, NULL); g_assert_cmpint (ssrc, >=, 20000); g_assert_cmpint (ssrc, <=, 20001); g_hash_table_add (tmp_set, GUINT_TO_POINTER (ssrc)); } gst_rtcp_buffer_unmap (&rtcp); gst_buffer_unref (buf); } /* now verify all received ssrcs have been reported */ fail_unless_equals_int (2, g_hash_table_size (sr_ssrcs)); fail_unless_equals_int (2, g_hash_table_size (rb_ssrcs)); for (i = 10000; i < 10002; i++) { tmp_set = g_hash_table_lookup (rb_ssrcs, GUINT_TO_POINTER (i)); g_assert (tmp_set); fail_unless_equals_int (2, g_hash_table_size (tmp_set)); } g_hash_table_unref (rb_ssrcs); g_hash_table_unref (sr_ssrcs); session_harness_free (h); } GST_END_TEST; GST_START_TEST (test_internal_sources_timeout) { SessionHarness *h = session_harness_new (); guint internal_ssrc; guint32 ssrc; GstBuffer *buf; GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT; GstRTCPPacket rtcp_packet; GstRTCPType rtcp_type; GstFlowReturn res; gint i, j; GstCaps *caps; gboolean seen_bye; g_object_set (h->internal_session, "internal-ssrc", 0xDEADBEEF, NULL); g_object_get (h->internal_session, "internal-ssrc", &internal_ssrc, NULL); fail_unless_equals_int (0xDEADBEEF, internal_ssrc); for (i = 1; i < 4; i++) { buf = generate_test_buffer (i, 0xBEEFDEAD); res = session_harness_recv_rtp (h, buf); fail_unless_equals_int (GST_FLOW_OK, res); } /* verify that rtpsession has sent RR for an internally-created * RTPSource that is using the internal-ssrc */ session_harness_produce_rtcp (h, 1); buf = session_harness_pull_rtcp (h); fail_unless (buf != NULL); fail_unless (gst_rtcp_buffer_validate (buf)); gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp); fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet)); fail_unless_equals_int (GST_RTCP_TYPE_RR, gst_rtcp_packet_get_type (&rtcp_packet)); ssrc = gst_rtcp_packet_rr_get_ssrc (&rtcp_packet); fail_unless_equals_int (ssrc, internal_ssrc); gst_rtcp_buffer_unmap (&rtcp); gst_buffer_unref (buf); /* ok, now let's push some RTP packets */ caps = gst_caps_new_simple ("application/x-rtp", "ssrc", G_TYPE_UINT, 0x01BADBAD, NULL); gst_harness_set_src_caps (h->send_rtp_h, caps); for (i = 1; i < 4; i++) { buf = generate_test_buffer (i, 0x01BADBAD); res = session_harness_send_rtp (h, buf); fail_unless_equals_int (GST_FLOW_OK, res); } /* internal ssrc must have changed already */ g_object_get (h->internal_session, "internal-ssrc", &internal_ssrc, NULL); fail_unless (internal_ssrc != ssrc); fail_unless_equals_int (0x01BADBAD, internal_ssrc); /* verify SR and RR */ j = 0; for (i = 0; i < 5; i++) { session_harness_produce_rtcp (h, 1); buf = session_harness_pull_rtcp (h); g_assert (buf != NULL); fail_unless (gst_rtcp_buffer_validate (buf)); gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp); fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet)); rtcp_type = gst_rtcp_packet_get_type (&rtcp_packet); if (rtcp_type == GST_RTCP_TYPE_SR) { gst_rtcp_packet_sr_get_sender_info (&rtcp_packet, &ssrc, NULL, NULL, NULL, NULL); fail_unless_equals_int (internal_ssrc, ssrc); fail_unless_equals_int (0x01BADBAD, ssrc); j |= 0x1; } else if (rtcp_type == GST_RTCP_TYPE_RR) { ssrc = gst_rtcp_packet_rr_get_ssrc (&rtcp_packet); if (internal_ssrc != ssrc) j |= 0x2; } gst_rtcp_buffer_unmap (&rtcp); gst_buffer_unref (buf); } fail_unless_equals_int (0x3, j); /* verify we got both SR and RR */ /* go 30 seconds in the future and observe both sources timing out: * 0xDEADBEEF -> BYE, 0x01BADBAD -> becomes receiver only */ fail_unless (session_harness_advance_and_crank (h, 30 * GST_SECOND)); /* verify BYE and RR */ j = 0; seen_bye = FALSE; while (!seen_bye) { session_harness_produce_rtcp (h, 1); buf = session_harness_pull_rtcp (h); fail_unless (buf != NULL); fail_unless (gst_rtcp_buffer_validate (buf)); gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp); fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet)); rtcp_type = gst_rtcp_packet_get_type (&rtcp_packet); if (rtcp_type == GST_RTCP_TYPE_RR) { ssrc = gst_rtcp_packet_rr_get_ssrc (&rtcp_packet); if (ssrc == 0x01BADBAD) { j |= 0x1; fail_unless_equals_int (internal_ssrc, ssrc); /* 2 => RR, SDES. There is no BYE here */ fail_unless_equals_int (2, gst_rtcp_buffer_get_packet_count (&rtcp)); } else if (ssrc == 0xDEADBEEF) { j |= 0x2; g_assert_cmpint (ssrc, !=, internal_ssrc); /* 3 => RR, SDES, BYE */ if (gst_rtcp_buffer_get_packet_count (&rtcp) == 3) { fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet)); fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet)); fail_unless_equals_int (GST_RTCP_TYPE_BYE, gst_rtcp_packet_get_type (&rtcp_packet)); seen_bye = TRUE; } } } gst_rtcp_buffer_unmap (&rtcp); gst_buffer_unref (buf); } fail_unless_equals_int (0x3, j); /* verify we got both BYE and RR */ session_harness_free (h); } GST_END_TEST; typedef struct { guint8 subtype; guint32 ssrc; gchar *name; GstBuffer *data; } RTCPAppResult; static void on_app_rtcp_cb (GObject * session, guint subtype, guint ssrc, const gchar * name, GstBuffer * data, RTCPAppResult * result) { result->subtype = subtype; result->ssrc = ssrc; result->name = g_strdup (name); result->data = data ? gst_buffer_ref (data) : NULL; } GST_START_TEST (test_receive_rtcp_app_packet) { SessionHarness *h = session_harness_new (); GstBuffer *buf; GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT; GstRTCPPacket packet; RTCPAppResult result = { 0 }; guint8 data[] = { 0x11, 0x22, 0x33, 0x44 }; g_signal_connect (h->internal_session, "on-app-rtcp", G_CALLBACK (on_app_rtcp_cb), &result); /* Push APP buffer with no data */ buf = gst_rtcp_buffer_new (1000); fail_unless (gst_rtcp_buffer_map (buf, GST_MAP_READWRITE, &rtcp)); fail_unless (gst_rtcp_buffer_add_packet (&rtcp, GST_RTCP_TYPE_APP, &packet)); gst_rtcp_packet_app_set_subtype (&packet, 21); gst_rtcp_packet_app_set_ssrc (&packet, 0x11111111); gst_rtcp_packet_app_set_name (&packet, "Test"); gst_rtcp_buffer_unmap (&rtcp); fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtcp (h, buf)); fail_unless_equals_int (21, result.subtype); fail_unless_equals_int (0x11111111, result.ssrc); fail_unless_equals_string ("Test", result.name); fail_unless_equals_pointer (NULL, result.data); g_free (result.name); /* Push APP buffer with data */ memset (&result, 0, sizeof (result)); buf = gst_rtcp_buffer_new (1000); fail_unless (gst_rtcp_buffer_map (buf, GST_MAP_READWRITE, &rtcp)); fail_unless (gst_rtcp_buffer_add_packet (&rtcp, GST_RTCP_TYPE_APP, &packet)); gst_rtcp_packet_app_set_subtype (&packet, 22); gst_rtcp_packet_app_set_ssrc (&packet, 0x22222222); gst_rtcp_packet_app_set_name (&packet, "Test"); gst_rtcp_packet_app_set_data_length (&packet, sizeof (data) / 4); memcpy (gst_rtcp_packet_app_get_data (&packet), data, sizeof (data)); gst_rtcp_buffer_unmap (&rtcp); fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtcp (h, buf)); fail_unless_equals_int (22, result.subtype); fail_unless_equals_int (0x22222222, result.ssrc); fail_unless_equals_string ("Test", result.name); fail_unless (gst_buffer_memcmp (result.data, 0, data, sizeof (data)) == 0); g_free (result.name); gst_buffer_unref (result.data); session_harness_free (h); } GST_END_TEST; static void stats_test_cb (GObject * object, GParamSpec * spec, gpointer data) { guint num_sources = 0; gboolean *cb_called = data; g_assert (*cb_called == FALSE); /* We should be able to get a rtpsession property without introducing the deadlock */ g_object_get (object, "num-sources", &num_sources, NULL); *cb_called = TRUE; } GST_START_TEST (test_dont_lock_on_stats) { SessionHarness *h = session_harness_new (); gboolean cb_called = FALSE; /* connect to the stats-reporting */ g_signal_connect (h->session, "notify::stats", G_CALLBACK (stats_test_cb), &cb_called); /* Push RTP buffer to make sure RTCP-thread have started */ fail_unless_equals_int (GST_FLOW_OK, session_harness_send_rtp (h, generate_test_buffer (0, 0xDEADBEEF))); /* crank the RTCP-thread and pull out rtcp, generating a stats-callback */ session_harness_crank_clock (h); gst_buffer_unref (session_harness_pull_rtcp (h)); fail_unless (cb_called); session_harness_free (h); } GST_END_TEST; static void suspicious_bye_cb (GObject * object, GParamSpec * spec, gpointer data) { GValueArray *stats_arr; GstStructure *stats, *internal_stats; gboolean *cb_called = data; gboolean internal = FALSE, sent_bye = TRUE; guint ssrc = 0; guint i; g_assert (*cb_called == FALSE); *cb_called = TRUE; g_object_get (object, "stats", &stats, NULL); stats_arr = g_value_get_boxed (gst_structure_get_value (stats, "source-stats")); g_assert (stats_arr != NULL); fail_unless (stats_arr->n_values >= 1); for (i = 0; i < stats_arr->n_values; i++) { internal_stats = g_value_get_boxed (g_value_array_get_nth (stats_arr, i)); g_assert (internal_stats != NULL); gst_structure_get (internal_stats, "ssrc", G_TYPE_UINT, &ssrc, "internal", G_TYPE_BOOLEAN, &internal, "received-bye", G_TYPE_BOOLEAN, &sent_bye, NULL); if (ssrc == 0xDEADBEEF) { fail_unless (internal); fail_unless (!sent_bye); break; } } fail_unless_equals_int (ssrc, 0xDEADBEEF); gst_structure_free (stats); } static GstBuffer * create_bye_rtcp (guint32 ssrc) { GstRTCPPacket packet; GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT; GSocketAddress *saddr; GstBuffer *buffer = gst_rtcp_buffer_new (1000); fail_unless (gst_rtcp_buffer_map (buffer, GST_MAP_READWRITE, &rtcp)); fail_unless (gst_rtcp_buffer_add_packet (&rtcp, GST_RTCP_TYPE_BYE, &packet)); gst_rtcp_packet_bye_add_ssrc (&packet, ssrc); gst_rtcp_buffer_unmap (&rtcp); /* Need to add meta to trigger collision detection */ saddr = g_inet_socket_address_new_from_string ("127.0.0.1", 3490); gst_buffer_add_net_address_meta (buffer, saddr); g_object_unref (saddr); return buffer; } GST_START_TEST (test_ignore_suspicious_bye) { SessionHarness *h = session_harness_new (); gboolean cb_called = FALSE; /* connect to the stats-reporting */ g_signal_connect (h->session, "notify::stats", G_CALLBACK (suspicious_bye_cb), &cb_called); /* Push RTP buffer making our internal SSRC=0xDEADBEEF */ fail_unless_equals_int (GST_FLOW_OK, session_harness_send_rtp (h, generate_test_buffer (0, 0xDEADBEEF))); /* Receive BYE RTCP referencing our internal SSRC(!?!) (0xDEADBEEF) */ fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtcp (h, create_bye_rtcp (0xDEADBEEF))); /* "crank" and check the stats */ session_harness_crank_clock (h); gst_buffer_unref (session_harness_pull_rtcp (h)); fail_unless (cb_called); session_harness_free (h); } GST_END_TEST; static GstBuffer * create_buffer (guint8 * data, gsize size) { return gst_buffer_new_wrapped_full (GST_MEMORY_FLAG_READONLY, data, size, 0, size, NULL, NULL); } GST_START_TEST (test_receive_regular_pli) { SessionHarness *h = session_harness_new (); GstEvent *ev; /* PLI packet */ guint8 rtcp_pkt[] = { 0x81, /* PLI */ 0xce, /* Type 206 Application layer feedback */ 0x00, 0x02, /* Length */ 0x37, 0x56, 0x93, 0xed, /* Sender SSRC */ 0x37, 0x56, 0x93, 0xed /* Media SSRC */ }; fail_unless_equals_int (GST_FLOW_OK, session_harness_send_rtp (h, generate_test_buffer (0, 928420845))); session_harness_recv_rtcp (h, create_buffer (rtcp_pkt, sizeof (rtcp_pkt))); fail_unless_equals_int (3, gst_harness_upstream_events_received (h->send_rtp_h)); /* Remove the first 2 reconfigure events */ fail_unless ((ev = gst_harness_pull_upstream_event (h->send_rtp_h)) != NULL); fail_unless_equals_int (GST_EVENT_RECONFIGURE, GST_EVENT_TYPE (ev)); gst_event_unref (ev); fail_unless ((ev = gst_harness_pull_upstream_event (h->send_rtp_h)) != NULL); fail_unless_equals_int (GST_EVENT_RECONFIGURE, GST_EVENT_TYPE (ev)); gst_event_unref (ev); /* Then pull and check the force key-unit event */ fail_unless ((ev = gst_harness_pull_upstream_event (h->send_rtp_h)) != NULL); fail_unless_equals_int (GST_EVENT_CUSTOM_UPSTREAM, GST_EVENT_TYPE (ev)); fail_unless (gst_video_event_is_force_key_unit (ev)); gst_event_unref (ev); session_harness_free (h); } GST_END_TEST; GST_START_TEST (test_receive_pli_no_sender_ssrc) { SessionHarness *h = session_harness_new (); GstEvent *ev; /* PLI packet */ guint8 rtcp_pkt[] = { 0x81, /* PLI */ 0xce, /* Type 206 Application layer feedback */ 0x00, 0x02, /* Length */ 0x00, 0x00, 0x00, 0x00, /* Sender SSRC */ 0x37, 0x56, 0x93, 0xed /* Media SSRC */ }; fail_unless_equals_int (GST_FLOW_OK, session_harness_send_rtp (h, generate_test_buffer (0, 928420845))); session_harness_recv_rtcp (h, create_buffer (rtcp_pkt, sizeof (rtcp_pkt))); fail_unless_equals_int (3, gst_harness_upstream_events_received (h->send_rtp_h)); /* Remove the first 2 reconfigure events */ fail_unless ((ev = gst_harness_pull_upstream_event (h->send_rtp_h)) != NULL); fail_unless_equals_int (GST_EVENT_RECONFIGURE, GST_EVENT_TYPE (ev)); gst_event_unref (ev); fail_unless ((ev = gst_harness_pull_upstream_event (h->send_rtp_h)) != NULL); fail_unless_equals_int (GST_EVENT_RECONFIGURE, GST_EVENT_TYPE (ev)); gst_event_unref (ev); /* Then pull and check the force key-unit event */ fail_unless ((ev = gst_harness_pull_upstream_event (h->send_rtp_h)) != NULL); fail_unless_equals_int (GST_EVENT_CUSTOM_UPSTREAM, GST_EVENT_TYPE (ev)); fail_unless (gst_video_event_is_force_key_unit (ev)); gst_event_unref (ev); session_harness_free (h); } GST_END_TEST; static void add_rtcp_sdes_packet (GstBuffer * gstbuf, guint32 ssrc, const char *cname) { GstRTCPPacket packet; GstRTCPBuffer buffer = GST_RTCP_BUFFER_INIT; gst_rtcp_buffer_map (gstbuf, GST_MAP_READWRITE, &buffer); fail_unless (gst_rtcp_buffer_add_packet (&buffer, GST_RTCP_TYPE_SDES, &packet) == TRUE); fail_unless (gst_rtcp_packet_sdes_add_item (&packet, ssrc) == TRUE); fail_unless (gst_rtcp_packet_sdes_add_entry (&packet, GST_RTCP_SDES_CNAME, strlen (cname), (const guint8 *) cname)); gst_rtcp_buffer_unmap (&buffer); } static void on_ssrc_collision_cb (GstElement * rtpsession, guint ssrc, gpointer user_data) { gboolean *had_collision = user_data; *had_collision = TRUE; } GST_START_TEST (test_ssrc_collision_when_sending) { SessionHarness *h = session_harness_new (); GstBuffer *buf; GstEvent *ev; GSocketAddress *saddr; gboolean had_collision = FALSE; g_signal_connect (h->internal_session, "on-ssrc-collision", G_CALLBACK (on_ssrc_collision_cb), &had_collision); /* Push SDES with identical SSRC as what we will use for sending RTP, establishing this as a non-internal SSRC */ buf = gst_rtcp_buffer_new (1400); add_rtcp_sdes_packet (buf, 0x12345678, "test@foo.bar"); saddr = g_inet_socket_address_new_from_string ("127.0.0.1", 8080); gst_buffer_add_net_address_meta (buf, saddr); g_object_unref (saddr); session_harness_recv_rtcp (h, buf); fail_unless (had_collision == FALSE); /* Push RTP buffer making our internal SSRC=0x12345678 */ buf = generate_test_buffer (0, 0x12345678); fail_unless_equals_int (GST_FLOW_OK, session_harness_send_rtp (h, buf)); fail_unless (had_collision == TRUE); /* Verify the packet we just sent is not being boomeranged back to us as a received packet! */ fail_unless_equals_int (0, gst_harness_buffers_in_queue (h->recv_rtp_h)); while ((ev = gst_harness_try_pull_upstream_event (h->send_rtp_h)) != NULL) { if (GST_EVENT_CUSTOM_UPSTREAM == GST_EVENT_TYPE (ev) && gst_event_has_name (ev, "GstRTPCollision")) break; gst_event_unref (ev); } fail_unless (ev != NULL); gst_event_unref (ev); session_harness_free (h); } GST_END_TEST; GST_START_TEST (test_ssrc_collision_when_sending_loopback) { SessionHarness *h = session_harness_new (); GstBuffer *buf; GstEvent *ev; GSocketAddress *saddr; gboolean had_collision = FALSE; guint new_ssrc; const GstStructure *s; g_signal_connect (h->internal_session, "on-ssrc-collision", G_CALLBACK (on_ssrc_collision_cb), &had_collision); /* Push SDES with identical SSRC as what we will use for sending RTP, establishing this as a non-internal SSRC */ buf = gst_rtcp_buffer_new (1400); add_rtcp_sdes_packet (buf, 0x12345678, "test@foo.bar"); saddr = g_inet_socket_address_new_from_string ("127.0.0.1", 8080); gst_buffer_add_net_address_meta (buf, saddr); g_object_unref (saddr); session_harness_recv_rtcp (h, buf); fail_unless (had_collision == FALSE); /* Push RTP buffer making our internal SSRC=0x12345678 */ buf = generate_test_buffer (0, 0x12345678); fail_unless_equals_int (GST_FLOW_OK, session_harness_send_rtp (h, buf)); fail_unless (had_collision == TRUE); /* Verify the packet we just sent is not being boomeranged back to us as a received packet! */ fail_unless_equals_int (0, gst_harness_buffers_in_queue (h->recv_rtp_h)); while ((ev = gst_harness_try_pull_upstream_event (h->send_rtp_h)) != NULL) { if (GST_EVENT_CUSTOM_UPSTREAM == GST_EVENT_TYPE (ev) && gst_event_has_name (ev, "GstRTPCollision")) break; gst_event_unref (ev); } fail_unless (ev != NULL); s = gst_event_get_structure (ev); fail_unless (gst_structure_get_uint (s, "ssrc", &new_ssrc)); gst_event_unref (ev); /* reset collision detection */ had_collision = FALSE; /* Push SDES from same address but with the new SSRC, as if someone * was looping back our packets to us */ buf = gst_rtcp_buffer_new (1400); add_rtcp_sdes_packet (buf, new_ssrc, "test@foo.bar"); saddr = g_inet_socket_address_new_from_string ("127.0.0.1", 8080); gst_buffer_add_net_address_meta (buf, saddr); g_object_unref (saddr); session_harness_recv_rtcp (h, buf); /* Make sure we didn't detect a collision */ fail_unless (had_collision == FALSE); /* Make sure there is no collision event either */ while ((ev = gst_harness_try_pull_upstream_event (h->send_rtp_h)) != NULL) { fail_if (GST_EVENT_CUSTOM_UPSTREAM == GST_EVENT_TYPE (ev) && gst_event_has_name (ev, "GstRTPCollision")); gst_event_unref (ev); } session_harness_free (h); } GST_END_TEST; GST_START_TEST (test_ssrc_collision_when_receiving) { SessionHarness *h = session_harness_new (); GstBuffer *buf; GstEvent *ev; GSocketAddress *saddr; gboolean had_collision = FALSE; g_signal_connect (h->internal_session, "on-ssrc-collision", G_CALLBACK (on_ssrc_collision_cb), &had_collision); /* Push RTP buffer making our internal SSRC=0x12345678 */ buf = generate_test_buffer (0, 0x12345678); fail_unless_equals_int (GST_FLOW_OK, session_harness_send_rtp (h, buf)); fail_unless (had_collision == FALSE); /* Push SDES with identical SSRC as what we used to send RTP, to create a collision */ buf = gst_rtcp_buffer_new (1400); add_rtcp_sdes_packet (buf, 0x12345678, "test@foo.bar"); saddr = g_inet_socket_address_new_from_string ("127.0.0.1", 8080); gst_buffer_add_net_address_meta (buf, saddr); g_object_unref (saddr); session_harness_recv_rtcp (h, buf); fail_unless (had_collision == TRUE); /* Verify the packet we just sent is not being boomeranged back to us as a received packet! */ fail_unless_equals_int (0, gst_harness_buffers_in_queue (h->recv_rtp_h)); while ((ev = gst_harness_try_pull_upstream_event (h->send_rtp_h)) != NULL) { if (GST_EVENT_CUSTOM_UPSTREAM == GST_EVENT_TYPE (ev)) break; gst_event_unref (ev); } fail_unless (ev != NULL); gst_event_unref (ev); session_harness_free (h); } GST_END_TEST; GST_START_TEST (test_ssrc_collision_third_party) { SessionHarness *h = session_harness_new (); GstBuffer *buf; GSocketAddress *saddr; gboolean had_collision = FALSE; guint i; g_signal_connect (h->internal_session, "on-ssrc-collision", G_CALLBACK (on_ssrc_collision_cb), &had_collision); for (i = 0; i < 4; i++) { /* Receive 4 buffers SSRC=0x12345678 from 127.0.0.1 to pass probation */ buf = generate_test_buffer (i, 0x12345678); saddr = g_inet_socket_address_new_from_string ("127.0.0.1", 8080); gst_buffer_add_net_address_meta (buf, saddr); g_object_unref (saddr); fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtp (h, buf)); } /* Check that we received the first 4 buffer */ for (i = 0; i < 4; i++) { buf = gst_harness_pull (h->recv_rtp_h); fail_unless (buf); gst_buffer_unref (buf); } fail_unless (had_collision == FALSE); /* Receive buffer SSRC=0x12345678 from 127.0.0.2 */ buf = generate_test_buffer (0, 0x12345678); saddr = g_inet_socket_address_new_from_string ("127.0.0.2", 8080); gst_buffer_add_net_address_meta (buf, saddr); g_object_unref (saddr); fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtp (h, buf)); /* Verify the packet we just sent has been dropped */ fail_unless_equals_int (0, gst_harness_buffers_in_queue (h->recv_rtp_h)); /* Receive another buffer SSRC=0x12345678 from 127.0.0.1 */ buf = generate_test_buffer (0, 0x12345678); saddr = g_inet_socket_address_new_from_string ("127.0.0.1", 8080); gst_buffer_add_net_address_meta (buf, saddr); g_object_unref (saddr); fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtp (h, buf)); /* Check that we received the other buffer */ buf = gst_harness_pull (h->recv_rtp_h); fail_unless (buf); gst_buffer_unref (buf); fail_unless (had_collision == FALSE); session_harness_free (h); } GST_END_TEST; GST_START_TEST (test_ssrc_collision_third_party_favor_new) { SessionHarness *h = session_harness_new (); GstBuffer *buf; GSocketAddress *saddr; gboolean had_collision = FALSE; guint i; g_object_set (h->internal_session, "favor-new", TRUE, NULL); g_signal_connect (h->internal_session, "on-ssrc-collision", G_CALLBACK (on_ssrc_collision_cb), &had_collision); for (i = 0; i < 4; i++) { /* Receive 4 buffers SSRC=0x12345678 from 127.0.0.1 to pass probation */ buf = generate_test_buffer (i, 0x12345678); saddr = g_inet_socket_address_new_from_string ("127.0.0.1", 8080); gst_buffer_add_net_address_meta (buf, saddr); g_object_unref (saddr); fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtp (h, buf)); } /* Check that we received the first 4 buffer */ for (i = 0; i < 4; i++) { buf = gst_harness_pull (h->recv_rtp_h); fail_unless (buf); gst_buffer_unref (buf); } fail_unless (had_collision == FALSE); /* Receive buffer SSRC=0x12345678 from 127.0.0.2 */ buf = generate_test_buffer (0, 0x12345678); saddr = g_inet_socket_address_new_from_string ("127.0.0.2", 8080); gst_buffer_add_net_address_meta (buf, saddr); g_object_unref (saddr); fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtp (h, buf)); /* Check that we received the other buffer */ buf = gst_harness_pull (h->recv_rtp_h); fail_unless (buf); gst_buffer_unref (buf); fail_unless (had_collision == FALSE); session_harness_free (h); } GST_END_TEST; GST_START_TEST (test_request_fir) { SessionHarness *h = session_harness_new (); GstBuffer *buf; GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT; GstRTCPPacket rtcp_packet; guint8 *fci_data; /* add FIR-capabilites to our caps */ gst_caps_set_simple (h->caps, "rtcp-fb-ccm-fir", G_TYPE_BOOLEAN, TRUE, NULL); /* clear pt-map to removed the cached caps without fir */ g_signal_emit_by_name (h->session, "clear-pt-map"); g_object_set (h->internal_session, "internal-ssrc", 0xDEADBEEF, NULL); /* Receive a RTP buffer from the wire from 2 different ssrcs */ fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtp (h, generate_test_buffer (0, 0x12345678))); fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtp (h, generate_test_buffer (0, 0x87654321))); /* fix to make the test deterministic: We need to wait for the RTCP-thread to have settled to ensure the key-unit will considered once released */ gst_test_clock_wait_for_next_pending_id (h->testclock, NULL); /* request FIR for both SSRCs */ session_harness_force_key_unit (h, 0, 0x12345678, TEST_BUF_PT, NULL, NULL); session_harness_force_key_unit (h, 0, 0x87654321, TEST_BUF_PT, NULL, NULL); session_harness_produce_rtcp (h, 1); buf = session_harness_pull_rtcp (h); fail_unless (gst_rtcp_buffer_validate (buf)); gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp); fail_unless_equals_int (3, gst_rtcp_buffer_get_packet_count (&rtcp)); fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet)); /* first a Receiver Report */ fail_unless_equals_int (GST_RTCP_TYPE_RR, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet)); /* then a SDES */ fail_unless_equals_int (GST_RTCP_TYPE_SDES, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet)); /* and then our FIR */ fail_unless_equals_int (GST_RTCP_TYPE_PSFB, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless_equals_int (GST_RTCP_PSFB_TYPE_FIR, gst_rtcp_packet_fb_get_type (&rtcp_packet)); /* FIR has sender-ssrc as normal, but media-ssrc set to 0, because it can have multiple media-ssrcs in its fci-data */ fail_unless_equals_int (0xDEADBEEF, gst_rtcp_packet_fb_get_sender_ssrc (&rtcp_packet)); fail_unless_equals_int (0, gst_rtcp_packet_fb_get_media_ssrc (&rtcp_packet)); fci_data = gst_rtcp_packet_fb_get_fci (&rtcp_packet); fail_unless_equals_int (16, gst_rtcp_packet_fb_get_fci_length (&rtcp_packet) * sizeof (guint32)); /* verify the FIR contains both SSRCs */ fail_unless_equals_int (0x87654321, GST_READ_UINT32_BE (fci_data)); fail_unless_equals_int (1, fci_data[4]); fail_unless_equals_int (0, fci_data[5]); fail_unless_equals_int (0, fci_data[6]); fail_unless_equals_int (0, fci_data[7]); fci_data += 8; fail_unless_equals_int (0x12345678, GST_READ_UINT32_BE (fci_data)); fail_unless_equals_int (1, fci_data[4]); fail_unless_equals_int (0, fci_data[5]); fail_unless_equals_int (0, fci_data[6]); fail_unless_equals_int (0, fci_data[7]); gst_rtcp_buffer_unmap (&rtcp); gst_buffer_unref (buf); session_harness_free (h); } GST_END_TEST; GST_START_TEST (test_request_pli) { SessionHarness *h = session_harness_new (); GstBuffer *buf; GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT; GstRTCPPacket rtcp_packet; /* add PLI-capabilites to our caps */ gst_caps_set_simple (h->caps, "rtcp-fb-nack-pli", G_TYPE_BOOLEAN, TRUE, NULL); /* clear pt-map to removed the cached caps without PLI */ g_signal_emit_by_name (h->session, "clear-pt-map"); g_object_set (h->internal_session, "internal-ssrc", 0xDEADBEEF, NULL); /* Receive a RTP buffer from the wire */ fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtp (h, generate_test_buffer (0, 0x12345678))); /* Wait for first regular RTCP to be sent so that we are clear to send early RTCP */ session_harness_produce_rtcp (h, 1); gst_buffer_unref (session_harness_pull_rtcp (h)); /* request PLI */ session_harness_force_key_unit (h, 0, 0x12345678, TEST_BUF_PT, NULL, NULL); /* PLI should be produced immediately as early RTCP is allowed. Pull buffer without advancing the clock to ensure this is the case */ buf = session_harness_pull_rtcp (h); fail_unless (gst_rtcp_buffer_validate (buf)); gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp); fail_unless_equals_int (3, gst_rtcp_buffer_get_packet_count (&rtcp)); fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet)); /* first a Receiver Report */ fail_unless_equals_int (GST_RTCP_TYPE_RR, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet)); /* then a SDES */ fail_unless_equals_int (GST_RTCP_TYPE_SDES, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet)); /* and then our PLI */ fail_unless_equals_int (GST_RTCP_TYPE_PSFB, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless_equals_int (GST_RTCP_PSFB_TYPE_PLI, gst_rtcp_packet_fb_get_type (&rtcp_packet)); fail_unless_equals_int (0xDEADBEEF, gst_rtcp_packet_fb_get_sender_ssrc (&rtcp_packet)); fail_unless_equals_int (0x12345678, gst_rtcp_packet_fb_get_media_ssrc (&rtcp_packet)); fail_unless_equals_int (0, gst_rtcp_packet_fb_get_fci_length (&rtcp_packet)); gst_rtcp_buffer_unmap (&rtcp); gst_buffer_unref (buf); session_harness_free (h); } GST_END_TEST; GST_START_TEST (test_request_fir_after_pli_in_caps) { SessionHarness *h = session_harness_new (); GstBuffer *buf; GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT; GstRTCPPacket rtcp_packet; guint8 *fci_data; /* add PLI-capabilites to our caps */ gst_caps_set_simple (h->caps, "rtcp-fb-nack-pli", G_TYPE_BOOLEAN, TRUE, NULL); /* clear pt-map to removed the cached caps without PLI */ g_signal_emit_by_name (h->session, "clear-pt-map"); g_object_set (h->internal_session, "internal-ssrc", 0xDEADBEEF, NULL); /* Receive a RTP buffer from the wire */ fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtp (h, generate_test_buffer (0, 0x12345678))); /* Wait for first regular RTCP to be sent so that we are clear to send early RTCP */ session_harness_produce_rtcp (h, 1); gst_buffer_unref (session_harness_pull_rtcp (h)); /* request PLI */ session_harness_force_key_unit (h, 0, 0x12345678, TEST_BUF_PT, NULL, NULL); /* PLI should be produced immediately as early RTCP is allowed. Pull buffer without advancing the clock to ensure this is the case */ buf = session_harness_pull_rtcp (h); fail_unless (gst_rtcp_buffer_validate (buf)); gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp); fail_unless_equals_int (3, gst_rtcp_buffer_get_packet_count (&rtcp)); fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet)); /* first a Receiver Report */ fail_unless_equals_int (GST_RTCP_TYPE_RR, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet)); /* then a SDES */ fail_unless_equals_int (GST_RTCP_TYPE_SDES, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet)); /* and then our PLI */ fail_unless_equals_int (GST_RTCP_TYPE_PSFB, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless_equals_int (GST_RTCP_PSFB_TYPE_PLI, gst_rtcp_packet_fb_get_type (&rtcp_packet)); fail_unless_equals_int (0xDEADBEEF, gst_rtcp_packet_fb_get_sender_ssrc (&rtcp_packet)); fail_unless_equals_int (0x12345678, gst_rtcp_packet_fb_get_media_ssrc (&rtcp_packet)); fail_unless_equals_int (0, gst_rtcp_packet_fb_get_fci_length (&rtcp_packet)); gst_rtcp_buffer_unmap (&rtcp); gst_buffer_unref (buf); /* Rebuild the caps */ gst_caps_unref (h->caps); h->caps = generate_caps (); /* add FIR-capabilites to our caps */ gst_caps_set_simple (h->caps, "rtcp-fb-ccm-fir", G_TYPE_BOOLEAN, TRUE, NULL); /* clear pt-map to removed the cached caps without fir */ g_signal_emit_by_name (h->session, "clear-pt-map"); /* Receive a RTP buffer from the wire */ fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtp (h, generate_test_buffer (0, 0x12345678))); /* fix to make the test deterministic: We need to wait for the RTCP-thread to have settled to ensure the key-unit will considered once released */ gst_test_clock_wait_for_next_pending_id (h->testclock, NULL); /* request FIR */ session_harness_force_key_unit (h, 0, 0x12345678, TEST_BUF_PT, NULL, NULL); session_harness_produce_rtcp (h, 1); buf = session_harness_pull_rtcp (h); fail_unless (gst_rtcp_buffer_validate (buf)); gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp); fail_unless_equals_int (3, gst_rtcp_buffer_get_packet_count (&rtcp)); fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet)); /* first a Receiver Report */ fail_unless_equals_int (GST_RTCP_TYPE_RR, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet)); /* then a SDES */ fail_unless_equals_int (GST_RTCP_TYPE_SDES, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet)); /* and then our FIR */ fail_unless_equals_int (GST_RTCP_TYPE_PSFB, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless_equals_int (GST_RTCP_PSFB_TYPE_FIR, gst_rtcp_packet_fb_get_type (&rtcp_packet)); /* FIR has sender-ssrc as normal, but media-ssrc set to 0, because it can have multiple media-ssrcs in its fci-data */ fail_unless_equals_int (0xDEADBEEF, gst_rtcp_packet_fb_get_sender_ssrc (&rtcp_packet)); fail_unless_equals_int (0, gst_rtcp_packet_fb_get_media_ssrc (&rtcp_packet)); fci_data = gst_rtcp_packet_fb_get_fci (&rtcp_packet); fail_unless_equals_int (8, gst_rtcp_packet_fb_get_fci_length (&rtcp_packet) * sizeof (guint32)); /* verify the FIR contains the SSRC */ fail_unless_equals_int (0x12345678, GST_READ_UINT32_BE (fci_data)); fail_unless_equals_int (1, fci_data[4]); fail_unless_equals_int (0, fci_data[5]); fail_unless_equals_int (0, fci_data[6]); fail_unless_equals_int (0, fci_data[7]); fail_unless_equals_int (0, fci_data[7]); gst_rtcp_buffer_unmap (&rtcp); gst_buffer_unref (buf); session_harness_free (h); } GST_END_TEST; GST_START_TEST (test_illegal_rtcp_fb_packet) { SessionHarness *h = session_harness_new (); GstBuffer *buf; /* Zero length RTCP feedback packet (reduced size) */ const guint8 rtcp_zero_fb_pkt[] = { 0x8f, 0xce, 0x00, 0x00 }; g_object_set (h->internal_session, "internal-ssrc", 0xDEADBEEF, NULL); buf = gst_buffer_new_and_alloc (sizeof (rtcp_zero_fb_pkt)); gst_buffer_fill (buf, 0, rtcp_zero_fb_pkt, sizeof (rtcp_zero_fb_pkt)); GST_BUFFER_DTS (buf) = GST_BUFFER_PTS (buf) = G_GUINT64_CONSTANT (0); /* Push the packet, this did previously crash because length of packet was * never validated. */ fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtcp (h, buf)); session_harness_free (h); } GST_END_TEST; typedef struct { GCond *cond; GMutex *mutex; gboolean fired; } FeedbackRTCPCallbackData; static void feedback_rtcp_cb (GstElement * element, guint fbtype, guint fmt, guint sender_ssrc, guint media_ssrc, GstBuffer * fci, FeedbackRTCPCallbackData * cb_data) { g_mutex_lock (cb_data->mutex); cb_data->fired = TRUE; g_cond_wait (cb_data->cond, cb_data->mutex); g_mutex_unlock (cb_data->mutex); } static void * send_feedback_rtcp (SessionHarness * h) { GstRTCPPacket packet; GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT; GstBuffer *buffer = gst_rtcp_buffer_new (1000); fail_unless (gst_rtcp_buffer_map (buffer, GST_MAP_READWRITE, &rtcp)); fail_unless (gst_rtcp_buffer_add_packet (&rtcp, GST_RTCP_TYPE_PSFB, &packet)); gst_rtcp_packet_fb_set_type (&packet, GST_RTCP_PSFB_TYPE_PLI); gst_rtcp_packet_fb_set_fci_length (&packet, 0); gst_rtcp_packet_fb_set_media_ssrc (&packet, 0xABE2B0B); gst_rtcp_packet_fb_set_media_ssrc (&packet, 0xDEADBEEF); gst_rtcp_buffer_unmap (&rtcp); fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtcp (h, buffer)); return NULL; } GST_START_TEST (test_feedback_rtcp_race) { SessionHarness *h = session_harness_new (); GCond cond; GMutex mutex; FeedbackRTCPCallbackData cb_data; GThread *send_rtcp_thread; g_cond_init (&cond); g_mutex_init (&mutex); cb_data.cond = &cond; cb_data.mutex = &mutex; cb_data.fired = FALSE; g_signal_connect (h->internal_session, "on-feedback-rtcp", G_CALLBACK (feedback_rtcp_cb), &cb_data); /* Push RTP buffer making external source with SSRC=0xDEADBEEF */ fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtp (h, generate_test_buffer (0, 0xDEADBEEF))); /* Push feedback RTCP with media SSRC=0xDEADBEEF */ send_rtcp_thread = g_thread_new (NULL, (GThreadFunc) send_feedback_rtcp, h); /* Waiting for feedback RTCP callback to fire */ while (!cb_data.fired) g_usleep (G_USEC_PER_SEC / 100); /* While send_rtcp_thread thread is waiting for our signal advance the clock by 30sec triggering removal of 0xDEADBEEF, as if the source was inactive for too long */ session_harness_advance_and_crank (h, GST_SECOND * 30); gst_buffer_unref (session_harness_pull_rtcp (h)); /* Let send_rtcp_thread finish */ g_mutex_lock (&mutex); g_cond_signal (&cond); g_mutex_unlock (&mutex); g_thread_join (send_rtcp_thread); session_harness_free (h); } GST_END_TEST; GST_START_TEST (test_dont_send_rtcp_while_idle) { SessionHarness *h = session_harness_new (); /* verify the RTCP thread has not started */ fail_unless_equals_int (0, gst_test_clock_peek_id_count (h->testclock)); /* and that no RTCP has been pushed */ fail_unless_equals_int (0, gst_harness_buffers_in_queue (h->rtcp_h)); session_harness_free (h); } GST_END_TEST; GST_START_TEST (test_send_rtcp_when_signalled) { SessionHarness *h = session_harness_new (); gboolean ret; /* verify the RTCP thread has not started */ fail_unless_equals_int (0, gst_test_clock_peek_id_count (h->testclock)); /* and that no RTCP has been pushed */ fail_unless_equals_int (0, gst_harness_buffers_in_queue (h->rtcp_h)); /* then ask explicitly to send RTCP */ g_signal_emit_by_name (h->internal_session, "send-rtcp-full", GST_SECOND, &ret); /* this is FALSE due to no next RTCP check time */ fail_unless (ret == FALSE); /* "crank" and verify RTCP now was sent */ session_harness_crank_clock (h); gst_buffer_unref (session_harness_pull_rtcp (h)); session_harness_free (h); } GST_END_TEST; static void validate_sdes_priv (GstBuffer * buf, const char *name_ref, const char *value) { GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT; GstRTCPPacket pkt; fail_unless (gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp)); fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &pkt)); do { if (gst_rtcp_packet_get_type (&pkt) == GST_RTCP_TYPE_SDES) { fail_unless (gst_rtcp_packet_sdes_first_entry (&pkt)); do { GstRTCPSDESType type; guint8 len; guint8 *data; fail_unless (gst_rtcp_packet_sdes_get_entry (&pkt, &type, &len, &data)); if (type == GST_RTCP_SDES_PRIV) { char *name = g_strndup ((const gchar *) &data[1], data[0]); len -= data[0] + 1; data += data[0] + 1; fail_unless_equals_int (len, strlen (value)); fail_unless (!strncmp (value, (char *) data, len)); fail_unless_equals_string (name, name_ref); g_free (name); goto sdes_done; } } while (gst_rtcp_packet_sdes_next_entry (&pkt)); g_assert_not_reached (); } } while (gst_rtcp_packet_move_to_next (&pkt)); g_assert_not_reached (); sdes_done: fail_unless (gst_rtcp_buffer_unmap (&rtcp)); } GST_START_TEST (test_change_sent_sdes) { SessionHarness *h = session_harness_new (); GstStructure *s; GstBuffer *buf; gboolean ret; GstFlowReturn res; /* verify the RTCP thread has not started */ fail_unless_equals_int (0, gst_test_clock_peek_id_count (h->testclock)); /* and that no RTCP has been pushed */ fail_unless_equals_int (0, gst_harness_buffers_in_queue (h->rtcp_h)); s = gst_structure_new ("application/x-rtp-source-sdes", "other", G_TYPE_STRING, "first", NULL); g_object_set (h->internal_session, "sdes", s, NULL); gst_structure_free (s); /* then ask explicitly to send RTCP */ g_signal_emit_by_name (h->internal_session, "send-rtcp-full", GST_SECOND, &ret); /* this is FALSE due to no next RTCP check time */ fail_unless (ret == FALSE); /* "crank" and verify RTCP now was sent */ session_harness_crank_clock (h); buf = session_harness_pull_rtcp (h); fail_unless (buf); validate_sdes_priv (buf, "other", "first"); gst_buffer_unref (buf); /* Change the SDES */ s = gst_structure_new ("application/x-rtp-source-sdes", "other", G_TYPE_STRING, "second", NULL); g_object_set (h->internal_session, "sdes", s, NULL); gst_structure_free (s); /* Send an RTP packet */ buf = generate_test_buffer (22, 10000); res = session_harness_send_rtp (h, buf); fail_unless_equals_int (GST_FLOW_OK, res); /* "crank" enough to ensure a RTCP packet has been produced ! */ session_harness_crank_clock (h); session_harness_crank_clock (h); session_harness_crank_clock (h); session_harness_crank_clock (h); session_harness_crank_clock (h); session_harness_crank_clock (h); session_harness_crank_clock (h); session_harness_crank_clock (h); session_harness_crank_clock (h); session_harness_crank_clock (h); /* and verify RTCP now was sent with new SDES */ buf = session_harness_pull_rtcp (h); validate_sdes_priv (buf, "other", "second"); gst_buffer_unref (buf); session_harness_free (h); } GST_END_TEST; GST_START_TEST (test_request_nack) { SessionHarness *h = session_harness_new (); GstBuffer *buf; GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT; GstRTCPPacket rtcp_packet; guint8 *fci_data; guint32 fci_length; g_object_set (h->internal_session, "internal-ssrc", 0xDEADBEEF, NULL); /* Receive a RTP buffer from the wire */ fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtp (h, generate_test_buffer (0, 0x12345678))); /* Wait for first regular RTCP to be sent so that we are clear to send early RTCP */ session_harness_produce_rtcp (h, 1); gst_buffer_unref (session_harness_pull_rtcp (h)); /* request NACK immediately */ session_harness_rtp_retransmission_request (h, 0x12345678, 1234, 0, 0, 0); /* NACK should be produced immediately as early RTCP is allowed. Pull buffer without advancing the clock to ensure this is the case */ buf = session_harness_pull_rtcp (h); fail_unless (gst_rtcp_buffer_validate (buf)); gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp); fail_unless_equals_int (3, gst_rtcp_buffer_get_packet_count (&rtcp)); fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet)); /* first a Receiver Report */ fail_unless_equals_int (GST_RTCP_TYPE_RR, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet)); /* then a SDES */ fail_unless_equals_int (GST_RTCP_TYPE_SDES, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet)); /* and then our NACK */ fail_unless_equals_int (GST_RTCP_TYPE_RTPFB, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless_equals_int (GST_RTCP_RTPFB_TYPE_NACK, gst_rtcp_packet_fb_get_type (&rtcp_packet)); fail_unless_equals_int (0xDEADBEEF, gst_rtcp_packet_fb_get_sender_ssrc (&rtcp_packet)); fail_unless_equals_int (0x12345678, gst_rtcp_packet_fb_get_media_ssrc (&rtcp_packet)); fci_data = gst_rtcp_packet_fb_get_fci (&rtcp_packet); fci_length = gst_rtcp_packet_fb_get_fci_length (&rtcp_packet) * sizeof (guint32); fail_unless_equals_int (4, fci_length); fail_unless_equals_int (GST_READ_UINT32_BE (fci_data), 1234L << 16); gst_rtcp_buffer_unmap (&rtcp); gst_buffer_unref (buf); session_harness_free (h); } GST_END_TEST; typedef struct { gulong id; GstPad *pad; GMutex mutex; GCond cond; gboolean blocked; } BlockingProbeData; static GstPadProbeReturn on_rtcp_pad_blocked (GstPad * pad, GstPadProbeInfo * info, gpointer user_data) { BlockingProbeData *probe = user_data; g_mutex_lock (&probe->mutex); probe->blocked = TRUE; g_cond_signal (&probe->cond); g_mutex_unlock (&probe->mutex); return GST_PAD_PROBE_OK; } static void session_harness_block_rtcp (SessionHarness * h, BlockingProbeData * probe) { probe->pad = gst_element_get_static_pad (h->session, "send_rtcp_src"); fail_unless (probe->pad); g_mutex_init (&probe->mutex); g_cond_init (&probe->cond); probe->blocked = FALSE; probe->id = gst_pad_add_probe (probe->pad, GST_PAD_PROBE_TYPE_BLOCK | GST_PAD_PROBE_TYPE_BUFFER | GST_PAD_PROBE_TYPE_BUFFER_LIST, on_rtcp_pad_blocked, probe, NULL); g_mutex_lock (&probe->mutex); while (!probe->blocked) { session_harness_crank_clock (h); g_cond_wait (&probe->cond, &probe->mutex); } g_mutex_unlock (&probe->mutex); } static void session_harness_unblock_rtcp (SessionHarness * h, BlockingProbeData * probe) { gst_pad_remove_probe (probe->pad, probe->id); gst_object_unref (probe->pad); g_mutex_clear (&probe->mutex); } GST_START_TEST (test_request_nack_surplus) { SessionHarness *h = session_harness_new (); GstRTCPPacket rtcp_packet; BlockingProbeData probe; GstBuffer *buf; GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT; guint8 *fci_data; gint i; GstStructure *sdes; g_object_set (h->internal_session, "internal-ssrc", 0xDEADBEEF, NULL); /* sdes cname has variable size, fix it */ g_object_get (h->internal_session, "sdes", &sdes, NULL); gst_structure_set (sdes, "cname", G_TYPE_STRING, "user@test", NULL); g_object_set (h->internal_session, "sdes", sdes, NULL); gst_structure_free (sdes); /* Receive a RTP buffer from the wire */ fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtp (h, generate_test_buffer (0, 0x12345678))); /* Block on first regular RTCP so we can fill the nack list */ session_harness_block_rtcp (h, &probe); /* request 400 NACK with 17 seqnum distance to optain the worst possible * packing */ for (i = 0; i < 350; i++) session_harness_rtp_retransmission_request (h, 0x12345678, 1234 + i * 17, 0, 0, 0); /* and the last 50 with a 2s deadline */ for (i = 350; i < 400; i++) session_harness_rtp_retransmission_request (h, 0x12345678, 1234 + i * 17, 0, 2000, 0); /* Unblock and wait for the regular and first early packet */ session_harness_unblock_rtcp (h, &probe); session_harness_produce_rtcp (h, 2); /* Move time forward, so that only the remaining 50 are still up to date */ session_harness_advance_and_crank (h, GST_SECOND); session_harness_produce_rtcp (h, 3); /* ignore the regular RTCP packet */ buf = session_harness_pull_rtcp (h); gst_buffer_unref (buf); /* validate the first early RTCP which should hold 335 Nack */ buf = session_harness_pull_rtcp (h); fail_unless (gst_rtcp_buffer_validate (buf)); gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp); fail_unless_equals_int (3, gst_rtcp_buffer_get_packet_count (&rtcp)); fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet)); /* first a Receiver Report */ fail_unless_equals_int (GST_RTCP_TYPE_RR, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet)); /* then a SDES */ fail_unless_equals_int (GST_RTCP_TYPE_SDES, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet)); /* and then our NACK */ fail_unless_equals_int (GST_RTCP_TYPE_RTPFB, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless_equals_int (GST_RTCP_RTPFB_TYPE_NACK, gst_rtcp_packet_fb_get_type (&rtcp_packet)); fail_unless_equals_int (0xDEADBEEF, gst_rtcp_packet_fb_get_sender_ssrc (&rtcp_packet)); fail_unless_equals_int (0x12345678, gst_rtcp_packet_fb_get_media_ssrc (&rtcp_packet)); fail_unless_equals_int (340, gst_rtcp_packet_fb_get_fci_length (&rtcp_packet)); fci_data = gst_rtcp_packet_fb_get_fci (&rtcp_packet); fail_unless_equals_int (GST_READ_UINT32_BE (fci_data), 1234L << 16); gst_rtcp_buffer_unmap (&rtcp); gst_buffer_unref (buf); /* validate the second early RTCP which should hold 50 Nack */ buf = session_harness_pull_rtcp (h); fail_unless (gst_rtcp_buffer_validate (buf)); gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp); fail_unless_equals_int (3, gst_rtcp_buffer_get_packet_count (&rtcp)); fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet)); /* first a Receiver Report */ fail_unless_equals_int (GST_RTCP_TYPE_RR, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet)); /* then a SDES */ fail_unless_equals_int (GST_RTCP_TYPE_SDES, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet)); /* and then our NACK */ fail_unless_equals_int (GST_RTCP_TYPE_RTPFB, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless_equals_int (GST_RTCP_RTPFB_TYPE_NACK, gst_rtcp_packet_fb_get_type (&rtcp_packet)); fail_unless_equals_int (0xDEADBEEF, gst_rtcp_packet_fb_get_sender_ssrc (&rtcp_packet)); fail_unless_equals_int (0x12345678, gst_rtcp_packet_fb_get_media_ssrc (&rtcp_packet)); fail_unless_equals_int (50, gst_rtcp_packet_fb_get_fci_length (&rtcp_packet)); fci_data = gst_rtcp_packet_fb_get_fci (&rtcp_packet); fail_unless_equals_int (GST_READ_UINT32_BE (fci_data), (guint16) (1234 + 350 * 17) << 16); gst_rtcp_buffer_unmap (&rtcp); gst_buffer_unref (buf); session_harness_free (h); } GST_END_TEST; GST_START_TEST (test_request_nack_packing) { SessionHarness *h = session_harness_new (); GstRTCPPacket rtcp_packet; BlockingProbeData probe; GstBuffer *buf; GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT; guint8 *fci_data; gint i; g_object_set (h->internal_session, "internal-ssrc", 0xDEADBEEF, NULL); /* Receive a RTP buffer from the wire */ fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtp (h, generate_test_buffer (0, 0x12345678))); /* Block on first regular RTCP so we can fill the nack list */ session_harness_block_rtcp (h, &probe); /* append 16 consecutive seqnum */ for (i = 1; i < 17; i++) session_harness_rtp_retransmission_request (h, 0x12345678, 1234 + i, 0, 0, 0); /* prepend one, still consecutive */ session_harness_rtp_retransmission_request (h, 0x12345678, 1234, 0, 0, 0); /* update it */ session_harness_rtp_retransmission_request (h, 0x12345678, 1234, 0, 0, 0); /* Unblock and wait for the regular and first early packet */ session_harness_unblock_rtcp (h, &probe); session_harness_produce_rtcp (h, 2); /* ignore the regular RTCP packet */ buf = session_harness_pull_rtcp (h); gst_buffer_unref (buf); /* validate the early RTCP which should hold 1 Nack */ buf = session_harness_pull_rtcp (h); fail_unless (gst_rtcp_buffer_validate (buf)); gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp); fail_unless_equals_int (3, gst_rtcp_buffer_get_packet_count (&rtcp)); fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet)); /* first a Receiver Report */ fail_unless_equals_int (GST_RTCP_TYPE_RR, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet)); /* then a SDES */ fail_unless_equals_int (GST_RTCP_TYPE_SDES, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet)); /* and then our NACK */ fail_unless_equals_int (GST_RTCP_TYPE_RTPFB, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless_equals_int (GST_RTCP_RTPFB_TYPE_NACK, gst_rtcp_packet_fb_get_type (&rtcp_packet)); fail_unless_equals_int (0xDEADBEEF, gst_rtcp_packet_fb_get_sender_ssrc (&rtcp_packet)); fail_unless_equals_int (0x12345678, gst_rtcp_packet_fb_get_media_ssrc (&rtcp_packet)); fail_unless_equals_int (1, gst_rtcp_packet_fb_get_fci_length (&rtcp_packet)); fci_data = gst_rtcp_packet_fb_get_fci (&rtcp_packet); fail_unless_equals_int (GST_READ_UINT32_BE (fci_data), 1234L << 16 | 0xFFFF); gst_rtcp_buffer_unmap (&rtcp); gst_buffer_unref (buf); session_harness_free (h); } GST_END_TEST; GST_START_TEST (test_disable_sr_timestamp) { SessionHarness *h = session_harness_new (); GstBuffer *buf; GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT; GstRTCPPacket rtcp_packet; guint64 ntptime; guint32 rtptime; g_object_set (h->internal_session, "disable-sr-timestamp", TRUE, NULL); /* Push RTP buffer to make sure RTCP-thread have started */ fail_unless_equals_int (GST_FLOW_OK, session_harness_send_rtp (h, generate_test_buffer (0, 0xDEADBEEF))); /* crank the RTCP-thread and pull out rtcp, generating a stats-callback */ session_harness_crank_clock (h); buf = session_harness_pull_rtcp (h); gst_rtcp_buffer_map (buf, GST_MAP_READWRITE, &rtcp); fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet)); fail_unless_equals_int (GST_RTCP_TYPE_SR, gst_rtcp_packet_get_type (&rtcp_packet)); gst_rtcp_packet_sr_get_sender_info (&rtcp_packet, NULL, &ntptime, &rtptime, NULL, NULL); fail_unless_equals_uint64 (ntptime, 0); fail_unless (rtptime == 0); gst_rtcp_buffer_unmap (&rtcp); gst_buffer_unref (buf); session_harness_free (h); } GST_END_TEST; static guint on_sending_nacks (GObject * internal_session, guint sender_ssrc, guint media_ssrc, GArray * nacks, GstBuffer * buffer) { GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT; GstRTCPPacket packet; guint16 seqnum = g_array_index (nacks, guint16, 0); guint8 *data; if (seqnum == 1235) return 0; fail_unless (gst_rtcp_buffer_map (buffer, GST_MAP_READWRITE, &rtcp)); fail_unless (gst_rtcp_buffer_add_packet (&rtcp, GST_RTCP_TYPE_APP, &packet)); gst_rtcp_packet_app_set_ssrc (&packet, media_ssrc); gst_rtcp_packet_app_set_name (&packet, "TEST"); fail_unless (gst_rtcp_packet_app_set_data_length (&packet, 1)); data = gst_rtcp_packet_app_get_data (&packet); GST_WRITE_UINT32_BE (data, seqnum); gst_rtcp_buffer_unmap (&rtcp); return 1; } GST_START_TEST (test_on_sending_nacks) { SessionHarness *h = session_harness_new (); BlockingProbeData probe; GstBuffer *buf; GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT; GstRTCPPacket rtcp_packet; guint8 *data; g_object_set (h->internal_session, "internal-ssrc", 0xDEADBEEF, NULL); /* Receive a RTP buffer from the wire */ fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtp (h, generate_test_buffer (0, 0x12345678))); /* Block on first regular RTCP so we can fill the nack list */ session_harness_block_rtcp (h, &probe); g_signal_connect (h->internal_session, "on-sending-nacks", G_CALLBACK (on_sending_nacks), NULL); /* request NACK immediately */ session_harness_rtp_retransmission_request (h, 0x12345678, 1234, 0, 0, 0); session_harness_rtp_retransmission_request (h, 0x12345678, 1235, 0, 0, 0); session_harness_unblock_rtcp (h, &probe); gst_buffer_unref (session_harness_pull_rtcp (h)); session_harness_produce_rtcp (h, 2); /* first packet only includes seqnum 1234 in an APP FB */ buf = session_harness_pull_rtcp (h); fail_unless (gst_rtcp_buffer_validate (buf)); gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp); fail_unless_equals_int (3, gst_rtcp_buffer_get_packet_count (&rtcp)); fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet)); /* first a Receiver Report */ fail_unless_equals_int (GST_RTCP_TYPE_RR, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet)); /* then a SDES */ fail_unless_equals_int (GST_RTCP_TYPE_SDES, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet)); /* and then our NACK */ fail_unless_equals_int (GST_RTCP_TYPE_APP, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless_equals_string ("TEST", gst_rtcp_packet_app_get_name (&rtcp_packet)); fail_unless_equals_int (0x12345678, gst_rtcp_packet_app_get_ssrc (&rtcp_packet)); fail_unless_equals_int (1, gst_rtcp_packet_app_get_data_length (&rtcp_packet)); data = gst_rtcp_packet_app_get_data (&rtcp_packet); fail_unless_equals_int (GST_READ_UINT32_BE (data), 1234L); gst_rtcp_buffer_unmap (&rtcp); gst_buffer_unref (buf); /* second will contain seqnum 1235 in a generic nack packet */ buf = session_harness_pull_rtcp (h); fail_unless (gst_rtcp_buffer_validate (buf)); gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp); fail_unless_equals_int (3, gst_rtcp_buffer_get_packet_count (&rtcp)); fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet)); /* first a Receiver Report */ fail_unless_equals_int (GST_RTCP_TYPE_RR, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet)); /* then a SDES */ fail_unless_equals_int (GST_RTCP_TYPE_SDES, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet)); /* and then our NACK */ fail_unless_equals_int (GST_RTCP_TYPE_RTPFB, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless_equals_int (GST_RTCP_RTPFB_TYPE_NACK, gst_rtcp_packet_fb_get_type (&rtcp_packet)); fail_unless_equals_int (0xDEADBEEF, gst_rtcp_packet_fb_get_sender_ssrc (&rtcp_packet)); fail_unless_equals_int (0x12345678, gst_rtcp_packet_fb_get_media_ssrc (&rtcp_packet)); fail_unless_equals_int (1, gst_rtcp_packet_fb_get_fci_length (&rtcp_packet)); data = gst_rtcp_packet_fb_get_fci (&rtcp_packet); fail_unless_equals_int (GST_READ_UINT32_BE (data), 1235L << 16); gst_rtcp_buffer_unmap (&rtcp); gst_buffer_unref (buf); session_harness_free (h); } GST_END_TEST; static void disable_probation_on_new_ssrc (GObject * session, GObject * source) { g_object_set (source, "probation", 0, NULL); } GST_START_TEST (test_disable_probation) { SessionHarness *h = session_harness_new (); g_object_set (h->internal_session, "internal-ssrc", 0xDEADBEEF, NULL); g_signal_connect (h->internal_session, "on-new-ssrc", G_CALLBACK (disable_probation_on_new_ssrc), NULL); /* Receive a RTP buffer from the wire */ fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtp (h, generate_test_buffer (0, 0x12345678))); /* When probation is disabled, the packet should be produced immediately */ fail_unless_equals_int (1, gst_harness_buffers_in_queue (h->recv_rtp_h)); session_harness_free (h); } GST_END_TEST; GST_START_TEST (test_request_late_nack) { SessionHarness *h = session_harness_new (); GstBuffer *buf; GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT; GstRTCPPacket rtcp_packet; guint8 *fci_data; guint32 fci_length; g_object_set (h->internal_session, "internal-ssrc", 0xDEADBEEF, NULL); /* Receive a RTP buffer from the wire */ fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtp (h, generate_test_buffer (0, 0x12345678))); /* Wait for first regular RTCP to be sent so that we are clear to send early RTCP */ session_harness_produce_rtcp (h, 1); gst_buffer_unref (session_harness_pull_rtcp (h)); /* request NACK immediately, but also advance the clock, so the request is * now late, but it should be kept to avoid sending an early rtcp without * NACK. This would otherwise lead to a stall if the late packet was cause * by high RTT, we need to send some RTX in order to update that statistic. */ session_harness_rtp_retransmission_request (h, 0x12345678, 1234, 0, 0, 0); gst_test_clock_advance_time (h->testclock, 100 * GST_USECOND); /* NACK should be produced immediately as early RTCP is allowed. Pull buffer without advancing the clock to ensure this is the case */ buf = session_harness_pull_rtcp (h); fail_unless (gst_rtcp_buffer_validate (buf)); gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp); fail_unless_equals_int (3, gst_rtcp_buffer_get_packet_count (&rtcp)); fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet)); /* first a Receiver Report */ fail_unless_equals_int (GST_RTCP_TYPE_RR, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet)); /* then a SDES */ fail_unless_equals_int (GST_RTCP_TYPE_SDES, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless (gst_rtcp_packet_move_to_next (&rtcp_packet)); /* and then our NACK */ fail_unless_equals_int (GST_RTCP_TYPE_RTPFB, gst_rtcp_packet_get_type (&rtcp_packet)); fail_unless_equals_int (GST_RTCP_RTPFB_TYPE_NACK, gst_rtcp_packet_fb_get_type (&rtcp_packet)); fail_unless_equals_int (0xDEADBEEF, gst_rtcp_packet_fb_get_sender_ssrc (&rtcp_packet)); fail_unless_equals_int (0x12345678, gst_rtcp_packet_fb_get_media_ssrc (&rtcp_packet)); fci_data = gst_rtcp_packet_fb_get_fci (&rtcp_packet); fci_length = gst_rtcp_packet_fb_get_fci_length (&rtcp_packet) * sizeof (guint32); fail_unless_equals_int (4, fci_length); fail_unless_equals_int (GST_READ_UINT32_BE (fci_data), 1234L << 16); gst_rtcp_buffer_unmap (&rtcp); gst_buffer_unref (buf); session_harness_free (h); } GST_END_TEST; static gpointer _push_caps_events (gpointer user_data) { SessionHarness *h = user_data; gint payload = 0; while (h->running) { GstCaps *caps = gst_caps_new_simple ("application/x-rtp", "payload", G_TYPE_INT, payload, NULL); gst_harness_set_src_caps (h->recv_rtp_h, caps); g_thread_yield (); payload++; } return NULL; } GST_START_TEST (test_clear_pt_map_stress) { SessionHarness *h = session_harness_new (); GThread *thread; guint i; h->running = TRUE; thread = g_thread_new (NULL, _push_caps_events, h); for (i = 0; i < 1000; i++) { g_signal_emit_by_name (h->session, "clear-pt-map"); g_thread_yield (); } h->running = FALSE; g_thread_join (thread); session_harness_free (h); } GST_END_TEST; static GstBuffer * generate_stepped_ts_buffer (guint i, gboolean stepped) { GstBuffer *buf; guint ts = (TEST_BUF_CLOCK_RATE * i) / 1000; if (stepped) { const int TEST_BUF_CLOCK_STEP = TEST_BUF_CLOCK_RATE / 30; ts /= TEST_BUF_CLOCK_STEP; ts *= TEST_BUF_CLOCK_STEP; } GST_LOG ("ts: %" GST_TIME_FORMAT " rtp: %u (%" GST_TIME_FORMAT "), seq: %u\n", GST_TIME_ARGS (i * GST_MSECOND), ts, GST_TIME_ARGS (gst_util_uint64_scale_int (GST_SECOND, ts, TEST_BUF_CLOCK_RATE)), i); buf = generate_test_buffer_full (i * GST_MSECOND, i, ts, 0xAAAA, FALSE, 0, 0); return buf; } static void test_packet_rate_impl (gboolean stepped) { SessionHarness *h = session_harness_new (); GstBuffer *buf; guint i; const int PROBATION_CNT = 5; GstStructure *stats; GObject *source; guint pktrate; /* First do probation */ for (i = 0; i < PROBATION_CNT; i++) { buf = generate_stepped_ts_buffer (i, stepped); fail_unless_equals_int (session_harness_recv_rtp (h, buf), GST_FLOW_OK); } for (i = 0; i < PROBATION_CNT; i++) { buf = gst_harness_pull (h->recv_rtp_h); fail_unless (buf); gst_buffer_unref (buf); } /* Now run the real test */ for (i = PROBATION_CNT; i < 10000; i++) { buf = generate_stepped_ts_buffer (i, stepped); fail_unless_equals_int (session_harness_recv_rtp (h, buf), GST_FLOW_OK); buf = gst_harness_pull (h->recv_rtp_h); fail_unless (buf); gst_buffer_unref (buf); } g_signal_emit_by_name (h->internal_session, "get-source-by-ssrc", 0xAAAA, &source); g_object_get (source, "stats", &stats, NULL); fail_unless (gst_structure_get_uint (stats, "recv-packet-rate", &pktrate)); fail_unless (pktrate > 900 && pktrate < 1100); /* Allow 10% of error */ gst_structure_free (stats); g_object_unref (source); session_harness_free (h); } GST_START_TEST (test_packet_rate) { test_packet_rate_impl (FALSE); } GST_END_TEST; GST_START_TEST (test_stepped_packet_rate) { test_packet_rate_impl (TRUE); } GST_END_TEST; /********************* TWCC-tests *********************/ static GstRTCPFBType _gst_buffer_get_rtcp_fbtype (GstBuffer * buf) { GstRTCPFBType ret = GST_RTCP_FB_TYPE_INVALID; GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT; GstRTCPPacket packet; if (!gst_rtcp_buffer_validate_reduced (buf)) return ret; if (!gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp)) return ret; if (!gst_rtcp_buffer_get_first_packet (&rtcp, &packet)) goto done; if (GST_RTCP_TYPE_RTPFB != gst_rtcp_packet_get_type (&packet)) goto done; ret = gst_rtcp_packet_fb_get_type (&packet); done: gst_rtcp_buffer_unmap (&rtcp); return ret; } static GstBuffer * session_harness_pull_twcc_rtcp (SessionHarness * h) { GstBuffer *ret = NULL; while (ret == NULL) { GstBuffer *buf = session_harness_pull_rtcp (h); if (GST_RTCP_RTPFB_TYPE_TWCC == _gst_buffer_get_rtcp_fbtype (buf)) { ret = buf; } else { gst_buffer_unref (buf); } } return ret; } typedef struct { guint16 base_seqnum; guint16 num_packets; GstClockTime base_time; GstClockTime duration; } TWCCTestData; static TWCCTestData twcc_header_and_run_lenght_test_data[] = { {0, 10, 0, 33 * GST_MSECOND}, {65530, 12, 37 * 64 * GST_MSECOND, 10 * GST_MSECOND}, /* seqnum wrap */ {99, 200, 1024 * 64 * GST_MSECOND, 10 * GST_MSECOND}, /* many packets */ {20000, 23, 0, 250 * GST_USECOND}, /* minimal duration */ {56000, 15, 1000 * 64 * GST_MSECOND, 10 * GST_MSECOND}, /* timestamp offset */ }; GST_START_TEST (test_twcc_header_and_run_length) { SessionHarness *h = session_harness_new (); gint i; GstFlowReturn res; GstBuffer *buf; GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT; GstRTCPPacket packet; guint8 *fci_data; guint16 run_length; TWCCTestData *td = &twcc_header_and_run_lenght_test_data[__i__]; /* enable twcc */ session_harness_set_twcc_recv_ext_id (h, TEST_TWCC_EXT_ID); /* receive some buffers */ for (i = 0; i < td->num_packets; i++) { gboolean last_packet = i == (td->num_packets - 1); buf = generate_twcc_recv_buffer (i + td->base_seqnum, td->base_time + i * td->duration, last_packet); res = session_harness_recv_rtp (h, buf); fail_unless_equals_int (GST_FLOW_OK, res); } session_harness_produce_rtcp (h, 1); buf = session_harness_pull_twcc_rtcp (h); fail_unless (buf); gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp); fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &packet)); fci_data = gst_rtcp_packet_fb_get_fci (&packet); /* base seqnum */ fail_unless_equals_int (td->base_seqnum, GST_READ_UINT16_BE (&fci_data[0])); /* packet count */ fail_unless_equals_int (td->num_packets, GST_READ_UINT16_BE (&fci_data[2])); /* reference time (in 64ms units) */ fail_unless_equals_int (td->base_time, GST_READ_UINT24_BE (&fci_data[4]) * 64 * GST_MSECOND); /* feedback packet number */ fail_unless_equals_int (0, fci_data[7]); /* run-length coding */ fail_unless_equals_int (0, fci_data[8] & 0x80); /* status: small-delta */ fail_unless_equals_int (0x20, fci_data[8] & 0x60); /* packets in run_length */ run_length = *((guint16 *) & fci_data[8]); run_length = run_length & ~0xE0; /* mask out the 3 last bits */ fail_unless_equals_int (td->num_packets, GST_READ_UINT16_BE (&run_length)); /* first recv-delta always 0 */ fail_unless_equals_int (0, fci_data[10]); /* following recv-delta equal to duration (in 250us units) */ fail_unless_equals_clocktime (td->duration, fci_data[11] * 250 * GST_USECOND); gst_rtcp_buffer_unmap (&rtcp); gst_buffer_unref (buf); session_harness_free (h); } GST_END_TEST; typedef struct { guint16 seqnum; GstClockTime timestamp; gboolean marker; } TWCCPacket; #define twcc_push_packets(h, packets) \ G_STMT_START { \ guint i; \ session_harness_set_twcc_recv_ext_id ((h), TEST_TWCC_EXT_ID); \ for (i = 0; i < G_N_ELEMENTS ((packets)); i++) { \ TWCCPacket *twcc_pkt = &(packets)[i]; \ fail_unless_equals_int (GST_FLOW_OK, \ session_harness_recv_rtp ((h), \ generate_twcc_recv_buffer (twcc_pkt->seqnum, \ twcc_pkt->timestamp, twcc_pkt->marker))); \ } \ } G_STMT_END #define twcc_verify_fci(buf, exp_fci) \ G_STMT_START { \ GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT; \ GstRTCPPacket packet; \ guint8 *fci_data; \ guint16 fci_length; \ fail_unless (gst_rtcp_buffer_validate_reduced (buf)); \ gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp); \ fail_unless (gst_rtcp_buffer_get_first_packet (&rtcp, &packet)); \ fail_unless_equals_int (GST_RTCP_TYPE_RTPFB, \ gst_rtcp_packet_get_type (&packet)); \ fail_unless_equals_int (GST_RTCP_RTPFB_TYPE_TWCC, \ gst_rtcp_packet_fb_get_type (&packet)); \ fci_data = gst_rtcp_packet_fb_get_fci (&packet); \ fci_length = gst_rtcp_packet_fb_get_fci_length (&packet) * sizeof (guint32); \ fail_unless_equals_int (fci_length, sizeof (exp_fci)); \ fail_unless_equals_int (0, memcmp (fci_data, (exp_fci), fci_length)); \ gst_rtcp_buffer_unmap (&rtcp); \ } G_STMT_END #define twcc_verify_packets_to_fci(h, packets, exp_fci) \ G_STMT_START { \ GstBuffer *buf; \ twcc_push_packets (h, packets); \ session_harness_produce_rtcp ((h), 1); \ buf = session_harness_pull_twcc_rtcp ((h)); \ twcc_verify_fci (buf, exp_fci); \ gst_buffer_unref (buf); \ } G_STMT_END #define twcc_verify_packets_to_event(packets, event) \ G_STMT_START { \ guint i; \ guint j = 0; \ GValueArray *packets_array = g_value_get_boxed ( \ gst_structure_get_value (gst_event_get_structure ((event)), "packets")); \ for (i = 0; i < packets_array->n_values; i++) { \ TWCCPacket *twcc_pkt; \ GstClockTime ts; \ guint seqnum; \ gboolean lost; \ const GstStructure *pkt_s = \ gst_value_get_structure (g_value_array_get_nth (packets_array, i)); \ fail_unless (gst_structure_get_boolean (pkt_s, "lost", &lost)); \ if (lost) \ continue; \ fail_unless (gst_structure_get_clock_time (pkt_s, "remote-ts", &ts)); \ fail_unless (gst_structure_get_uint (pkt_s, "seqnum", &seqnum)); \ twcc_pkt = &(packets)[j++]; \ fail_unless_equals_int (twcc_pkt->seqnum, seqnum); \ fail_unless_equals_clocktime (twcc_pkt->timestamp, ts); \ } \ gst_event_unref (event); \ } G_STMT_END #define twcc_verify_packets_to_packets(send_h, recv_h, packets) \ G_STMT_START { \ guint i; \ GstEvent *event; \ twcc_push_packets ((recv_h), packets); \ session_harness_produce_rtcp ((recv_h), 1); \ session_harness_recv_rtcp ((send_h), \ session_harness_pull_twcc_rtcp ((recv_h))); \ for (i = 0; i < 2; i++) \ gst_event_unref (gst_harness_pull_upstream_event ((send_h)->send_rtp_h)); \ event = gst_harness_pull_upstream_event ((send_h)->send_rtp_h); \ twcc_verify_packets_to_event (packets, event); \ } G_STMT_END GST_START_TEST (test_twcc_1_bit_status_vector) { SessionHarness *h0 = session_harness_new (); SessionHarness *h1 = session_harness_new (); TWCCPacket packets[] = { {10, 0 * GST_MSECOND, FALSE}, {12, 12 * GST_MSECOND, FALSE}, {14, 14 * GST_MSECOND, FALSE}, {15, 15 * GST_MSECOND, FALSE}, {17, 17 * GST_MSECOND, FALSE}, {20, 20 * GST_MSECOND, FALSE}, {21, 21 * GST_MSECOND, FALSE}, {23, 23 * GST_MSECOND, TRUE}, }; guint8 exp_fci[] = { 0x00, 0x0a, /* base sequence number: 10 */ 0x00, 0x0e, /* packet status count: 14 */ 0x00, 0x00, 0x00, /* reference time: 0 */ 0x00, /* feedback packet count: 0 */ 0xab, 0x4d, /* packet chunk: 1 0 1 0 1 0 1 1 | 0 1 0 0 1 1 0 1 */ 0x00, /* recv delta: +0:00:00.000000000 */ 0x30, /* recv delta: +0:00:00.012000000 */ 0x08, /* recv delta: +0:00:00.002000000 */ 0x04, /* recv delta: +0:00:00.001000000 */ 0x08, /* recv delta: +0:00:00.002000000 */ 0x0c, /* recv delta: +0:00:00.003000000 */ 0x04, /* recv delta: +0:00:00.001000000 */ 0x08, /* recv delta: +0:00:00.002000000 */ 0x00, 0x00, /* padding */ }; /* check we get the expected fci */ twcc_verify_packets_to_fci (h0, packets, exp_fci); /* and check we can parse this back to the original packets */ twcc_verify_packets_to_packets (h1, h1, packets); session_harness_free (h0); session_harness_free (h1); } GST_END_TEST; GST_START_TEST (test_twcc_2_bit_status_vector) { SessionHarness *h0 = session_harness_new (); SessionHarness *h1 = session_harness_new (); TWCCPacket packets[] = { {5, 5 * 64 * GST_MSECOND, FALSE}, {7, 7 * 64 * GST_MSECOND, FALSE}, {8, 8 * 64 * GST_MSECOND, FALSE}, {11, 12 * 64 * GST_MSECOND, TRUE}, }; guint8 exp_fci[] = { 0x00, 0x05, /* base sequence number: 5 */ 0x00, 0x07, /* packet status count: 7 */ 0x00, 0x00, 0x05, /* reference time: 5 */ 0x00, /* feedback packet count: 0 */ 0xd2, 0x82, /* packet chunk: 1 1 0 1 0 0 1 0 | 1 0 0 0 0 0 1 0 */ /* normal, missing, large, large, missing, missing, large */ 0x00, /* recv delta: +0:00:00.000000000 */ 0x02, 0x00, /* recv delta: +0:00:00.128000000 */ 0x01, 0x00, /* recv delta: +0:00:00.064000000 */ 0x04, 0x00, /* recv delta: +0:00:00.256000000 */ 0x00, 0x00, 0x00, /* padding */ }; twcc_verify_packets_to_fci (h0, packets, exp_fci); twcc_verify_packets_to_packets (h1, h1, packets); session_harness_free (h0); session_harness_free (h1); } GST_END_TEST; GST_START_TEST (test_twcc_various_gaps) { SessionHarness *h = session_harness_new (); guint16 seq = 1 + __i__; TWCCPacket packets[] = { {0, 0 * 250 * GST_USECOND, FALSE}, {seq, seq * 250 * GST_USECOND, TRUE}, }; twcc_verify_packets_to_packets (h, h, packets); session_harness_free (h); } GST_END_TEST; GST_START_TEST (test_twcc_negative_delta) { SessionHarness *h0 = session_harness_new (); SessionHarness *h1 = session_harness_new (); TWCCPacket packets[] = { {0, 0 * 250 * GST_USECOND, FALSE}, {1, 2 * 250 * GST_USECOND, FALSE}, {2, 1 * 250 * GST_USECOND, FALSE}, {3, 3 * 250 * GST_USECOND, TRUE}, }; guint8 exp_fci[] = { 0x00, 0x00, /* base sequence number: 0 */ 0x00, 0x04, /* packet status count: 4 */ 0x00, 0x00, 0x00, /* reference time: 0 */ 0x00, /* feedback packet count: 0 */ 0xd6, 0x40, /* packet chunk: 1 1 0 1 0 1 1 0 | 0 1 0 0 0 0 0 0 */ 0x00, /* recv delta: +0:00:00.000000000 */ 0x02, /* recv delta: +0:00:00.000500000 */ 0xff, 0xff, /* recv delta: -0:00:00.000250000 */ 0x02, /* recv delta: +0:00:00.000500000 */ 0x00, /* padding */ }; twcc_verify_packets_to_fci (h0, packets, exp_fci); twcc_verify_packets_to_packets (h1, h1, packets); session_harness_free (h0); session_harness_free (h1); } GST_END_TEST; GST_START_TEST (test_twcc_seqnum_wrap) { SessionHarness *h0 = session_harness_new (); SessionHarness *h1 = session_harness_new (); TWCCPacket packets[] = { {65534, 0 * 250 * GST_USECOND, FALSE}, {65535, 1 * 250 * GST_USECOND, FALSE}, {0, 2 * 250 * GST_USECOND, FALSE}, {1, 3 * 250 * GST_USECOND, TRUE}, }; guint8 exp_fci[] = { 0xff, 0xfe, /* base sequence number: 65534 */ 0x00, 0x04, /* packet status count: 4 */ 0x00, 0x00, 0x00, /* reference time: 0 */ 0x00, /* feedback packet count: 0 */ 0x20, 0x04, /* packet chunk: 0 0 1 0 0 0 0 0 | 0 0 0 0 0 1 0 0 */ 0x00, /* recv delta: +0:00:00.000000000 */ 0x01, /* recv delta: +0:00:00.000250000 */ 0x01, /* recv delta: +0:00:00.000250000 */ 0x01, /* recv delta: +0:00:00.000250000 */ 0x00, 0x00, /* padding */ }; twcc_verify_packets_to_fci (h0, packets, exp_fci); twcc_verify_packets_to_packets (h1, h1, packets); session_harness_free (h0); session_harness_free (h1); } GST_END_TEST; GST_START_TEST (test_twcc_double_packets) { SessionHarness *h = session_harness_new (); TWCCPacket packets0[] = { {11, 11 * GST_MSECOND, FALSE}, {12, 12 * GST_MSECOND, TRUE}, }; TWCCPacket packets1[] = { {13, 13 * GST_MSECOND, FALSE}, {14, 14 * GST_MSECOND, FALSE}, {15, 15 * GST_MSECOND, TRUE}, }; guint8 exp_fci0[] = { 0x00, 0x0b, /* base sequence number: 11 */ 0x00, 0x02, /* packet status count: 14 */ 0x00, 0x00, 0x00, /* reference time: 0 */ 0x00, /* feedback packet count: 0 */ 0x20, 0x02, /* packet chunk: 0 0 1 0 0 0 0 0 | 0 0 0 0 0 0 1 0 */ 0x2c, 0x04, /* recv deltas */ }; guint8 exp_fci1[] = { 0x00, 0x0d, /* base sequence number: 13 */ 0x00, 0x03, /* packet status count: 3 */ 0x00, 0x00, 0x00, /* reference time: 0 */ 0x01, /* feedback packet count: 1 */ 0x20, 0x03, /* packet chunk: 0 0 1 0 0 0 0 0 | 0 0 0 0 0 0 1 1 */ 0x34, 0x04, 0x04, /* recv deltas */ 0x00, 0x00, 0x00, /* padding */ }; twcc_verify_packets_to_fci (h, packets0, exp_fci0); twcc_verify_packets_to_fci (h, packets1, exp_fci1); session_harness_free (h); } GST_END_TEST; GST_START_TEST (test_twcc_huge_seqnum_gap) { SessionHarness *h = session_harness_new (); GstBuffer *buf; TWCCPacket packets[] = { {9, 4 * 32 * GST_MSECOND, FALSE}, {10, 5 * 32 * GST_MSECOND, FALSE}, {30011, 6 * 32 * GST_MSECOND, FALSE}, {30012, 7 * 32 * GST_MSECOND, FALSE}, {30013, 8 * 32 * GST_MSECOND, TRUE}, }; guint8 exp_fci0[] = { 0x00, 0x09, /* base sequence number: 9 */ 0x00, 0x02, /* packet status count: 2 */ 0x00, 0x00, 0x02, /* reference time: 2 * 64ms */ 0x00, /* feedback packet count: 0 */ /* packet chunks: */ 0x20, 0x02, /* 0 0 1 0 0 0 0 0 | 0 0 0 0 0 0 1 0 */ 0x00, 0x80, /* recv deltas, +0, +32ms */ }; guint8 exp_fci1[] = { 0x75, 0x3b, /* base sequence number: 300011 */ 0x00, 0x03, /* packet status count: 3 */ 0x00, 0x00, 0x03, /* reference time: 3 * 64ms */ 0x01, /* feedback packet count: 1 */ /* packet chunks: */ 0x20, 0x03, /* 0 0 1 0 0 0 0 0 | 0 0 0 0 0 0 1 1 */ 0x00, 0x80, 0x80, /* recv deltas, +4, +32ms, +32ms */ 0x00, 0x00, 0x00, /* padding */ }; /* The sequence-number does a huge leap. Instead of encoding this as a massive run-lenght sequence, like so */ #if 0 guint8 exp_fci[] = { 0x00, 0x09, /* base sequence number: 9 */ 0x75, 0x35, /* packet status count: 30005 */ 0x00, 0x00, 0x02, /* reference time: 2 */ 0x00, /* feedback packet count: 0 */ /* packet chunks: */ 0xb0, 0x00, /* 1 bit 2 there, 12 lost: 1 0 1 1 0 0 0 0 | 0 0 0 0 0 0 0 0 */ 0x1f, 0xff, /* run-length: 8191 lost: 0 0 0 1 1 1 1 1 | 1 1 1 1 1 1 1 1 */ 0x1f, 0xff, /* run-length: 8191 lost: 0 0 0 1 1 1 1 1 | 1 1 1 1 1 1 1 1 */ 0x1f, 0xff, /* run-length: 8191 lost: 0 0 0 1 1 1 1 1 | 1 1 1 1 1 1 1 1 */ 0x15, 0x27, /* run-length: 5415 lost: 0 0 0 1 0 1 0 1 | 0 0 1 0 0 1 1 1 */ /* 12 + 8191 + 8191 + 8191 + 5415 = 30000 lost packets */ 0xb8, 0x00, /* 1 bit 3 there : 1 0 1 1 1 0 0 0 | 0 0 0 0 0 0 0 0 */ 0x00, 0x80, 0x80, 0x80, 0x80, /* recv deltas */ 0x00, 0x00, 0x00, /* padding */ }; #endif /* ...just send two feedback-packets, with the second one starting from the new sequence-number. */ twcc_push_packets (h, packets); session_harness_produce_rtcp (h, 1); buf = session_harness_pull_twcc_rtcp (h); twcc_verify_fci (buf, exp_fci0); gst_buffer_unref (buf); buf = session_harness_pull_twcc_rtcp (h); twcc_verify_fci (buf, exp_fci1); gst_buffer_unref (buf); session_harness_free (h); } GST_END_TEST; GST_START_TEST (test_twcc_duplicate_seqnums) { SessionHarness *h = session_harness_new (); GstBuffer *buf; /* A duplicate seqnum can be interpreted as a gap of 65536 packets. Whatever the cause might be, we will follow the behavior of reordered packets, and drop it */ TWCCPacket packets[] = { {1, 4 * 32 * GST_MSECOND, FALSE}, {2, 5 * 32 * GST_MSECOND, FALSE}, {2, 6 * 32 * GST_MSECOND, FALSE}, {3, 7 * 32 * GST_MSECOND, TRUE}, }; guint8 exp_fci0[] = { 0x00, 0x01, /* base sequence number: 1 */ 0x00, 0x02, /* packet status count: 2 */ 0x00, 0x00, 0x02, /* reference time: 2 * 64ms */ 0x00, /* feedback packet count: 0 */ /* packet chunks: */ 0x20, 0x02, /* 0 0 1 0 0 0 0 0 | 0 0 0 0 0 0 1 0 */ 0x00, 0x80, /* recv deltas: +0, +32ms */ }; guint8 exp_fci1[] = { 0x00, 0x03, /* base sequence number: 3 */ 0x00, 0x01, /* packet status count: 1 */ 0x00, 0x00, 0x03, /* reference time: 3 * 64ms */ 0x01, /* feedback packet count: 1 */ /* packet chunks: */ 0x20, 0x01, /* 0 0 1 0 0 0 0 0 | 0 0 0 0 0 0 0 1 */ 0x80, /* recv deltas: +32ms */ 0x00, /* padding */ }; twcc_push_packets (h, packets); session_harness_produce_rtcp (h, 1); buf = session_harness_pull_twcc_rtcp (h); twcc_verify_fci (buf, exp_fci0); gst_buffer_unref (buf); buf = session_harness_pull_twcc_rtcp (h); twcc_verify_fci (buf, exp_fci1); gst_buffer_unref (buf); session_harness_free (h); } GST_END_TEST; GST_START_TEST (test_twcc_multiple_markers) { SessionHarness *h = session_harness_new (); GstBuffer *buf; /* for this test, notice how the first recv-delta should relate back to the reference-time, which is 0 in this case. The packets are incrementing in timestamps equal to the smallest unit for TWCC (250 microseconds) */ TWCCPacket packets[] = { {1, 1 * 250 * GST_USECOND, FALSE}, {2, 2 * 250 * GST_USECOND, FALSE}, {3, 3 * 250 * GST_USECOND, TRUE}, {4, 4 * 250 * GST_USECOND, FALSE}, {5, 5 * 250 * GST_USECOND, TRUE}, {6, 6 * 250 * GST_USECOND, FALSE}, {7, 7 * 250 * GST_USECOND, FALSE}, {8, 8 * 250 * GST_USECOND, FALSE}, {9, 9 * 250 * GST_USECOND, TRUE}, }; guint8 exp_fci0[] = { 0x00, 0x01, /* base sequence number: 1 */ 0x00, 0x03, /* packet status count: 3 */ 0x00, 0x00, 0x00, /* reference time: 0 */ 0x00, /* feedback packet count: 0 */ /* packet chunks: */ 0x20, 0x03, /* 0 0 1 0 0 0 0 0 | 0 0 0 0 0 0 1 1 */ 0x01, 0x01, 0x01, /* recv deltas, +1, +1, +1 */ 0x00, 0x00, 0x00, /* padding */ }; guint8 exp_fci1[] = { 0x00, 0x04, /* base sequence number: 4 */ 0x00, 0x02, /* packet status count: 2 */ 0x00, 0x00, 0x00, /* reference time: 0 */ 0x01, /* feedback packet count: 1 */ /* packet chunks: */ 0x20, 0x02, /* 0 0 1 0 0 0 0 0 | 0 0 0 0 0 0 1 0 */ 0x04, 0x01, /* recv deltas, +4, +1, +1 */ }; guint8 exp_fci2[] = { 0x00, 0x06, /* base sequence number: 6 */ 0x00, 0x04, /* packet status count: 4 */ 0x00, 0x00, 0x00, /* reference time: 0 */ 0x02, /* feedback packet count: 2 */ /* packet chunks: */ 0x20, 0x04, /* 0 0 1 0 0 0 0 0 | 0 0 0 0 0 1 0 0 */ 0x06, 0x01, 0x01, 0x01, /* recv deltas, +6, +1, +1, +1 */ 0x00, 0x00, }; twcc_push_packets (h, packets); /* we should get 1 SR/RR, and then 3x TWCC packets */ session_harness_produce_rtcp (h, 1); buf = session_harness_pull_twcc_rtcp (h); twcc_verify_fci (buf, exp_fci0); gst_buffer_unref (buf); buf = session_harness_pull_twcc_rtcp (h); twcc_verify_fci (buf, exp_fci1); gst_buffer_unref (buf); buf = session_harness_pull_twcc_rtcp (h); twcc_verify_fci (buf, exp_fci2); gst_buffer_unref (buf); session_harness_free (h); } GST_END_TEST; GST_START_TEST (test_twcc_no_marker_and_gaps) { SessionHarness *h = session_harness_new (); guint i; g_object_set (h->internal_session, "probation", 1, NULL); /* Push packets with gaps and no marker bit. This should not prevent the feedback packets from being sent at all. */ for (i = 0; i < 80; i += 10) { TWCCPacket packets[] = { {i, i * 250 * GST_USECOND, FALSE} }; twcc_push_packets (h, packets); } /* verify we did receive some feedback for these packets */ session_harness_produce_rtcp (h, 1); for (i = 0; i < 2; i++) { gst_buffer_unref (session_harness_pull_twcc_rtcp (h)); } session_harness_free (h); } GST_END_TEST; static GstBuffer * generate_twcc_feedback_rtcp (guint8 * fci_data, guint16 fci_length) { GstRTCPPacket packet; GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT; GstBuffer *buffer = gst_rtcp_buffer_new (1000); guint8 *fci; fail_unless (gst_rtcp_buffer_map (buffer, GST_MAP_READWRITE, &rtcp)); fail_unless (gst_rtcp_buffer_add_packet (&rtcp, GST_RTCP_TYPE_RTPFB, &packet)); gst_rtcp_packet_fb_set_type (&packet, GST_RTCP_RTPFB_TYPE_TWCC); gst_rtcp_packet_fb_set_fci_length (&packet, fci_length); fci = gst_rtcp_packet_fb_get_fci (&packet); memcpy (fci, fci_data, fci_length); gst_rtcp_packet_fb_set_sender_ssrc (&packet, TEST_BUF_SSRC); gst_rtcp_packet_fb_set_media_ssrc (&packet, 0); gst_rtcp_buffer_unmap (&rtcp); return buffer; } GST_START_TEST (test_twcc_bad_rtcp) { SessionHarness *h = session_harness_new (); guint i; GstBuffer *buf; GstEvent *event; GValueArray *packets_array; guint8 fci[] = { 0xff, 0xff, /* base sequence number: max */ 0xff, 0xff, /* packet status count: max */ 0xff, 0xff, 0xff, /* reference time: max */ 0xff, /* feedback packet count: max */ 0x3f, 0xff, /* packet chunk: run-length, max */ 0x00, /* only 1 recv-delta */ }; buf = generate_twcc_feedback_rtcp (fci, sizeof (fci)); session_harness_recv_rtcp (h, buf); /* two reconfigure events */ for (i = 0; i < 2; i++) gst_event_unref (gst_harness_pull_upstream_event (h->send_rtp_h)); event = gst_harness_pull_upstream_event (h->send_rtp_h); packets_array = g_value_get_boxed (gst_structure_get_value (gst_event_get_structure (event), "packets")); /* this ends up with 0 packets, due to completely invalid data */ fail_unless_equals_int (packets_array->n_values, 0); gst_event_unref (event); session_harness_free (h); } GST_END_TEST; GST_START_TEST (test_twcc_delta_ts_rounding) { SessionHarness *h = session_harness_new (); guint i, j = 0; GstEvent *event; GstBuffer *buf; GValueArray *packets_array; TWCCPacket packets[] = { {2002, 9 * GST_SECOND + 366458177, FALSE} , {2003, 9 * GST_SECOND + 366497068, FALSE} , {2017, 9 * GST_SECOND + 366929482, FALSE} , {2019, 9 * GST_SECOND + 391595309, FALSE} , {2020, 9 * GST_SECOND + 426883507, FALSE} , {2025, 9 * GST_SECOND + 427021638, TRUE} , }; TWCCPacket exp_packets[] = { {2002, 9 * GST_SECOND + 366250000, FALSE} , {2003, 9 * GST_SECOND + 366250000, FALSE} , {2017, 9 * GST_SECOND + 366750000, FALSE} , {2019, 9 * GST_SECOND + 391500000, FALSE} , {2020, 9 * GST_SECOND + 426750000, FALSE} , {2025, 9 * GST_SECOND + 427000000, TRUE} , }; guint8 exp_fci[] = { 0x07, 0xd2, /* base sequence number: 2002 */ 0x00, 0x18, /* packet status count: 24 */ 0x00, 0x00, 0x92, /* reference time: 0:00:09.344000000 */ 0x00, /* feedback packet count: 0 */ 0xb0, 0x00, /* packet chunk: 1 0 1 1 0 0 0 0 | 0 0 0 0 0 0 0 0 */ 0x96, 0x10, /* packet chunk: 1 0 0 1 0 1 1 0 | 0 0 0 1 0 0 0 0 */ 0x59, /* recv delta: 0:00:00.022250000 abs: 0:00:09.366250000 */ 0x00, /* recv delta: 0:00:00.000000000 abs: 0:00:09.366250000 */ 0x02, /* recv delta: 0:00:00.000500000 abs: 0:00:09.366750000 */ 0x63, /* recv delta: 0:00:00.024750000 abs: 0:00:09.391500000 */ 0x8d, /* recv delta: 0:00:00.035250000 abs: 0:00:09.426750000 */ 0x01, /* recv delta: 0:00:00.000250000 abs: 0:00:09.427000000 */ 0x00, 0x00, /* padding */ }; twcc_push_packets (h, packets); session_harness_produce_rtcp (h, 1); buf = session_harness_pull_twcc_rtcp (h); twcc_verify_fci (buf, exp_fci); session_harness_recv_rtcp (h, buf); for (i = 0; i < 2; i++) gst_event_unref (gst_harness_pull_upstream_event (h->send_rtp_h)); event = gst_harness_pull_upstream_event (h->send_rtp_h); packets_array = g_value_get_boxed (gst_structure_get_value (gst_event_get_structure (event), "packets")); for (i = 0; i < packets_array->n_values; i++) { TWCCPacket *twcc_pkt; const GstStructure *pkt_s = gst_value_get_structure (g_value_array_get_nth (packets_array, i)); GstClockTime ts; guint seqnum; gboolean lost; fail_unless (gst_structure_get_boolean (pkt_s, "lost", &lost)); if (lost) continue; twcc_pkt = &exp_packets[j++]; fail_unless (gst_structure_get_clock_time (pkt_s, "remote-ts", &ts)); fail_unless (gst_structure_get_uint (pkt_s, "seqnum", &seqnum)); fail_unless_equals_int (twcc_pkt->seqnum, seqnum); fail_unless_equals_clocktime (twcc_pkt->timestamp, ts); } gst_event_unref (event); session_harness_free (h); } GST_END_TEST; GST_START_TEST (test_twcc_double_gap) { SessionHarness *h0 = session_harness_new (); SessionHarness *h1 = session_harness_new (); TWCCPacket packets[] = { {1202, 5 * GST_SECOND + 717000000, FALSE} , {1215, 5 * GST_SECOND + 760250000, FALSE} , {1221, 5 * GST_SECOND + 775500000, TRUE} , }; guint8 exp_fci[] = { 0x04, 0xb2, /* base sequence number: 1202 */ 0x00, 0x14, /* packet status count: 20 */ 0x00, 0x00, 0x59, /* reference time: 0:00:05.696000000 */ 0x00, /* feedback packet count: 0 */ 0xa0, 0x01, /* packet chunk: 1 0 1 0 0 0 0 0 | 0 0 0 0 0 0 0 1 */ 0x81, 0x00, /* packet chunk: 1 0 0 0 0 0 0 1 | 0 0 0 0 0 0 0 0 */ 0x54, /* recv delta: +0:00:00.021000000 */ 0xad, /* recv delta: +0:00:00.043250000 */ 0x3d, /* recv delta: +0:00:00.015250000 */ 0x00, /* padding */ }; twcc_verify_packets_to_fci (h0, packets, exp_fci); twcc_verify_packets_to_packets (h1, h1, packets); session_harness_free (h0); session_harness_free (h1); } GST_END_TEST; GST_START_TEST (test_twcc_recv_packets_reordered) { SessionHarness *h = session_harness_new (); GstBuffer *buf; /* a reordered seqence, with marker-bits for #3 and #4 */ TWCCPacket packets[] = { {1, 1 * 250 * GST_USECOND, FALSE} , {3, 2 * 250 * GST_USECOND, TRUE} , {2, 3 * 250 * GST_USECOND, FALSE} , {4, 4 * 250 * GST_USECOND, TRUE} , }; /* first we expect #2 to be reported lost */ guint8 exp_fci0[] = { 0x00, 0x01, /* base sequence number: 1 */ 0x00, 0x03, /* packet status count: 3 */ 0x00, 0x00, 0x00, /* reference time: 0 */ 0x00, /* feedback packet count: 0 */ /* packet chunks: */ 0xa8, 0x00, /* 1 0 1 0 1 0 0 0 | 0 0 0 0 0 0 0 0 */ 0x01, 0x01, /* recv deltas, +1, +1 */ }; /* and then when 2 actually arrives, it is already reported lost, so we will not re-report it, but drop it */ guint8 exp_fci1[] = { 0x00, 0x04, /* base sequence number: 4 */ 0x00, 0x01, /* packet status count: 1 */ 0x00, 0x00, 0x00, /* reference time: 0 */ 0x01, /* feedback packet count: 1 */ /* packet chunks: */ 0x20, 0x01, /* 0 0 1 0 0 0 0 0 | 0 0 0 0 0 0 0 1 */ 0x04, /* recv deltas, +4 */ 0x00, /* padding */ }; twcc_push_packets (h, packets); session_harness_produce_rtcp (h, 1); buf = session_harness_pull_twcc_rtcp (h); twcc_verify_fci (buf, exp_fci0); gst_buffer_unref (buf); buf = session_harness_pull_twcc_rtcp (h); twcc_verify_fci (buf, exp_fci1); gst_buffer_unref (buf); session_harness_free (h); } GST_END_TEST; GST_START_TEST (test_twcc_recv_rtcp_reordered) { SessionHarness *send_h = session_harness_new (); SessionHarness *recv_h = session_harness_new (); GstBuffer *buf[4]; GstEvent *event; guint i; /* three frames, two packets each */ TWCCPacket packets[] = { {1, 1 * GST_SECOND, FALSE} , {2, 2 * GST_SECOND, TRUE} , {3, 3 * GST_SECOND, FALSE} , {4, 4 * GST_SECOND, TRUE} , {5, 5 * GST_SECOND, FALSE} , {6, 6 * GST_SECOND, TRUE} , {7, 7 * GST_SECOND, FALSE} , {8, 8 * GST_SECOND, TRUE} , }; /* TWCCPacket expected_packets0[] = { {1, 1 * 250 * GST_USECOND, FALSE}, {2, 2 * 250 * GST_USECOND, TRUE}, }; */ twcc_push_packets (recv_h, packets); session_harness_produce_rtcp (recv_h, 1); buf[0] = session_harness_pull_twcc_rtcp (recv_h); buf[1] = session_harness_pull_twcc_rtcp (recv_h); buf[2] = session_harness_pull_twcc_rtcp (recv_h); buf[3] = session_harness_pull_twcc_rtcp (recv_h); /* reorder the twcc-feedback */ session_harness_recv_rtcp (send_h, buf[0]); session_harness_recv_rtcp (send_h, buf[2]); session_harness_recv_rtcp (send_h, buf[1]); session_harness_recv_rtcp (send_h, buf[3]); for (i = 0; i < 2; i++) gst_event_unref (gst_harness_pull_upstream_event (send_h->send_rtp_h)); event = gst_harness_pull_upstream_event (send_h->send_rtp_h); twcc_verify_packets_to_event (&packets[0 * 2], event); event = gst_harness_pull_upstream_event (send_h->send_rtp_h); twcc_verify_packets_to_event (&packets[2 * 2], event); event = gst_harness_pull_upstream_event (send_h->send_rtp_h); twcc_verify_packets_to_event (&packets[1 * 2], event); event = gst_harness_pull_upstream_event (send_h->send_rtp_h); twcc_verify_packets_to_event (&packets[3 * 2], event); session_harness_free (send_h); session_harness_free (recv_h); } GST_END_TEST; GST_START_TEST (test_twcc_no_exthdr_in_buffer) { SessionHarness *h = session_harness_new (); session_harness_set_twcc_recv_ext_id (h, TEST_TWCC_EXT_ID); fail_unless_equals_int (GST_FLOW_OK, session_harness_recv_rtp (h, generate_test_buffer (0, 1234))); session_harness_free (h); } GST_END_TEST; GST_START_TEST (test_twcc_send_and_recv) { SessionHarness *h_send = session_harness_new (); SessionHarness *h_recv = session_harness_new (); guint frame; const guint num_frames = 2; const guint num_slices = 15; /* enable twcc */ session_harness_set_twcc_recv_ext_id (h_recv, TEST_TWCC_EXT_ID); session_harness_set_twcc_send_ext_id (h_send, TEST_TWCC_EXT_ID); for (frame = 0; frame < num_frames; frame++) { GstBuffer *buf; for (guint slice = 0; slice < num_slices; slice++) { GstFlowReturn res; guint seq = frame * num_slices + slice; /* from payloder to rtpbin */ buf = generate_twcc_send_buffer (seq, slice == num_slices - 1); res = session_harness_send_rtp (h_send, buf); fail_unless_equals_int (GST_FLOW_OK, res); /* get the buffer ready for the network */ buf = session_harness_pull_send_rtp (h_send); /* buffer arrives at the receiver */ res = session_harness_recv_rtp (h_recv, buf); fail_unless_equals_int (GST_FLOW_OK, res); } /* receiver sends a TWCC packet to the sender */ session_harness_produce_rtcp (h_recv, 1); buf = session_harness_pull_twcc_rtcp (h_recv); /* sender receives the TWCC packet */ session_harness_recv_rtcp (h_send, buf); if (frame > 0) { GstStructure *twcc_stats; guint bitrate_sent; guint bitrate_recv; guint packets_sent; guint packets_recv; gdouble packet_loss_pct; GstClockTimeDiff avg_delta_of_delta; twcc_stats = session_harness_get_last_twcc_stats (h_send); fail_unless (gst_structure_get (twcc_stats, "bitrate-sent", G_TYPE_UINT, &bitrate_sent, "bitrate-recv", G_TYPE_UINT, &bitrate_recv, "packets-sent", G_TYPE_UINT, &packets_sent, "packets-recv", G_TYPE_UINT, &packets_recv, "packet-loss-pct", G_TYPE_DOUBLE, &packet_loss_pct, "avg-delta-of-delta", G_TYPE_INT64, &avg_delta_of_delta, NULL)); fail_unless_equals_int (TEST_BUF_BPS, bitrate_sent); fail_unless_equals_int (TEST_BUF_BPS, bitrate_recv); fail_unless_equals_int (num_slices, packets_sent); fail_unless_equals_int (num_slices, packets_recv); fail_unless_equals_float (0.0f, packet_loss_pct); fail_unless_equals_int64 (0, avg_delta_of_delta); gst_structure_free (twcc_stats); } } session_harness_free (h_send); session_harness_free (h_recv); } GST_END_TEST; static Suite * rtpsession_suite (void) { Suite *s = suite_create ("rtpsession"); TCase *tc_chain = tcase_create ("general"); suite_add_tcase (s, tc_chain); tcase_add_test (tc_chain, test_multiple_ssrc_rr); tcase_add_test (tc_chain, test_multiple_senders_roundrobin_rbs); tcase_add_test (tc_chain, test_no_rbs_for_internal_senders); tcase_add_test (tc_chain, test_internal_sources_timeout); tcase_add_test (tc_chain, test_receive_rtcp_app_packet); tcase_add_test (tc_chain, test_dont_lock_on_stats); tcase_add_test (tc_chain, test_ignore_suspicious_bye); tcase_add_test (tc_chain, test_ssrc_collision_when_sending); tcase_add_test (tc_chain, test_ssrc_collision_when_sending_loopback); tcase_add_test (tc_chain, test_ssrc_collision_when_receiving); tcase_add_test (tc_chain, test_ssrc_collision_third_party); tcase_add_test (tc_chain, test_ssrc_collision_third_party_favor_new); tcase_add_test (tc_chain, test_request_fir); tcase_add_test (tc_chain, test_request_pli); tcase_add_test (tc_chain, test_request_fir_after_pli_in_caps); tcase_add_test (tc_chain, test_request_nack); tcase_add_test (tc_chain, test_request_nack_surplus); tcase_add_test (tc_chain, test_request_nack_packing); tcase_add_test (tc_chain, test_illegal_rtcp_fb_packet); tcase_add_test (tc_chain, test_feedback_rtcp_race); tcase_add_test (tc_chain, test_receive_regular_pli); tcase_add_test (tc_chain, test_receive_pli_no_sender_ssrc); tcase_add_test (tc_chain, test_dont_send_rtcp_while_idle); tcase_add_test (tc_chain, test_send_rtcp_when_signalled); tcase_add_test (tc_chain, test_change_sent_sdes); tcase_add_test (tc_chain, test_disable_sr_timestamp); tcase_add_test (tc_chain, test_on_sending_nacks); tcase_add_test (tc_chain, test_disable_probation); tcase_add_test (tc_chain, test_request_late_nack); tcase_add_test (tc_chain, test_clear_pt_map_stress); tcase_add_test (tc_chain, test_packet_rate); tcase_add_test (tc_chain, test_stepped_packet_rate); /* twcc */ tcase_add_loop_test (tc_chain, test_twcc_header_and_run_length, 0, G_N_ELEMENTS (twcc_header_and_run_lenght_test_data)); tcase_add_test (tc_chain, test_twcc_1_bit_status_vector); tcase_add_test (tc_chain, test_twcc_2_bit_status_vector); tcase_add_loop_test (tc_chain, test_twcc_various_gaps, 0, 50); tcase_add_test (tc_chain, test_twcc_negative_delta); tcase_add_test (tc_chain, test_twcc_seqnum_wrap); tcase_add_test (tc_chain, test_twcc_huge_seqnum_gap); tcase_add_test (tc_chain, test_twcc_double_packets); tcase_add_test (tc_chain, test_twcc_duplicate_seqnums); tcase_add_test (tc_chain, test_twcc_multiple_markers); tcase_add_test (tc_chain, test_twcc_no_marker_and_gaps); tcase_add_test (tc_chain, test_twcc_bad_rtcp); tcase_add_test (tc_chain, test_twcc_delta_ts_rounding); tcase_add_test (tc_chain, test_twcc_double_gap); tcase_add_test (tc_chain, test_twcc_recv_packets_reordered); tcase_add_test (tc_chain, test_twcc_recv_rtcp_reordered); tcase_add_test (tc_chain, test_twcc_no_exthdr_in_buffer); tcase_add_test (tc_chain, test_twcc_send_and_recv); return s; } GST_CHECK_MAIN (rtpsession);