tests: add unit test for GstMemory

Add a GstMemory unit test
Add some more asserts in GstMemory to catch invalid cases.

diff --git a/gst/gstmemory.c b/gst/gstmemory.c
index 26efc88..e2f5e48 100644 (file)
--- a/gst/gstmemory.c
+++ b/gst/gstmemory.c
@@ -192,6 +192,7 @@ _default_mem_get_sizes (GstMemoryDefault * mem, gsize * offset, gsize * maxsize)
 static void
 _default_mem_resize (GstMemoryDefault * mem, gssize offset, gsize size)
 {
+  g_return_if_fail (offset >= 0 || mem->offset >= -offset);
   g_return_if_fail (size + mem->offset + offset <= mem->maxsize);
 
   mem->offset += offset;
@@ -213,8 +214,10 @@ _default_mem_map (GstMemoryDefault * mem, gsize * size, gsize * maxsize,
 static gboolean
 _default_mem_unmap (GstMemoryDefault * mem, gpointer data, gsize size)
 {
-  if (size != -1)
+  if (size != -1) {
+    g_return_val_if_fail (size <= mem->maxsize, FALSE);
     mem->size = size;
+  }
   return TRUE;
 }
 

diff --git a/tests/check/Makefile.am b/tests/check/Makefile.am
index 7103d8e..1892d98 100644 (file)
--- a/tests/check/Makefile.am
+++ b/tests/check/Makefile.am
@@ -105,6 +105,7 @@ check_PROGRAMS =                            \
        gst/gstbuffer                           \
        gst/gstbufferlist                       \
        gst/gstmeta                             \
+       gst/gstmemory                           \
        gst/gstbus                              \
        gst/gstcaps                             \
        $(CXX_CHECKS)                           \

diff --git a/tests/check/gst/gstmemory.c b/tests/check/gst/gstmemory.c
new file mode 100644 (file)
index 0000000..11bd9ec
--- /dev/null
+++ b/tests/check/gst/gstmemory.c
@@ -0,0 +1,388 @@
+/* GStreamer
+ *
+ * unit test for GstMemory
+ *
+ * Copyright (C) <2012> Wim Taymans <wim.taymans at gmail.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
+
+#ifdef HAVE_VALGRIND_H
+# include <valgrind/valgrind.h>
+#else
+# define RUNNING_ON_VALGRIND FALSE
+#endif
+
+#include <gst/check/gstcheck.h>
+
+GST_START_TEST (test_submemory)
+{
+  GstMemory *memory, *sub;
+  gsize size, maxsize, ssize;
+  guint8 *data, *sdata;
+
+  memory = gst_allocator_alloc (NULL, 4, 0);
+
+  /* check sizes, memory starts out empty */
+  data = gst_memory_map (memory, &size, &maxsize, GST_MAP_WRITE);
+  fail_unless (size == 4, "memory has wrong size");
+  fail_unless (maxsize >= 4, "memory has wrong size");
+  memset (data, 0, 4);
+  gst_memory_unmap (memory, data, 4);
+
+  data = gst_memory_map (memory, &size, NULL, GST_MAP_READ);
+
+  sub = gst_memory_share (memory, 1, 2);
+  fail_if (sub == NULL, "share of memory returned NULL");
+
+  sdata = gst_memory_map (sub, &ssize, NULL, GST_MAP_READ);
+  fail_unless (ssize == 2, "submemory has wrong size");
+  fail_unless (memcmp (data + 1, sdata, 2) == 0,
+      "submemory contains the wrong data");
+  ASSERT_MEMORY_REFCOUNT (sub, "submemory", 1);
+  gst_memory_unmap (sub, sdata, ssize);
+  gst_memory_unref (sub);
+
+  /* create a submemory of size 0 */
+  sub = gst_memory_share (memory, 1, 0);
+  fail_if (sub == NULL, "share memory returned NULL");
+  sdata = gst_memory_map (sub, &ssize, NULL, GST_MAP_READ);
+  fail_unless (ssize == 0, "submemory has wrong size");
+  fail_unless (memcmp (data + 1, sdata, 0) == 0,
+      "submemory contains the wrong data");
+  ASSERT_MEMORY_REFCOUNT (sub, "submemory", 1);
+  gst_memory_unmap (sub, sdata, ssize);
+  gst_memory_unref (sub);
+
+  /* test if metadata is coppied, not a complete memory copy so only the
+   * timestamp and offset fields are copied. */
+  sub = gst_memory_share (memory, 0, 1);
+  fail_if (sub == NULL, "share of memory returned NULL");
+  fail_unless (gst_memory_get_sizes (sub, NULL, NULL) == 1,
+      "submemory has wrong size");
+  gst_memory_unref (sub);
+
+  /* test if metadata is coppied, a complete memory is copied so all the timing
+   * fields should be copied. */
+  sub = gst_memory_share (memory, 0, 4);
+  fail_if (sub == NULL, "share of memory returned NULL");
+  fail_unless (gst_memory_get_sizes (sub, NULL, NULL) == 4,
+      "submemory has wrong size");
+
+  /* clean up */
+  gst_memory_unref (sub);
+
+  gst_memory_unmap (memory, data, size);
+  gst_memory_unref (memory);
+}
+
+GST_END_TEST;
+
+GST_START_TEST (test_is_span)
+{
+  GstMemory *memory, *sub1, *sub2;
+
+  memory = gst_allocator_alloc (NULL, 4, 0);
+
+  sub1 = gst_memory_share (memory, 0, 2);
+  fail_if (sub1 == NULL, "share of memory returned NULL");
+
+  sub2 = gst_memory_share (memory, 2, 2);
+  fail_if (sub2 == NULL, "share of memory returned NULL");
+
+  fail_if (gst_memory_is_span (memory, sub2, NULL) == TRUE,
+      "a parent memory can't be span");
+
+  fail_if (gst_memory_is_span (sub1, memory, NULL) == TRUE,
+      "a parent memory can't be span");
+
+  fail_if (gst_memory_is_span (sub1, sub2, NULL) == FALSE,
+      "two submemorys next to each other should be span");
+
+  /* clean up */
+  gst_memory_unref (sub1);
+  gst_memory_unref (sub2);
+  gst_memory_unref (memory);
+}
+
+GST_END_TEST;
+
+static const char ro_memory[] = "abcdefghijklmnopqrstuvwxyz";
+
+static GstMemory *
+create_read_only_memory (void)
+{
+  GstMemory *mem;
+
+  /* assign some read-only data to the new memory */
+  mem = gst_memory_new_wrapped (GST_MEMORY_FLAG_READONLY,
+      (gpointer) ro_memory, NULL, sizeof (ro_memory), 0, sizeof (ro_memory));
+  fail_if (GST_MEMORY_IS_WRITABLE (mem));
+
+  return mem;
+}
+
+GST_START_TEST (test_writable)
+{
+  GstMemory *mem, *mem2;
+  guint8 *data;
+  gsize size;
+
+  /* create read-only memory and try to write */
+  mem = create_read_only_memory ();
+
+  ASSERT_CRITICAL (gst_memory_map (mem, &size, NULL, GST_MAP_WRITE));
+  fail_if (GST_MEMORY_IS_WRITABLE (mem));
+
+  mem2 = gst_memory_copy (mem, 0, -1);
+  fail_if (GST_MEMORY_IS_WRITABLE (mem));
+  fail_unless (GST_MEMORY_IS_WRITABLE (mem2));
+
+  data = gst_memory_map (mem2, &size, NULL, GST_MAP_WRITE);
+  data[4] = 'a';
+  gst_memory_unmap (mem2, data, size);
+  gst_memory_unref (mem2);
+
+  gst_memory_unref (mem);
+}
+
+GST_END_TEST;
+
+GST_START_TEST (test_submemory_writable)
+{
+  GstMemory *mem, *sub_mem;
+  gsize size;
+
+  /* create sub-memory of read-only memory and make it writable */
+  mem = create_read_only_memory ();
+
+  sub_mem = gst_memory_share (mem, 0, 8);
+  fail_if (GST_MEMORY_IS_WRITABLE (sub_mem));
+
+  ASSERT_CRITICAL (gst_memory_map (sub_mem, &size, NULL, GST_MAP_WRITE));
+
+  gst_memory_unref (sub_mem);
+  gst_memory_unref (mem);
+}
+
+GST_END_TEST;
+
+GST_START_TEST (test_copy)
+{
+  GstMemory *memory, *copy;
+  gsize size, ssize;
+  guint8 *data, *sdata;
+
+  memory = gst_allocator_alloc (NULL, 4, 0);
+  ASSERT_MEMORY_REFCOUNT (memory, "memory", 1);
+
+  copy = gst_memory_copy (memory, 0, -1);
+  ASSERT_MEMORY_REFCOUNT (memory, "memory", 1);
+  ASSERT_MEMORY_REFCOUNT (copy, "copy", 1);
+  /* memorys are copied and must point to different memory */
+  fail_if (memory == copy);
+
+  data = gst_memory_map (memory, &size, NULL, GST_MAP_READ);
+  sdata = gst_memory_map (copy, &ssize, NULL, GST_MAP_READ);
+
+  /* NOTE that data is refcounted */
+  fail_unless (size == ssize);
+
+  gst_memory_unmap (copy, sdata, ssize);
+  gst_memory_unmap (memory, data, size);
+
+  gst_memory_unref (copy);
+  gst_memory_unref (memory);
+
+  memory = gst_allocator_alloc (NULL, 0, 0);
+  data = gst_memory_map (memory, &size, NULL, GST_MAP_READ);
+  fail_unless (size == 0);
+  gst_memory_unmap (memory, data, size);
+
+  /* copying a 0-sized memory should not crash */
+  copy = gst_memory_copy (memory, 0, -1);
+  data = gst_memory_map (copy, &size, NULL, GST_MAP_READ);
+  fail_unless (size == 0);
+  gst_memory_unmap (copy, data, size);
+
+  gst_memory_unref (copy);
+  gst_memory_unref (memory);
+}
+
+GST_END_TEST;
+
+GST_START_TEST (test_try_new_and_alloc)
+{
+  GstMemory *mem;
+  gsize size;
+  guint8 *data;
+
+  mem = gst_allocator_alloc (NULL, 0, 0);
+  fail_unless (mem != NULL);
+  data = gst_memory_map (mem, &size, NULL, GST_MAP_READ);
+  fail_unless (size == 0);
+  gst_memory_unmap (mem, data, size);
+  gst_memory_unref (mem);
+
+  /* normal alloc should still work */
+  mem = gst_allocator_alloc (NULL, 640 * 480 * 4, 0);
+  fail_unless (mem != NULL);
+  data = gst_memory_map (mem, &size, NULL, GST_MAP_WRITE);
+  fail_unless (data != NULL);
+  fail_unless (size == (640 * 480 * 4));
+  data[640 * 479 * 4 + 479] = 0xff;
+  gst_memory_unmap (mem, data, size);
+
+  gst_memory_unref (mem);
+
+#if 0
+  /* Disabled this part of the test, because it happily succeeds on 64-bit
+   * machines that have enough memory+swap, because the address space is large
+   * enough. There's not really any way to test the failure case except by 
+   * allocating chunks of memory until it fails, which would suck. */
+
+  /* now this better fail (don't run in valgrind, it will abort
+   * or warn when passing silly arguments to malloc) */
+  if (!RUNNING_ON_VALGRIND) {
+    buf = gst_allocator_alloc (NULL, (guint) - 1, 0);
+    fail_unless (buf == NULL);
+  }
+#endif
+}
+
+GST_END_TEST;
+
+GST_START_TEST (test_resize)
+{
+  GstMemory *buf;
+  gsize maxalloc;
+  gsize size, maxsize, offset;
+
+  /* one memory block */
+  buf = gst_allocator_alloc (NULL, 100, 0);
+
+  size = gst_memory_get_sizes (buf, &offset, &maxalloc);
+  fail_unless (size == 100);
+  fail_unless (offset == 0);
+  fail_unless (maxalloc >= 100);
+
+  ASSERT_CRITICAL (gst_memory_resize (buf, 200, 50));
+  ASSERT_CRITICAL (gst_memory_resize (buf, 0, 150));
+  ASSERT_CRITICAL (gst_memory_resize (buf, 1, maxalloc));
+  ASSERT_CRITICAL (gst_memory_resize (buf, maxalloc, 1));
+
+  /* this does nothing */
+  gst_memory_resize (buf, 0, 100);
+
+  /* nothing should have changed */
+  size = gst_memory_get_sizes (buf, &offset, &maxsize);
+  fail_unless (size == 100);
+  fail_unless (offset == 0);
+  fail_unless (maxsize == maxalloc);
+
+  gst_memory_resize (buf, 0, 50);
+  size = gst_memory_get_sizes (buf, &offset, &maxsize);
+  fail_unless (size == 50);
+  fail_unless (offset == 0);
+  fail_unless (maxsize == maxalloc);
+
+  gst_memory_resize (buf, 0, 100);
+  size = gst_memory_get_sizes (buf, &offset, &maxsize);
+  fail_unless (size == 100);
+  fail_unless (offset == 0);
+  fail_unless (maxsize == maxalloc);
+
+  gst_memory_resize (buf, 1, 99);
+  size = gst_memory_get_sizes (buf, &offset, &maxsize);
+  fail_unless (size == 99);
+  fail_unless (offset == 1);
+  fail_unless (maxsize == maxalloc);
+
+  ASSERT_CRITICAL (gst_memory_resize (buf, 1, maxalloc - 1));
+
+  gst_memory_resize (buf, 0, 99);
+  size = gst_memory_get_sizes (buf, &offset, &maxsize);
+  fail_unless (size == 99);
+  fail_unless (offset == 1);
+  fail_unless (maxsize == maxalloc);
+
+  gst_memory_resize (buf, -1, 100);
+  size = gst_memory_get_sizes (buf, &offset, &maxsize);
+  fail_unless (size == 100);
+  fail_unless (offset == 0);
+  fail_unless (maxsize == maxalloc);
+
+  /* can't set offset below 0 */
+  ASSERT_CRITICAL (gst_memory_resize (buf, -1, 100));
+
+  gst_memory_resize (buf, 50, 40);
+  size = gst_memory_get_sizes (buf, &offset, &maxsize);
+  fail_unless (size == 40);
+  fail_unless (offset == 50);
+  fail_unless (maxsize == maxalloc);
+
+  gst_memory_resize (buf, -50, 100);
+  size = gst_memory_get_sizes (buf, &offset, &maxsize);
+  fail_unless (size == 100);
+  fail_unless (offset == 0);
+  fail_unless (maxsize == maxalloc);
+
+  gst_memory_resize (buf, 0, 0);
+  size = gst_memory_get_sizes (buf, &offset, &maxsize);
+  fail_unless (size == 0);
+  fail_unless (offset == 0);
+  fail_unless (maxsize == maxalloc);
+
+  gst_memory_resize (buf, 0, 100);
+  size = gst_memory_get_sizes (buf, &offset, &maxsize);
+  fail_unless (size == 100);
+  fail_unless (offset == 0);
+  fail_unless (maxsize == maxalloc);
+
+  gst_memory_resize (buf, 0, 100);
+  size = gst_memory_get_sizes (buf, &offset, &maxsize);
+  fail_unless (size == 100);
+  fail_unless (offset == 0);
+  fail_unless (maxsize == maxalloc);
+
+  gst_memory_unref (buf);
+}
+
+GST_END_TEST;
+
+static Suite *
+gst_memory_suite (void)
+{
+  Suite *s = suite_create ("GstMemory");
+  TCase *tc_chain = tcase_create ("general");
+
+  suite_add_tcase (s, tc_chain);
+  tcase_add_test (tc_chain, test_submemory);
+  tcase_add_test (tc_chain, test_submemory_writable);
+  tcase_add_test (tc_chain, test_writable);
+  tcase_add_test (tc_chain, test_is_span);
+  tcase_add_test (tc_chain, test_copy);
+  tcase_add_test (tc_chain, test_try_new_and_alloc);
+  tcase_add_test (tc_chain, test_resize);
+
+  return s;
+}
+
+GST_CHECK_MAIN (gst_memory);