From c0d85cda53fd9ea056641d5e91cc8f736b34c0e7 Mon Sep 17 00:00:00 2001 From: Rui He <118280419+ruihe774@users.noreply.github.com> Date: Thu, 30 Jan 2025 23:42:15 +0800 Subject: [PATCH] std.fs.File: limit initial_cap according to max_bytes in readToEndAllocOptions --- lib/std/fs/File.zig | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/std/fs/File.zig b/lib/std/fs/File.zig index 69c3553ac3..9797a1b896 100644 --- a/lib/std/fs/File.zig +++ b/lib/std/fs/File.zig @@ -1156,7 +1156,7 @@ pub fn readToEndAllocOptions( // The file size returned by stat is used as hint to set the buffer // size. If the reported size is zero, as it happens on Linux for files // in /proc, a small buffer is allocated instead. - const initial_cap = (if (size > 0) size else 1024) + @intFromBool(optional_sentinel != null); + const initial_cap = @min((if (size > 0) size else 1024), max_bytes) + @intFromBool(optional_sentinel != null); var array_list = try std.ArrayListAligned(u8, alignment).initCapacity(allocator, initial_cap); defer array_list.deinit();