tests: rtph264depay: add test for using downstream memory allocator

4 files changed, 316 insertions, 0 deletions

diff --git a/tests/check/Makefile.am b/tests/check/Makefile.am
index 5085eea5f..c17c90797 100644
--- a/tests/check/Makefile.am
+++ b/tests/check/Makefile.am
@@ -235,6 +235,7 @@ check_rtp = \
 	elements/rtp-payloading \
 	elements/rtph261 \
 	elements/rtph263 \
+	elements/rtph264 \
 	elements/rtpvp9
 else
 check_rtp =
@@ -571,6 +572,9 @@ elements_rtph261_LDADD = $(GST_PLUGINS_BASE_LIBS) -lgstrtp-$(GST_API_VERSION) $(
 elements_rtph263_CFLAGS = $(GST_PLUGINS_BASE_CFLAGS) $(GST_BASE_CFLAGS) $(AM_CFLAGS)
 elements_rtph263_LDADD = $(GST_PLUGINS_BASE_LIBS) -lgstrtp-$(GST_API_VERSION) $(GST_BASE_LIBS) $(LDADD)
 
+elements_rtph264_CFLAGS = $(GST_PLUGINS_BASE_CFLAGS) $(GST_BASE_CFLAGS) $(AM_CFLAGS)
+elements_rtph264_LDADD = $(GST_PLUGINS_BASE_LIBS) -lgstapp-$(GST_API_VERSION) $(GST_BASE_LIBS) $(LDADD)
+
 elements_rtpmux_CFLAGS = $(GST_PLUGINS_BASE_CFLAGS) $(GST_BASE_CFLAGS) $(AM_CFLAGS)
 elements_rtpmux_LDADD = $(GST_PLUGINS_BASE_LIBS) -lgstrtp-$(GST_API_VERSION) $(GST_BASE_LIBS) $(LDADD)
 
diff --git a/tests/check/elements/rtph264.c b/tests/check/elements/rtph264.c
new file mode 100644
index 000000000..323e1bfa5
--- /dev/null
+++ b/tests/check/elements/rtph264.c
@@ -0,0 +1,311 @@
+/* GStreamer RTP H.264 unit test
+ *
+ * Copyright (C) 2017 Centricular Ltd
+ *   @author: Tim-Philipp Müller <tim@centricular.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., 51 Franklin St, Fifth Floor,
+ * Boston, MA 02110-1301, USA.
+ */
+
+#include <gst/check/check.h>
+#include <gst/app/app.h>
+
+#define ALLOCATOR_CUSTOM_SYSMEM "CustomSysMem"
+
+static GstAllocator *custom_sysmem_allocator;   /* NULL */
+
+/* Custom memory */
+
+typedef struct
+{
+  GstMemory mem;
+  guint8 *data;
+  guint8 *allocdata;
+} CustomSysmem;
+
+static CustomSysmem *
+custom_sysmem_new (GstMemoryFlags flags, gsize maxsize, gsize align,
+    gsize offset, gsize size)
+{
+  gsize aoffset, padding;
+  CustomSysmem *mem;
+
+  /* ensure configured alignment */
+  align |= gst_memory_alignment;
+  /* allocate more to compensate for alignment */
+  maxsize += align;
+
+  mem = g_new0 (CustomSysmem, 1);
+
+  mem->allocdata = g_malloc (maxsize);
+
+  mem->data = mem->allocdata;
+
+  /* do alignment */
+  if ((aoffset = ((guintptr) mem->data & align))) {
+    aoffset = (align + 1) - aoffset;
+    mem->data += aoffset;
+    maxsize -= aoffset;
+  }
+
+  if (offset && (flags & GST_MEMORY_FLAG_ZERO_PREFIXED))
+    memset (mem->data, 0, offset);
+
+  padding = maxsize - (offset + size);
+  if (padding && (flags & GST_MEMORY_FLAG_ZERO_PADDED))
+    memset (mem->data + offset + size, 0, padding);
+
+  gst_memory_init (GST_MEMORY_CAST (mem), flags, custom_sysmem_allocator,
+      NULL, maxsize, align, offset, size);
+
+  return mem;
+}
+
+static gpointer
+custom_sysmem_map (CustomSysmem * mem, gsize maxsize, GstMapFlags flags)
+{
+  return mem->data;
+}
+
+static gboolean
+custom_sysmem_unmap (CustomSysmem * mem)
+{
+  return TRUE;
+}
+
+static CustomSysmem *
+custom_sysmem_copy (CustomSysmem * mem, gssize offset, gsize size)
+{
+  g_return_val_if_reached (NULL);
+}
+
+static CustomSysmem *
+custom_sysmem_share (CustomSysmem * mem, gssize offset, gsize size)
+{
+  g_return_val_if_reached (NULL);
+}
+
+static gboolean
+custom_sysmem_is_span (CustomSysmem * mem1, CustomSysmem * mem2, gsize * offset)
+{
+  g_return_val_if_reached (FALSE);
+}
+
+/* Custom allocator */
+
+typedef struct
+{
+  GstAllocator allocator;
+} CustomSysmemAllocator;
+
+typedef struct
+{
+  GstAllocatorClass allocator_class;
+} CustomSysmemAllocatorClass;
+
+GType custom_sysmem_allocator_get_type (void);
+G_DEFINE_TYPE (CustomSysmemAllocator, custom_sysmem_allocator,
+    GST_TYPE_ALLOCATOR);
+
+static GstMemory *
+custom_sysmem_allocator_alloc (GstAllocator * allocator, gsize size,
+    GstAllocationParams * params)
+{
+  gsize maxsize = size + params->prefix + params->padding;
+
+  return (GstMemory *) custom_sysmem_new (params->flags,
+      maxsize, params->align, params->prefix, size);
+}
+
+static void
+custom_sysmem_allocator_free (GstAllocator * allocator, GstMemory * mem)
+{
+  CustomSysmem *csmem = (CustomSysmem *) mem;
+
+  g_free (csmem->allocdata);
+  g_free (csmem);
+}
+
+static void
+custom_sysmem_allocator_class_init (CustomSysmemAllocatorClass * klass)
+{
+  GstAllocatorClass *allocator_class = (GstAllocatorClass *) klass;
+
+  allocator_class->alloc = custom_sysmem_allocator_alloc;
+  allocator_class->free = custom_sysmem_allocator_free;
+}
+
+static void
+custom_sysmem_allocator_init (CustomSysmemAllocator * allocator)
+{
+  GstAllocator *alloc = GST_ALLOCATOR_CAST (allocator);
+
+  alloc->mem_type = ALLOCATOR_CUSTOM_SYSMEM;
+  alloc->mem_map = (GstMemoryMapFunction) custom_sysmem_map;
+  alloc->mem_unmap = (GstMemoryUnmapFunction) custom_sysmem_unmap;
+  alloc->mem_copy = (GstMemoryCopyFunction) custom_sysmem_copy;
+  alloc->mem_share = (GstMemoryShareFunction) custom_sysmem_share;
+  alloc->mem_is_span = (GstMemoryIsSpanFunction) custom_sysmem_is_span;
+}
+
+/* AppSink subclass proposing our custom allocator to upstream */
+
+typedef struct
+{
+  GstAppSink appsink;
+} CMemAppSink;
+
+typedef struct
+{
+  GstAppSinkClass appsink;
+} CMemAppSinkClass;
+
+GType c_mem_app_sink_get_type (void);
+
+G_DEFINE_TYPE (CMemAppSink, c_mem_app_sink, GST_TYPE_APP_SINK);
+
+static void
+c_mem_app_sink_init (CMemAppSink * cmemsink)
+{
+}
+
+static gboolean
+c_mem_app_sink_propose_allocation (GstBaseSink * sink, GstQuery * query)
+{
+  gst_query_add_allocation_param (query, custom_sysmem_allocator, NULL);
+  return TRUE;
+}
+
+static void
+c_mem_app_sink_class_init (CMemAppSinkClass * klass)
+{
+  GstBaseSinkClass *basesink_class = (GstBaseSinkClass *) klass;
+
+  basesink_class->propose_allocation = c_mem_app_sink_propose_allocation;
+}
+
+#define RTP_H264_FILE GST_TEST_FILES_PATH G_DIR_SEPARATOR_S "h264.rtp"
+
+GST_START_TEST (test_rtph264depay_with_downstream_allocator)
+{
+  GstElement *pipeline, *src, *depay, *sink;
+  GstMemory *mem;
+  GstSample *sample;
+  GstBuffer *buf;
+  GstCaps *caps;
+
+  custom_sysmem_allocator =
+      g_object_new (custom_sysmem_allocator_get_type (), NULL);
+
+  pipeline = gst_pipeline_new ("pipeline");
+
+  src = gst_element_factory_make ("appsrc", NULL);
+
+  caps = gst_caps_new_simple ("application/x-rtp",
+      "media", G_TYPE_STRING, "video",
+      "payload", G_TYPE_INT, 96,
+      "clock-rate", G_TYPE_INT, 90000,
+      "encoding-name", G_TYPE_STRING, "H264",
+      "ssrc", G_TYPE_UINT, 1990683810,
+      "timestamp-offset", G_TYPE_UINT, 3697583446,
+      "seqnum-offset", G_TYPE_UINT, 15568,
+      "a-framerate", G_TYPE_STRING, "30", NULL);
+  g_object_set (src, "format", GST_FORMAT_TIME, "caps", caps, NULL);
+  gst_bin_add (GST_BIN (pipeline), src);
+  gst_caps_unref (caps);
+
+  depay = gst_element_factory_make ("rtph264depay", NULL);
+  gst_bin_add (GST_BIN (pipeline), depay);
+
+  sink = g_object_new (c_mem_app_sink_get_type (), NULL);
+  gst_bin_add (GST_BIN (pipeline), sink);
+
+  gst_element_link_many (src, depay, sink, NULL);
+
+  gst_element_set_state (pipeline, GST_STATE_PAUSED);
+
+  {
+    gchar *data, *pdata;
+    gsize len;
+
+    fail_unless (g_file_get_contents (RTP_H264_FILE, &data, &len, NULL));
+    fail_unless (len > 2);
+
+    pdata = data;
+    while (len > 2) {
+      GstFlowReturn flow;
+      guint16 packet_len;
+
+      packet_len = GST_READ_UINT16_BE (pdata);
+      GST_INFO ("rtp packet length: %u (bytes left: %u)", packet_len,
+          (guint) len);
+      fail_unless (len >= 2 + packet_len);
+
+      flow = gst_app_src_push_buffer (GST_APP_SRC (src),
+          gst_buffer_new_wrapped (g_memdup (pdata + 2, packet_len),
+              packet_len));
+
+      fail_unless_equals_int (flow, GST_FLOW_OK);
+
+      pdata += 2 + packet_len;
+      len -= 2 + packet_len;
+    }
+
+    g_free (data);
+  }
+
+  gst_app_src_end_of_stream (GST_APP_SRC (src));
+
+  sample = gst_app_sink_pull_preroll (GST_APP_SINK (sink));
+  fail_unless (sample != NULL);
+
+  buf = gst_sample_get_buffer (sample);
+
+  GST_LOG ("buffer has %u memories", gst_buffer_n_memory (buf));
+  GST_LOG ("buffer size: %u", (guint) gst_buffer_get_size (buf));
+
+  fail_unless (gst_buffer_n_memory (buf) > 0);
+  mem = gst_buffer_peek_memory (buf, 0);
+  fail_unless (mem != NULL);
+
+  GST_LOG ("buffer memory type: %s", mem->allocator->mem_type);
+  fail_unless (gst_memory_is_type (mem, ALLOCATOR_CUSTOM_SYSMEM));
+
+  gst_sample_unref (sample);
+
+  gst_element_set_state (pipeline, GST_STATE_NULL);
+
+  gst_object_unref (pipeline);
+
+  g_object_unref (custom_sysmem_allocator);
+  custom_sysmem_allocator = NULL;
+}
+
+GST_END_TEST;
+
+static Suite *
+rtph264_suite (void)
+{
+  Suite *s = suite_create ("rtph264");
+  TCase *tc_chain;
+
+  tc_chain = tcase_create ("rtph264depay");
+  suite_add_tcase (s, tc_chain);
+  tcase_add_test (tc_chain, test_rtph264depay_with_downstream_allocator);
+
+  return s;
+}
+
+GST_CHECK_MAIN (rtph264);
diff --git a/tests/files/Makefile.am b/tests/files/Makefile.am
index 7c4ab60a8..7b3b4dc1d 100644
--- a/tests/files/Makefile.am
+++ b/tests/files/Makefile.am
@@ -3,6 +3,7 @@ EXTRA_DIST = \
 	audiotestsrc.flac \
 	audiotestsrc.wav \
 	gradient.j2k \
+	h264.rtp \
 	id3-407349-1.tag \
 	id3-407349-2.tag \
 	id3-447000-wcop.tag \
diff --git a/tests/files/h264.rtp b/tests/files/h264.rtp
new file mode 100644
index 000000000..53ad97e6d
--- /dev/null
+++ b/tests/files/h264.rtp