mirrors/gstreamer
mirrors/gstreamer
1
1
Fork 0
mirror of https://gitlab.freedesktop.org/gstreamer/gstreamer.git synced 2025-02-02 12:32:29 +00:00
Code Issues 1 Projects Releases Wiki Activity

tests: Add h264parser SEI checks

Browse source 
Add some tests around SEI parsing.
This commit is contained in:
Jan Schmidt 2019-07-05 00:17:10 +10:00
parent de115dac4e
commit 15f149a990
1 changed files with 85 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

85
tests/check/libs/h264parser.c
View file

@ -229,6 +229,90 @@ GST_START_TEST (test_h264_parse_slice_5bytes)
GST_END_TEST;
static guint8 nalu_sps_with_vui[] = {
0x00, 0x00, 0x00, 0x01, 0x67, 0x64, 0x00, 0x28,
0xac, 0xd9, 0x40, 0x78, 0x04, 0x4f, 0xde, 0x03,
0xd2, 0x02, 0x02, 0x02, 0x80, 0x00, 0x01, 0xf4,
0x80, 0x00, 0x75, 0x30, 0x4f, 0x8b, 0x16, 0xcb
};
static guint8 nalu_sei_pic_timing[] = {
0x00, 0x00, 0x01, 0x06, 0x01, 0x02, 0x32, 0x80
};
static guint8 nalu_chained_sei[] = {
0x00, 0x00, 0x01, 0x06, 0x01, 0x02, 0x32, 0x80,
0x06, 0x01, 0xc4, 0x80
};
GST_START_TEST (test_h264_parse_invalid_sei)
{
GstH264ParserResult res;
GstH264NalUnit nalu;
GstH264NalParser *const parser = gst_h264_nal_parser_new ();
const guint8 *buf = nalu_sps_with_vui;
GArray *seis = NULL;
GstH264SEIMessage *sei;
/* First try parsing the SEI, which will fail because there's no SPS yet */
res =
gst_h264_parser_identify_nalu (parser, nalu_sei_pic_timing, 0,
sizeof (nalu_sei_pic_timing), &nalu);
assert_equals_int (res, GST_H264_PARSER_NO_NAL_END);
assert_equals_int (nalu.type, GST_H264_NAL_SEI);
res = gst_h264_parser_parse_sei (parser, &nalu, &seis);
assert_equals_int (res, GST_H264_PARSER_BROKEN_LINK);
g_array_free (seis, TRUE);
/* Inject SPS */
res =
gst_h264_parser_identify_nalu (parser, buf, 0, sizeof (nalu_sps_with_vui),
&nalu);
assert_equals_int (res, GST_H264_PARSER_NO_NAL_END);
assert_equals_int (nalu.type, GST_H264_NAL_SPS);
assert_equals_int (nalu.size, 28);
res = gst_h264_parser_parse_nal (parser, &nalu);
assert_equals_int (res, GST_H264_PARSER_OK);
/* Parse the SEI again */
res =
gst_h264_parser_identify_nalu (parser, nalu_sei_pic_timing, 0,
sizeof (nalu_sei_pic_timing), &nalu);
assert_equals_int (res, GST_H264_PARSER_NO_NAL_END);
assert_equals_int (nalu.type, GST_H264_NAL_SEI);
res = gst_h264_parser_parse_sei (parser, &nalu, &seis);
assert_equals_int (res, GST_H264_PARSER_OK);
fail_if (seis == NULL);
assert_equals_int (seis->len, 1);
g_array_free (seis, TRUE);
/* Parse NALU with 2 chained SEI */
res =
gst_h264_parser_identify_nalu (parser, nalu_chained_sei, 0,
sizeof (nalu_chained_sei), &nalu);
assert_equals_int (res, GST_H264_PARSER_NO_NAL_END);
assert_equals_int (nalu.type, GST_H264_NAL_SEI);
res = gst_h264_parser_parse_sei (parser, &nalu, &seis);
assert_equals_int (res, GST_H264_PARSER_OK);
fail_if (seis == NULL);
assert_equals_int (seis->len, 2);
sei = &g_array_index (seis, GstH264SEIMessage, 0);
assert_equals_int (sei->payloadType, GST_H264_SEI_PIC_TIMING);
sei = &g_array_index (seis, GstH264SEIMessage, 1);
assert_equals_int (sei->payloadType, GST_H264_SEI_RECOVERY_POINT);
g_array_free (seis, TRUE);
gst_h264_nal_parser_free (parser);
}
GST_END_TEST;
static Suite *
h264parser_suite (void)
{
@ -240,6 +324,7 @@ h264parser_suite (void)
tcase_add_test (tc_chain, test_h264_parse_slice_dpa);
tcase_add_test (tc_chain, test_h264_parse_slice_eoseq_slice);
tcase_add_test (tc_chain, test_h264_parse_slice_5bytes);
tcase_add_test (tc_chain, test_h264_parse_invalid_sei);
return s;
}