gstreamer/subprojects/gst-plugins-good/tests/check/elements/udpsrc.c
Seungha Yang 3374f2f44d udpsrc: Add support for IGMPv3 SSM
Adding "multicast-source" property to support Source Specific Muliticast
RFC 4604. The source can be multiple address with '+' (for positive
filter) or '-' (negative filter) prefix, or URI query can be used.
Note that negative filter is not implemented yet and it will be
ignored

Example:
gst-launch-1.0 uridecodebin \
  uri=udp://{ADDRESS}:PORT?multicast-source=+SOURCE0+SOURCE1

Inspired by:
https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/2620

Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/3485>
2023-04-12 16:32:07 +00:00

313 lines
9.3 KiB
C

/* GStreamer UDP source unit tests
* Copyright (C) 2011 Tim-Philipp Müller <tim centricular net>
*
* 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/check/gstcheck.h>
#include <gio/gio.h>
#include <stdlib.h>
static GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink",
GST_PAD_SINK,
GST_PAD_ALWAYS,
GST_STATIC_CAPS_ANY);
static gboolean
udpsrc_setup (GstElement ** udpsrc, GSocket ** socket,
GstPad ** sinkpad, GSocketAddress ** sa)
{
GInetAddress *ia;
int port = 0;
gchar *s;
*udpsrc = gst_check_setup_element ("udpsrc");
fail_unless (*udpsrc != NULL);
g_object_set (*udpsrc, "port", 0, NULL);
*sinkpad = gst_check_setup_sink_pad_by_name (*udpsrc, &sinktemplate, "src");
fail_unless (*sinkpad != NULL);
gst_pad_set_active (*sinkpad, TRUE);
gst_element_set_state (*udpsrc, GST_STATE_PLAYING);
g_object_get (*udpsrc, "port", &port, NULL);
GST_INFO ("udpsrc port = %d", port);
*socket = g_socket_new (G_SOCKET_FAMILY_IPV4, G_SOCKET_TYPE_DATAGRAM,
G_SOCKET_PROTOCOL_UDP, NULL);
if (*socket == NULL) {
GST_WARNING ("Could not create IPv4 UDP socket for unit test");
return FALSE;
}
ia = g_inet_address_new_loopback (G_SOCKET_FAMILY_IPV4);
s = g_inet_address_to_string (ia);
GST_LOG ("inet address %s", s);
g_free (s);
*sa = g_inet_socket_address_new (ia, port);
g_object_unref (ia);
return TRUE;
}
GST_START_TEST (test_udpsrc_empty_packet)
{
GSocketAddress *sa = NULL;
GstElement *udpsrc = NULL;
GSocket *socket = NULL;
GstPad *sinkpad = NULL;
if (!udpsrc_setup (&udpsrc, &socket, &sinkpad, &sa))
goto no_socket;
if (g_socket_send_to (socket, sa, "HeLL0", 0, NULL, NULL) == 0) {
GST_INFO ("sent 0 bytes");
if (g_socket_send_to (socket, sa, "HeLL0", 6, NULL, NULL) == 6) {
GstMapInfo map;
GstBuffer *buf;
guint len = 0;
GST_INFO ("sent 6 bytes");
g_mutex_lock (&check_mutex);
len = g_list_length (buffers);
while (len < 1) {
g_cond_wait (&check_cond, &check_mutex);
len = g_list_length (buffers);
GST_INFO ("%u buffers", len);
}
/* wait a bit more for a second buffer */
if (len < 2) {
g_cond_wait_until (&check_cond, &check_mutex,
g_get_monotonic_time () + G_TIME_SPAN_SECOND / 100);
len = g_list_length (buffers);
GST_INFO ("%u buffers", len);
}
fail_unless (len == 1 || len == 2);
/* last buffer should be our HeLL0 string */
buf = GST_BUFFER (g_list_nth_data (buffers, len - 1));
gst_buffer_map (buf, &map, GST_MAP_READ);
fail_unless_equals_int (map.size, 6);
fail_unless_equals_string ((gchar *) map.data, "HeLL0");
gst_buffer_unmap (buf, &map);
/* if there's another buffer, it should be 0 bytes */
if (len == 2) {
buf = GST_BUFFER (g_list_nth_data (buffers, 0));
fail_unless_equals_int (gst_buffer_get_size (buf), 0);
}
g_mutex_unlock (&check_mutex);
} else {
GST_WARNING ("send_to(6 bytes) failed");
}
} else {
GST_WARNING ("send_to(0 bytes) failed");
}
no_socket:
gst_element_set_state (udpsrc, GST_STATE_NULL);
gst_check_drop_buffers ();
gst_check_teardown_pad_by_name (udpsrc, "src");
gst_check_teardown_element (udpsrc);
g_object_unref (socket);
g_object_unref (sa);
}
GST_END_TEST;
GST_START_TEST (test_udpsrc)
{
GSocketAddress *sa = NULL;
GstElement *udpsrc = NULL;
GSocket *socket = NULL;
GstPad *sinkpad = NULL;
GstBuffer *buf;
GstMemory *mem;
gchar data[48000];
gsize max_size;
int i, len = 0;
gssize sent;
GError *err = NULL;
for (i = 0; i < G_N_ELEMENTS (data); ++i)
data[i] = i & 0xff;
if (!udpsrc_setup (&udpsrc, &socket, &sinkpad, &sa))
goto no_socket;
if ((sent = g_socket_send_to (socket, sa, data, 48000, NULL, &err)) == -1)
goto send_failure;
fail_unless_equals_int (sent, 48000);
if ((sent = g_socket_send_to (socket, sa, data, 21000, NULL, &err)) == -1)
goto send_failure;
fail_unless_equals_int (sent, 21000);
if ((sent = g_socket_send_to (socket, sa, data, 500, NULL, &err)) == -1)
goto send_failure;
fail_unless_equals_int (sent, 500);
if ((sent = g_socket_send_to (socket, sa, data, 1600, NULL, &err)) == -1)
goto send_failure;
fail_unless_equals_int (sent, 1600);
if ((sent = g_socket_send_to (socket, sa, data, 1400, NULL, &err)) == -1)
goto send_failure;
fail_unless_equals_int (sent, 1400);
GST_INFO ("sent some packets");
g_mutex_lock (&check_mutex);
len = g_list_length (buffers);
while (len < 5) {
g_cond_wait (&check_cond, &check_mutex);
len = g_list_length (buffers);
GST_INFO ("%u buffers", len);
}
/* check that large packets are made up of multiple memory chunks and that
* the first one is fairly small */
buf = GST_BUFFER (g_list_nth_data (buffers, 0));
fail_unless_equals_int (gst_buffer_get_size (buf), 48000);
fail_unless_equals_int (gst_buffer_n_memory (buf), 2);
mem = gst_buffer_peek_memory (buf, 0);
gst_memory_get_sizes (mem, NULL, &max_size);
fail_unless (max_size <= 2000);
buf = GST_BUFFER (g_list_nth_data (buffers, 1));
fail_unless_equals_int (gst_buffer_get_size (buf), 21000);
fail_unless_equals_int (gst_buffer_n_memory (buf), 2);
mem = gst_buffer_peek_memory (buf, 0);
gst_memory_get_sizes (mem, NULL, &max_size);
fail_unless (max_size <= 2000);
buf = GST_BUFFER (g_list_nth_data (buffers, 2));
fail_unless_equals_int (gst_buffer_get_size (buf), 500);
fail_unless_equals_int (gst_buffer_n_memory (buf), 1);
mem = gst_buffer_peek_memory (buf, 0);
gst_memory_get_sizes (mem, NULL, &max_size);
fail_unless (max_size <= 2000);
buf = GST_BUFFER (g_list_nth_data (buffers, 3));
fail_unless_equals_int (gst_buffer_get_size (buf), 1600);
fail_unless_equals_int (gst_buffer_n_memory (buf), 2);
mem = gst_buffer_peek_memory (buf, 0);
gst_memory_get_sizes (mem, NULL, &max_size);
fail_unless (max_size <= 2000);
buf = GST_BUFFER (g_list_nth_data (buffers, 4));
fail_unless_equals_int (gst_buffer_get_size (buf), 1400);
fail_unless_equals_int (gst_buffer_n_memory (buf), 1);
mem = gst_buffer_peek_memory (buf, 0);
gst_memory_get_sizes (mem, NULL, &max_size);
fail_unless (max_size <= 2000);
g_list_foreach (buffers, (GFunc) gst_buffer_unref, NULL);
g_list_free (buffers);
buffers = NULL;
g_mutex_unlock (&check_mutex);
no_socket:
send_failure:
if (err) {
GST_WARNING ("Socket send error, skipping test: %s", err->message);
g_clear_error (&err);
}
gst_element_set_state (udpsrc, GST_STATE_NULL);
gst_check_drop_buffers ();
gst_check_teardown_pad_by_name (udpsrc, "src");
gst_check_teardown_element (udpsrc);
g_object_unref (socket);
g_object_unref (sa);
}
GST_END_TEST;
static void
on_multicast_source_updated (GObject * src, GParamSpec * pspec, guint * count)
{
*count += 1;
}
GST_START_TEST (test_udpsrc_multicast_source)
{
GstElement *src;
guint count = 0;
gchar *multicast_source = NULL;
src = gst_check_setup_element ("udpsrc");
g_signal_connect (G_OBJECT (src), "notify::multicast-source",
(GCallback) on_multicast_source_updated, &count);
/* Set uri without multicast-source */
g_object_set (src, "uri", "udp://127.0.0.1:5004", NULL);
fail_unless_equals_int (count, 0);
g_object_get (src, "multicast-source", &multicast_source, NULL);
fail_unless (multicast_source == NULL);
/* Sets source filter explicitly */
g_object_set (src, "multicast-source", "+127.0.0.2+127.0.0.3", NULL);
fail_unless_equals_int (count, 1);
g_object_get (src, "multicast-source", &multicast_source, NULL);
fail_unless_equals_string (multicast_source, "+127.0.0.2+127.0.0.3");
g_clear_pointer (&multicast_source, g_free);
/* Uri with source filters */
g_object_set (src, "uri", "udp://127.0.0.1:5004?multicast-source=+127.0.0.2",
NULL);
fail_unless_equals_int (count, 2);
g_object_get (src, "multicast-source", &multicast_source, NULL);
fail_unless_equals_string (multicast_source, "+127.0.0.2");
g_clear_pointer (&multicast_source, g_free);
/* New uri will reset source filters */
g_object_set (src, "uri", "udp://127.0.0.1:5004", NULL);
fail_unless_equals_int (count, 3);
g_object_get (src, "multicast-source", &multicast_source, NULL);
fail_unless (multicast_source == NULL);
gst_object_unref (src);
}
GST_END_TEST;
static Suite *
udpsrc_suite (void)
{
Suite *s = suite_create ("udpsrc");
TCase *tc_chain = tcase_create ("udpsrc");
suite_add_tcase (s, tc_chain);
tcase_add_test (tc_chain, test_udpsrc_empty_packet);
tcase_add_test (tc_chain, test_udpsrc);
tcase_add_test (tc_chain, test_udpsrc_multicast_source);
return s;
}
GST_CHECK_MAIN (udpsrc)