changeset 2944:3785497656c9 libavcodec

Correctly skip unknown matrices (patch by Matthieu Castet) and disable comment reading
author alex
date Sun, 13 Nov 2005 20:06:31 +0000
parents 6bf62c1cf1b6
children 394ea2703387
files vp3.c
diffstat 1 files changed, 21 insertions(+), 12 deletions(-) [+]
line wrap: on
line diff
--- a/vp3.c	Sun Nov 13 17:59:21 2005 +0000
+++ b/vp3.c	Sun Nov 13 20:06:31 2005 +0000
@@ -63,7 +63,7 @@
 
 #define KEYFRAMES_ONLY 0
 
-#define DEBUG_VP3 0
+#define DEBUG_VP3 1
 #define DEBUG_INIT 0
 #define DEBUG_DEQUANTIZERS 0
 #define DEBUG_BLOCK_CODING 0
@@ -2780,7 +2780,7 @@
 static int theora_decode_tables(AVCodecContext *avctx, GetBitContext gb)
 {
     Vp3DecodeContext *s = avctx->priv_data;
-    int i, n;
+    int i, n, matrices;
 
     if (s->theora >= 0x030200) {
         n = get_bits(&gb, 3);
@@ -2806,12 +2806,12 @@
 	s->coded_dc_scale_factor[i] = get_bits(&gb, n);
 
     if (s->theora >= 0x030200)
-        n = get_bits(&gb, 9) + 1;
+        matrices = get_bits(&gb, 9) + 1;
     else
-        n = 3;
-    if (n != 3) {
-        av_log(avctx,AV_LOG_ERROR, "unsupported nbms : %d\n", n);
-        return -1;
+        matrices = 3;
+    if (matrices != 3) {
+        av_log(avctx,AV_LOG_ERROR, "unsupported matrices: %d\n", matrices);
+//        return -1;
     }
     /* y coeffs */
     for (i = 0; i < 64; i++)
@@ -2825,7 +2825,12 @@
     for (i = 0; i < 64; i++)
 	s->coded_inter_dequant[i] = get_bits(&gb, 8);
 
-    /* Huffman tables */
+    /* skip unknown matrices */
+    n = matrices - 3;
+    while(n--)
+	for (i = 0; i < 64; i++)
+	    skip_bits(&gb, 8);
+
     for (i = 0; i <= 1; i++) {
         for (n = 0; n <= 2; n++) {
             int newqr;
@@ -2839,17 +2844,20 @@
             }
             else {
                 int qi = 0;
-                skip_bits(&gb, av_log2(2)+1);
+                skip_bits(&gb, av_log2(matrices-1)+1);
                 while (qi < 63) {
                     qi += get_bits(&gb, av_log2(63-qi)+1) + 1;
-                    skip_bits(&gb, av_log2(2)+1);
+                    skip_bits(&gb, av_log2(matrices-1)+1);
                 }
-                if (qi > 63)
+                if (qi > 63) {
                     av_log(avctx, AV_LOG_ERROR, "invalid qi %d > 63\n", qi);
+		    return -1;
+		}
             }
         }
     }
 
+    /* Huffman tables */
     for (s->hti = 0; s->hti < 80; s->hti++) {
         s->entries = 0;
         s->huff_code_size = 1;
@@ -2907,7 +2915,8 @@
             theora_decode_header(avctx, gb);
     	    break;
 	case 0x81:
-	    theora_decode_comments(avctx, gb);
+// FIXME: is this needed? it breaks sometimes
+//	    theora_decode_comments(avctx, gb);
 	    break;
 	case 0x82:
 	    theora_decode_tables(avctx, gb);