Commit a625df4636

dweiller <4678790+dweiller@users.noreplay.github.com>
2023-02-04 01:39:48
std.compress.zstandard: fix fse decoding crash
master
1 parent 1c509f4
Changed files (1)
lib
std
compress
zstandard
decode
fse.zig
lib/std/compress/zstandard/decode/fse.zig
@@ -46,6 +46,7 @@ pub fn decodeFseTable(
         if (value == 1) {
             while (true) {
                 const repeat_flag = try bit_reader.readBitsNoEof(u2, 2);
+                if (repeat_flag + value_count > 256) return error.MalformedFseTable;
                 var i: usize = 0;
                 while (i < repeat_flag) : (i += 1) {
                     values[value_count] = 1;
@@ -54,6 +55,7 @@ pub fn decodeFseTable(
                 if (repeat_flag < 3) break;
             }
         }
+        if (value_count == 256) break;
     }
     bit_reader.alignToByte();