gstreamer/tests/check/elements/dtls.c

225 lines
6.6 KiB
C
Raw Normal View History

/* GStreamer unit tests for the srtp elements
* Copyright (C) 2007 Tim-Philipp Müller <tim centricular net>
* Copyright (C) 2008 Sebastian Dröge <slomo@circular-chaos.org>
* Copyright (C) 2016 Collabora Ltd <vincent.penquerch@collabora.co.uk>
*
* 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.
*/
#ifdef HAVE_CONFIG_H
# include "config.h"
#endif
#ifdef HAVE_VALGRIND
# include <valgrind/valgrind.h>
#endif
#include <gst/check/gstcheck.h>
#include <gst/check/gstharness.h>
GST_START_TEST (test_create_and_unref)
{
GstElement *e;
e = gst_element_factory_make ("dtlsenc", NULL);
fail_unless (e != NULL);
gst_element_set_state (e, GST_STATE_NULL);
gst_object_unref (e);
e = gst_element_factory_make ("dtlsdec", NULL);
fail_unless (e != NULL);
gst_element_set_state (e, GST_STATE_NULL);
gst_object_unref (e);
}
GST_END_TEST;
static GMutex key_lock;
static GCond key_cond;
static int key_count;
static gboolean errored;
static GstBusSyncReply
bus_msg_handler (GstBus * bus, GstMessage * message, gpointer user_data)
{
if (GST_MESSAGE_TYPE (message) == GST_MESSAGE_ERROR) {
g_mutex_lock (&key_lock);
errored = TRUE;
g_cond_broadcast (&key_cond);
g_mutex_unlock (&key_lock);
}
return GST_BUS_PASS;
}
static void
_on_key_received (GstElement * element, gpointer user_data)
{
g_mutex_lock (&key_lock);
key_count++;
g_cond_broadcast (&key_cond);
g_mutex_unlock (&key_lock);
}
static void
_wait_for_key_count_to_reach (int n)
{
g_mutex_lock (&key_lock);
while (key_count < n) {
g_cond_wait (&key_cond, &key_lock);
/* Check if any errors were posted */
if (errored) {
g_mutex_unlock (&key_lock);
fail ("DTLS element posted an error");
g_mutex_lock (&key_lock);
}
}
g_mutex_unlock (&key_lock);
}
static gchar data[] = {
0x00, 0x01, 0x02, 0x03,
};
GST_START_TEST (test_data_transfer)
{
GstHarness *server, *client;
GstElement *s_enc, *s_dec, *c_enc, *c_dec, *s_bin, *c_bin;
GstPad *target, *ghost;
GstBuffer *buffer, *buf2;
GstBus *bus;
/* setup a server and client for dtls negotiation */
s_bin = gst_bin_new (NULL);
c_bin = gst_bin_new (NULL);
/* XXX: the element set states are needed to avoid a runtime warning:
*
* 'gst_dtls_connection_process: runtime check failed: (!priv->bio_buffer)'
*
* where the encoder needs to be started (and SSL initialized) before the
* associated decoder receives any data and calls gst_dtls_connection_process().
*/
s_dec = gst_element_factory_make ("dtlsdec", "server_dec");
g_object_set (s_dec, "connection-id", "server", NULL);
g_signal_connect (s_dec, "on-key-received", G_CALLBACK (_on_key_received),
NULL);
gst_element_set_state (s_dec, GST_STATE_PAUSED);
gst_bin_add (GST_BIN (s_bin), s_dec);
s_enc = gst_element_factory_make ("dtlsenc", "server_enc");
g_object_set (s_enc, "connection-id", "server", NULL);
g_signal_connect (s_enc, "on-key-received", G_CALLBACK (_on_key_received),
NULL);
gst_element_set_state (s_enc, GST_STATE_PAUSED);
gst_bin_add (GST_BIN (c_bin), s_enc);
c_dec = gst_element_factory_make ("dtlsdec", "client_dec");
g_object_set (c_dec, "connection-id", "client", NULL);
g_signal_connect (c_dec, "on-key-received", G_CALLBACK (_on_key_received),
NULL);
gst_element_set_state (c_dec, GST_STATE_PAUSED);
gst_bin_add (GST_BIN (c_bin), c_dec);
c_enc = gst_element_factory_make ("dtlsenc", "client_enc");
g_object_set (c_enc, "connection-id", "client", "is-client", TRUE, NULL);
g_signal_connect (c_enc, "on-key-received", G_CALLBACK (_on_key_received),
NULL);
gst_bin_add (GST_BIN (s_bin), c_enc);
gst_element_link_pads (s_enc, "src", c_dec, "sink");
gst_element_link_pads (c_enc, "src", s_dec, "sink");
gst_element_set_state (c_enc, GST_STATE_PAUSED);
target = gst_element_request_pad_simple (c_dec, "src");
ghost = gst_ghost_pad_new ("src", target);
gst_element_add_pad (s_bin, ghost);
gst_object_unref (target);
target = gst_element_request_pad_simple (s_enc, "sink");
ghost = gst_ghost_pad_new ("sink", target);
gst_element_add_pad (s_bin, ghost);
gst_object_unref (target);
target = gst_element_request_pad_simple (s_dec, "src");
ghost = gst_ghost_pad_new ("src", target);
gst_element_add_pad (c_bin, ghost);
gst_object_unref (target);
target = gst_element_request_pad_simple (c_enc, "sink");
ghost = gst_ghost_pad_new ("sink", target);
gst_element_add_pad (c_bin, ghost);
gst_object_unref (target);
server = gst_harness_new_with_element (s_bin, "sink", "src");
client = gst_harness_new_with_element (c_bin, "sink", "src");
gst_harness_set_src_caps_str (server, "application/data");
gst_harness_set_src_caps_str (client, "application/data");
bus = gst_bus_new ();
gst_bus_set_sync_handler (bus, bus_msg_handler, NULL, NULL);
gst_element_set_bus (s_bin, bus);
gst_element_set_bus (c_bin, bus);
gst_object_unref (bus);
_wait_for_key_count_to_reach (4);
buffer = gst_buffer_new_wrapped_full (GST_MEMORY_FLAG_READONLY, data,
G_N_ELEMENTS (data), 0, G_N_ELEMENTS (data), NULL, NULL);
gst_harness_push (server, gst_buffer_ref (buffer));
buf2 = gst_harness_pull (server);
fail_unless_equals_int (0, gst_buffer_memcmp (buf2, 0, data,
G_N_ELEMENTS (data)));
gst_buffer_unref (buf2);
gst_harness_play (client);
gst_harness_push (client, gst_buffer_ref (buffer));
buf2 = gst_harness_pull (client);
fail_unless_equals_int (0, gst_buffer_memcmp (buf2, 0, data,
G_N_ELEMENTS (data)));
gst_buffer_unref (buf2);
gst_object_unref (s_bin);
gst_object_unref (c_bin);
gst_buffer_unref (buffer);
gst_harness_teardown (server);
gst_harness_teardown (client);
}
GST_END_TEST;
static Suite *
dtls_suite (void)
{
Suite *s = suite_create ("dtls");
TCase *tc_chain = tcase_create ("general");
suite_add_tcase (s, tc_chain);
tcase_add_test (tc_chain, test_create_and_unref);
tcase_add_test (tc_chain, test_data_transfer);
return s;
}
GST_CHECK_MAIN (dtls);