gstreamer/tests/check/libs/rtphdrext.c
Matthew Waters 427c3f4442 rtp: add base object for reading/writing rtp header extensions (RFC5285)
Facilitates the creation of rtp header extension implementations that
can be reused across applications.

Implementations are registered into the GStreamer registry as elements
(idea from GstRTSPExtension) and can be retrieved by URI or filtered
manually.  RTP header extensions must have the classification
"Network/Extension/RTPHeader" to be considered as a RTP Header
extension.

Fixes: https://gitlab.freedesktop.org/gstreamer/gst-plugins-base/-/issues/777
Part-of: <https://gitlab.freedesktop.org/gstreamer/gst-plugins-base/-/merge_requests/748>
2020-12-03 10:19:32 +00:00

146 lines
4.7 KiB
C

/* GStreamer RTP header extension unit tests
* Copyright (C) 2020 Matthew Waters <matthew@centricular.com>
*
* 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.
*/
#include "gst/gstcaps.h"
#include "gst/gstvalue.h"
#include "gst/rtp/gstrtphdrext.h"
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include <gst/gst.h>
#include <gst/check/check.h>
#include <gst/rtp/rtp.h>
#include "rtpdummyhdrextimpl.c"
GST_START_TEST (rtp_header_ext_write)
{
GstRTPHeaderExtension *dummy;
GstBuffer *buffer;
guint8 *data;
gsize size, written;
dummy = rtp_dummy_hdr_ext_new ();
gst_rtp_header_extension_set_id (dummy, 1);
buffer = gst_buffer_new ();
size = gst_rtp_header_extension_get_max_size (dummy, buffer);
fail_unless (size > 0);
data = g_malloc0 (size);
fail_unless (data != NULL);
written =
gst_rtp_header_extension_write (dummy, buffer, 0, buffer, data, size);
fail_unless (written > 0 && written <= size);
fail_unless_equals_int (GST_RTP_DUMMY_HDR_EXT (dummy)->write_count, 1);
fail_unless (gst_rtp_header_extension_read (dummy, 0, data, size, buffer));
fail_unless_equals_int (GST_RTP_DUMMY_HDR_EXT (dummy)->read_count, 1);
g_free (data);
gst_buffer_unref (buffer);
g_object_unref (dummy);
}
GST_END_TEST;
GST_START_TEST (rtp_header_ext_create_from_uri)
{
GstElementFactory *factory;
GstRTPHeaderExtension *dummy;
fail_unless (gst_element_register (NULL, "test-dummyrtphdrext",
GST_RANK_MARGINAL, GST_TYPE_RTP_DUMMY_HDR_EXT));
dummy = gst_rtp_header_extension_create_from_uri (DUMMY_HDR_EXT_URI);
fail_unless (GST_IS_RTP_DUMMY_HDR_EXT (dummy));
factory = gst_element_get_factory (GST_ELEMENT (dummy));
gst_registry_remove_feature (gst_registry_get (),
GST_PLUGIN_FEATURE (factory));
gst_object_unref (dummy);
}
GST_END_TEST;
GST_START_TEST (rtp_header_ext_caps_with_attributes)
{
GstRTPHeaderExtension *dummy;
GstCaps *caps = gst_caps_new_empty_simple ("application/x-rtp");
GstStructure *s = gst_caps_get_structure (caps, 0);
const GValue *arr, *val;
const gchar *attributes = "attr0 attr1";
const gchar *direction = "recvonly";
dummy = rtp_dummy_hdr_ext_new ();
gst_rtp_header_extension_set_id (dummy, 1);
GST_RTP_DUMMY_HDR_EXT (dummy)->direction = g_strdup (direction);
GST_RTP_DUMMY_HDR_EXT (dummy)->attributes = g_strdup (attributes);
fail_unless (gst_rtp_header_extension_set_caps_from_attributes (dummy, caps));
fail_unless (gst_structure_has_field_typed (s, "extmap-1", GST_TYPE_ARRAY));
arr = gst_structure_get_value (s, "extmap-1");
fail_unless (GST_VALUE_HOLDS_ARRAY (arr));
fail_unless_equals_int (gst_value_array_get_size (arr), 3);
val = gst_value_array_get_value (arr, 0);
fail_unless_equals_int (g_strcmp0 (g_value_get_string (val), direction), 0);
val = gst_value_array_get_value (arr, 1);
fail_unless_equals_int (g_strcmp0 (g_value_get_string (val),
gst_rtp_header_extension_get_uri (dummy)), 0);
val = gst_value_array_get_value (arr, 2);
fail_unless_equals_int (g_strcmp0 (g_value_get_string (val), attributes), 0);
g_free (GST_RTP_DUMMY_HDR_EXT (dummy)->direction);
GST_RTP_DUMMY_HDR_EXT (dummy)->direction = NULL;
g_free (GST_RTP_DUMMY_HDR_EXT (dummy)->attributes);
GST_RTP_DUMMY_HDR_EXT (dummy)->attributes = NULL;
fail_unless (gst_rtp_header_extension_set_attributes_from_caps (dummy, caps));
fail_unless_equals_int (g_strcmp0 (GST_RTP_DUMMY_HDR_EXT (dummy)->attributes,
attributes), 0);
fail_unless_equals_int (g_strcmp0 (GST_RTP_DUMMY_HDR_EXT (dummy)->direction,
direction), 0);
gst_caps_unref (caps);
gst_object_unref (dummy);
}
GST_END_TEST;
static Suite *
rtp_header_extension_suite (void)
{
Suite *s = suite_create ("rtp_header_extension_test");
TCase *tc_chain = tcase_create ("header extension test");
suite_add_tcase (s, tc_chain);
tcase_add_test (tc_chain, rtp_header_ext_write);
tcase_add_test (tc_chain, rtp_header_ext_create_from_uri);
tcase_add_test (tc_chain, rtp_header_ext_caps_with_attributes);
return s;
}
GST_CHECK_MAIN (rtp_header_extension)