gstreamer/tests/check/elements/uvch264demux.c
2012-09-10 16:09:26 -04:00

696 lines
24 KiB
C

/* GStreamer
*
* unit test for uvch264_demux
*
* Copyright (C) <2012> Collabora Ltd.
* Author: Youness Alaoui <youness.alaoui@collabora.co.uk>
* Copyright (C) <2008> Sebastian Dröge <sebastian.droege@collabora.co.uk>
*
* 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 <gst/check/gstcheck.h>
#include <string.h>
static GstElement *demux;
static GstPad *mjpg_pad, *h264_pad, *yuy2_pad, *nv12_pad, *jpg_pad;
static gboolean have_h264_eos, have_yuy2_eos, have_nv12_eos, have_jpg_eos;
static GstBuffer *buffer_h264, *buffer_yuy2, *buffer_nv12, *buffer_jpg;
static GError *gerror;
static gchar *error_debug;
static GstStaticPadTemplate mjpg_template =
GST_STATIC_PAD_TEMPLATE ("src", GST_PAD_SRC, GST_PAD_ALWAYS,
GST_STATIC_CAPS ("image/jpeg, width=640, height=480, framerate=15/1"));
static GstStaticPadTemplate sink_template =
GST_STATIC_PAD_TEMPLATE ("sink", GST_PAD_SINK, GST_PAD_ALWAYS,
GST_STATIC_CAPS_ANY);
#define STRINGIFY_(x) #x
#define STRINGIFY(x) STRINGIFY_ (x)
#define DATADIR STRINGIFY (UVCH264DEMUX_DATADIR)
#define VALID_H264_JPG_MJPG_FILENAME DATADIR "/valid_h264_jpg.mjpg"
#define VALID_H264_JPG_JPG_FILENAME DATADIR "/valid_h264_jpg.jpg"
#define VALID_H264_JPG_H264_FILENAME DATADIR "/valid_h264_jpg.h264"
#define VALID_H264_YUY2_MJPG_FILENAME DATADIR "/valid_h264_yuy2.mjpg"
#define VALID_H264_YUY2_YUY2_FILENAME DATADIR "/valid_h264_yuy2.yuy2"
#define VALID_H264_YUY2_H264_FILENAME DATADIR "/valid_h264_yuy2.h264"
#define _sink_chain_func(type) \
static GstFlowReturn \
_sink_##type##_chain (GstPad * pad, GstBuffer * buffer) \
{ \
fail_unless (GST_BUFFER_CAPS (buffer) != NULL); \
\
buffer_##type = buffer; \
\
return GST_FLOW_OK; \
}
#define _sink_event_func(type) \
static gboolean \
_sink_##type##_event (GstPad * pad, GstEvent * event) \
{ \
if (GST_EVENT_TYPE (event) == GST_EVENT_EOS) \
have_##type##_eos = TRUE; \
\
gst_event_unref (event); \
\
return TRUE; \
}
_sink_chain_func (h264);
_sink_chain_func (yuy2);
_sink_chain_func (nv12);
_sink_chain_func (jpg);
_sink_event_func (h264);
_sink_event_func (yuy2);
_sink_event_func (nv12);
_sink_event_func (jpg);
static GstBusSyncReply
_bus_sync_handler (GstBus * bus, GstMessage * message, gpointer data)
{
if (GST_MESSAGE_TYPE (message) == GST_MESSAGE_ERROR) {
fail_unless (gerror == NULL && error_debug == NULL);
fail_unless (GST_MESSAGE_SRC (message) == GST_OBJECT (demux));
gst_message_parse_error (message, &gerror, &error_debug);
}
return GST_BUS_PASS;
}
static void
_teardown_test (void)
{
GstBus *bus;
gst_element_set_state (demux, GST_STATE_NULL);
bus = GST_ELEMENT_BUS (demux);
gst_bus_set_flushing (bus, TRUE);
gst_object_unref (bus);
gst_pad_set_active (mjpg_pad, FALSE);
gst_object_unref (mjpg_pad);
if (h264_pad) {
gst_pad_set_active (h264_pad, FALSE);
gst_object_unref (h264_pad);
}
if (yuy2_pad) {
gst_pad_set_active (yuy2_pad, FALSE);
gst_object_unref (yuy2_pad);
}
if (nv12_pad) {
gst_pad_set_active (nv12_pad, FALSE);
gst_object_unref (nv12_pad);
}
if (jpg_pad) {
gst_pad_set_active (jpg_pad, FALSE);
gst_object_unref (jpg_pad);
}
if (gerror) {
g_error_free (gerror);
gerror = NULL;
}
if (error_debug) {
g_free (error_debug);
error_debug = NULL;
}
gst_object_unref (demux);
mjpg_pad = h264_pad = yuy2_pad = nv12_pad = jpg_pad = NULL;
demux = NULL;
}
static void
_setup_test (gboolean link_h264, gboolean link_yuy2, gboolean link_nv12,
gboolean link_jpg)
{
GstBus *bus = gst_bus_new ();
GstPad *sinkpad, *h264pad, *yuy2pad, *nv12pad, *jpgpad;
have_h264_eos = have_yuy2_eos = have_nv12_eos = have_jpg_eos = FALSE;
buffer_h264 = buffer_yuy2 = buffer_nv12 = buffer_jpg = NULL;
demux = gst_element_factory_make ("uvch264_mjpgdemux", NULL);
fail_unless (demux != NULL);
gst_element_set_bus (demux, bus);
gst_bus_set_sync_handler (bus, _bus_sync_handler, NULL);
mjpg_pad = gst_pad_new_from_static_template (&mjpg_template, "src");
fail_unless (mjpg_pad != NULL);
sinkpad = gst_element_get_static_pad (demux, "sink");
fail_unless (sinkpad != NULL);
fail_unless (gst_pad_link (mjpg_pad, sinkpad) == GST_PAD_LINK_OK);
gst_object_unref (sinkpad);
gst_pad_set_active (mjpg_pad, TRUE);
if (link_h264) {
h264pad = gst_element_get_static_pad (demux, "h264");
fail_unless (h264pad != NULL);
h264_pad = gst_pad_new_from_static_template (&sink_template, "h264");
fail_unless (h264_pad != NULL);
gst_pad_set_chain_function (h264_pad, _sink_h264_chain);
gst_pad_set_event_function (h264_pad, _sink_h264_event);
fail_unless (gst_pad_link (h264pad, h264_pad) == GST_PAD_LINK_OK);
gst_object_unref (h264pad);
gst_pad_set_active (h264_pad, TRUE);
}
if (link_yuy2) {
yuy2pad = gst_element_get_static_pad (demux, "yuy2");
fail_unless (yuy2pad != NULL);
yuy2_pad = gst_pad_new_from_static_template (&sink_template, "yuy2");
fail_unless (yuy2_pad != NULL);
gst_pad_set_chain_function (yuy2_pad, _sink_yuy2_chain);
gst_pad_set_event_function (yuy2_pad, _sink_yuy2_event);
fail_unless (gst_pad_link (yuy2pad, yuy2_pad) == GST_PAD_LINK_OK);
gst_object_unref (yuy2pad);
gst_pad_set_active (yuy2_pad, TRUE);
}
if (link_nv12) {
nv12pad = gst_element_get_static_pad (demux, "nv12");
fail_unless (nv12pad != NULL);
nv12_pad = gst_pad_new_from_static_template (&sink_template, "nv12");
fail_unless (nv12_pad != NULL);
gst_pad_set_chain_function (nv12_pad, _sink_nv12_chain);
gst_pad_set_event_function (nv12_pad, _sink_nv12_event);
fail_unless (gst_pad_link (nv12pad, nv12_pad) == GST_PAD_LINK_OK);
gst_object_unref (nv12pad);
gst_pad_set_active (nv12_pad, TRUE);
}
if (link_jpg) {
jpgpad = gst_element_get_static_pad (demux, "jpeg");
fail_unless (jpgpad != NULL);
jpg_pad = gst_pad_new_from_static_template (&sink_template, "jpeg");
fail_unless (jpg_pad != NULL);
gst_pad_set_chain_function (jpg_pad, _sink_jpg_chain);
gst_pad_set_event_function (jpg_pad, _sink_jpg_event);
fail_unless (gst_pad_link (jpgpad, jpg_pad) == GST_PAD_LINK_OK);
gst_object_unref (jpgpad);
gst_pad_set_active (jpg_pad, TRUE);
}
gst_element_set_state (demux, GST_STATE_PLAYING);
}
static GstBuffer *
_buffer_from_file (const gchar * filename)
{
GstBuffer *buffer = gst_buffer_new ();
gchar *contents = NULL;
gsize length = 0;
fail_unless (g_file_get_contents (filename, &contents, &length, NULL));
GST_BUFFER_MALLOCDATA (buffer) = (guint8 *) contents;
GST_BUFFER_DATA (buffer) = (guint8 *) contents;
GST_BUFFER_SIZE (buffer) = length;
GST_BUFFER_OFFSET (buffer) = 0;
return buffer;
}
GST_START_TEST (test_valid_h264_jpg)
{
GstCaps *mjpg_caps = gst_static_pad_template_get_caps (&mjpg_template);
GstCaps *h264_caps;
GstBuffer *buffer;
gchar *h264_data, *jpg_data;
gsize h264_size, jpg_size;
_setup_test (TRUE, TRUE, TRUE, TRUE);
h264_caps = gst_caps_new_simple ("video/x-h264",
"width", G_TYPE_INT, 640, "height", G_TYPE_INT, 480,
"framerate", GST_TYPE_FRACTION, 15, 1, NULL);
buffer = _buffer_from_file (VALID_H264_JPG_MJPG_FILENAME);
gst_buffer_set_caps (buffer, mjpg_caps);
fail_unless (g_file_get_contents (VALID_H264_JPG_H264_FILENAME,
&h264_data, &h264_size, NULL));
fail_unless (g_file_get_contents (VALID_H264_JPG_JPG_FILENAME,
&jpg_data, &jpg_size, NULL));
fail_unless (gst_pad_push (mjpg_pad, buffer) == GST_FLOW_OK);
fail_unless (gst_pad_push_event (mjpg_pad, gst_event_new_eos ()));
fail_unless (have_h264_eos);
fail_unless (have_yuy2_eos);
fail_unless (have_nv12_eos);
fail_unless (have_jpg_eos);
fail_unless (buffer_h264 != NULL);
fail_unless (buffer_jpg != NULL);
fail_unless (buffer_nv12 == NULL);
fail_unless (buffer_yuy2 == NULL);
fail_unless (gerror == NULL && error_debug == NULL);
fail_unless (gst_caps_is_always_compatible (GST_BUFFER_CAPS (buffer_h264),
h264_caps));
fail_unless (gst_caps_is_always_compatible (GST_BUFFER_CAPS (buffer_jpg),
mjpg_caps));
fail_unless (GST_BUFFER_SIZE (buffer_h264) == h264_size);
fail_unless (GST_BUFFER_SIZE (buffer_jpg) == jpg_size);
fail_unless (memcmp (GST_BUFFER_DATA (buffer_h264), h264_data,
h264_size) == 0);
fail_unless (memcmp (GST_BUFFER_DATA (buffer_jpg), jpg_data, jpg_size) == 0);
gst_caps_unref (mjpg_caps);
gst_caps_unref (h264_caps);
g_free (h264_data);
g_free (jpg_data);
gst_buffer_unref (buffer_h264);
gst_buffer_unref (buffer_jpg);
_teardown_test ();
}
GST_END_TEST;
GST_START_TEST (test_valid_h264_yuy2)
{
GstCaps *mjpg_caps = gst_static_pad_template_get_caps (&mjpg_template);
GstCaps *h264_caps;
GstCaps *yuy2_caps;
GstBuffer *buffer;
gchar *h264_data, *yuy2_data;
gsize h264_size, yuy2_size;
_setup_test (TRUE, TRUE, TRUE, TRUE);
h264_caps = gst_caps_new_simple ("video/x-h264",
"width", G_TYPE_INT, 640, "height", G_TYPE_INT, 480,
"framerate", GST_TYPE_FRACTION, 15, 1, NULL);
yuy2_caps = gst_caps_new_simple ("video/x-raw-yuv",
"format", GST_TYPE_FOURCC, GST_MAKE_FOURCC ('Y', 'U', 'Y', '2'),
"width", G_TYPE_INT, 160, "height", G_TYPE_INT, 90,
"framerate", GST_TYPE_FRACTION, 15, 1, NULL);
buffer = _buffer_from_file (VALID_H264_YUY2_MJPG_FILENAME);
gst_buffer_set_caps (buffer, mjpg_caps);
fail_unless (g_file_get_contents (VALID_H264_YUY2_H264_FILENAME,
&h264_data, &h264_size, NULL));
fail_unless (g_file_get_contents (VALID_H264_YUY2_YUY2_FILENAME,
&yuy2_data, &yuy2_size, NULL));
fail_unless (gst_pad_push (mjpg_pad, buffer) == GST_FLOW_OK);
fail_unless (gst_pad_push_event (mjpg_pad, gst_event_new_eos ()));
fail_unless (have_h264_eos);
fail_unless (have_yuy2_eos);
fail_unless (have_nv12_eos);
fail_unless (have_jpg_eos);
fail_unless (buffer_h264 != NULL);
fail_unless (buffer_jpg == NULL);
fail_unless (buffer_nv12 == NULL);
fail_unless (buffer_yuy2 != NULL);
fail_unless (gerror == NULL && error_debug == NULL);
fail_unless (gst_caps_is_always_compatible (GST_BUFFER_CAPS (buffer_h264),
h264_caps));
fail_unless (gst_caps_is_always_compatible (GST_BUFFER_CAPS (buffer_yuy2),
yuy2_caps));
fail_unless (GST_BUFFER_SIZE (buffer_h264) == h264_size);
fail_unless (GST_BUFFER_SIZE (buffer_yuy2) == yuy2_size);
fail_unless (memcmp (GST_BUFFER_DATA (buffer_h264), h264_data,
h264_size) == 0);
fail_unless (memcmp (GST_BUFFER_DATA (buffer_yuy2), yuy2_data,
yuy2_size) == 0);
gst_caps_unref (mjpg_caps);
gst_caps_unref (yuy2_caps);
gst_caps_unref (h264_caps);
g_free (h264_data);
g_free (yuy2_data);
gst_buffer_unref (buffer_h264);
gst_buffer_unref (buffer_yuy2);
_teardown_test ();
}
GST_END_TEST;
GST_START_TEST (test_no_data)
{
GstCaps *mjpg_caps = gst_static_pad_template_get_caps (&mjpg_template);
GstBuffer *buffer = gst_buffer_new ();
_setup_test (TRUE, TRUE, TRUE, TRUE);
gst_buffer_set_caps (buffer, mjpg_caps);
fail_unless (gst_pad_push (mjpg_pad, buffer) == GST_FLOW_OK);
fail_unless (gst_pad_push_event (mjpg_pad, gst_event_new_eos ()));
fail_unless (have_h264_eos && have_yuy2_eos && have_nv12_eos && have_jpg_eos);
fail_unless (buffer_h264 == NULL && buffer_jpg != NULL);
fail_unless (buffer_nv12 == NULL && buffer_yuy2 == NULL);
fail_unless (gerror == NULL && error_debug == NULL);
_teardown_test ();
}
GST_END_TEST;
GST_START_TEST (test_data_zero)
{
GstCaps *mjpg_caps = gst_static_pad_template_get_caps (&mjpg_template);
GstBuffer *buffer = gst_buffer_new_and_alloc (1024);
_setup_test (TRUE, TRUE, TRUE, TRUE);
memset (GST_BUFFER_DATA (buffer), 0, 1024);
GST_BUFFER_SIZE (buffer) = 1024;
gst_buffer_set_caps (buffer, mjpg_caps);
fail_unless (gst_pad_push (mjpg_pad, buffer) == GST_FLOW_OK);
fail_unless (gst_pad_push_event (mjpg_pad, gst_event_new_eos ()));
fail_unless (have_h264_eos && have_yuy2_eos && have_nv12_eos && have_jpg_eos);
fail_unless (buffer_h264 == NULL && buffer_jpg == NULL);
fail_unless (buffer_nv12 == NULL && buffer_yuy2 == NULL);
_teardown_test ();
}
GST_END_TEST;
GST_START_TEST (test_no_marker_size)
{
GstCaps *mjpg_caps = gst_static_pad_template_get_caps (&mjpg_template);
GstBuffer *buffer = gst_buffer_new_and_alloc (1024);
const guchar data[] = {
0xff, 0xd8, 0xff, 0xe4, 0x00
};
_setup_test (TRUE, TRUE, TRUE, TRUE);
memcpy (GST_BUFFER_DATA (buffer), data, sizeof (data));
GST_BUFFER_SIZE (buffer) = sizeof (data);
gst_buffer_set_caps (buffer, mjpg_caps);
fail_unless (gst_pad_push (mjpg_pad, buffer) == GST_FLOW_ERROR);
fail_unless (gst_pad_push_event (mjpg_pad, gst_event_new_eos ()));
fail_unless (have_h264_eos && have_yuy2_eos && have_nv12_eos && have_jpg_eos);
fail_unless (buffer_h264 == NULL && buffer_jpg == NULL);
fail_unless (buffer_nv12 == NULL && buffer_yuy2 == NULL);
fail_unless (gerror != NULL);
fail_unless (gerror->domain == GST_STREAM_ERROR);
fail_unless (gerror->code == GST_STREAM_ERROR_DEMUX);
fail_unless (memcmp (gerror->message,
"Not enough data to read marker size",
strlen (gerror->message)) == 0);
_teardown_test ();
}
GST_END_TEST;
GST_START_TEST (test_not_enough_data)
{
GstCaps *mjpg_caps = gst_static_pad_template_get_caps (&mjpg_template);
GstBuffer *buffer = gst_buffer_new_and_alloc (1024);
const guchar data[] = {
0xff, 0xd8, 0xff, 0xe4, 0x00, 0xff, 0x00, 0x00
};
_setup_test (TRUE, TRUE, TRUE, TRUE);
memcpy (GST_BUFFER_DATA (buffer), data, sizeof (data));
GST_BUFFER_SIZE (buffer) = sizeof (data);
gst_buffer_set_caps (buffer, mjpg_caps);
fail_unless (gst_pad_push (mjpg_pad, buffer) == GST_FLOW_ERROR);
fail_unless (gst_pad_push_event (mjpg_pad, gst_event_new_eos ()));
fail_unless (have_h264_eos && have_yuy2_eos && have_nv12_eos && have_jpg_eos);
fail_unless (buffer_h264 == NULL && buffer_jpg == NULL);
fail_unless (buffer_nv12 == NULL && buffer_yuy2 == NULL);
fail_unless (gerror != NULL);
fail_unless (gerror->domain == GST_STREAM_ERROR);
fail_unless (gerror->code == GST_STREAM_ERROR_DEMUX);
fail_unless (memcmp (gerror->message,
"Not enough data to read marker content",
strlen (gerror->message)) == 0);
_teardown_test ();
}
GST_END_TEST;
GST_START_TEST (test_no_aux_header)
{
GstCaps *mjpg_caps = gst_static_pad_template_get_caps (&mjpg_template);
GstBuffer *buffer = gst_buffer_new_and_alloc (1024);
const guchar data[] = {
0xff, 0xd8, 0xff, 0xe4, 0x00, 0x02, 0x00, 0x00,
0xff, 0xd9
};
_setup_test (TRUE, TRUE, TRUE, TRUE);
memcpy (GST_BUFFER_DATA (buffer), data, sizeof (data));
GST_BUFFER_SIZE (buffer) = sizeof (data);
gst_buffer_set_caps (buffer, mjpg_caps);
fail_unless (gst_pad_push (mjpg_pad, buffer) == GST_FLOW_ERROR);
fail_unless (gst_pad_push_event (mjpg_pad, gst_event_new_eos ()));
fail_unless (have_h264_eos && have_yuy2_eos && have_nv12_eos && have_jpg_eos);
fail_unless (buffer_h264 == NULL && buffer_jpg == NULL);
fail_unless (buffer_nv12 == NULL && buffer_yuy2 == NULL);
fail_unless (gerror != NULL);
fail_unless (gerror->domain == GST_STREAM_ERROR);
fail_unless (gerror->code == GST_STREAM_ERROR_DEMUX);
fail_unless (memcmp (gerror->message,
"Not enough data to read aux header", strlen (gerror->message)) == 0);
_teardown_test ();
}
GST_END_TEST;
GST_START_TEST (test_empty_aux_data)
{
GstCaps *mjpg_caps = gst_static_pad_template_get_caps (&mjpg_template);
GstBuffer *buffer = gst_buffer_new_and_alloc (1024);
const guchar data[] = {
0xff, 0xd8, 0xff, 0xe4, 0x00, 0x1C, 0x00, 0x01,
0x16, 0x00, 0x48, 0x32, 0x36, 0x34, 0x80, 0x07,
0x38, 0x04, 0x2a, 0x2c, 0x0a, 0x00, 0x1b, 0x00,
0x40, 0x62, 0xcb, 0x0a, 0x00, 0x00, 0x00, 0x00,
0xff, 0xd9
};
_setup_test (TRUE, TRUE, TRUE, TRUE);
memcpy (GST_BUFFER_DATA (buffer), data, sizeof (data));
GST_BUFFER_SIZE (buffer) = sizeof (data);
gst_buffer_set_caps (buffer, mjpg_caps);
fail_unless (gst_pad_push (mjpg_pad, buffer) == GST_FLOW_OK);
fail_unless (gst_pad_push_event (mjpg_pad, gst_event_new_eos ()));
fail_unless (have_h264_eos && have_yuy2_eos && have_nv12_eos && have_jpg_eos);
fail_unless (buffer_h264 == NULL && buffer_jpg == NULL);
fail_unless (buffer_nv12 == NULL && buffer_yuy2 == NULL);
fail_unless (gerror == NULL);
_teardown_test ();
}
GST_END_TEST;
GST_START_TEST (test_unknown_fcc)
{
GstCaps *mjpg_caps = gst_static_pad_template_get_caps (&mjpg_template);
GstBuffer *buffer = gst_buffer_new_and_alloc (1024);
const guchar data[] = {
0xff, 0xd8, 0xff, 0xe4, 0x00, 0x2C, 0x00, 0x01,
0x16, 0x00, 0x48, 0x30, 0x30, 0x30, 0x80, 0x07,
0x38, 0x04, 0x2a, 0x2c, 0x0a, 0x00, 0x1b, 0x00,
0x40, 0x62, 0xcb, 0x0a, 0x10, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0xff, 0xd9
};
_setup_test (TRUE, TRUE, TRUE, TRUE);
memcpy (GST_BUFFER_DATA (buffer), data, sizeof (data));
GST_BUFFER_SIZE (buffer) = sizeof (data);
gst_buffer_set_caps (buffer, mjpg_caps);
fail_unless (gst_pad_push (mjpg_pad, buffer) == GST_FLOW_ERROR);
fail_unless (gst_pad_push_event (mjpg_pad, gst_event_new_eos ()));
fail_unless (have_h264_eos && have_yuy2_eos && have_nv12_eos && have_jpg_eos);
fail_unless (buffer_h264 == NULL && buffer_jpg == NULL);
fail_unless (buffer_nv12 == NULL && buffer_yuy2 == NULL);
fail_unless (gerror != NULL);
fail_unless (gerror->domain == GST_STREAM_ERROR);
fail_unless (gerror->code == GST_STREAM_ERROR_DEMUX);
fail_unless (memcmp (gerror->message,
"Unknown auxiliary stream format : H000",
strlen (gerror->message)) == 0);
_teardown_test ();
}
GST_END_TEST;
GST_START_TEST (test_not_enough_aux_data)
{
GstCaps *mjpg_caps = gst_static_pad_template_get_caps (&mjpg_template);
GstBuffer *buffer = gst_buffer_new_and_alloc (1024);
const guchar data[] = {
0xff, 0xd8, 0xff, 0xe4, 0x00, 0x1C, 0x00, 0x01,
0x16, 0x00, 0x48, 0x32, 0x36, 0x34, 0x80, 0x07,
0x38, 0x04, 0x2a, 0x2c, 0x0a, 0x00, 0x1b, 0x00,
0x40, 0x62, 0xcb, 0x0a, 0x10, 0x00, 0x00, 0x00,
0xff, 0xd9
};
_setup_test (TRUE, TRUE, TRUE, TRUE);
memcpy (GST_BUFFER_DATA (buffer), data, sizeof (data));
GST_BUFFER_SIZE (buffer) = sizeof (data);
gst_buffer_set_caps (buffer, mjpg_caps);
fail_unless (gst_pad_push (mjpg_pad, buffer) == GST_FLOW_ERROR);
fail_unless (gst_pad_push_event (mjpg_pad, gst_event_new_eos ()));
fail_unless (have_h264_eos && have_yuy2_eos && have_nv12_eos && have_jpg_eos);
fail_unless (buffer_h264 == NULL && buffer_jpg == NULL);
fail_unless (buffer_nv12 == NULL && buffer_yuy2 == NULL);
fail_unless (gerror != NULL);
fail_unless (gerror->domain == GST_STREAM_ERROR);
fail_unless (gerror->code == GST_STREAM_ERROR_DEMUX);
fail_unless (memcmp (gerror->message,
"Incomplete auxiliary stream. 16 bytes missing",
strlen (gerror->message)) == 0);
_teardown_test ();
}
GST_END_TEST;
GST_START_TEST (test_too_much_aux_data)
{
GstCaps *mjpg_caps = gst_static_pad_template_get_caps (&mjpg_template);
GstBuffer *buffer = gst_buffer_new_and_alloc (1024);
const guchar data[] = {
0xff, 0xd8, 0xff, 0xe4, 0x00, 0x3C, 0x00, 0x01,
0x16, 0x00, 0x48, 0x32, 0x36, 0x34, 0x80, 0x07,
0x38, 0x04, 0x2a, 0x2c, 0x0a, 0x00, 0x1b, 0x00,
0x40, 0x62, 0xcb, 0x0a, 0x10, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0xff, 0xd9
};
_setup_test (TRUE, TRUE, TRUE, TRUE);
memcpy (GST_BUFFER_DATA (buffer), data, sizeof (data));
GST_BUFFER_SIZE (buffer) = sizeof (data);
gst_buffer_set_caps (buffer, mjpg_caps);
fail_unless (gst_pad_push (mjpg_pad, buffer) == GST_FLOW_ERROR);
fail_unless (gst_pad_push_event (mjpg_pad, gst_event_new_eos ()));
fail_unless (have_h264_eos && have_yuy2_eos && have_nv12_eos && have_jpg_eos);
fail_unless (buffer_h264 == NULL && buffer_jpg == NULL);
fail_unless (buffer_nv12 == NULL && buffer_yuy2 == NULL);
fail_unless (gerror != NULL);
fail_unless (gerror->domain == GST_STREAM_ERROR);
fail_unless (gerror->code == GST_STREAM_ERROR_DEMUX);
fail_unless (memcmp (gerror->message,
"Expected 16 auxiliary data, got 32 bytes",
strlen (gerror->message)) == 0);
_teardown_test ();
}
GST_END_TEST;
GST_START_TEST (test_no_sos_marker)
{
GstCaps *mjpg_caps = gst_static_pad_template_get_caps (&mjpg_template);
GstBuffer *buffer = gst_buffer_new_and_alloc (1024);
const guchar data[] = {
0xff, 0xd8, 0xff, 0xe4, 0x00, 0x2C, 0x00, 0x01,
0x16, 0x00, 0x48, 0x32, 0x36, 0x34, 0x80, 0x07,
0x38, 0x04, 0x2a, 0x2c, 0x0a, 0x00, 0x1b, 0x00,
0x40, 0x62, 0xcb, 0x0a, 0x10, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0xff, 0xd9
};
const guchar h264_data[] = {
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
};
_setup_test (TRUE, TRUE, TRUE, TRUE);
memcpy (GST_BUFFER_DATA (buffer), data, sizeof (data));
GST_BUFFER_SIZE (buffer) = sizeof (data);
gst_buffer_set_caps (buffer, mjpg_caps);
fail_unless (gst_pad_push (mjpg_pad, buffer) == GST_FLOW_OK);
fail_unless (gst_pad_push_event (mjpg_pad, gst_event_new_eos ()));
fail_unless (have_h264_eos && have_yuy2_eos && have_nv12_eos && have_jpg_eos);
fail_unless (buffer_h264 != NULL && buffer_jpg == NULL);
fail_unless (buffer_nv12 == NULL && buffer_yuy2 == NULL);
fail_unless (gerror == NULL);
fail_unless (GST_BUFFER_SIZE (buffer_h264) == sizeof (h264_data));
fail_unless (memcmp (GST_BUFFER_DATA (buffer_h264), h264_data,
sizeof (h264_data)) == 0);
_teardown_test ();
}
GST_END_TEST;
static Suite *
uvch264demux_suite (void)
{
Suite *s = suite_create ("uvch264demux");
TCase *tc_chain = tcase_create ("general");
suite_add_tcase (s, tc_chain);
tcase_set_timeout (tc_chain, 180);
tcase_add_test (tc_chain, test_valid_h264_jpg);
tcase_add_test (tc_chain, test_valid_h264_yuy2);
tcase_add_test (tc_chain, test_no_data);
tcase_add_test (tc_chain, test_data_zero);
tcase_add_test (tc_chain, test_no_marker_size);
tcase_add_test (tc_chain, test_not_enough_data);
tcase_add_test (tc_chain, test_no_aux_header);
tcase_add_test (tc_chain, test_empty_aux_data);
tcase_add_test (tc_chain, test_unknown_fcc);
tcase_add_test (tc_chain, test_no_sos_marker);
tcase_add_test (tc_chain, test_not_enough_aux_data);
tcase_add_test (tc_chain, test_too_much_aux_data);
return s;
}
GST_CHECK_MAIN (uvch264demux);