mirror of
https://github.com/ziglang/zig.git
synced 2025-12-30 18:13:19 +00:00
c81345c8ae
* rework os.sendfile and add macosx support, and a fallback implementation for any OS. * fix sendto compile error * std.os write functions support partial writes. closes #3443. * std.os pread / pwrite functions can now return `error.Unseekable`. * std.fs.File read/write functions now have readAll/writeAll variants which loop to complete operations even when partial reads/writes happen. * Audit std.os read/write functions with respect to Linux returning EINVAL for lengths greater than 0x7fff0000. * std.os read/write shim functions do not unnecessarily loop. Since partial reads/writes are part of the API, the caller will be forced to loop anyway, and so that would just be code bloat. * Improve doc comments * Add a non-trivial test for std.os.sendfile * Fix std.os.pread on 32 bit Linux * Add missing SYS_sendfile bit on aarch64
72 lines
1.9 KiB
Zig
72 lines
1.9 KiB
Zig
const std = @import("std");
|
|
const io = std.io;
|
|
const process = std.process;
|
|
const fs = std.fs;
|
|
const mem = std.mem;
|
|
const warn = std.debug.warn;
|
|
const allocator = std.testing.allocator;
|
|
|
|
pub fn main() !void {
|
|
var args_it = process.args();
|
|
const exe = try unwrapArg(args_it.next(allocator).?);
|
|
var catted_anything = false;
|
|
const stdout_file = io.getStdOut();
|
|
|
|
const cwd = fs.cwd();
|
|
|
|
while (args_it.next(allocator)) |arg_or_err| {
|
|
const arg = try unwrapArg(arg_or_err);
|
|
if (mem.eql(u8, arg, "-")) {
|
|
catted_anything = true;
|
|
try cat_file(stdout_file, io.getStdIn());
|
|
} else if (arg[0] == '-') {
|
|
return usage(exe);
|
|
} else {
|
|
const file = cwd.openFile(arg, .{}) catch |err| {
|
|
warn("Unable to open file: {}\n", .{@errorName(err)});
|
|
return err;
|
|
};
|
|
defer file.close();
|
|
|
|
catted_anything = true;
|
|
try cat_file(stdout_file, file);
|
|
}
|
|
}
|
|
if (!catted_anything) {
|
|
try cat_file(stdout_file, io.getStdIn());
|
|
}
|
|
}
|
|
|
|
fn usage(exe: []const u8) !void {
|
|
warn("Usage: {} [FILE]...\n", .{exe});
|
|
return error.Invalid;
|
|
}
|
|
|
|
// TODO use copy_file_range
|
|
fn cat_file(stdout: fs.File, file: fs.File) !void {
|
|
var buf: [1024 * 4]u8 = undefined;
|
|
|
|
while (true) {
|
|
const bytes_read = file.read(buf[0..]) catch |err| {
|
|
warn("Unable to read from stream: {}\n", .{@errorName(err)});
|
|
return err;
|
|
};
|
|
|
|
if (bytes_read == 0) {
|
|
break;
|
|
}
|
|
|
|
stdout.writeAll(buf[0..bytes_read]) catch |err| {
|
|
warn("Unable to write to stdout: {}\n", .{@errorName(err)});
|
|
return err;
|
|
};
|
|
}
|
|
}
|
|
|
|
fn unwrapArg(arg: anyerror![]u8) ![]u8 {
|
|
return arg catch |err| {
|
|
warn("Unable to parse command line: {}\n", .{err});
|
|
return err;
|
|
};
|
|
}
|