lavf/vc1dec: Reduce probe score for streams with invalid frames.
authorCarl Eugen Hoyos <ceffmpeg@gmail.com>
Fri, 19 Apr 2019 12:06:29 +0000 (14:06 +0200)
committerCarl Eugen Hoyos <ceffmpeg@gmail.com>
Fri, 19 Apr 2019 12:06:29 +0000 (14:06 +0200)
Fixes ticket #7853.

libavformat/vc1dec.c

index f4101ae..fdbb05d 100644 (file)
@@ -26,7 +26,7 @@
 
 static int vc1_probe(const AVProbeData *p)
 {
 
 static int vc1_probe(const AVProbeData *p)
 {
-    int seq = 0, entry = 0, frame = 0, i;
+    int seq = 0, entry = 0, invalid = 0, frame = 0, i;
 
     for (i = 0; i < p->buf_size + 5; i++) {
         uint32_t code = AV_RB32(p->buf + i);
 
     for (i = 0; i < p->buf_size + 5; i++) {
         uint32_t code = AV_RB32(p->buf + i);
@@ -39,16 +39,19 @@ static int vc1_probe(const AVProbeData *p)
                 profile = (p->buf[i] & 0xc0) >> 6;
                 if (profile != PROFILE_ADVANCED) {
                     seq = 0;
                 profile = (p->buf[i] & 0xc0) >> 6;
                 if (profile != PROFILE_ADVANCED) {
                     seq = 0;
+                    invalid++;
                     continue;
                 }
                 level = (p->buf[i] & 0x38) >> 3;
                 if (level >= 5) {
                     seq = 0;
                     continue;
                 }
                 level = (p->buf[i] & 0x38) >> 3;
                 if (level >= 5) {
                     seq = 0;
+                    invalid++;
                     continue;
                 }
                 chromaformat = (p->buf[i] & 0x6) >> 1;
                 if (chromaformat != 1) {
                     seq = 0;
                     continue;
                 }
                 chromaformat = (p->buf[i] & 0x6) >> 1;
                 if (chromaformat != 1) {
                     seq = 0;
+                    invalid++;
                     continue;
                 }
                 seq++;
                     continue;
                 }
                 seq++;
@@ -56,8 +59,10 @@ static int vc1_probe(const AVProbeData *p)
                 break;
             }
             case VC1_CODE_ENTRYPOINT:
                 break;
             }
             case VC1_CODE_ENTRYPOINT:
-                if (!seq)
+                if (!seq) {
+                    invalid++;
                     continue;
                     continue;
+                }
                 entry++;
                 i += 2;
                 break;
                 entry++;
                 i += 2;
                 break;
@@ -71,9 +76,9 @@ static int vc1_probe(const AVProbeData *p)
         }
     }
 
         }
     }
 
-    if (frame > 1)
+    if (frame > 1 && frame >> 1 > invalid)
         return AVPROBE_SCORE_EXTENSION / 2 + 1;
         return AVPROBE_SCORE_EXTENSION / 2 + 1;
-    if (frame == 1)
+    if (frame >= 1)
         return AVPROBE_SCORE_EXTENSION / 4;
     return 0;
 }
         return AVPROBE_SCORE_EXTENSION / 4;
     return 0;
 }