]> git.sesse.net Git - ffmpeg/commitdiff
avcodec/vp3: Do not initialize unused tables for keyframes in unpack_superblock()
authorMichael Niedermayer <michael@niedermayer.cc>
Sun, 28 Oct 2018 11:46:52 +0000 (12:46 +0100)
committerMichael Niedermayer <michael@niedermayer.cc>
Mon, 29 Oct 2018 16:00:49 +0000 (17:00 +0100)
Fixes: Timeout (139sec -> 102sec)
Fixes: 9642/clusterfuzz-testcase-minimized-ffmpeg_AV_CODEC_ID_VP3_fuzzer-6676767875006464
Found-by: continuous fuzzing process https://github.com/google/oss-fuzz/tree/master/projects/ffmpeg
Signed-off-by: Michael Niedermayer <michael@niedermayer.cc>
libavcodec/vp3.c

index 0e6da89abbcad811727bba6ad029b0e3c6405273..348416b25d9968f8f13e29e61d4b8702b205ea80 100644 (file)
@@ -544,8 +544,21 @@ static int unpack_superblocks(Vp3DecodeContext *s, GetBitContext *gb)
                                          : s->y_superblock_count);
         int num_coded_frags = 0;
 
+        if (s->keyframe) {
+            for (i = sb_start; i < sb_end; i++) {
+                /* iterate through all 16 fragments in a superblock */
+                for (j = 0; j < 16; j++) {
+                    /* if the fragment is in bounds, check its coding status */
+                    current_fragment = s->superblock_fragments[i * 16 + j];
+                    if (current_fragment != -1) {
+                        s->coded_fragment_list[plane][num_coded_frags++] =
+                            current_fragment;
+                    }
+                }
+            }
+        } else {
         for (i = sb_start; i < sb_end && get_bits_left(gb) > 0; i++) {
-            if (s->keyframe == 0 && get_bits_left(gb) < plane0_num_coded_frags >> 2) {
+            if (get_bits_left(gb) < plane0_num_coded_frags >> 2) {
                 return AVERROR_INVALIDDATA;
             }
             /* iterate through all 16 fragments in a superblock */
@@ -580,6 +593,7 @@ static int unpack_superblocks(Vp3DecodeContext *s, GetBitContext *gb)
                 }
             }
         }
+        }
         if (!plane)
             plane0_num_coded_frags = num_coded_frags;
         s->total_num_coded_frags += num_coded_frags;