FaSTTUBe/GST-Tensordecoder-ov_ep
35
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

h265parse: Fix wrong NALU minimum length check

Browse Source 
Fixes a problem where an EOS/EOB NALU placed at the end of
an AU is detected as an other AU and create a buffer that
does not have valid pts.
This commit is contained in:
Yeongjin Jeong 2019-10-12 04:04:59 +09:00
parent 0318458b0e
commit 732735cab0
1 changed files with 18 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
gst/videoparsers/gsth265parse.c
View File

@ -276,7 +276,7 @@ gst_h265_parse_start (GstBaseParse * parse)
h265parse->nalparser = gst_h265_parser_new (); h265parse->nalparser = gst_h265_parser_new ();
h265parse->state = 0; h265parse->state = 0;
gst_base_parse_set_min_frame_size (parse, 7); gst_base_parse_set_min_frame_size (parse, 5);
return TRUE; return TRUE;
} }
@ -1023,14 +1023,15 @@ gst_h265_parse_collect_nal (GstH265Parse * h265parse, const guint8 * data,
GST_LOG_OBJECT (h265parse, "next nal type: %d %s", nal_type, GST_LOG_OBJECT (h265parse, "next nal type: %d %s", nal_type,
_nal_name (nal_type)); _nal_name (nal_type));
if (nnalu.size > nnalu.header_bytes) {
/* Any VCL Nal unit with first_slice_segment_in_pic_flag == 1 considered start of frame */ /* Any VCL Nal unit with first_slice_segment_in_pic_flag == 1 considered start of frame */
complete |= h265parse->picture_start complete |= h265parse->picture_start
&& (((nal_type >= GST_H265_NAL_SLICE_TRAIL_N && (((nal_type >= GST_H265_NAL_SLICE_TRAIL_N
&& nal_type <= GST_H265_NAL_SLICE_RASL_R) && nal_type <= GST_H265_NAL_SLICE_RASL_R)
|| (nal_type >= GST_H265_NAL_SLICE_BLA_W_LP || (nal_type >= GST_H265_NAL_SLICE_BLA_W_LP
&& nal_type <= RESERVED_IRAP_NAL_TYPE_MAX)) && nal_type <= RESERVED_IRAP_NAL_TYPE_MAX))
&& (nnalu.data[nnalu.offset + 2] & 0x80)); && (nnalu.data[nnalu.offset + 2] & 0x80));
}
GST_LOG_OBJECT (h265parse, "au complete: %d", complete); GST_LOG_OBJECT (h265parse, "au complete: %d", complete);
return complete; return complete;
@ -1159,8 +1160,10 @@ gst_h265_parse_handle_frame (GstBaseParse * parse,
data = map.data; data = map.data;
size = map.size; size = map.size;
/* expect at least 3 bytes startcode == sc, and 3 bytes NALU payload */ /* expect at least 3 bytes start_code, and 2 bytes NALU header.
if (G_UNLIKELY (size < 6)) { * the length of the NALU payload can be zero.
* (e.g. EOS/EOB placed at the end of an AU.) */
if (G_UNLIKELY (size < 5)) {
gst_buffer_unmap (buffer, &map); gst_buffer_unmap (buffer, &map);
*skipsize = 1; *skipsize = 1;
return GST_FLOW_OK; return GST_FLOW_OK;
@ -1279,7 +1282,10 @@ gst_h265_parse_handle_frame (GstBaseParse * parse,
nonext = nonext || (h265parse->align == GST_H265_PARSE_ALIGN_NAL); nonext = nonext || (h265parse->align == GST_H265_PARSE_ALIGN_NAL);
if (!nonext) { if (!nonext) {
if (nalu.offset + nalu.size + 5 + 2 > size) { /* expect at least 3 bytes start_code, and 2 bytes NALU header.
* the length of the NALU payload can be zero.
* (e.g. EOS/EOB placed at the end of an AU.) */
if (nalu.offset + nalu.size + 3 + 2 > size) {
GST_DEBUG_OBJECT (h265parse, "not enough data for next NALU"); GST_DEBUG_OBJECT (h265parse, "not enough data for next NALU");
if (drain) { if (drain) {
GST_DEBUG_OBJECT (h265parse, "but draining anyway"); GST_DEBUG_OBJECT (h265parse, "but draining anyway");