mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-09-08 13:18:52 +00:00
Make it so that one can specify a buffer for get/pull_range where the downstream element should write into. When passing NULL, upstream should allocate a buffer, like in 0.10. We also need to change the probes a little because before the pull probe, there could already be a buffer passed. This then allows us to use the same PROBE macro for before and after pulling. While we're at the probes, make the query probe more powerful by handling the GST_PAD_PROBE_DROP return value. Returning _DROP from a query probe will now return TRUE upstream and will not forward the probe to the peer or handler. Also handle _DROP for get/pull_range properly by not dispatching to the peer/handler or by generating EOS when the probe returns DROP and no buffer. Make filesrc handle the non-NULL buffer passed in the get_range function and skip the allocation in that case, writing directly into the downstream provided buffer. Update tests because now we need to make sure to not pass a random value in the buffer pointer to get/pull_range
520 lines
15 KiB
C
520 lines
15 KiB
C
/* GStreamer
|
|
*
|
|
* Copyright (C) 2006 Thomas Vander Stichele <thomas at apestaart dot org>
|
|
*
|
|
* 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., 59 Temple Place - Suite 330,
|
|
* Boston, MA 02111-1307, USA.
|
|
*/
|
|
|
|
#include <unistd.h>
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <fcntl.h>
|
|
|
|
#include <gst/check/gstcheck.h>
|
|
|
|
static gboolean have_eos = FALSE;
|
|
static GCond *eos_cond;
|
|
static GMutex *event_mutex;
|
|
|
|
static GstPad *mysinkpad;
|
|
|
|
static GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink",
|
|
GST_PAD_SINK,
|
|
GST_PAD_ALWAYS,
|
|
GST_STATIC_CAPS_ANY);
|
|
|
|
static gboolean
|
|
event_func (GstPad * pad, GstObject * parent, GstEvent * event)
|
|
{
|
|
gboolean res = TRUE;
|
|
|
|
g_mutex_lock (event_mutex);
|
|
if (GST_EVENT_TYPE (event) == GST_EVENT_EOS) {
|
|
have_eos = TRUE;
|
|
GST_DEBUG ("signal EOS");
|
|
g_cond_broadcast (eos_cond);
|
|
}
|
|
g_mutex_unlock (event_mutex);
|
|
|
|
gst_event_unref (event);
|
|
|
|
return res;
|
|
}
|
|
|
|
static void
|
|
wait_eos (void)
|
|
{
|
|
g_mutex_lock (event_mutex);
|
|
GST_DEBUG ("waiting for EOS");
|
|
while (!have_eos) {
|
|
g_cond_wait (eos_cond, event_mutex);
|
|
}
|
|
GST_DEBUG ("received EOS");
|
|
g_mutex_unlock (event_mutex);
|
|
}
|
|
|
|
static GstElement *
|
|
setup_filesrc (void)
|
|
{
|
|
GstElement *filesrc;
|
|
|
|
GST_DEBUG ("setup_filesrc");
|
|
filesrc = gst_check_setup_element ("filesrc");
|
|
mysinkpad = gst_check_setup_sink_pad (filesrc, &sinktemplate);
|
|
gst_pad_set_event_function (mysinkpad, event_func);
|
|
gst_pad_set_active (mysinkpad, TRUE);
|
|
|
|
eos_cond = g_cond_new ();
|
|
event_mutex = g_mutex_new ();
|
|
|
|
return filesrc;
|
|
}
|
|
|
|
static void
|
|
cleanup_filesrc (GstElement * filesrc)
|
|
{
|
|
gst_pad_set_active (mysinkpad, FALSE);
|
|
gst_check_teardown_sink_pad (filesrc);
|
|
gst_check_teardown_element (filesrc);
|
|
|
|
g_cond_free (eos_cond);
|
|
g_mutex_free (event_mutex);
|
|
}
|
|
|
|
GST_START_TEST (test_seeking)
|
|
{
|
|
GstElement *src;
|
|
GstQuery *seeking_query;
|
|
gboolean seekable;
|
|
|
|
#ifndef TESTFILE
|
|
#error TESTFILE not defined
|
|
#endif
|
|
src = setup_filesrc ();
|
|
|
|
g_object_set (G_OBJECT (src), "location", TESTFILE, NULL);
|
|
fail_unless (gst_element_set_state (src,
|
|
GST_STATE_PAUSED) == GST_STATE_CHANGE_SUCCESS,
|
|
"could not set to paused");
|
|
|
|
/* Test that filesrc is seekable with a file fd */
|
|
fail_unless ((seeking_query = gst_query_new_seeking (GST_FORMAT_BYTES))
|
|
!= NULL);
|
|
fail_unless (gst_element_query (src, seeking_query) == TRUE);
|
|
gst_query_parse_seeking (seeking_query, NULL, &seekable, NULL, NULL);
|
|
fail_unless (seekable == TRUE);
|
|
gst_query_unref (seeking_query);
|
|
|
|
fail_unless (gst_element_set_state (src,
|
|
GST_STATE_NULL) == GST_STATE_CHANGE_SUCCESS, "could not set to null");
|
|
|
|
/* cleanup */
|
|
cleanup_filesrc (src);
|
|
}
|
|
|
|
GST_END_TEST;
|
|
|
|
GST_START_TEST (test_reverse)
|
|
{
|
|
GstElement *src;
|
|
|
|
#ifndef TESTFILE
|
|
#error TESTFILE not defined
|
|
#endif
|
|
src = setup_filesrc ();
|
|
|
|
g_object_set (G_OBJECT (src), "location", TESTFILE, NULL);
|
|
/* we're going to perform the seek in ready */
|
|
fail_unless (gst_element_set_state (src,
|
|
GST_STATE_READY) == GST_STATE_CHANGE_SUCCESS,
|
|
"could not set to ready");
|
|
|
|
/* reverse seek from end to start */
|
|
gst_element_seek (src, -1.0, GST_FORMAT_BYTES, 0, GST_SEEK_TYPE_SET, 100,
|
|
GST_SEEK_TYPE_SET, -1);
|
|
|
|
fail_unless (gst_element_set_state (src,
|
|
GST_STATE_PAUSED) == GST_STATE_CHANGE_SUCCESS,
|
|
"could not set to paused");
|
|
|
|
/* wait for EOS */
|
|
wait_eos ();
|
|
|
|
fail_unless (gst_element_set_state (src,
|
|
GST_STATE_NULL) == GST_STATE_CHANGE_SUCCESS, "could not set to null");
|
|
|
|
/* cleanup */
|
|
cleanup_filesrc (src);
|
|
}
|
|
|
|
GST_END_TEST;
|
|
|
|
GST_START_TEST (test_pull)
|
|
{
|
|
GstElement *src;
|
|
GstQuery *seeking_query;
|
|
gboolean res, seekable;
|
|
gint64 start, stop;
|
|
GstPad *pad;
|
|
GstFlowReturn ret;
|
|
GstBuffer *buffer1, *buffer2;
|
|
GstMapInfo info1, info2;
|
|
|
|
src = setup_filesrc ();
|
|
|
|
g_object_set (G_OBJECT (src), "location", TESTFILE, NULL);
|
|
fail_unless (gst_element_set_state (src,
|
|
GST_STATE_READY) == GST_STATE_CHANGE_SUCCESS,
|
|
"could not set to ready");
|
|
|
|
/* get the source pad */
|
|
pad = gst_element_get_static_pad (src, "src");
|
|
fail_unless (pad != NULL);
|
|
|
|
/* activate the pad in pull mode */
|
|
res = gst_pad_activate_mode (pad, GST_PAD_MODE_PULL, TRUE);
|
|
fail_unless (res == TRUE);
|
|
|
|
/* not start playing */
|
|
fail_unless (gst_element_set_state (src,
|
|
GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,
|
|
"could not set to paused");
|
|
|
|
/* Test that filesrc is seekable with a file fd */
|
|
fail_unless ((seeking_query = gst_query_new_seeking (GST_FORMAT_BYTES))
|
|
!= NULL);
|
|
fail_unless (gst_element_query (src, seeking_query) == TRUE);
|
|
|
|
/* get the seeking capabilities */
|
|
gst_query_parse_seeking (seeking_query, NULL, &seekable, &start, &stop);
|
|
fail_unless (seekable == TRUE);
|
|
fail_unless (start == 0);
|
|
fail_unless (start != -1);
|
|
gst_query_unref (seeking_query);
|
|
|
|
/* do some pulls */
|
|
buffer1 = NULL;
|
|
ret = gst_pad_get_range (pad, 0, 100, &buffer1);
|
|
fail_unless (ret == GST_FLOW_OK);
|
|
fail_unless (buffer1 != NULL);
|
|
fail_unless (gst_buffer_get_size (buffer1) == 100);
|
|
|
|
buffer2 = NULL;
|
|
ret = gst_pad_get_range (pad, 0, 50, &buffer2);
|
|
fail_unless (ret == GST_FLOW_OK);
|
|
fail_unless (buffer2 != NULL);
|
|
fail_unless (gst_buffer_get_size (buffer2) == 50);
|
|
|
|
/* this should be the same */
|
|
fail_unless (gst_buffer_map (buffer1, &info1, GST_MAP_READ));
|
|
fail_unless (gst_buffer_map (buffer2, &info2, GST_MAP_READ));
|
|
fail_unless (memcmp (info1.data, info2.data, 50) == 0);
|
|
gst_buffer_unmap (buffer2, &info2);
|
|
|
|
gst_buffer_unref (buffer2);
|
|
|
|
/* read next 50 bytes */
|
|
buffer2 = NULL;
|
|
ret = gst_pad_get_range (pad, 50, 50, &buffer2);
|
|
fail_unless (ret == GST_FLOW_OK);
|
|
fail_unless (buffer2 != NULL);
|
|
fail_unless (gst_buffer_get_size (buffer2) == 50);
|
|
|
|
/* compare with previously read data */
|
|
fail_unless (gst_buffer_map (buffer2, &info2, GST_MAP_READ));
|
|
fail_unless (memcmp ((guint8 *) info1.data + 50, info2.data, 50) == 0);
|
|
gst_buffer_unmap (buffer2, &info2);
|
|
|
|
gst_buffer_unmap (buffer1, &info1);
|
|
gst_buffer_unref (buffer1);
|
|
gst_buffer_unref (buffer2);
|
|
|
|
/* read 10 bytes at end-10 should give exactly 10 bytes */
|
|
buffer1 = NULL;
|
|
ret = gst_pad_get_range (pad, stop - 10, 10, &buffer1);
|
|
fail_unless (ret == GST_FLOW_OK);
|
|
fail_unless (buffer1 != NULL);
|
|
fail_unless (gst_buffer_get_size (buffer1) == 10);
|
|
gst_buffer_unref (buffer1);
|
|
|
|
/* read 20 bytes at end-10 should give exactly 10 bytes */
|
|
buffer1 = NULL;
|
|
ret = gst_pad_get_range (pad, stop - 10, 20, &buffer1);
|
|
fail_unless (ret == GST_FLOW_OK);
|
|
fail_unless (buffer1 != NULL);
|
|
fail_unless (gst_buffer_get_size (buffer1) == 10);
|
|
gst_buffer_unref (buffer1);
|
|
|
|
/* read 0 bytes at end-1 should return 0 bytes */
|
|
buffer1 = NULL;
|
|
ret = gst_pad_get_range (pad, stop - 1, 0, &buffer1);
|
|
fail_unless (ret == GST_FLOW_OK);
|
|
fail_unless (buffer1 != NULL);
|
|
fail_unless (gst_buffer_get_size (buffer1) == 0);
|
|
gst_buffer_unref (buffer1);
|
|
|
|
/* read 10 bytes at end-1 should return 1 byte */
|
|
buffer1 = NULL;
|
|
ret = gst_pad_get_range (pad, stop - 1, 10, &buffer1);
|
|
fail_unless (ret == GST_FLOW_OK);
|
|
fail_unless (buffer1 != NULL);
|
|
fail_unless (gst_buffer_get_size (buffer1) == 1);
|
|
gst_buffer_unref (buffer1);
|
|
|
|
/* read 0 bytes at end should EOS */
|
|
buffer1 = NULL;
|
|
ret = gst_pad_get_range (pad, stop, 0, &buffer1);
|
|
fail_unless (ret == GST_FLOW_EOS);
|
|
|
|
/* read 10 bytes before end should EOS */
|
|
buffer1 = NULL;
|
|
ret = gst_pad_get_range (pad, stop, 10, &buffer1);
|
|
fail_unless (ret == GST_FLOW_EOS);
|
|
|
|
/* read 0 bytes after end should EOS */
|
|
buffer1 = NULL;
|
|
ret = gst_pad_get_range (pad, stop + 10, 0, &buffer1);
|
|
fail_unless (ret == GST_FLOW_EOS);
|
|
|
|
/* read 10 bytes after end should EOS too */
|
|
buffer1 = NULL;
|
|
ret = gst_pad_get_range (pad, stop + 10, 10, &buffer1);
|
|
fail_unless (ret == GST_FLOW_EOS);
|
|
|
|
fail_unless (gst_element_set_state (src,
|
|
GST_STATE_NULL) == GST_STATE_CHANGE_SUCCESS, "could not set to null");
|
|
|
|
/* cleanup */
|
|
gst_object_unref (pad);
|
|
cleanup_filesrc (src);
|
|
}
|
|
|
|
GST_END_TEST;
|
|
|
|
GST_START_TEST (test_coverage)
|
|
{
|
|
GstElement *src;
|
|
gchar *location;
|
|
GstBus *bus;
|
|
GstMessage *message;
|
|
|
|
src = setup_filesrc ();
|
|
bus = gst_bus_new ();
|
|
|
|
gst_element_set_bus (src, bus);
|
|
|
|
g_object_set (G_OBJECT (src), "location", "/i/do/not/exist", NULL);
|
|
g_object_get (G_OBJECT (src), "location", &location, NULL);
|
|
fail_unless_equals_string (location, "/i/do/not/exist");
|
|
g_free (location);
|
|
fail_unless (gst_element_set_state (src,
|
|
GST_STATE_PLAYING) == GST_STATE_CHANGE_FAILURE,
|
|
"could set to playing with wrong location");
|
|
|
|
/* a state change and an error */
|
|
fail_if ((message = gst_bus_pop (bus)) == NULL);
|
|
gst_message_unref (message);
|
|
fail_if ((message = gst_bus_pop (bus)) == NULL);
|
|
fail_unless_message_error (message, RESOURCE, NOT_FOUND);
|
|
gst_message_unref (message);
|
|
|
|
g_object_set (G_OBJECT (src), "location", NULL, NULL);
|
|
g_object_get (G_OBJECT (src), "location", &location, NULL);
|
|
fail_if (location);
|
|
|
|
/* cleanup */
|
|
gst_element_set_bus (src, NULL);
|
|
gst_object_unref (GST_OBJECT (bus));
|
|
cleanup_filesrc (src);
|
|
}
|
|
|
|
GST_END_TEST;
|
|
|
|
GST_START_TEST (test_uri_interface)
|
|
{
|
|
GstElement *src;
|
|
gchar *location;
|
|
GstBus *bus;
|
|
|
|
src = setup_filesrc ();
|
|
bus = gst_bus_new ();
|
|
|
|
gst_element_set_bus (src, bus);
|
|
|
|
g_object_set (G_OBJECT (src), "location", "/i/do/not/exist", NULL);
|
|
g_object_get (G_OBJECT (src), "location", &location, NULL);
|
|
fail_unless_equals_string (location, "/i/do/not/exist");
|
|
g_free (location);
|
|
|
|
location = gst_uri_handler_get_uri (GST_URI_HANDLER (src));
|
|
fail_unless_equals_string (location, "file:///i/do/not/exist");
|
|
g_free (location);
|
|
|
|
/* should accept file:///foo/bar URIs */
|
|
fail_unless (gst_uri_handler_set_uri (GST_URI_HANDLER (src),
|
|
"file:///foo/bar", NULL));
|
|
location = gst_uri_handler_get_uri (GST_URI_HANDLER (src));
|
|
fail_unless_equals_string (location, "file:///foo/bar");
|
|
g_free (location);
|
|
g_object_get (G_OBJECT (src), "location", &location, NULL);
|
|
fail_unless_equals_string (location, "/foo/bar");
|
|
g_free (location);
|
|
|
|
/* should accept file://localhost/foo/bar URIs */
|
|
fail_unless (gst_uri_handler_set_uri (GST_URI_HANDLER (src),
|
|
"file://localhost/foo/baz", NULL));
|
|
location = gst_uri_handler_get_uri (GST_URI_HANDLER (src));
|
|
fail_unless_equals_string (location, "file:///foo/baz");
|
|
g_free (location);
|
|
g_object_get (G_OBJECT (src), "location", &location, NULL);
|
|
fail_unless_equals_string (location, "/foo/baz");
|
|
g_free (location);
|
|
|
|
/* should escape non-uri characters for the URI but not for the location */
|
|
g_object_set (G_OBJECT (src), "location", "/foo/b?r", NULL);
|
|
g_object_get (G_OBJECT (src), "location", &location, NULL);
|
|
fail_unless_equals_string (location, "/foo/b?r");
|
|
g_free (location);
|
|
location = gst_uri_handler_get_uri (GST_URI_HANDLER (src));
|
|
fail_unless_equals_string (location, "file:///foo/b%3Fr");
|
|
g_free (location);
|
|
|
|
/* should fail with other hostnames */
|
|
fail_if (gst_uri_handler_set_uri (GST_URI_HANDLER (src),
|
|
"file://hostname/foo/foo", NULL));
|
|
|
|
/* cleanup */
|
|
gst_element_set_bus (src, NULL);
|
|
gst_object_unref (GST_OBJECT (bus));
|
|
cleanup_filesrc (src);
|
|
}
|
|
|
|
GST_END_TEST;
|
|
|
|
static void
|
|
check_uri_for_uri (GstElement * e, const gchar * in_uri, const gchar * uri)
|
|
{
|
|
GstQuery *query;
|
|
gchar *query_uri = NULL;
|
|
|
|
gst_uri_handler_set_uri (GST_URI_HANDLER (e), in_uri, NULL);
|
|
|
|
query = gst_query_new_uri ();
|
|
fail_unless (gst_element_query (e, query));
|
|
gst_query_parse_uri (query, &query_uri);
|
|
gst_query_unref (query);
|
|
|
|
if (uri != NULL) {
|
|
fail_unless_equals_string (query_uri, uri);
|
|
} else {
|
|
gchar *fn;
|
|
|
|
fail_unless (gst_uri_is_valid (query_uri));
|
|
fn = g_filename_from_uri (query_uri, NULL, NULL);
|
|
fail_unless (g_path_is_absolute (fn));
|
|
fail_unless (fn != NULL);
|
|
g_free (fn);
|
|
}
|
|
|
|
g_free (query_uri);
|
|
}
|
|
|
|
static void
|
|
check_uri_for_location (GstElement * e, const gchar * location,
|
|
const gchar * uri)
|
|
{
|
|
GstQuery *query;
|
|
gchar *query_uri = NULL;
|
|
|
|
g_object_set (e, "location", location, NULL);
|
|
query = gst_query_new_uri ();
|
|
fail_unless (gst_element_query (e, query));
|
|
gst_query_parse_uri (query, &query_uri);
|
|
gst_query_unref (query);
|
|
|
|
if (uri != NULL) {
|
|
fail_unless_equals_string (query_uri, uri);
|
|
} else {
|
|
gchar *fn;
|
|
|
|
fail_unless (gst_uri_is_valid (query_uri));
|
|
fn = g_filename_from_uri (query_uri, NULL, NULL);
|
|
fail_unless (g_path_is_absolute (fn));
|
|
fail_unless (fn != NULL);
|
|
g_free (fn);
|
|
}
|
|
|
|
g_free (query_uri);
|
|
}
|
|
|
|
GST_START_TEST (test_uri_query)
|
|
{
|
|
GstElement *src;
|
|
|
|
src = setup_filesrc ();
|
|
|
|
#ifdef G_OS_UNIX
|
|
{
|
|
GST_INFO ("*nix");
|
|
check_uri_for_location (src, "/i/do/not/exist", "file:///i/do/not/exist");
|
|
check_uri_for_location (src, "/i/do/not/../exist", "file:///i/do/exist");
|
|
check_uri_for_location (src, "/i/do/not/.././exist", "file:///i/do/exist");
|
|
check_uri_for_location (src, "/i/./do/not/../exist", "file:///i/do/exist");
|
|
check_uri_for_location (src, "/i/do/./not/../exist", "file:///i/do/exist");
|
|
check_uri_for_location (src, "/i/do/not/./../exist", "file:///i/do/exist");
|
|
check_uri_for_location (src, "/i/./do/./././././exist",
|
|
"file:///i/do/exist");
|
|
check_uri_for_location (src, "/i/do/not/../../exist", "file:///i/exist");
|
|
check_uri_for_location (src, "/i/../not/../exist", "file:///exist");
|
|
/* hard to test relative URIs, just make sure it returns an URI of sorts */
|
|
check_uri_for_location (src, "foo", NULL);
|
|
check_uri_for_location (src, "foo/../bar", NULL);
|
|
check_uri_for_location (src, "./foo", NULL);
|
|
check_uri_for_location (src, "../foo", NULL);
|
|
check_uri_for_location (src, "foo/./bar", NULL);
|
|
/* make sure non-ASCII characters are escaped properly (U+00F6 here) */
|
|
check_uri_for_location (src, "/i/./d\303\266/not/../exist",
|
|
"file:///i/d%C3%B6/exist");
|
|
/* let's see what happens if we set a malformed URI with ISO-8859-1 chars,
|
|
* i.e. one that the input characters haven't been escaped properly. We
|
|
* should get back a properly escaped URI */
|
|
check_uri_for_uri (src, "file:///M\366t\366r", "file:///M%F6t%F6r");
|
|
}
|
|
#endif
|
|
|
|
cleanup_filesrc (src);
|
|
}
|
|
|
|
GST_END_TEST;
|
|
|
|
static Suite *
|
|
filesrc_suite (void)
|
|
{
|
|
Suite *s = suite_create ("filesrc");
|
|
TCase *tc_chain = tcase_create ("general");
|
|
|
|
suite_add_tcase (s, tc_chain);
|
|
tcase_add_test (tc_chain, test_seeking);
|
|
tcase_add_test (tc_chain, test_reverse);
|
|
tcase_add_test (tc_chain, test_pull);
|
|
tcase_add_test (tc_chain, test_coverage);
|
|
tcase_add_test (tc_chain, test_uri_interface);
|
|
tcase_add_test (tc_chain, test_uri_query);
|
|
|
|
return s;
|
|
}
|
|
|
|
GST_CHECK_MAIN (filesrc);
|