mirror of
https://github.com/ziglang/zig.git
synced 2025-12-06 06:13:07 +00:00
std: std.fs.File fix sendFile with buffered data
fixes #25196 Co-authored-by: Andrew Kelley <andrew@ziglang.org>
This commit is contained in:
parent
bf58b4e419
commit
e5a55f6144
@ -1732,7 +1732,7 @@ pub const Writer = struct {
|
||||
) std.Io.Writer.FileError!usize {
|
||||
const reader_buffered = file_reader.interface.buffered();
|
||||
if (reader_buffered.len >= @intFromEnum(limit))
|
||||
return sendFileBuffered(io_w, file_reader, reader_buffered);
|
||||
return sendFileBuffered(io_w, file_reader, limit.slice(reader_buffered));
|
||||
const writer_buffered = io_w.buffered();
|
||||
const file_limit = @intFromEnum(limit) - reader_buffered.len;
|
||||
const w: *Writer = @alignCast(@fieldParentPtr("interface", io_w));
|
||||
|
||||
@ -1525,6 +1525,41 @@ test "sendfile" {
|
||||
try testing.expectEqualStrings("header1\nsecond header\nine1\nsecontrailer1\nsecond trailer\n", written_buf[0..amt]);
|
||||
}
|
||||
|
||||
test "sendfile with buffered data" {
|
||||
var tmp = tmpDir(.{});
|
||||
defer tmp.cleanup();
|
||||
|
||||
try tmp.dir.makePath("os_test_tmp");
|
||||
|
||||
var dir = try tmp.dir.openDir("os_test_tmp", .{});
|
||||
defer dir.close();
|
||||
|
||||
var src_file = try dir.createFile("sendfile1.txt", .{ .read = true });
|
||||
defer src_file.close();
|
||||
|
||||
try src_file.writeAll("AAAABBBB");
|
||||
|
||||
var dest_file = try dir.createFile("sendfile2.txt", .{ .read = true });
|
||||
defer dest_file.close();
|
||||
|
||||
var src_buffer: [32]u8 = undefined;
|
||||
var file_reader = src_file.reader(&src_buffer);
|
||||
|
||||
try file_reader.seekTo(0);
|
||||
try file_reader.interface.fill(8);
|
||||
|
||||
var fallback_buffer: [32]u8 = undefined;
|
||||
var file_writer = dest_file.writer(&fallback_buffer);
|
||||
|
||||
try std.testing.expectEqual(4, try file_writer.interface.sendFileAll(&file_reader, .limited(4)));
|
||||
|
||||
var written_buf: [8]u8 = undefined;
|
||||
const amt = try dest_file.preadAll(&written_buf, 0);
|
||||
|
||||
try std.testing.expectEqual(4, amt);
|
||||
try std.testing.expectEqualSlices(u8, "AAAA", written_buf[0..amt]);
|
||||
}
|
||||
|
||||
test "copyRangeAll" {
|
||||
var tmp = tmpDir(.{});
|
||||
defer tmp.cleanup();
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user