Commit d7bebe48 authored by Anton Khirnov's avatar Anton Khirnov

hevc: skip invalid/ignored NALUs when splitting the packet

There is no need to wait until actually decoding the NALU. This will
allow to get rid of the second hls_nal_unit() call later.
parent b11acd57
...@@ -2503,13 +2503,7 @@ static int decode_nal_unit(HEVCContext *s, const HEVCNAL *nal) ...@@ -2503,13 +2503,7 @@ static int decode_nal_unit(HEVCContext *s, const HEVCNAL *nal)
if (ret < 0) if (ret < 0)
return ret; return ret;
ret = hls_nal_unit(s); hls_nal_unit(s);
if (ret < 0) {
av_log(s->avctx, AV_LOG_ERROR, "Invalid NAL unit %d, skipping.\n",
s->nal_unit_type);
goto fail;
} else if (!ret)
return 0;
switch (s->nal_unit_type) { switch (s->nal_unit_type) {
case NAL_VPS: case NAL_VPS:
...@@ -2709,12 +2703,22 @@ static int decode_nal_units(HEVCContext *s, const uint8_t *buf, int length) ...@@ -2709,12 +2703,22 @@ static int decode_nal_units(HEVCContext *s, const uint8_t *buf, int length)
ret = init_get_bits8(&s->HEVClc.gb, nal->data, nal->size); ret = init_get_bits8(&s->HEVClc.gb, nal->data, nal->size);
if (ret < 0) if (ret < 0)
goto fail; goto fail;
hls_nal_unit(s);
ret = hls_nal_unit(s);
if (ret <= 0) {
if (ret < 0) {
av_log(s->avctx, AV_LOG_ERROR, "Invalid NAL unit %d, skipping.\n",
s->nal_unit_type);
}
s->nb_nals--;
goto skip_nal;
}
if (s->nal_unit_type == NAL_EOB_NUT || if (s->nal_unit_type == NAL_EOB_NUT ||
s->nal_unit_type == NAL_EOS_NUT) s->nal_unit_type == NAL_EOS_NUT)
s->eos = 1; s->eos = 1;
skip_nal:
buf += consumed; buf += consumed;
length -= consumed; length -= consumed;
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment