mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-04 23:46:43 +00:00
434bb14995
We were writing unitialized stack memory to the file.
227 lines
6.2 KiB
C
227 lines
6.2 KiB
C
/* GStreamer
|
|
*
|
|
* unit test for cachefile helper
|
|
*
|
|
* Copyright (C) 2014 Wim Taymans <wtaymans@redhat.com>
|
|
*
|
|
* 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.
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include "config.h"
|
|
#endif
|
|
|
|
#include <glib/gstdio.h>
|
|
|
|
#include <gst/check/gstcheck.h>
|
|
|
|
/* not public API for now */
|
|
#include "../../../plugins/elements/gstsparsefile.c"
|
|
|
|
static void
|
|
expect_range_before (GstSparseFile * file, gsize offset, gsize start,
|
|
gsize stop)
|
|
{
|
|
gsize tstart, tstop;
|
|
|
|
fail_unless (gst_sparse_file_get_range_before (file, offset, &tstart,
|
|
&tstop) == TRUE);
|
|
fail_unless (tstart == start);
|
|
fail_unless (tstop == stop);
|
|
}
|
|
|
|
static void
|
|
expect_range_after (GstSparseFile * file, gsize offset, gsize start, gsize stop)
|
|
{
|
|
gsize tstart, tstop;
|
|
|
|
fail_unless (gst_sparse_file_get_range_after (file, offset, &tstart,
|
|
&tstop) == TRUE);
|
|
fail_unless (tstart == start);
|
|
fail_unless (tstop == stop);
|
|
}
|
|
|
|
static gboolean
|
|
expect_write (GstSparseFile * file, gsize offset, gsize count, gsize result,
|
|
gsize avail)
|
|
{
|
|
GError *error = NULL;
|
|
gchar buffer[200] = { 0, };
|
|
gsize res, a;
|
|
|
|
res = gst_sparse_file_write (file, offset, buffer, count, &a, &error);
|
|
if (res != result)
|
|
return FALSE;
|
|
if (res == 0) {
|
|
if (error == NULL)
|
|
return FALSE;
|
|
g_clear_error (&error);
|
|
} else if (a != avail)
|
|
return FALSE;
|
|
return TRUE;
|
|
}
|
|
|
|
static gboolean
|
|
expect_read (GstSparseFile * file, gsize offset, gsize count, gsize result,
|
|
gsize avail)
|
|
{
|
|
GError *error = NULL;
|
|
gchar buffer[200];
|
|
gsize res, a;
|
|
|
|
res = gst_sparse_file_read (file, offset, buffer, count, &a, &error);
|
|
if (res != result)
|
|
return FALSE;
|
|
if (res == 0) {
|
|
if (error == NULL)
|
|
return FALSE;
|
|
g_clear_error (&error);
|
|
} else if (a != avail)
|
|
return FALSE;
|
|
return TRUE;
|
|
}
|
|
|
|
GST_START_TEST (test_write_read)
|
|
{
|
|
GstSparseFile *file;
|
|
gint fd;
|
|
gchar *name;
|
|
gsize start, stop;
|
|
|
|
name = g_strdup ("cachefile-testXXXXXX");
|
|
fd = g_mkstemp (name);
|
|
fail_if (fd == -1);
|
|
|
|
file = gst_sparse_file_new ();
|
|
fail_unless (file != NULL);
|
|
fail_unless (gst_sparse_file_set_fd (file, fd));
|
|
fail_unless (gst_sparse_file_n_ranges (file) == 0);
|
|
|
|
/* should fail, we didn't write anything yet */
|
|
fail_unless (expect_read (file, 0, 100, 0, 0));
|
|
|
|
/* no ranges, searching for a range should fail */
|
|
fail_unless (gst_sparse_file_n_ranges (file) == 0);
|
|
fail_unless (gst_sparse_file_get_range_before (file, 0, &start,
|
|
&stop) == FALSE);
|
|
fail_unless (gst_sparse_file_get_range_before (file, 10, &start,
|
|
&stop) == FALSE);
|
|
fail_unless (gst_sparse_file_get_range_after (file, 0, &start,
|
|
&stop) == FALSE);
|
|
fail_unless (gst_sparse_file_get_range_after (file, 10, &start,
|
|
&stop) == FALSE);
|
|
|
|
/* now write some data */
|
|
fail_unless (expect_write (file, 0, 100, 100, 0));
|
|
|
|
/* we have 1 range now */
|
|
fail_unless (gst_sparse_file_n_ranges (file) == 1);
|
|
expect_range_before (file, 0, 0, 100);
|
|
expect_range_after (file, 0, 0, 100);
|
|
expect_range_before (file, 100, 0, 100);
|
|
expect_range_before (file, 50, 0, 100);
|
|
expect_range_before (file, 200, 0, 100);
|
|
fail_unless (gst_sparse_file_get_range_after (file, 100, &start,
|
|
&stop) == FALSE);
|
|
expect_range_after (file, 50, 0, 100);
|
|
|
|
/* we can read all data now */
|
|
fail_unless (expect_read (file, 0, 100, 100, 0));
|
|
/* we can read less */
|
|
fail_unless (expect_read (file, 0, 50, 50, 50));
|
|
/* but we can't read more than what is written */
|
|
fail_unless (expect_read (file, 0, 101, 0, 0));
|
|
|
|
g_unlink (name);
|
|
gst_sparse_file_free (file);
|
|
g_free (name);
|
|
}
|
|
|
|
GST_END_TEST;
|
|
|
|
GST_START_TEST (test_write_merge)
|
|
{
|
|
GstSparseFile *file;
|
|
gint fd;
|
|
gchar *name;
|
|
gsize start, stop;
|
|
|
|
name = g_strdup ("cachefile-testXXXXXX");
|
|
fd = g_mkstemp (name);
|
|
fail_if (fd == -1);
|
|
|
|
file = gst_sparse_file_new ();
|
|
gst_sparse_file_set_fd (file, fd);
|
|
|
|
/* write something at offset 0 */
|
|
fail_unless (expect_write (file, 0, 100, 100, 0));
|
|
/* we have 1 range now */
|
|
fail_unless (gst_sparse_file_n_ranges (file) == 1);
|
|
expect_range_before (file, 110, 0, 100);
|
|
expect_range_after (file, 50, 0, 100);
|
|
fail_unless (gst_sparse_file_get_range_after (file, 100, &start,
|
|
&stop) == FALSE);
|
|
|
|
/* read should fail */
|
|
fail_unless (expect_read (file, 50, 150, 0, 0));
|
|
|
|
/* write something at offset 150 */
|
|
fail_unless (expect_write (file, 150, 100, 100, 0));
|
|
/* we have 2 ranges now */
|
|
fail_unless (gst_sparse_file_n_ranges (file) == 2);
|
|
expect_range_before (file, 110, 0, 100);
|
|
expect_range_after (file, 50, 0, 100);
|
|
expect_range_after (file, 100, 150, 250);
|
|
expect_range_before (file, 150, 150, 250);
|
|
|
|
/* read should still fail */
|
|
fail_unless (expect_read (file, 50, 150, 0, 0));
|
|
|
|
/* fill the hole */
|
|
fail_unless (expect_write (file, 100, 50, 50, 100));
|
|
/* we have 1 range now */
|
|
fail_unless (gst_sparse_file_n_ranges (file) == 1);
|
|
expect_range_before (file, 110, 0, 250);
|
|
expect_range_after (file, 50, 0, 250);
|
|
expect_range_after (file, 100, 0, 250);
|
|
expect_range_before (file, 150, 0, 250);
|
|
fail_unless (gst_sparse_file_get_range_after (file, 250, &start,
|
|
&stop) == FALSE);
|
|
|
|
/* read work */
|
|
fail_unless (expect_read (file, 50, 150, 150, 50));
|
|
|
|
g_unlink (name);
|
|
gst_sparse_file_free (file);
|
|
g_free (name);
|
|
}
|
|
|
|
GST_END_TEST;
|
|
|
|
static Suite *
|
|
gst_cachefile_suite (void)
|
|
{
|
|
Suite *s = suite_create ("cachefile");
|
|
TCase *tc_chain = tcase_create ("general");
|
|
|
|
suite_add_tcase (s, tc_chain);
|
|
tcase_add_test (tc_chain, test_write_read);
|
|
tcase_add_test (tc_chain, test_write_merge);
|
|
|
|
return s;
|
|
}
|
|
|
|
GST_CHECK_MAIN (gst_cachefile);
|