Address issue #44 (#45)

Address issue #44
RustAudio · Mar 2, 2019 · 9078060 · 9078060
1 parent 0d13bc4
commit 9078060
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/header.rs b/src/header.rs
@@ -792,7 +792,11 @@ fn read_floor(rdr :&mut BitpackCursor, codebook_cnt :u16, blocksizes :(u8, u8))
 			let mut floor0_book_list = Vec::with_capacity(
 				convert_to_usize!(floor0_number_of_books, u8));
 			for _ in 0 .. floor0_number_of_books {
-				floor0_book_list.push(try!(rdr.read_u8()));
+				let value = try!(rdr.read_u8());
+				if value > codebook_cnt as u8 {
+					try!(Err(HeaderReadError::HeaderBadFormat));
+				}
+				floor0_book_list.push(value);
 			}
 			Ok(Floor::TypeZero(FloorTypeZero {
 				floor0_order,