tests: Add tests for client-requested multicast address

diff --git a/tests/check/gst/client.c b/tests/check/gst/client.c
index 59d5972..e59e06a 100644 (file)
--- a/tests/check/gst/client.c
+++ b/tests/check/gst/client.c
@@ -215,6 +215,328 @@ GST_START_TEST (test_describe)
 
 GST_END_TEST;
 
+gchar *expected_transport = NULL;;
+
+static gboolean
+test_setup_response_200_multicast (GstRTSPClient * client,
+    GstRTSPMessage * response, gboolean close, gpointer user_data)
+{
+  GstRTSPStatusCode code;
+  const gchar *reason;
+  GstRTSPVersion version;
+  gchar *str;
+  GstRTSPSessionPool *session_pool;
+  GstRTSPSession *session;
+
+  fail_unless (expected_transport != NULL);
+
+  fail_unless (gst_rtsp_message_get_type (response) ==
+      GST_RTSP_MESSAGE_RESPONSE);
+
+  fail_unless (gst_rtsp_message_parse_response (response, &code, &reason,
+          &version)
+      == GST_RTSP_OK);
+  fail_unless (code == GST_RTSP_STS_OK);
+  fail_unless (g_str_equal (reason, "OK"));
+  fail_unless (version == GST_RTSP_VERSION_1_0);
+
+  fail_unless (gst_rtsp_message_get_header (response, GST_RTSP_HDR_CSEQ, &str,
+          0) == GST_RTSP_OK);
+  fail_unless (atoi (str) == cseq++);
+
+  fail_unless (gst_rtsp_message_get_header (response, GST_RTSP_HDR_TRANSPORT,
+          &str, 0) == GST_RTSP_OK);
+
+  fail_unless (!strcmp (str, expected_transport));
+
+  fail_unless (gst_rtsp_message_get_header (response, GST_RTSP_HDR_SESSION,
+          &str, 0) == GST_RTSP_OK);
+
+  session_pool = gst_rtsp_client_get_session_pool (client);
+  fail_unless (session_pool != NULL);
+
+  fail_unless (gst_rtsp_session_pool_get_n_sessions (session_pool) == 1);
+  session = gst_rtsp_session_pool_find (session_pool, str);
+  fail_unless (session != NULL);
+  g_object_unref (session);
+
+  g_object_unref (session_pool);
+
+
+  return TRUE;
+}
+
+static GstRTSPClient *
+setup_multicast_client (void)
+{
+  GstRTSPClient *client;
+  GstRTSPSessionPool *session_pool;
+  GstRTSPMountPoints *mount_points;
+  GstRTSPMediaFactory *factory;
+  GstRTSPAddressPool *address_pool;
+
+  client = gst_rtsp_client_new ();
+
+  session_pool = gst_rtsp_session_pool_new ();
+  gst_rtsp_client_set_session_pool (client, session_pool);
+
+  mount_points = gst_rtsp_mount_points_new ();
+  factory = gst_rtsp_media_factory_new ();
+  gst_rtsp_media_factory_set_launch (factory,
+      "audiotestsrc ! audio/x-raw,rate=44100 ! audioconvert ! rtpL16pay name=pay0");
+  address_pool = gst_rtsp_address_pool_new ();
+  fail_unless (gst_rtsp_address_pool_add_range (address_pool,
+          "233.252.0.1", "233.252.0.1", 5000, 5010, 1));
+  gst_rtsp_media_factory_set_address_pool (factory, address_pool);
+  gst_rtsp_mount_points_add_factory (mount_points, "/test", factory);
+  gst_rtsp_client_set_mount_points (client, mount_points);
+
+  g_object_unref (mount_points);
+  g_object_unref (session_pool);
+  g_object_unref (address_pool);
+
+  return client;
+}
+
+GST_START_TEST (test_client_multicast_transport_404)
+{
+  GstRTSPClient *client;
+  GstRTSPMessage request = { 0, };
+  gchar *str;
+
+  client = setup_multicast_client ();
+
+  /* simple SETUP for non-existing url */
+  fail_unless (gst_rtsp_message_init_request (&request, GST_RTSP_SETUP,
+          "rtsp://localhost/test2/stream=0") == GST_RTSP_OK);
+  str = g_strdup_printf ("%d", cseq);
+  gst_rtsp_message_take_header (&request, GST_RTSP_HDR_CSEQ, str);
+  gst_rtsp_message_add_header (&request, GST_RTSP_HDR_TRANSPORT,
+      "RTP/AVP;multicast");
+
+  gst_rtsp_client_set_send_func (client, test_response_404, NULL, NULL);
+  fail_unless (gst_rtsp_client_handle_message (client,
+          &request) == GST_RTSP_OK);
+  gst_rtsp_message_unset (&request);
+
+  g_object_unref (client);
+}
+
+GST_END_TEST;
+
+GST_START_TEST (test_client_multicast_transport)
+{
+  GstRTSPClient *client;
+  GstRTSPMessage request = { 0, };
+  gchar *str;
+
+  client = setup_multicast_client ();
+
+  /* simple SETUP with a valid URI and multicast */
+  fail_unless (gst_rtsp_message_init_request (&request, GST_RTSP_SETUP,
+          "rtsp://localhost/test/stream=0") == GST_RTSP_OK);
+  str = g_strdup_printf ("%d", cseq);
+  gst_rtsp_message_take_header (&request, GST_RTSP_HDR_CSEQ, str);
+  gst_rtsp_message_add_header (&request, GST_RTSP_HDR_TRANSPORT,
+      "RTP/AVP;multicast");
+
+  expected_transport = "RTP/AVP;multicast;destination=233.252.0.1;"
+      "ttl=1;port=5000-5001;mode=\"PLAY\"";
+  gst_rtsp_client_set_send_func (client, test_setup_response_200_multicast,
+      NULL, NULL);
+  fail_unless (gst_rtsp_client_handle_message (client,
+          &request) == GST_RTSP_OK);
+  gst_rtsp_message_unset (&request);
+  expected_transport = NULL;
+
+  g_object_unref (client);
+}
+
+GST_END_TEST;
+
+GST_START_TEST (test_client_multicast_ignore_transport_specific)
+{
+  GstRTSPClient *client;
+  GstRTSPMessage request = { 0, };
+  gchar *str;
+
+  client = setup_multicast_client ();
+
+  /* simple SETUP with a valid URI and multicast and a specific dest,
+   * but ignore it  */
+  fail_unless (gst_rtsp_message_init_request (&request, GST_RTSP_SETUP,
+          "rtsp://localhost/test/stream=0") == GST_RTSP_OK);
+  str = g_strdup_printf ("%d", cseq);
+  gst_rtsp_message_take_header (&request, GST_RTSP_HDR_CSEQ, str);
+  gst_rtsp_message_add_header (&request, GST_RTSP_HDR_TRANSPORT,
+      "RTP/AVP;multicast;destination=233.252.0.2;ttl=2;port=5001-5006;");
+
+  expected_transport = "RTP/AVP;multicast;destination=233.252.0.1;"
+      "ttl=1;port=5000-5001;mode=\"PLAY\"";
+  gst_rtsp_client_set_send_func (client, test_setup_response_200_multicast,
+      NULL, NULL);
+  fail_unless (gst_rtsp_client_handle_message (client,
+          &request) == GST_RTSP_OK);
+  gst_rtsp_message_unset (&request);
+  expected_transport = NULL;
+
+  g_object_unref (client);
+}
+
+GST_END_TEST;
+
+static gboolean
+test_setup_response_461 (GstRTSPClient * client,
+    GstRTSPMessage * response, gboolean close, gpointer user_data)
+{
+  GstRTSPStatusCode code;
+  const gchar *reason;
+  GstRTSPVersion version;
+  gchar *str;
+
+  fail_unless (expected_transport == NULL);
+
+  fail_unless (gst_rtsp_message_get_type (response) ==
+      GST_RTSP_MESSAGE_RESPONSE);
+
+  fail_unless (gst_rtsp_message_parse_response (response, &code, &reason,
+          &version)
+      == GST_RTSP_OK);
+  fail_unless (code == GST_RTSP_STS_UNSUPPORTED_TRANSPORT);
+  fail_unless (g_str_equal (reason, "Unsupported transport"));
+  fail_unless (version == GST_RTSP_VERSION_1_0);
+
+  fail_unless (gst_rtsp_message_get_header (response, GST_RTSP_HDR_CSEQ, &str,
+          0) == GST_RTSP_OK);
+  fail_unless (atoi (str) == cseq++);
+
+
+  return TRUE;
+}
+
+GST_START_TEST (test_client_multicast_invalid_transport_specific)
+{
+  GstRTSPClient *client;
+  GstRTSPMessage request = { 0, };
+  gchar *str;
+  GstRTSPSessionPool *session_pool;
+
+  client = setup_multicast_client ();
+
+  gst_rtsp_client_set_use_client_settings (client, TRUE);
+  fail_unless (gst_rtsp_client_get_use_client_settings (client));
+
+
+  /* simple SETUP with a valid URI and multicast, but an invalid ip */
+  fail_unless (gst_rtsp_message_init_request (&request, GST_RTSP_SETUP,
+          "rtsp://localhost/test/stream=0") == GST_RTSP_OK);
+  str = g_strdup_printf ("%d", cseq);
+  gst_rtsp_message_take_header (&request, GST_RTSP_HDR_CSEQ, str);
+  gst_rtsp_message_add_header (&request, GST_RTSP_HDR_TRANSPORT,
+      "RTP/AVP;multicast;destination=233.252.0.2;ttl=1;port=5000-5001;");
+
+  gst_rtsp_client_set_send_func (client, test_setup_response_461, NULL, NULL);
+  fail_unless (gst_rtsp_client_handle_message (client,
+          &request) == GST_RTSP_OK);
+  gst_rtsp_message_unset (&request);
+
+  session_pool = gst_rtsp_client_get_session_pool (client);
+  fail_unless (session_pool != NULL);
+  /* FIXME: There seems to be a leak of a session here ! */
+  /* fail_unless (gst_rtsp_session_pool_get_n_sessions (session_pool) == 0); */
+  g_object_unref (session_pool);
+
+
+
+  /* simple SETUP with a valid URI and multicast, but an invalid prt */
+  fail_unless (gst_rtsp_message_init_request (&request, GST_RTSP_SETUP,
+          "rtsp://localhost/test/stream=0") == GST_RTSP_OK);
+  str = g_strdup_printf ("%d", cseq);
+  gst_rtsp_message_take_header (&request, GST_RTSP_HDR_CSEQ, str);
+  gst_rtsp_message_add_header (&request, GST_RTSP_HDR_TRANSPORT,
+      "RTP/AVP;multicast;destination=233.252.0.1;ttl=1;port=6000-6001;");
+
+  gst_rtsp_client_set_send_func (client, test_setup_response_461, NULL, NULL);
+  fail_unless (gst_rtsp_client_handle_message (client,
+          &request) == GST_RTSP_OK);
+  gst_rtsp_message_unset (&request);
+
+  session_pool = gst_rtsp_client_get_session_pool (client);
+  fail_unless (session_pool != NULL);
+  /* FIXME: There seems to be a leak of a session here ! */
+  /* fail_unless (gst_rtsp_session_pool_get_n_sessions (session_pool) == 0); */
+  g_object_unref (session_pool);
+
+
+
+  /* simple SETUP with a valid URI and multicast, but an invalid ttl */
+  fail_unless (gst_rtsp_message_init_request (&request, GST_RTSP_SETUP,
+          "rtsp://localhost/test/stream=0") == GST_RTSP_OK);
+  str = g_strdup_printf ("%d", cseq);
+  gst_rtsp_message_take_header (&request, GST_RTSP_HDR_CSEQ, str);
+  gst_rtsp_message_add_header (&request, GST_RTSP_HDR_TRANSPORT,
+      "RTP/AVP;multicast;destination=233.252.0.1;ttl=2;port=5000-5001;");
+
+  gst_rtsp_client_set_send_func (client, test_setup_response_461, NULL, NULL);
+  fail_unless (gst_rtsp_client_handle_message (client,
+          &request) == GST_RTSP_OK);
+  gst_rtsp_message_unset (&request);
+
+  session_pool = gst_rtsp_client_get_session_pool (client);
+  fail_unless (session_pool != NULL);
+  /* FIXME: There seems to be a leak of a session here ! */
+  /* fail_unless (gst_rtsp_session_pool_get_n_sessions (session_pool) == 0); */
+  g_object_unref (session_pool);
+
+
+  g_object_unref (client);
+}
+
+GST_END_TEST;
+
+GST_START_TEST (test_client_multicast_transport_specific)
+{
+  GstRTSPClient *client;
+  GstRTSPMessage request = { 0, };
+  gchar *str;
+  GstRTSPSessionPool *session_pool;
+
+  client = setup_multicast_client ();
+
+  gst_rtsp_client_set_use_client_settings (client, TRUE);
+  fail_unless (gst_rtsp_client_get_use_client_settings (client));
+
+  expected_transport = "RTP/AVP;multicast;destination=233.252.0.1;"
+      "ttl=1;port=5000-5001;mode=\"PLAY\"";
+
+  /* simple SETUP with a valid URI and multicast, but an invalid ip */
+  fail_unless (gst_rtsp_message_init_request (&request, GST_RTSP_SETUP,
+          "rtsp://localhost/test/stream=0") == GST_RTSP_OK);
+  str = g_strdup_printf ("%d", cseq);
+  gst_rtsp_message_take_header (&request, GST_RTSP_HDR_CSEQ, str);
+  gst_rtsp_message_add_header (&request, GST_RTSP_HDR_TRANSPORT,
+      expected_transport);
+
+  gst_rtsp_client_set_send_func (client, test_setup_response_200_multicast,
+      NULL, NULL);
+  fail_unless (gst_rtsp_client_handle_message (client,
+          &request) == GST_RTSP_OK);
+  gst_rtsp_message_unset (&request);
+  expected_transport = NULL;
+
+  gst_rtsp_client_set_send_func (client, test_setup_response_200_multicast,
+      NULL, NULL);
+  session_pool = gst_rtsp_client_get_session_pool (client);
+  fail_unless (session_pool != NULL);
+  fail_unless (gst_rtsp_session_pool_get_n_sessions (session_pool) == 1);
+  g_object_unref (session_pool);
+
+  g_object_unref (client);
+}
+
+GST_END_TEST;
+
+
 static Suite *
 rtspclient_suite (void)
 {
@@ -226,6 +548,11 @@ rtspclient_suite (void)
   tcase_add_test (tc, test_request);
   tcase_add_test (tc, test_options);
   tcase_add_test (tc, test_describe);
+  tcase_add_test (tc, test_client_multicast_transport_404);
+  tcase_add_test (tc, test_client_multicast_transport);
+  tcase_add_test (tc, test_client_multicast_ignore_transport_specific);
+  tcase_add_test (tc, test_client_multicast_invalid_transport_specific);
+  tcase_add_test (tc, test_client_multicast_transport_specific);
 
   return s;
 }