| ... | @@ -46,6 +46,7 @@ pub fn decodeFseTable( | ... | @@ -46,6 +46,7 @@ pub fn decodeFseTable( |
| 46 | if (value == 1) { | 46 | if (value == 1) { |
| 47 | while (true) { | 47 | while (true) { |
| 48 | const repeat_flag = try bit_reader.readBitsNoEof(u2, 2); | 48 | const repeat_flag = try bit_reader.readBitsNoEof(u2, 2); |
| | 49 | if (repeat_flag + value_count > 256) return error.MalformedFseTable; |
| 49 | var i: usize = 0; | 50 | var i: usize = 0; |
| 50 | while (i < repeat_flag) : (i += 1) { | 51 | while (i < repeat_flag) : (i += 1) { |
| 51 | values[value_count] = 1; | 52 | values[value_count] = 1; |
| ... | @@ -54,6 +55,7 @@ pub fn decodeFseTable( | ... | @@ -54,6 +55,7 @@ pub fn decodeFseTable( |
| 54 | if (repeat_flag < 3) break; | 55 | if (repeat_flag < 3) break; |
| 55 | } | 56 | } |
| 56 | } | 57 | } |
| | 58 | if (value_count == 256) break; |
| 57 | } | 59 | } |
| 58 | bit_reader.alignToByte(); | 60 | bit_reader.alignToByte(); |
| 59 | | 61 | |