mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-04 23:46:43 +00:00
120 lines
3.3 KiB
C
120 lines
3.3 KiB
C
|
/* GStreamer
|
|||
|
*
|
|||
|
* Copyright (C) 2020 Seungha Yang <seungha@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/gstcheck.h>
|
|||
|
#include <gst/codecparsers/nalutils.h>
|
|||
|
#include <string.h>
|
|||
|
|
|||
|
GST_START_TEST (test_nal_writer_init)
|
|||
|
{
|
|||
|
NalWriter nw;
|
|||
|
|
|||
|
/* init with invalid params */
|
|||
|
ASSERT_CRITICAL (nal_writer_init (&nw, 0, TRUE));
|
|||
|
ASSERT_CRITICAL (nal_writer_init (&nw, 0, FALSE));
|
|||
|
ASSERT_CRITICAL (nal_writer_init (&nw, 5, TRUE));
|
|||
|
ASSERT_CRITICAL (nal_writer_init (&nw, 5, FALSE));
|
|||
|
|
|||
|
nal_writer_init (&nw, 4, FALSE);
|
|||
|
nal_writer_reset (&nw);
|
|||
|
}
|
|||
|
|
|||
|
GST_END_TEST;
|
|||
|
|
|||
|
GST_START_TEST (test_nal_writer_emulation_preventation)
|
|||
|
{
|
|||
|
NalWriter nw;
|
|||
|
gint i, j;
|
|||
|
static guint8 patterns[][3] = {
|
|||
|
{0x00, 0x00, 0x00},
|
|||
|
{0x00, 0x00, 0x01},
|
|||
|
{0x00, 0x00, 0x02},
|
|||
|
{0x00, 0x00, 0x03},
|
|||
|
};
|
|||
|
static guint8 expected_rst[][4] = {
|
|||
|
{0x00, 0x00, 0x03, 0x00},
|
|||
|
{0x00, 0x00, 0x03, 0x01},
|
|||
|
{0x00, 0x00, 0x03, 0x02},
|
|||
|
{0x00, 0x00, 0x03, 0x03},
|
|||
|
};
|
|||
|
|
|||
|
/* Within the NAL unit, the following three-byte sequences shall not occur
|
|||
|
* at any byte-aligned position:
|
|||
|
* – 0x000000
|
|||
|
* – 0x000001
|
|||
|
* – 0x000002
|
|||
|
* Within the NAL unit, any four-byte sequence that starts with 0x000003
|
|||
|
* other than the following sequences shall not occur
|
|||
|
* at any byte-aligned position:
|
|||
|
* – 0x00000300
|
|||
|
* – 0x00000301
|
|||
|
* – 0x00000302
|
|||
|
* – 0x00000303
|
|||
|
*/
|
|||
|
|
|||
|
for (i = 0; i < G_N_ELEMENTS (patterns); i++) {
|
|||
|
GstMemory *mem;
|
|||
|
GstMapInfo info;
|
|||
|
|
|||
|
nal_writer_init (&nw, 4, FALSE);
|
|||
|
|
|||
|
/* forbidden_zero_bit */
|
|||
|
fail_unless (nal_writer_put_bits_uint8 (&nw, 0, 1));
|
|||
|
/* nal_ref_idc, just set zero for test */
|
|||
|
fail_unless (nal_writer_put_bits_uint8 (&nw, 0, 2));
|
|||
|
/* nal_unit_type, unknown h264 nal type */
|
|||
|
fail_unless (nal_writer_put_bits_uint8 (&nw, 0x1f, 5));
|
|||
|
|
|||
|
for (j = 0; j < 3; j++)
|
|||
|
fail_unless (nal_writer_put_bits_uint8 (&nw, patterns[i][j], 8));
|
|||
|
|
|||
|
mem = nal_writer_reset_and_get_memory (&nw);
|
|||
|
|
|||
|
fail_unless (mem != NULL);
|
|||
|
fail_unless (gst_memory_map (mem, &info, GST_MAP_READ));
|
|||
|
|
|||
|
/* start code prefix 4 + nalu header 1 + written bytes 3 +
|
|||
|
* emulation prevention byte 1 */
|
|||
|
assert_equals_int (info.size, (4 + 1 + 3 + 1));
|
|||
|
|
|||
|
fail_if (memcmp (info.data + 5, expected_rst[i], 4));
|
|||
|
gst_memory_unmap (mem, &info);
|
|||
|
gst_memory_unref (mem);
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
GST_END_TEST;
|
|||
|
|
|||
|
static Suite *
|
|||
|
nalutils_suite (void)
|
|||
|
{
|
|||
|
Suite *s = suite_create ("H264/H264 Nal Utils");
|
|||
|
|
|||
|
TCase *tc_chain = tcase_create ("general");
|
|||
|
|
|||
|
suite_add_tcase (s, tc_chain);
|
|||
|
tcase_add_test (tc_chain, test_nal_writer_init);
|
|||
|
tcase_add_test (tc_chain, test_nal_writer_emulation_preventation);
|
|||
|
|
|||
|
return s;
|
|||
|
}
|
|||
|
|
|||
|
GST_CHECK_MAIN (nalutils);
|