/* 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);