From a625df463648b7f6ff2d15de8be5e168c8bc7363 Mon Sep 17 00:00:00 2001
From: dweiller <4678790+dweiller@users.noreplay.github.com>
Date: Sat, 4 Feb 2023 11:39:48 +1100
Subject: [PATCH] std.compress.zstandard: fix fse decoding crash

---
 lib/std/compress/zstandard/decode/fse.zig | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/lib/std/compress/zstandard/decode/fse.zig b/lib/std/compress/zstandard/decode/fse.zig
index 5f87c1f81b..726891873c 100644
--- a/lib/std/compress/zstandard/decode/fse.zig
+++ b/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();