rtspconnection: Add more tests

Fixes https://bugzilla.gnome.org/show_bug.cgi?id=728907
This commit is contained in:
Ognyan Tonchev 2014-03-16 16:09:36 +01:00 committed by Wim Taymans
parent 954db90918
commit 6750531504

View file

@ -33,21 +33,19 @@
static const gchar *get_msg =
"GET /example/url HTTP/1.0\r\n"
"Host: 127.0.0.1\r\n"
"x-sessioncookie: 805849328\r\n"
"\r\n";
"Host: 127.0.0.1\r\n" "x-sessioncookie: 805849328\r\n\r\n";
static const gchar *post_msg =
"POST /example/url HTTP/1.0\r\n"
"Host: 127.0.0.1\r\n"
"x-sessioncookie: 805849328\r\n"
"Content-Length: 0\r\n"
"Content-Type: application/x-rtsp-tunnelled\r\n"
"\r\n";
"Content-Type: application/x-rtsp-tunnelled\r\n\r\n";
static guint tunnel_start_count;
static guint tunnel_complete_count;
static guint tunnel_lost_count;
static guint closed_count;
static guint message_sent_count;
typedef struct
{
@ -60,8 +58,8 @@ typedef struct
} ServiceData;
static gboolean
incoming_callback (GSocketService *service, GSocketConnection *connection,
GObject *source_object, gpointer user_data)
incoming_callback (GSocketService * service, GSocketConnection * connection,
GObject * source_object, gpointer user_data)
{
ServiceData *data = user_data;
@ -85,7 +83,7 @@ service_thread_func (gpointer user_data)
/* find available port and start service */
service = g_socket_service_new ();
data->port = g_socket_listener_add_any_inet_port ((GSocketListener *)service,
data->port = g_socket_listener_add_any_inet_port ((GSocketListener *) service,
NULL, NULL);
fail_unless (data->port != 0);
@ -112,14 +110,14 @@ service_thread_func (gpointer user_data)
}
static void
create_connection (GSocketConnection **client_conn,
GSocketConnection **server_conn)
create_connection (GSocketConnection ** client_conn,
GSocketConnection ** server_conn)
{
ServiceData *data;
GThread *service_thread;
GSocketClient * client = g_socket_client_new ();
GSocketClient *client = g_socket_client_new ();
data = g_new0 (ServiceData, 1);
data = g_new0 (ServiceData, 1);
g_mutex_init (&data->mutex);
g_cond_init (&data->cond);
@ -134,7 +132,7 @@ create_connection (GSocketConnection **client_conn,
g_mutex_unlock (&data->mutex);
/* create the tcp link */
*client_conn = g_socket_client_connect_to_host (client, (gchar *)"localhost",
*client_conn = g_socket_client_connect_to_host (client, (gchar *) "localhost",
data->port, NULL, NULL);
fail_unless (*client_conn != NULL);
fail_unless (g_socket_connection_is_connected (*client_conn));
@ -151,36 +149,43 @@ create_connection (GSocketConnection **client_conn,
}
static GstRTSPStatusCode
tunnel_start (GstRTSPWatch *watch, gpointer user_data)
tunnel_start (GstRTSPWatch * watch, gpointer user_data)
{
tunnel_start_count++;
return GST_RTSP_STS_OK;
}
static GstRTSPResult
tunnel_complete (GstRTSPWatch *watch, gpointer user_data)
tunnel_complete (GstRTSPWatch * watch, gpointer user_data)
{
tunnel_complete_count++;
return GST_RTSP_OK;
}
static GstRTSPResult
tunnel_lost (GstRTSPWatch *watch, gpointer user_data)
tunnel_lost (GstRTSPWatch * watch, gpointer user_data)
{
tunnel_lost_count++;
return GST_RTSP_OK;
}
static GstRTSPResult
closed (GstRTSPWatch *watch, gpointer user_data)
closed (GstRTSPWatch * watch, gpointer user_data)
{
closed_count++;
return GST_RTSP_OK;
}
static GstRTSPResult
message_sent (GstRTSPWatch * watch, guint id, gpointer user_data)
{
message_sent_count++;
return GST_RTSP_OK;
}
static GstRTSPWatchFuncs watch_funcs = {
NULL,
NULL,
message_sent,
closed,
NULL,
tunnel_start,
@ -222,6 +227,7 @@ GST_START_TEST (test_rtspconnection_tunnel_setup)
watch1 = gst_rtsp_watch_new (rtsp_conn1, &watch_funcs, NULL, NULL);
fail_unless (watch1 != NULL);
fail_unless (gst_rtsp_watch_attach (watch1, NULL) > 0);
g_source_unref ((GSource *) watch1);
ostream_get = g_io_stream_get_output_stream (G_IO_STREAM (client_get));
fail_unless (ostream_get != NULL);
@ -231,7 +237,7 @@ GST_START_TEST (test_rtspconnection_tunnel_setup)
/* initiate the tunnel by sending HTTP GET */
fail_unless (g_output_stream_write_all (ostream_get, get_msg,
strlen (get_msg), &size, NULL, NULL));
strlen (get_msg), &size, NULL, NULL));
fail_unless (size == strlen (get_msg));
while (!g_main_context_iteration (NULL, TRUE));
@ -259,13 +265,14 @@ GST_START_TEST (test_rtspconnection_tunnel_setup)
watch2 = gst_rtsp_watch_new (rtsp_conn2, &watch_funcs, NULL, NULL);
fail_unless (watch2 != NULL);
fail_unless (gst_rtsp_watch_attach (watch2, NULL) > 0);
g_source_unref ((GSource *) watch2);
ostream_post = g_io_stream_get_output_stream (G_IO_STREAM (client_post));
fail_unless (ostream_post != NULL);
/* complete the tunnel by sending HTTP POST */
fail_unless (g_output_stream_write_all (ostream_post, post_msg,
strlen (post_msg), &size, NULL, NULL));
strlen (post_msg), &size, NULL, NULL));
fail_unless (size == strlen (post_msg));
while (!g_main_context_iteration (NULL, TRUE));
@ -278,7 +285,7 @@ GST_START_TEST (test_rtspconnection_tunnel_setup)
fail_unless (gst_rtsp_connection_do_tunnel (rtsp_conn1, rtsp_conn2) ==
GST_RTSP_OK);
gst_rtsp_watch_reset (watch1);
g_source_destroy ((GSource *)watch2);
g_source_destroy ((GSource *) watch2);
gst_rtsp_connection_free (rtsp_conn2);
rtsp_conn2 = NULL;
@ -307,13 +314,14 @@ GST_START_TEST (test_rtspconnection_tunnel_setup)
watch2 = gst_rtsp_watch_new (rtsp_conn2, &watch_funcs, NULL, NULL);
fail_unless (watch2 != NULL);
fail_unless (gst_rtsp_watch_attach (watch2, NULL) > 0);
g_source_unref ((GSource *) watch2);
ostream_post = g_io_stream_get_output_stream (G_IO_STREAM (client_post));
fail_unless (ostream_post != NULL);
/* complete the tunnel by sending HTTP POST */
fail_unless (g_output_stream_write_all (ostream_post, post_msg,
strlen (post_msg), &size, NULL, NULL));
strlen (post_msg), &size, NULL, NULL));
fail_unless (size == strlen (post_msg));
while (!g_main_context_iteration (NULL, TRUE));
@ -326,7 +334,7 @@ GST_START_TEST (test_rtspconnection_tunnel_setup)
fail_unless (gst_rtsp_connection_do_tunnel (rtsp_conn1, rtsp_conn2) ==
GST_RTSP_OK);
gst_rtsp_watch_reset (watch1);
g_source_destroy ((GSource *)watch2);
g_source_destroy ((GSource *) watch2);
gst_rtsp_connection_free (rtsp_conn2);
rtsp_conn2 = NULL;
@ -348,6 +356,285 @@ GST_START_TEST (test_rtspconnection_tunnel_setup)
GST_END_TEST;
GST_START_TEST (test_rtspconnection_send_receive)
{
GSocketConnection *input_conn = NULL;
GSocketConnection *output_conn = NULL;
GSocket *input_sock;
GSocket *output_sock;
GstRTSPConnection *rtsp_output_conn;
GstRTSPConnection *rtsp_input_conn;
GstRTSPMessage *msg;
gchar body[] = "message body";
gchar *recv_body;
guint recv_body_len;
create_connection (&input_conn, &output_conn);
input_sock = g_socket_connection_get_socket (input_conn);
fail_unless (input_sock != NULL);
output_sock = g_socket_connection_get_socket (output_conn);
fail_unless (output_sock != NULL);
fail_unless (gst_rtsp_connection_create_from_socket (input_sock, "127.0.0.1",
4444, NULL, &rtsp_input_conn) == GST_RTSP_OK);
fail_unless (rtsp_input_conn != NULL);
fail_unless (gst_rtsp_connection_create_from_socket (output_sock, "127.0.0.1",
4444, NULL, &rtsp_output_conn) == GST_RTSP_OK);
fail_unless (rtsp_output_conn != NULL);
/* send data message */
fail_unless (gst_rtsp_message_new_data (&msg, 1) == GST_RTSP_OK);
fail_unless (gst_rtsp_message_set_body (msg, (guint8 *) body,
sizeof (body)) == GST_RTSP_OK);
fail_unless (gst_rtsp_connection_send (rtsp_output_conn, msg,
NULL) == GST_RTSP_OK);
fail_unless (gst_rtsp_message_free (msg) == GST_RTSP_OK);
msg = NULL;
/* receive data message and make sure it is correct */
fail_unless (gst_rtsp_message_new (&msg) == GST_RTSP_OK);
fail_unless (gst_rtsp_connection_receive (rtsp_input_conn, msg, NULL) ==
GST_RTSP_OK);
fail_unless (gst_rtsp_message_get_type (msg) == GST_RTSP_MESSAGE_DATA);
fail_unless (gst_rtsp_message_get_body (msg, (guint8 **) & recv_body,
&recv_body_len) == GST_RTSP_OK);
/* RTSPConnection adds an extra byte for the trailing '\0' */
fail_unless_equals_int (recv_body_len, sizeof (body) + 1);
fail_unless_equals_string (recv_body, body);
fail_unless (gst_rtsp_message_free (msg) == GST_RTSP_OK);
msg = NULL;
/* send request message */
fail_unless (gst_rtsp_message_new_request (&msg, GST_RTSP_OPTIONS,
"example.org") == GST_RTSP_OK);
fail_unless (gst_rtsp_message_set_body (msg, (guint8 *) body,
sizeof (body)) == GST_RTSP_OK);
fail_unless (gst_rtsp_connection_send (rtsp_output_conn, msg,
NULL) == GST_RTSP_OK);
fail_unless (gst_rtsp_message_free (msg) == GST_RTSP_OK);
msg = NULL;
/* receive request message and make sure it is correct */
fail_unless (gst_rtsp_message_new (&msg) == GST_RTSP_OK);
fail_unless (gst_rtsp_connection_receive (rtsp_input_conn, msg, NULL) ==
GST_RTSP_OK);
fail_unless (gst_rtsp_message_get_type (msg) == GST_RTSP_MESSAGE_REQUEST);
fail_unless (gst_rtsp_message_get_body (msg, (guint8 **) & recv_body,
&recv_body_len) == GST_RTSP_OK);
/* RTSPConnection adds an extra byte for the trailing '\0' */
fail_unless_equals_int (recv_body_len, sizeof (body) + 1);
fail_unless_equals_string (recv_body, body);
fail_unless (gst_rtsp_message_free (msg) == GST_RTSP_OK);
msg = NULL;
fail_unless (gst_rtsp_connection_close (rtsp_input_conn) == GST_RTSP_OK);
fail_unless (gst_rtsp_connection_free (rtsp_input_conn) == GST_RTSP_OK);
fail_unless (gst_rtsp_connection_close (rtsp_output_conn) == GST_RTSP_OK);
fail_unless (gst_rtsp_connection_free (rtsp_output_conn) == GST_RTSP_OK);
g_object_unref (input_conn);
g_object_unref (output_conn);
}
GST_END_TEST;
GST_START_TEST (test_rtspconnection_connect)
{
ServiceData *data;
GThread *service_thread;
GSocketConnection *socket_conn;
GstRTSPConnection *rtsp_conn = NULL;
GstRTSPUrl *url = NULL;
gchar *path;
data = g_new0 (ServiceData, 1);
g_mutex_init (&data->mutex);
g_cond_init (&data->cond);
/* create socket service */
service_thread = g_thread_new ("service thread", service_thread_func, data);
fail_unless (service_thread != NULL);
/* wait for the service to start */
g_mutex_lock (&data->mutex);
while (!data->started) {
g_cond_wait (&data->cond, &data->mutex);
}
g_mutex_unlock (&data->mutex);
/* connect to our service using the RTSPConnection API */
path = g_strdup_printf ("rtsp://localhost:%d", data->port);
fail_unless (gst_rtsp_url_parse (path, &url) == GST_RTSP_OK);
fail_unless (gst_rtsp_connection_create (url, &rtsp_conn) == GST_RTSP_OK);
fail_unless (gst_rtsp_connection_connect (rtsp_conn, NULL) == GST_RTSP_OK);
g_free (path);
gst_rtsp_url_free (url);
/* wait for the other end and check whether it is connected */
g_thread_join (service_thread);
socket_conn = data->conn;
data->conn = NULL;
fail_unless (g_socket_connection_is_connected (socket_conn));
fail_unless (gst_rtsp_connection_close (rtsp_conn) == GST_RTSP_OK);
fail_unless (gst_rtsp_connection_free (rtsp_conn) == GST_RTSP_OK);
g_object_unref (socket_conn);
g_mutex_clear (&data->mutex);
g_cond_clear (&data->cond);
g_free (data);
}
GST_END_TEST;
GST_START_TEST (test_rtspconnection_poll)
{
GSocketConnection *conn1 = NULL;
GSocketConnection *conn2 = NULL;
GSocket *sock;
GstRTSPConnection *rtsp_conn;
GstRTSPEvent event;
GOutputStream *ostream;
gsize size;
GTimeVal tv;
create_connection (&conn1, &conn2);
sock = g_socket_connection_get_socket (conn1);
fail_unless (sock != NULL);
ostream = g_io_stream_get_output_stream (G_IO_STREAM (conn2));
fail_unless (ostream != NULL);
fail_unless (gst_rtsp_connection_create_from_socket (sock, "127.0.0.1",
4444, NULL, &rtsp_conn) == GST_RTSP_OK);
fail_unless (rtsp_conn != NULL);
/* should be possible to write on socket */
fail_unless (gst_rtsp_connection_poll (rtsp_conn, GST_RTSP_EV_WRITE, &event,
NULL) == GST_RTSP_OK);
fail_unless (event & GST_RTSP_EV_WRITE);
/* but not read, add timeout so that we don't block forever */
tv.tv_sec = 1;
tv.tv_usec = 0;
fail_unless (gst_rtsp_connection_poll (rtsp_conn, GST_RTSP_EV_READ, &event,
&tv) == GST_RTSP_ETIMEOUT);
fail_if (event & GST_RTSP_EV_READ);
/* write on the other end and make sure socket can be read */
fail_unless (g_output_stream_write_all (ostream, "data", 5, &size, NULL,
NULL));
fail_unless (gst_rtsp_connection_poll (rtsp_conn, GST_RTSP_EV_READ, &event,
NULL) == GST_RTSP_OK);
fail_unless (event & GST_RTSP_EV_READ);
fail_unless (gst_rtsp_connection_close (rtsp_conn) == GST_RTSP_OK);
fail_unless (gst_rtsp_connection_free (rtsp_conn) == GST_RTSP_OK);
g_object_unref (conn1);
g_object_unref (conn2);
}
GST_END_TEST;
GST_START_TEST (test_rtspconnection_backlog)
{
GSocketConnection *conn1 = NULL;
GSocketConnection *conn2 = NULL;
GSocket *sock;
GstRTSPConnection *rtsp_conn = NULL;
GstRTSPWatch *watch;
GInputStream *istream;
guint8 *buffer;
guint8 recv[1024];
gsize count;
GstRTSPResult res = GST_RTSP_OK;
guint num_queued;
guint num_sent;
create_connection (&conn1, &conn2);
sock = g_socket_connection_get_socket (conn1);
fail_unless (sock != NULL);
fail_unless (gst_rtsp_connection_create_from_socket (sock, "127.0.0.1",
4444, NULL, &rtsp_conn) == GST_RTSP_OK);
fail_unless (rtsp_conn != NULL);
watch = gst_rtsp_watch_new (rtsp_conn, &watch_funcs, NULL, NULL);
fail_unless (watch != NULL);
fail_unless (gst_rtsp_watch_attach (watch, NULL) > 0);
g_source_unref ((GSource *) watch);
gst_rtsp_watch_set_send_backlog (watch, 1024, 0);
/* write until we fill tcp window and writes result in would_block,
* data will then start getting queued until the backlog also gets full */
num_queued = 0;
num_sent = 0;
while (res == GST_RTSP_OK) {
guint id = 0;
buffer = malloc (1024);
memset (buffer, 0, 1024);
res = gst_rtsp_watch_write_data (watch, buffer, 1024, &id);
if (id > 0)
num_queued++;
if (res == GST_RTSP_OK)
num_sent++;
}
/* make sure we got enomem and at least 1 message got queued */
fail_unless (res == GST_RTSP_ENOMEM);
fail_unless (num_queued > 0);
istream = g_io_stream_get_input_stream (G_IO_STREAM (conn2));
fail_unless (istream != NULL);
/* read a bit from the socket and make sure queued data gets sent */
while (num_queued > 0) {
fail_unless (g_input_stream_read_all (istream, recv, 1024, &count, NULL,
NULL));
num_sent--;
g_main_context_iteration (NULL, FALSE);
num_queued -= message_sent_count;
fail_unless (num_queued >= 0);
}
/* make sure we can read the rest of the data */
while (num_sent > 0) {
fail_unless (g_input_stream_read_all (istream, recv, 1024, &count, NULL,
NULL));
num_sent--;
}
g_source_destroy ((GSource *) watch);
fail_unless (gst_rtsp_connection_close (rtsp_conn) == GST_RTSP_OK);
fail_unless (gst_rtsp_connection_free (rtsp_conn) == GST_RTSP_OK);
g_object_unref (conn1);
g_object_unref (conn2);
}
GST_END_TEST;
GST_START_TEST (test_rtspconnection_ip)
{
GstRTSPConnection *conn = NULL;
GstRTSPUrl *url = NULL;
fail_unless (gst_rtsp_url_parse ("rtsp://127.0.0.1:42", &url) == GST_RTSP_OK);
fail_unless (url != NULL);
fail_unless (gst_rtsp_connection_create (url, &conn) == GST_RTSP_OK);
fail_unless (conn != NULL);
gst_rtsp_connection_set_ip (conn, "127.0.0.1");
fail_unless_equals_string (gst_rtsp_connection_get_ip (conn), "127.0.0.1");
gst_rtsp_url_free (url);
fail_unless (gst_rtsp_connection_free (conn) == GST_RTSP_OK);
}
GST_END_TEST;
static Suite *
rtspconnection_suite (void)
{
@ -356,6 +643,11 @@ rtspconnection_suite (void)
suite_add_tcase (s, tc_chain);
tcase_add_test (tc_chain, test_rtspconnection_tunnel_setup);
tcase_add_test (tc_chain, test_rtspconnection_send_receive);
tcase_add_test (tc_chain, test_rtspconnection_connect);
tcase_add_test (tc_chain, test_rtspconnection_poll);
tcase_add_test (tc_chain, test_rtspconnection_backlog);
tcase_add_test (tc_chain, test_rtspconnection_ip);
return s;
}