mirror/zig
1
0
Fork 0
mirror of https://github.com/ziglang/zig.git synced 2026-01-21 06:45:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

std.io.Writer.Allocating: extra ensure byte

Browse Source 
So that when returning from drain there is always capacity for at least
one more byte.
This commit is contained in:
Andrew Kelley 2025-06-30 15:52:13 -07:00
parent d03d273117
commit 8f4229158b
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
lib/std/io/Writer.zig
View File

@ -2158,8 +2158,11 @@ pub const Allocating = struct {
var list = a.toArrayList();
defer setArrayList(a, list);
const start_len = list.items.len;
// Even if we append no data, this function needs to ensure there is more
// capacity in the buffer to avoid infinite loop, hence the +1 in this loop.
assert(data.len != 0);
for (data) |bytes| {
list.ensureUnusedCapacity(gpa, bytes.len + splat_len) catch return error.WriteFailed;
list.ensureUnusedCapacity(gpa, bytes.len + splat_len + 1) catch return error.WriteFailed;
list.appendSliceAssumeCapacity(bytes);
}
if (splat == 0) {