test: h264/h265: Add test for four bytes start code initial skip

This test detects if the parser have skipped too much and dropped meaninful
NALs.

Part-of: <https://gitlab.freedesktop.org/gstreamer/gst-plugins-bad/-/merge_requests/1251>

diff --git a/tests/check/elements/h264parse.c b/tests/check/elements/h264parse.c
index e399a30dcaf6c0572eae3b4c465a560339951380..de5297c974adfb9f6c69e963907c1cdbaedb733a 100644 (file)
--- a/tests/check/elements/h264parse.c
+++ b/tests/check/elements/h264parse.c
@@ -1148,6 +1148,46 @@ GST_START_TEST (test_parse_sei_closedcaptions)
 
 GST_END_TEST;
 
+GST_START_TEST (test_parse_skip_to_4bytes_sc)
+{
+  GstHarness *h;
+  GstBuffer *buf1, *buf2;
+  const guint8 initial_bytes[] = { 0x00, 0x00, 0x00, 0x00, 0x01 };
+  GstMapInfo map;
+
+  h = gst_harness_new ("h264parse");
+
+  gst_harness_set_caps_str (h, "video/x-h264, stream-format=byte-stream",
+      "video/x-h264, stream-format=byte-stream, alignment=nal");
+
+  /* padding bytes, four bytes start code. */
+  buf1 = wrap_buffer (initial_bytes, sizeof (initial_bytes), 100, 0);
+
+  /* The second contains the an AUD, starting from NAL identification byte,
+   * and is followed by SPS, PPS and IDR */
+  buf2 = composite_buffer (100, 0, 4, h264_aud + 4, sizeof (h264_aud) - 4,
+      h264_sps, sizeof (h264_sps), h264_pps, sizeof (h264_pps),
+      h264_idrframe, sizeof (h264_idrframe));
+
+  fail_unless_equals_int (gst_harness_push (h, buf1), GST_FLOW_OK);
+  fail_unless_equals_int (gst_harness_buffers_in_queue (h), 0);
+
+  fail_unless_equals_int (gst_harness_push (h, buf2), GST_FLOW_OK);
+  /* The parser will deliver AUD, SPS, PPS as it now have complete caps */
+  fail_unless_equals_int (gst_harness_buffers_in_queue (h), 3);
+
+  buf1 = gst_harness_pull (h);
+  gst_buffer_map (buf1, &map, GST_MAP_READ);
+  fail_unless_equals_int (gst_buffer_get_size (buf1), sizeof (h264_aud));
+  gst_buffer_unmap (buf1, &map);
+  gst_buffer_unref (buf1);
+
+  gst_harness_teardown (h);
+}
+
+GST_END_TEST;
+
+
 /*
  * TODO:
  *   - Both push- and pull-modes need to be tested
@@ -1258,6 +1298,7 @@ main (int argc, char **argv)
     suite_add_tcase (s, tc_chain);
     tcase_add_test (tc_chain, test_parse_sei_closedcaptions);
     tcase_add_test (tc_chain, test_parse_compatible_caps);
+    tcase_add_test (tc_chain, test_parse_skip_to_4bytes_sc);
     nf += gst_check_run_suite (s, "h264parse", __FILE__);
   }
 

diff --git a/tests/check/elements/h265parse.c b/tests/check/elements/h265parse.c
index ff7a4902a935eac259642d22199d63d983391940..f32c7327d3915cf12491bc7d852da2c9d5d14818 100644 (file)
--- a/tests/check/elements/h265parse.c
+++ b/tests/check/elements/h265parse.c
@@ -955,6 +955,48 @@ GST_START_TEST (test_sliced_au_au)
 
 GST_END_TEST;
 
+GST_START_TEST (test_parse_skip_to_4bytes_sc)
+{
+  GstHarness *h;
+  GstBuffer *buf1, *buf2;
+  const guint8 initial_bytes[] = { 0x00, 0x00, 0x00, 0x00, 0x01, h265_vps[4] };
+  GstMapInfo map;
+
+  h = gst_harness_new ("h265parse");
+
+  gst_harness_set_caps_str (h, "video/x-h265, stream-format=byte-stream",
+      "video/x-h265, stream-format=byte-stream, alignment=nal");
+
+  /* padding bytes, four bytes start code and 1 of the two identification
+   * bytes. */
+  buf1 = wrap_buffer (initial_bytes, sizeof (initial_bytes), 100, 0);
+
+  /* The second contains the an VPS, starting from second NAL identification
+   * byte and is followed by an SPS, IDR to ensure that the NAL end can be
+   * found */
+  buf2 = composite_buffer (100, 0, 4, h265_vps + 5, sizeof (h265_vps) - 5,
+      h265_sps, sizeof (h265_sps), h265_pps, sizeof (h265_pps),
+      h265_idr, sizeof (h265_idr));
+
+  fail_unless_equals_int (gst_harness_push (h, buf1), GST_FLOW_OK);
+  fail_unless_equals_int (gst_harness_buffers_in_queue (h), 0);
+
+  fail_unless_equals_int (gst_harness_push (h, buf2), GST_FLOW_OK);
+  /* The parser will deliver VPS, SPS, PPS as it now have complete cpas */
+  fail_unless_equals_int (gst_harness_buffers_in_queue (h), 3);
+
+  buf1 = gst_harness_pull (h);
+  gst_buffer_map (buf1, &map, GST_MAP_READ);
+  fail_unless_equals_int (gst_buffer_get_size (buf1), sizeof (h265_vps));
+  gst_buffer_unmap (buf1, &map);
+  gst_buffer_unref (buf1);
+
+  gst_harness_teardown (h);
+}
+
+GST_END_TEST;
+
+
 /* nal->au has latency, but EOS should force the last AU out */
 GST_START_TEST (test_drain)
 {
@@ -1006,6 +1048,8 @@ h265parse_harnessed_suite (void)
   tcase_add_test (tc_chain, test_sliced_nal_au);
   tcase_add_test (tc_chain, test_sliced_au_au);
 
+  tcase_add_test (tc_chain, test_parse_skip_to_4bytes_sc);
+
   tcase_add_test (tc_chain, test_drain);
 
   return s;