Commit 31d1cae8c6

dweiller <4678790+dweiller@users.noreplay.github.com>
2023-01-24 03:14:06
std.compress.zstandard: validate fse table value count
master
1 parent 95953e1
Changed files (1)
lib
std
compress
zstandard
decompress.zig
lib/std/compress/zstandard/decompress.zig
@@ -1418,8 +1418,7 @@ fn decodeFseTable(
     }
     bit_reader.alignToByte();
 
-    // TODO: check there are at least 2 non-zero probabilities
-
+    if (value_count < 2) return error.MalformedFseTable;
     if (accumulated_probability != total_probability) return error.MalformedFseTable;
     if (value_count > expected_symbol_count) return error.MalformedFseTable;