Fix crash in vapoursynth decoder

rust-av · Dec 2, 2022 · 2205b29 · 2205b29
1 parent 22bc881
commit 2205b29
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/av_metrics_decoders/Cargo.toml b/av_metrics_decoders/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "av-metrics-decoders"
-version = "0.2.0"
+version = "0.2.1"
 authors = ["Josh Holmer <[email protected]>"]
 edition = "2021"
 description = "Decoders for use with av-metrics"

diff --git a/av_metrics_decoders/src/vapoursynth.rs b/av_metrics_decoders/src/vapoursynth.rs
@@ -101,7 +101,13 @@ impl Decoder for VapoursynthDecoder {
  &mut self,
  ) -> Option<av_metrics::video::Frame<T>> {
  let details = self.get_video_details();
- assert!(details.bit_depth == size_of::<T>());
+ if details.bit_depth <= 8 {
+ assert!(size_of::<T>() == 1);
+ } else if details.bit_depth <= 16 {
+ assert!(size_of::<T>() == 2);
+ } else {
+ panic!("Unsupported bit depth");
+ }
 
  let mut f: av_metrics::video::Frame<T> = av_metrics::video::Frame::new_with_padding(
  details.width,