]> git.nihav.org Git - nihav.git/commitdiff
realmedia/rv3040: ignore slices with wrong headers
authorKostya Shishkov <kostya.shishkov@gmail.com>
Wed, 11 Mar 2020 13:59:29 +0000 (14:59 +0100)
committerKostya Shishkov <kostya.shishkov@gmail.com>
Wed, 11 Mar 2020 13:59:29 +0000 (14:59 +0100)
nihav-realmedia/src/codecs/rv3040.rs

index d2ff13954d13eb8e0a53b491bf8465cad418e3b4..e45b62b0682204306551761a04f147b1a3871264 100644 (file)
@@ -539,10 +539,23 @@ fn decode_slice_header(br: &mut BitReader, bd: &mut RV34BitstreamDecoder, slice_
     if slice_no < slice_offs.len() - 1 {
         let cur_pos = br.tell() as u32;
         br.seek((slice_offs[slice_no + 1] * 8) as u32)?;
-        let nhdr = bd.decode_slice_header(br, shdr.width, shdr.height)?;
+        if let Ok(nhdr) = bd.decode_slice_header(br, shdr.width, shdr.height) {
+            validate!(nhdr.start > shdr.start);
+            shdr.end = nhdr.start;
+        } else {
+            if slice_no + 2 < slice_offs.len() {
+                br.seek((slice_offs[slice_no + 2] * 8) as u32)?;
+                if let Ok(nhdr) = bd.decode_slice_header(br, shdr.width, shdr.height) {
+                    validate!(nhdr.start > shdr.start);
+                    shdr.end = nhdr.start;
+                } else {
+                    shdr.end = ((shdr.width + 15) >> 4) * ((shdr.height + 15) >> 4);
+                }
+            } else {
+                shdr.end = ((shdr.width + 15) >> 4) * ((shdr.height + 15) >> 4);
+            }
+        }
         br.seek(cur_pos)?;
-        validate!(nhdr.start > shdr.start);
-        shdr.end = nhdr.start;
     } else {
         shdr.end = ((shdr.width + 15) >> 4) * ((shdr.height + 15) >> 4);
     }