diff --git a/tests/check/elements/ristrtpext.c b/tests/check/elements/ristrtpext.c new file mode 100644 index 0000000000..3022c8deb0 --- /dev/null +++ b/tests/check/elements/ristrtpext.c @@ -0,0 +1,1084 @@ +/* + * rtpristext.c + * + * Copyright (C) 2019 Net Insight AB + * Author: Olivier Crete + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 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 + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this library; if not, see . + */ + +#include +#include + +static const guint8 ts_packet[] = { + 0x47, 0x40, 0x41, 0x12, 0x00, 0x00, 0x01, 0xe0, 0x0f, 0x96, 0x81, 0xc0, + 0x0a, 0x31, 0x4d, 0x41, 0x0f, 0xbf, 0x11, 0x4d, 0x3f, 0x9a, 0x93, 0x00, + 0x00, 0x00, 0x01, 0x09, 0x30, 0x00, 0x00, 0x01, 0x41, 0x9a, 0x24, 0x6c, + 0x41, 0xaf, 0xfe, 0xda, 0xa6, 0x58, 0x00, 0x09, 0xcf, 0x64, 0x41, 0xf5, + 0x7c, 0x67, 0x65, 0x1d, 0x00, 0x23, 0xd3, 0x7c, 0xf2, 0xd2, 0xf8, 0x2f, + 0x30, 0x20, 0xfe, 0x2b, 0xad, 0x61, 0x0b, 0xd4, 0x47, 0x22, 0x82, 0x2a, + 0x46, 0xe2, 0xc3, 0x4c, 0x6a, 0xb4, 0x1d, 0x07, 0xc9, 0x77, 0x6c, 0xc9, + 0xc3, 0x6d, 0x37, 0x14, 0x86, 0x45, 0xb1, 0x0b, 0x44, 0xc4, 0xee, 0x03, + 0x95, 0xd6, 0x7f, 0x09, 0x54, 0x51, 0xb9, 0xcb, 0xe4, 0xea, 0x6b, 0xc9, + 0x2f, 0xfc, 0xa2, 0xb3, 0xef, 0x46, 0x86, 0xa0, 0xd9, 0x72, 0x93, 0x20, + 0xee, 0x5d, 0x31, 0xe2, 0xa1, 0x59, 0x9a, 0xbd, 0x17, 0x25, 0x77, 0x72, + 0x2d, 0xc4, 0xc4, 0x29, 0xf8, 0x6e, 0x36, 0x9c, 0xe8, 0x3f, 0x61, 0x3b, + 0x83, 0xc8, 0xc1, 0x0c, 0x53, 0xc9, 0xe1, 0x6a, 0x99, 0xcb, 0x0f, 0xb4, + 0x2f, 0x53, 0x30, 0x4a, 0xec, 0xec, 0x3d, 0xe4, 0x8f, 0x3c, 0xe3, 0xe4, + 0xec, 0x13, 0x18, 0x87, 0xed, 0xc4, 0x3f, 0xee, 0x26, 0xcf, 0xd4, 0x5b, + 0xfd, 0x1c, 0x32, 0x5f, 0xc5, 0xb9, 0xc0, 0x4b +}; + +static const guint8 null_ts_packet[] = { + 0x47, 0x1f, 0xff, 0x10, 0x55, 0x33, 0x41, 0xd8, 0x99, 0x92, 0x09, 0xc5, + 0xd9, 0x74, 0x2f, 0xaf, 0x61, 0xa6, 0xda, 0x36, 0x95, 0xac, 0x72, 0x82, + 0xa7, 0xda, 0xb9, 0x57, 0x91, 0x66, 0x6e, 0x64, 0xec, 0x75, 0xa4, 0x51, + 0x31, 0xac, 0x10, 0x4a, 0x33, 0xa6, 0xb9, 0x3f, 0x50, 0x7c, 0xb5, 0x81, + 0x57, 0x9c, 0x00, 0x32, 0x61, 0x77, 0x70, 0x4e, 0xe6, 0x95, 0x9b, 0xe3, + 0xe9, 0xd1, 0x9b, 0xa5, 0x81, 0xbc, 0x95, 0x03, 0x24, 0x7a, 0x60, 0x36, + 0x0d, 0xbf, 0x0d, 0xfd, 0x56, 0x7f, 0xec, 0x73, 0x47, 0x88, 0x5c, 0x52, + 0x77, 0x24, 0xdc, 0xcb, 0xba, 0x24, 0xc3, 0xbb, 0xa4, 0xa5, 0x2e, 0xd8, + 0x5b, 0x85, 0x0f, 0x98, 0x1d, 0xb6, 0xe4, 0xb2, 0x5c, 0x14, 0x57, 0x54, + 0xb2, 0xce, 0xe0, 0x76, 0x86, 0x0b, 0x90, 0xbf, 0x1b, 0x54, 0x98, 0x4f, + 0xae, 0x77, 0x18, 0x3d, 0x81, 0x10, 0x3e, 0xe6, 0x73, 0xf1, 0xb9, 0xed, + 0x5e, 0xde, 0x8b, 0xe0, 0x5f, 0x6b, 0xc7, 0xe8, 0x9b, 0xe6, 0x53, 0xf3, + 0xa0, 0x85, 0x13, 0xcb, 0x46, 0x56, 0x07, 0xe7, 0xfa, 0xb5, 0x3d, 0x5f, + 0xa4, 0x74, 0x4b, 0xf1, 0x84, 0xdb, 0x94, 0xb4, 0xd7, 0x25, 0x99, 0xa3, + 0xbe, 0xcb, 0x11, 0x5d, 0xcb, 0x69, 0xe0, 0xb5, 0xd1, 0xda, 0x50, 0x24, + 0xca, 0x96, 0x09, 0x23, 0xcb, 0x1f, 0xbe, 0x00 +}; + +static GstBuffer * +alloc_ts_buffer (guint num_ts_packets) +{ + GstBuffer *buf; + GstRTPBuffer rtp = GST_RTP_BUFFER_INIT; + + buf = gst_rtp_buffer_new_allocate (188 * num_ts_packets, 0, 0); + gst_rtp_buffer_map (buf, GST_MAP_READWRITE, &rtp); + gst_rtp_buffer_set_version (&rtp, 2); + gst_rtp_buffer_set_ssrc (&rtp, 12); + gst_rtp_buffer_set_seq (&rtp, 44); + gst_rtp_buffer_set_timestamp (&rtp, 55); + gst_rtp_buffer_set_payload_type (&rtp, 33); + gst_rtp_buffer_unmap (&rtp); + + return buf; +} + +static void +validate_ts_buffer_full (GstRTPBuffer * rtp, guint num_ts_packets, guint16 seq, + gboolean extension) +{ + fail_unless_equals_int (gst_rtp_buffer_get_payload_len (rtp), + 188 * num_ts_packets); + fail_unless_equals_int (gst_rtp_buffer_get_ssrc (rtp), 12); + fail_unless_equals_int (gst_rtp_buffer_get_payload_type (rtp), 33); + fail_unless_equals_int (gst_rtp_buffer_get_seq (rtp), seq); + fail_unless_equals_int (gst_rtp_buffer_get_timestamp (rtp), 55); + fail_unless_equals_int (gst_rtp_buffer_get_extension (rtp), extension); +} + +static void +validate_ts_buffer_seq (GstRTPBuffer * rtp, guint num_ts_packets, guint16 seq) +{ + validate_ts_buffer_full (rtp, num_ts_packets, seq, TRUE); +} + +static void +validate_ts_buffer (GstRTPBuffer * rtp, guint num_ts_packets) +{ + validate_ts_buffer_full (rtp, num_ts_packets, 44, TRUE); +} + +static void +validate_ts_buffer_noext (GstRTPBuffer * rtp, guint num_ts_packets) +{ + validate_ts_buffer_full (rtp, num_ts_packets, 44, FALSE); +} + +static void +validate_ext (GstRTPBuffer * rtp, gboolean wanted_has_drop_null, + gboolean wanted_has_seqnum_ext, guint wanted_orig_ts_packet_count, + guint wanted_ts_packet_size, guint wanted_npd_bits, guint16 wanted_ext) +{ + guint extlen; + gpointer extdata; + guint16 bits; + gboolean has_seqnum_ext; + gboolean has_drop_null; + guint orig_ts_packet_count; + gboolean ts_packet_size; + guint8 npd_bits; + + fail_unless (gst_rtp_buffer_get_extension_data (rtp, &bits, &extdata, + &extlen)); + + has_drop_null = (bits >> 15) & 1; /* N */ + has_seqnum_ext = (bits >> 14) & 1; /* E */ + orig_ts_packet_count = (bits >> 10) & 7; /* Size */ + ts_packet_size = ((bits >> 7) & 1) ? 204 : 188; + npd_bits = bits & 0x7F; + + fail_unless_equals_int (has_drop_null, wanted_has_drop_null); + fail_unless_equals_int (has_seqnum_ext, wanted_has_seqnum_ext); + fail_unless_equals_int (orig_ts_packet_count, wanted_orig_ts_packet_count); + fail_unless_equals_int (ts_packet_size, wanted_ts_packet_size); + fail_unless_equals_int (npd_bits, wanted_npd_bits); + + if (wanted_has_seqnum_ext) { + guint16 ext; + + fail_unless (extlen == 1); + ext = GST_READ_UINT16_BE (extdata); + + fail_unless_equals_int (ext, wanted_ext); + } +} + +/* Don't touch this */ +GST_START_TEST (test_noop) +{ + GstHarness *h = gst_harness_new ("ristrtpext"); + GstRTPBuffer rtp = GST_RTP_BUFFER_INIT; + GstBuffer *ibuf; + GstBuffer *obuf; + guint i; + guint8 *payload; + + gst_harness_set_src_caps_str (h, "application/x-rtp, payload=33," + "clock-rate=90000, encoding-name=MP2T"); + + ibuf = alloc_ts_buffer (7); + gst_rtp_buffer_map (ibuf, GST_MAP_READWRITE, &rtp); + payload = gst_rtp_buffer_get_payload (&rtp); + for (i = 0; i < 7; i++) { + memcpy (payload + (i * 188), ts_packet, 188); + payload[(i * 188) + 187] = i; + } + gst_rtp_buffer_unmap (&rtp); + + obuf = gst_harness_push_and_pull (h, ibuf); + + gst_rtp_buffer_map (obuf, GST_MAP_READ, &rtp); + validate_ts_buffer_noext (&rtp, 7); + payload = gst_rtp_buffer_get_payload (&rtp); + for (i = 0; i < 7; i++) { + fail_unless_equals_int (memcmp (payload + (188 * i), ts_packet, 187), 0); + fail_unless_equals_int (payload[(i * 188) + 187], i); + } + gst_rtp_buffer_unmap (&rtp); + gst_buffer_unref (obuf); + + gst_harness_teardown (h); +} + +GST_END_TEST; + +/* No null packets */ +GST_START_TEST (test_remove_null_none) +{ + GstHarness *h = gst_harness_new ("ristrtpext"); + GstRTPBuffer rtp = GST_RTP_BUFFER_INIT; + GstBuffer *ibuf; + GstBuffer *obuf; + guint i; + guint8 *payload; + + g_object_set (h->element, "drop-null-ts-packets", TRUE, NULL); + gst_harness_set_src_caps_str (h, "application/x-rtp, payload=33," + "clock-rate=90000, encoding-name=MP2T"); + + ibuf = alloc_ts_buffer (7); + gst_rtp_buffer_map (ibuf, GST_MAP_READWRITE, &rtp); + payload = gst_rtp_buffer_get_payload (&rtp); + for (i = 0; i < 7; i++) { + memcpy (payload + (i * 188), ts_packet, 188); + payload[(i * 188) + 187] = i; + } + gst_rtp_buffer_unmap (&rtp); + + obuf = gst_harness_push_and_pull (h, ibuf); + + gst_rtp_buffer_map (obuf, GST_MAP_READ, &rtp); + validate_ts_buffer (&rtp, 7); + validate_ext (&rtp, TRUE, FALSE, 7, 188, 0, 0); + payload = gst_rtp_buffer_get_payload (&rtp); + for (i = 0; i < 7; i++) { + fail_unless_equals_int (memcmp (payload + (188 * i), ts_packet, 187), 0); + fail_unless_equals_int (payload[(i * 188) + 187], i); + } + gst_rtp_buffer_unmap (&rtp); + gst_buffer_unref (obuf); + + gst_harness_teardown (h); +} + +GST_END_TEST; + +/* 1 null packet in middle */ +GST_START_TEST (test_remove_null_middle) +{ + GstHarness *h = gst_harness_new ("ristrtpext"); + GstRTPBuffer rtp = GST_RTP_BUFFER_INIT; + GstBuffer *ibuf; + GstBuffer *obuf; + guint i; + guint8 *payload; + + g_object_set (h->element, "drop-null-ts-packets", TRUE, NULL); + gst_harness_set_src_caps_str (h, "application/x-rtp, payload=33," + "clock-rate=90000, encoding-name=MP2T"); + + ibuf = alloc_ts_buffer (7); + gst_rtp_buffer_map (ibuf, GST_MAP_READWRITE, &rtp); + payload = gst_rtp_buffer_get_payload (&rtp); + for (i = 0; i < 3; i++) { + memcpy (payload + (i * 188), ts_packet, 188); + payload[(i * 188) + 187] = i; + } + memcpy (payload + (3 * 188), null_ts_packet, 188); + for (i = 4; i < 7; i++) { + memcpy (payload + (i * 188), ts_packet, 188); + payload[(i * 188) + 187] = i; + } + gst_rtp_buffer_unmap (&rtp); + + obuf = gst_harness_push_and_pull (h, ibuf); + + gst_rtp_buffer_map (obuf, GST_MAP_READ, &rtp); + validate_ts_buffer (&rtp, 6); + validate_ext (&rtp, TRUE, FALSE, 7, 188, 1 << 3, 0); + payload = gst_rtp_buffer_get_payload (&rtp); + for (i = 0; i < 6; i++) { + fail_unless_equals_int (memcmp (payload + (188 * i), ts_packet, 187), 0); + if (i < 3) { + fail_unless_equals_int (payload[(i * 188) + 187], i); + } else { + fail_unless_equals_int (payload[(i * 188) + 187], i + 1); + } + } + + gst_rtp_buffer_unmap (&rtp); + gst_buffer_unref (obuf); + + gst_harness_teardown (h); +} + +GST_END_TEST; + +/* one null packet at the start and one in the end */ +GST_START_TEST (test_remove_null_start_and_end) +{ + GstHarness *h = gst_harness_new ("ristrtpext"); + GstRTPBuffer rtp = GST_RTP_BUFFER_INIT; + GstBuffer *ibuf; + GstBuffer *obuf; + guint i; + guint8 *payload; + + g_object_set (h->element, "drop-null-ts-packets", TRUE, NULL); + gst_harness_set_src_caps_str (h, "application/x-rtp, payload=33," + "clock-rate=90000, encoding-name=MP2T"); + + ibuf = alloc_ts_buffer (7); + gst_rtp_buffer_map (ibuf, GST_MAP_READWRITE, &rtp); + payload = gst_rtp_buffer_get_payload (&rtp); + memcpy (payload + (0 * 188), null_ts_packet, 188); + for (i = 1; i < 6; i++) { + memcpy (payload + (i * 188), ts_packet, 188); + payload[(i * 188) + 187] = i; + } + memcpy (payload + (6 * 188), null_ts_packet, 188); + gst_rtp_buffer_unmap (&rtp); + + obuf = gst_harness_push_and_pull (h, ibuf); + + gst_rtp_buffer_map (obuf, GST_MAP_READ, &rtp); + validate_ts_buffer (&rtp, 5); + validate_ext (&rtp, TRUE, FALSE, 7, 188, 1 << 6 | 1, 0); + payload = gst_rtp_buffer_get_payload (&rtp); + for (i = 0; i < 5; i++) { + fail_unless_equals_int (memcmp (payload + (188 * i), ts_packet, 187), 0); + fail_unless_equals_int (payload[(i * 188) + 187], i + 1); + } + gst_rtp_buffer_unmap (&rtp); + gst_buffer_unref (obuf); + + gst_harness_teardown (h); +} + +GST_END_TEST; + + +/* All null packets */ +GST_START_TEST (test_remove_null_all) +{ + GstHarness *h = gst_harness_new ("ristrtpext"); + GstRTPBuffer rtp = GST_RTP_BUFFER_INIT; + GstBuffer *ibuf; + GstBuffer *obuf; + guint i; + guint8 *payload; + + g_object_set (h->element, "drop-null-ts-packets", TRUE, NULL); + gst_harness_set_src_caps_str (h, "application/x-rtp, payload=33," + "clock-rate=90000, encoding-name=MP2T"); + + ibuf = alloc_ts_buffer (7); + gst_rtp_buffer_map (ibuf, GST_MAP_READWRITE, &rtp); + payload = gst_rtp_buffer_get_payload (&rtp); + for (i = 0; i < 7; i++) { + memcpy (payload + (i * 188), null_ts_packet, 188); + payload[(i * 188) + 187] = i; + } + gst_rtp_buffer_unmap (&rtp); + + obuf = gst_harness_push_and_pull (h, ibuf); + + gst_rtp_buffer_map (obuf, GST_MAP_READ, &rtp); + validate_ts_buffer (&rtp, 0); + validate_ext (&rtp, TRUE, FALSE, 7, 188, 0x7F, 0); + gst_rtp_buffer_unmap (&rtp); + gst_buffer_unref (obuf); + + gst_harness_teardown (h); +} + +GST_END_TEST; + +GST_START_TEST (test_remove_null_not_ts) +{ + GstHarness *h = gst_harness_new ("ristrtpext"); + GstRTPBuffer rtp = GST_RTP_BUFFER_INIT; + GstBuffer *ibuf; + GstBuffer *obuf; + + g_object_set (h->element, "drop-null-ts-packets", TRUE, NULL); + gst_harness_set_src_caps_str (h, "application/x-rtp, payload=33," + "clock-rate=90000, encoding-name=MP2T"); + + ibuf = alloc_ts_buffer (7); + gst_rtp_buffer_map (ibuf, GST_MAP_READWRITE, &rtp); + gst_rtp_buffer_set_payload_type (&rtp, 96); + gst_rtp_buffer_unmap (&rtp); + + obuf = gst_harness_push_and_pull (h, ibuf); + + fail_unless (obuf == ibuf); + gst_buffer_unref (obuf); + + gst_harness_teardown (h); +} + +GST_END_TEST; + + +/* Eight null packets */ +GST_START_TEST (test_remove_null_all_8_packets) +{ + GstHarness *h = gst_harness_new ("ristrtpext"); + GstRTPBuffer rtp = GST_RTP_BUFFER_INIT; + GstBuffer *ibuf; + GstBuffer *obuf; + guint i; + guint8 *payload; + + g_object_set (h->element, "drop-null-ts-packets", TRUE, NULL); + gst_harness_set_src_caps_str (h, "application/x-rtp, payload=33," + "clock-rate=90000, encoding-name=MP2T"); + + ibuf = alloc_ts_buffer (8); + gst_rtp_buffer_map (ibuf, GST_MAP_READWRITE, &rtp); + payload = gst_rtp_buffer_get_payload (&rtp); + for (i = 0; i < 8; i++) { + memcpy (payload + (i * 188), null_ts_packet, 188); + payload[(i * 188) + 187] = i; + } + gst_rtp_buffer_unmap (&rtp); + + obuf = gst_harness_push_and_pull (h, ibuf); + + gst_rtp_buffer_map (obuf, GST_MAP_READ, &rtp); + validate_ts_buffer (&rtp, 1); + validate_ext (&rtp, TRUE, FALSE, 0, 188, 0x7F, 0); + gst_rtp_buffer_unmap (&rtp); + gst_buffer_unref (obuf); + + gst_harness_teardown (h); +} + +GST_END_TEST; + +/* 204 bytes null packets */ +GST_START_TEST (test_remove_null_all_204bytes) +{ + GstHarness *h = gst_harness_new ("ristrtpext"); + GstRTPBuffer rtp = GST_RTP_BUFFER_INIT; + GstBuffer *ibuf; + GstBuffer *obuf; + guint i; + guint8 *payload; + + g_object_set (h->element, "drop-null-ts-packets", TRUE, NULL); + gst_harness_set_src_caps_str (h, "application/x-rtp, payload=33," + "clock-rate=90000, encoding-name=MP2T"); + + ibuf = alloc_ts_buffer (7); + gst_buffer_append_memory (ibuf, gst_allocator_alloc (NULL, (204 - 188) * 7, + NULL)); + gst_rtp_buffer_map (ibuf, GST_MAP_READWRITE, &rtp); + payload = gst_rtp_buffer_get_payload (&rtp); + for (i = 0; i < 7; i++) { + memcpy (payload + (i * 204), null_ts_packet, 188); + payload[(i * 204) + 187] = i; + } + gst_rtp_buffer_unmap (&rtp); + + obuf = gst_harness_push_and_pull (h, ibuf); + + gst_rtp_buffer_map (obuf, GST_MAP_READ, &rtp); + validate_ts_buffer (&rtp, 0); + validate_ext (&rtp, TRUE, FALSE, 7, 204, 0x7F, 0); + gst_rtp_buffer_unmap (&rtp); + gst_buffer_unref (obuf); + + gst_harness_teardown (h); +} + +GST_END_TEST; + +/* 2 null packet out of 3 */ +GST_START_TEST (test_remove_null_two_of_three) +{ + GstHarness *h = gst_harness_new ("ristrtpext"); + GstRTPBuffer rtp = GST_RTP_BUFFER_INIT; + GstBuffer *ibuf; + GstBuffer *obuf; + guint8 *payload; + + g_object_set (h->element, "drop-null-ts-packets", TRUE, NULL); + gst_harness_set_src_caps_str (h, "application/x-rtp, payload=33," + "clock-rate=90000, encoding-name=MP2T"); + + ibuf = alloc_ts_buffer (3); + gst_rtp_buffer_map (ibuf, GST_MAP_READWRITE, &rtp); + payload = gst_rtp_buffer_get_payload (&rtp); + memcpy (payload + (0 * 188), null_ts_packet, 188); + memcpy (payload + (1 * 188), ts_packet, 188); + payload[188 + 187] = 33; + memcpy (payload + (2 * 188), null_ts_packet, 188); + + gst_rtp_buffer_unmap (&rtp); + + obuf = gst_harness_push_and_pull (h, ibuf); + + gst_rtp_buffer_map (obuf, GST_MAP_READ, &rtp); + validate_ts_buffer (&rtp, 1); + validate_ext (&rtp, TRUE, FALSE, 3, 188, 1 << 6 | 1 << 4, 0); + payload = gst_rtp_buffer_get_payload (&rtp); + fail_unless_equals_int (memcmp (payload, ts_packet, 187), 0); + fail_unless_equals_int (payload[187], 33); + + gst_rtp_buffer_unmap (&rtp); + gst_buffer_unref (obuf); + + gst_harness_teardown (h); +} + +GST_END_TEST; + +static void +push_one_seqnum (GstHarness * h, guint16 seqnum, guint16 wanted_ext) +{ + GstBuffer *ibuf; + GstBuffer *obuf; + static const int NUM_PACKETS = 5; + guint i; + guint8 *payload; + GstRTPBuffer rtp = GST_RTP_BUFFER_INIT; + + ibuf = alloc_ts_buffer (NUM_PACKETS); + gst_rtp_buffer_map (ibuf, GST_MAP_READWRITE, &rtp); + gst_rtp_buffer_set_seq (&rtp, seqnum); + payload = gst_rtp_buffer_get_payload (&rtp); + for (i = 0; i < NUM_PACKETS; i++) { + memcpy (payload + (i * 188), ts_packet, 188); + payload[(i * 188) + 187] = i; + } + + gst_rtp_buffer_unmap (&rtp); + + obuf = gst_harness_push_and_pull (h, ibuf); + + gst_rtp_buffer_map (obuf, GST_MAP_READ, &rtp); + validate_ts_buffer_seq (&rtp, NUM_PACKETS, seqnum); + validate_ext (&rtp, FALSE, TRUE, 0, 188, 0, wanted_ext); + payload = gst_rtp_buffer_get_payload (&rtp); + for (i = 0; i < NUM_PACKETS; i++) { + fail_unless_equals_int (memcmp (payload + (188 * i), ts_packet, 187), 0); + fail_unless_equals_int (payload[(i * 188) + 187], i); + } + gst_rtp_buffer_unmap (&rtp); + gst_buffer_unref (obuf); +} + +GST_START_TEST (test_add_seqnum_ext) +{ + GstHarness *h = gst_harness_new ("ristrtpext"); + + g_object_set (h->element, "sequence-number-extension", TRUE, NULL); + gst_harness_set_src_caps_str (h, "application/x-rtp, payload=33," + "clock-rate=90000, encoding-name=MP2T"); + + push_one_seqnum (h, 44, 0); + + gst_harness_teardown (h); +} + +GST_END_TEST; + +GST_START_TEST (test_add_seqnum_ext_roll_over) +{ + GstHarness *h = gst_harness_new ("ristrtpext"); + + g_object_set (h->element, "sequence-number-extension", TRUE, NULL); + gst_harness_set_src_caps_str (h, "application/x-rtp, payload=33," + "clock-rate=90000, encoding-name=MP2T"); + + /* push one */ + push_one_seqnum (h, 0xA123, 0); + + /* Now roll over */ + push_one_seqnum (h, 0x0123, 1); + + gst_harness_teardown (h); +} + +GST_END_TEST; + +GST_START_TEST (test_add_seqnum_ext_roll_back) +{ + GstHarness *h = gst_harness_new ("ristrtpext"); + + g_object_set (h->element, "sequence-number-extension", TRUE, NULL); + gst_harness_set_src_caps_str (h, "application/x-rtp, payload=33," + "clock-rate=90000, encoding-name=MP2T"); + + /* Send one packet */ + push_one_seqnum (h, 0xA123, 0); + + /* Now roll over */ + push_one_seqnum (h, 0x0123, 1); + + /* Now roll back */ + push_one_seqnum (h, 0xF123, 0); + + gst_harness_teardown (h); +} + +GST_END_TEST; + +GST_START_TEST (test_add_seqnum_ext_roll_over_twice) +{ + GstHarness *h = gst_harness_new ("ristrtpext"); + + g_object_set (h->element, "sequence-number-extension", TRUE, NULL); + gst_harness_set_src_caps_str (h, "application/x-rtp, payload=33," + "clock-rate=90000, encoding-name=MP2T"); + + /* Send one packet */ + push_one_seqnum (h, 0xF123, 0); + + /* Now roll over */ + push_one_seqnum (h, 0x2123, 1); + + /* Now go foward */ + push_one_seqnum (h, 0x9123, 1); + + /* Now roll back */ + push_one_seqnum (h, 0x0123, 2); + + gst_harness_teardown (h); +} + +GST_END_TEST; + + +static GstBuffer * +alloc_ts_buffer_with_ext (guint num_ts_packets, gboolean has_drop_null, + gboolean has_seqnum_ext, guint orig_ts_packet_count, + guint ts_packet_size, guint npd_bits, guint16 extseq) +{ + GstBuffer *buf = alloc_ts_buffer (num_ts_packets); + GstRTPBuffer rtp = GST_RTP_BUFFER_INIT; + guint8 *payload; + guint16 bits; + guint i; + + bits = 0; + bits |= has_drop_null << 15; /* N */ + bits |= has_seqnum_ext << 14; /* E */ + bits |= (MIN (orig_ts_packet_count, 7) & 7) << 10; /* Size */ + bits |= (ts_packet_size == 204) << 7; /* T */ + bits |= (npd_bits & 0x7F); + + gst_rtp_buffer_map (buf, GST_MAP_READWRITE, &rtp); + gst_rtp_buffer_set_extension_data (&rtp, bits, + (has_seqnum_ext && extseq != 0xFFFF) ? 1 : 0); + if (has_seqnum_ext && extseq != 0xFFFF) { + guint8 *data; + guint wordlen; + + gst_rtp_buffer_get_extension_data (&rtp, &bits, (void **) &data, &wordlen); + GST_WRITE_UINT16_BE (data, extseq); + data[2] = data[3] = 0; + } + + payload = gst_rtp_buffer_get_payload (&rtp); + for (i = 0; i < num_ts_packets; i++) { + memcpy (payload + (i * 188), ts_packet, 188); + payload[(i * 188) + 187] = i; + } + + gst_rtp_buffer_unmap (&rtp); + + return buf; +} + +GST_START_TEST (test_deext_noop) +{ + GstHarness *h = gst_harness_new ("ristrtpdeext"); + GstBuffer *ibuf, *obuf; + GstRTPBuffer rtp = GST_RTP_BUFFER_INIT; + + gst_harness_set_src_caps_str (h, "application/x-rtp, payload=33," + "clock-rate=90000, encoding-name=MP2T"); + + ibuf = alloc_ts_buffer_with_ext (7, FALSE, FALSE, 7, 188, 0, 0); + + obuf = gst_harness_push_and_pull (h, ibuf); + + gst_rtp_buffer_map (obuf, GST_MAP_READ, &rtp); + validate_ts_buffer_noext (&rtp, 7); + gst_rtp_buffer_unmap (&rtp); + + gst_buffer_unref (obuf); + + gst_harness_teardown (h); +} + +GST_END_TEST; + +GST_START_TEST (test_deext_restore_middle) +{ + GstHarness *h = gst_harness_new ("ristrtpdeext"); + GstBuffer *ibuf, *obuf; + GstRTPBuffer rtp = GST_RTP_BUFFER_INIT; + guint8 *payload; + guint i; + + gst_harness_set_src_caps_str (h, "application/x-rtp, payload=33," + "clock-rate=90000, encoding-name=MP2T"); + + ibuf = alloc_ts_buffer_with_ext (6, TRUE, FALSE, 7, 188, 1 << 3, 0); + + obuf = gst_harness_push_and_pull (h, ibuf); + + gst_rtp_buffer_map (obuf, GST_MAP_READ, &rtp); + validate_ts_buffer_noext (&rtp, 7); + payload = gst_rtp_buffer_get_payload (&rtp); + for (i = 0; i < 7; i++) { + if (i < 3) { + fail_unless_equals_int (memcmp (payload + (188 * i), ts_packet, 187), 0); + fail_unless_equals_int (payload[(i * 188) + 187], i); + } else if (i > 3) { + fail_unless_equals_int (memcmp (payload + (188 * i), ts_packet, 187), 0); + fail_unless_equals_int (payload[(i * 188) + 187], i - 1); + } else { + fail_unless_equals_int (memcmp (payload + (188 * i), null_ts_packet, 4), + 0); + } + } + gst_rtp_buffer_unmap (&rtp); + gst_buffer_unref (obuf); + + gst_harness_teardown (h); +} + +GST_END_TEST; + + +GST_START_TEST (test_deext_restore_start_and_end) +{ + GstHarness *h = gst_harness_new ("ristrtpdeext"); + GstBuffer *ibuf, *obuf; + GstRTPBuffer rtp = GST_RTP_BUFFER_INIT; + guint8 *payload; + guint i; + + gst_harness_set_src_caps_str (h, "application/x-rtp, payload=33," + "clock-rate=90000, encoding-name=MP2T"); + + ibuf = alloc_ts_buffer_with_ext (5, TRUE, FALSE, 7, 188, 1 << 6 | 1, 0); + + obuf = gst_harness_push_and_pull (h, ibuf); + + gst_rtp_buffer_map (obuf, GST_MAP_READ, &rtp); + validate_ts_buffer_noext (&rtp, 7); + payload = gst_rtp_buffer_get_payload (&rtp); + fail_unless_equals_int (memcmp (payload, null_ts_packet, 4), 0); + for (i = 1; i < 6; i++) { + fail_unless_equals_int (memcmp (payload + (188 * i), ts_packet, 187), 0); + fail_unless_equals_int (payload[(i * 188) + 187], i - 1); + } + fail_unless_equals_int (memcmp (payload + (188 * 6), null_ts_packet, 4), 0); + gst_rtp_buffer_unmap (&rtp); + gst_buffer_unref (obuf); + + gst_harness_teardown (h); +} + +GST_END_TEST; + +GST_START_TEST (test_deext_restore_middle_no_origcnt) +{ + GstHarness *h = gst_harness_new ("ristrtpdeext"); + GstBuffer *ibuf, *obuf; + GstRTPBuffer rtp = GST_RTP_BUFFER_INIT; + guint8 *payload; + guint i; + + gst_harness_set_src_caps_str (h, "application/x-rtp, payload=33," + "clock-rate=90000, encoding-name=MP2T"); + + ibuf = alloc_ts_buffer_with_ext (6, TRUE, FALSE, 0, 188, 1 << 3, 0); + + obuf = gst_harness_push_and_pull (h, ibuf); + + gst_rtp_buffer_map (obuf, GST_MAP_READ, &rtp); + validate_ts_buffer_noext (&rtp, 7); + payload = gst_rtp_buffer_get_payload (&rtp); + for (i = 0; i < 7; i++) { + if (i < 3) { + fail_unless_equals_int (memcmp (payload + (188 * i), ts_packet, 187), 0); + fail_unless_equals_int (payload[(i * 188) + 187], i); + } else if (i > 3) { + fail_unless_equals_int (memcmp (payload + (188 * i), ts_packet, 187), 0); + fail_unless_equals_int (payload[(i * 188) + 187], i - 1); + } else { + fail_unless_equals_int (memcmp (payload + (188 * i), null_ts_packet, 4), + 0); + } + } + gst_rtp_buffer_unmap (&rtp); + gst_buffer_unref (obuf); + + gst_harness_teardown (h); +} + +GST_END_TEST; + +GST_START_TEST (test_deext_restore_all) +{ + GstHarness *h = gst_harness_new ("ristrtpdeext"); + GstBuffer *ibuf, *obuf; + GstRTPBuffer rtp = GST_RTP_BUFFER_INIT; + guint8 *payload; + guint i; + + gst_harness_set_src_caps_str (h, "application/x-rtp, payload=33," + "clock-rate=90000, encoding-name=MP2T"); + + ibuf = alloc_ts_buffer_with_ext (0, TRUE, FALSE, 7, 188, 0x7F, 0); + + obuf = gst_harness_push_and_pull (h, ibuf); + + gst_rtp_buffer_map (obuf, GST_MAP_READ, &rtp); + validate_ts_buffer_noext (&rtp, 7); + payload = gst_rtp_buffer_get_payload (&rtp); + for (i = 0; i < 7; i++) + fail_unless_equals_int (memcmp (payload + (188 * i), null_ts_packet, 4), 0); + gst_rtp_buffer_unmap (&rtp); + gst_buffer_unref (obuf); + + gst_harness_teardown (h); +} + +GST_END_TEST; + +GST_START_TEST (test_deext_restore_all_8) +{ + GstHarness *h = gst_harness_new ("ristrtpdeext"); + GstBuffer *ibuf, *obuf; + GstRTPBuffer rtp = GST_RTP_BUFFER_INIT; + guint8 *payload; + guint i; + + gst_harness_set_src_caps_str (h, "application/x-rtp, payload=33," + "clock-rate=90000, encoding-name=MP2T"); + + ibuf = alloc_ts_buffer_with_ext (1, TRUE, FALSE, 0, 188, 0x7F, 0); + + obuf = gst_harness_push_and_pull (h, ibuf); + + gst_rtp_buffer_map (obuf, GST_MAP_READ, &rtp); + validate_ts_buffer_noext (&rtp, 8); + payload = gst_rtp_buffer_get_payload (&rtp); + for (i = 0; i < 7; i++) + fail_unless_equals_int (memcmp (payload + (188 * i), null_ts_packet, 4), 0); + fail_unless_equals_int (memcmp (payload + (188 * i), ts_packet, 187), 0); + gst_rtp_buffer_unmap (&rtp); + gst_buffer_unref (obuf); + + gst_harness_teardown (h); +} + +GST_END_TEST; + +GST_START_TEST (test_deext_restore_all_204bytes) +{ + GstHarness *h = gst_harness_new ("ristrtpdeext"); + GstBuffer *ibuf, *obuf; + GstRTPBuffer rtp = GST_RTP_BUFFER_INIT; + guint8 *payload; + guint i; + + gst_harness_set_src_caps_str (h, "application/x-rtp, payload=33," + "clock-rate=90000, encoding-name=MP2T"); + + ibuf = alloc_ts_buffer_with_ext (0, TRUE, FALSE, 7, 204, 0x7F, 0); + + obuf = gst_harness_push_and_pull (h, ibuf); + + gst_rtp_buffer_map (obuf, GST_MAP_READ, &rtp); + + fail_unless_equals_int (gst_rtp_buffer_get_payload_len (&rtp), 204 * 7); + fail_unless_equals_int (gst_rtp_buffer_get_ssrc (&rtp), 12); + fail_unless_equals_int (gst_rtp_buffer_get_payload_type (&rtp), 33); + fail_unless_equals_int (gst_rtp_buffer_get_seq (&rtp), 44); + fail_unless_equals_int (gst_rtp_buffer_get_timestamp (&rtp), 55); + fail_unless_equals_int (gst_rtp_buffer_get_extension (&rtp), FALSE); + + payload = gst_rtp_buffer_get_payload (&rtp); + for (i = 0; i < 7; i++) + fail_unless_equals_int (memcmp (payload + (204 * i), null_ts_packet, 4), 0); + gst_rtp_buffer_unmap (&rtp); + gst_buffer_unref (obuf); + + gst_harness_teardown (h); +} + +GST_END_TEST; + + +GST_START_TEST (test_deext_restore_empty) +{ + GstHarness *h = gst_harness_new ("ristrtpdeext"); + GstBuffer *ibuf, *obuf; + GstRTPBuffer rtp = GST_RTP_BUFFER_INIT; + + gst_harness_set_src_caps_str (h, "application/x-rtp, payload=33," + "clock-rate=90000, encoding-name=MP2T"); + + ibuf = alloc_ts_buffer_with_ext (0, TRUE, FALSE, 0, 188, 0, 0); + + obuf = gst_harness_push_and_pull (h, ibuf); + + gst_rtp_buffer_map (obuf, GST_MAP_READ, &rtp); + validate_ts_buffer_noext (&rtp, 0); + gst_rtp_buffer_unmap (&rtp); + gst_buffer_unref (obuf); + + gst_harness_teardown (h); +} + +GST_END_TEST; + + +GST_START_TEST (test_deext_restore_invalid_origcnt) +{ + GstHarness *h = gst_harness_new ("ristrtpdeext"); + GstBuffer *ibuf, *obuf; + GstRTPBuffer rtp = GST_RTP_BUFFER_INIT; + guint8 *payload; + + gst_harness_set_src_caps_str (h, "application/x-rtp, payload=33," + "clock-rate=90000, encoding-name=MP2T"); + + ibuf = alloc_ts_buffer_with_ext (2, TRUE, FALSE, 5, 188, 1 << 6 | 1 << 4, 0); + + obuf = gst_harness_push_and_pull (h, ibuf); + + gst_rtp_buffer_map (obuf, GST_MAP_READ, &rtp); + validate_ts_buffer_noext (&rtp, 4); + payload = gst_rtp_buffer_get_payload (&rtp); + + fail_unless_equals_int (memcmp (payload + (188 * 0), null_ts_packet, 4), 0); + + fail_unless_equals_int (memcmp (payload + (188 * 1), ts_packet, 187), 0); + fail_unless_equals_int (payload[(188 * 1) + 187], 0); + + fail_unless_equals_int (memcmp (payload + (188 * 2), null_ts_packet, 4), 0); + + fail_unless_equals_int (memcmp (payload + (188 * 3), ts_packet, 187), 0); + fail_unless_equals_int (payload[(188 * 3) + 187], 1); + + gst_rtp_buffer_unmap (&rtp); + gst_buffer_unref (obuf); + + gst_harness_teardown (h); +} + +GST_END_TEST; + + +GST_START_TEST (test_deext_noop_invalid_size) +{ + GstHarness *h = gst_harness_new ("ristrtpdeext"); + GstBuffer *ibuf, *obuf; + GstRTPBuffer rtp = GST_RTP_BUFFER_INIT; + + gst_harness_set_src_caps_str (h, "application/x-rtp, payload=33," + "clock-rate=90000, encoding-name=MP2T"); + + ibuf = alloc_ts_buffer_with_ext (7, FALSE, FALSE, 7, 188, 0, 0); + + gst_buffer_append_memory (ibuf, gst_allocator_alloc (NULL, 5, NULL)); + + obuf = gst_harness_push_and_pull (h, ibuf); + + gst_rtp_buffer_map (obuf, GST_MAP_READ, &rtp); + + fail_unless_equals_int (gst_rtp_buffer_get_payload_len (&rtp), 188 * 7 + 5); + fail_unless_equals_int (gst_rtp_buffer_get_ssrc (&rtp), 12); + fail_unless_equals_int (gst_rtp_buffer_get_payload_type (&rtp), 33); + fail_unless_equals_int (gst_rtp_buffer_get_seq (&rtp), 44); + fail_unless_equals_int (gst_rtp_buffer_get_timestamp (&rtp), 55); + fail_unless_equals_int (gst_rtp_buffer_get_extension (&rtp), FALSE); + + gst_rtp_buffer_unmap (&rtp); + + gst_buffer_unref (obuf); + + gst_harness_teardown (h); +} + +GST_END_TEST; + +GST_START_TEST (test_deext_seq_base) +{ + GstHarness *h = gst_harness_new ("ristrtpdeext"); + GstBuffer *ibuf, *obuf; + GstRTPBuffer rtp = GST_RTP_BUFFER_INIT; + + gst_harness_set_src_caps_str (h, "application/x-rtp, payload=33," + "clock-rate=90000, encoding-name=MP2T"); + + ibuf = alloc_ts_buffer_with_ext (7, FALSE, TRUE, 7, 188, 0, 0); + obuf = gst_harness_push_and_pull (h, ibuf); + gst_rtp_buffer_map (obuf, GST_MAP_READ, &rtp); + validate_ts_buffer_noext (&rtp, 7); + gst_rtp_buffer_unmap (&rtp); + gst_buffer_unref (obuf); + + ibuf = alloc_ts_buffer_with_ext (7, FALSE, TRUE, 7, 188, 0, 1); + obuf = gst_harness_push_and_pull (h, ibuf); + gst_rtp_buffer_map (obuf, GST_MAP_READ, &rtp); + validate_ts_buffer_noext (&rtp, 7); + gst_rtp_buffer_unmap (&rtp); + gst_buffer_unref (obuf); + + gst_harness_teardown (h); +} + +GST_END_TEST; + +GST_START_TEST (test_deext_seq_drop) +{ + GstHarness *h = gst_harness_new ("ristrtpdeext"); + GstBuffer *ibuf, *obuf; + GstRTPBuffer rtp = GST_RTP_BUFFER_INIT; + + gst_harness_set_src_caps_str (h, "application/x-rtp, payload=33," + "clock-rate=90000, encoding-name=MP2T"); + + ibuf = alloc_ts_buffer_with_ext (7, FALSE, TRUE, 7, 188, 0, 0); + obuf = gst_harness_push_and_pull (h, ibuf); + gst_rtp_buffer_map (obuf, GST_MAP_READ, &rtp); + validate_ts_buffer_noext (&rtp, 7); + gst_rtp_buffer_unmap (&rtp); + gst_buffer_unref (obuf); + + ibuf = alloc_ts_buffer_with_ext (7, FALSE, TRUE, 7, 188, 0, 2); + obuf = gst_harness_push_and_pull (h, ibuf); + gst_rtp_buffer_map (obuf, GST_MAP_READ, &rtp); + validate_ts_buffer_noext (&rtp, 7); + gst_rtp_buffer_unmap (&rtp); + gst_buffer_unref (obuf); + + ibuf = alloc_ts_buffer_with_ext (7, FALSE, TRUE, 7, 188, 0, 0); + fail_unless_equals_int (gst_harness_push (h, ibuf), GST_FLOW_OK); + fail_unless_equals_int (gst_harness_buffers_in_queue (h), 0); + + gst_harness_teardown (h); +} + +GST_END_TEST; + +static Suite * +ristrtpext_suite (void) +{ + Suite *s = suite_create ("ristrtpext"); + TCase *tc; + + tc = tcase_create ("ext"); + suite_add_tcase (s, tc); + + tcase_add_test (tc, test_noop); + tcase_add_test (tc, test_remove_null_none); + tcase_add_test (tc, test_remove_null_middle); + tcase_add_test (tc, test_remove_null_start_and_end); + tcase_add_test (tc, test_remove_null_all); + tcase_add_test (tc, test_remove_null_not_ts); + tcase_add_test (tc, test_remove_null_all_8_packets); + tcase_add_test (tc, test_remove_null_all_204bytes); + tcase_add_test (tc, test_remove_null_two_of_three); + + tcase_add_test (tc, test_add_seqnum_ext); + tcase_add_test (tc, test_add_seqnum_ext_roll_over); + tcase_add_test (tc, test_add_seqnum_ext_roll_back); + tcase_add_test (tc, test_add_seqnum_ext_roll_over_twice); + + tc = tcase_create ("deext"); + suite_add_tcase (s, tc); + + tcase_add_test (tc, test_deext_noop); + tcase_add_test (tc, test_deext_restore_middle); + tcase_add_test (tc, test_deext_restore_start_and_end); + tcase_add_test (tc, test_deext_restore_middle_no_origcnt); + tcase_add_test (tc, test_deext_restore_all); + tcase_add_test (tc, test_deext_restore_all_8); + tcase_add_test (tc, test_deext_restore_all_204bytes); + tcase_add_test (tc, test_deext_restore_empty); + tcase_add_test (tc, test_deext_restore_invalid_origcnt); + tcase_add_test (tc, test_deext_noop_invalid_size); + + tcase_add_test (tc, test_deext_seq_base); + tcase_add_test (tc, test_deext_seq_drop); + + return s; +} + +GST_CHECK_MAIN (ristrtpext); diff --git a/tests/check/meson.build b/tests/check/meson.build index bc1b714a8c..1502350fc1 100644 --- a/tests/check/meson.build +++ b/tests/check/meson.build @@ -46,6 +46,7 @@ base_tests = [ [['elements/svthevcenc.c'], not svthevcenc_dep.found(), [svthevcenc_dep]], [['elements/pcapparse.c'], false, [libparser_dep]], [['elements/pnm.c']], + [['elements/ristrtpext.c']], [['elements/rtponvifparse.c']], [['elements/rtponviftimestamp.c']], [['elements/rtpsrc.c']],