mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-09-09 05:38:58 +00:00
262 lines
7.6 KiB
C
262 lines
7.6 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 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);
|
|
return s;
|
|
}
|
|
|
|
GST_CHECK_MAIN (udpsrc)
|