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.

1 files changed, 12 insertions, 8 deletions

diff --git a/libavcodec/hevc.c b/libavcodec/hevc.c
index 9b1cd66495..7080d00eb8 100644
--- a/libavcodec/hevc.c
+++ b/libavcodec/hevc.c
@@ -2503,13 +2503,7 @@ static int decode_nal_unit(HEVCContext *s, const HEVCNAL *nal)
     if (ret < 0)
         return ret;
 
-    ret = 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;
+    hls_nal_unit(s);
 
     switch (s->nal_unit_type) {
     case NAL_VPS:
@@ -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);
         if (ret < 0)
             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 ||
             s->nal_unit_type == NAL_EOS_NUT)
             s->eos = 1;
 
+skip_nal:
         buf    += consumed;
         length -= consumed;
     }