From fcc00c9a276eca1213dcad7ea4d49bd329a29437 Mon Sep 17 00:00:00 2001 From: Brian Gianforcaro Date: Sat, 27 Nov 2021 13:02:33 -0800 Subject: [PATCH] LibCore+cat: Switch Core::System::read/write to take a Span of bytes In the spirit of the Core::System name space having "modern" facades for classically C functions / Kernel interfaces, it seems appropriate that we should take Span's of data instead of raw pointer + length arguments. --- Userland/Libraries/LibCore/System.cpp | 8 ++++---- Userland/Libraries/LibCore/System.h | 4 ++-- Userland/Utilities/cat.cpp | 13 ++++++++----- 3 files changed, 14 insertions(+), 11 deletions(-) diff --git a/Userland/Libraries/LibCore/System.cpp b/Userland/Libraries/LibCore/System.cpp index b6051d08511..4289b22ed59 100644 --- a/Userland/Libraries/LibCore/System.cpp +++ b/Userland/Libraries/LibCore/System.cpp @@ -157,17 +157,17 @@ ErrorOr stat(StringView path) #endif } -ErrorOr read(int fd, void* buffer, size_t buffer_size) +ErrorOr read(int fd, Bytes buffer) { - ssize_t rc = ::read(fd, buffer, buffer_size); + ssize_t rc = ::read(fd, buffer.data(), buffer.size()); if (rc < 0) return Error::from_syscall("read"sv, -errno); return rc; } -ErrorOr write(int fd, void const* data, size_t data_size) +ErrorOr write(int fd, ReadonlyBytes buffer) { - ssize_t rc = ::write(fd, data, data_size); + ssize_t rc = ::write(fd, buffer.data(), buffer.size()); if (rc < 0) return Error::from_syscall("write"sv, -errno); return rc; diff --git a/Userland/Libraries/LibCore/System.h b/Userland/Libraries/LibCore/System.h index d5e39d67ed4..c90e2f0c9f4 100644 --- a/Userland/Libraries/LibCore/System.h +++ b/Userland/Libraries/LibCore/System.h @@ -27,8 +27,8 @@ ErrorOr open(StringView path, int options, ...); ErrorOr close(int fd); ErrorOr ftruncate(int fd, off_t length); ErrorOr stat(StringView path); -ErrorOr read(int fd, void* buffer, size_t buffer_size); -ErrorOr write(int fd, void const* data, size_t data_size); +ErrorOr read(int fd, Bytes buffer); +ErrorOr write(int fd, ReadonlyBytes buffer); ErrorOr kill(pid_t, int signal); ErrorOr dup2(int source_fd, int destination_fd); ErrorOr ptsname(int fd); diff --git a/Userland/Utilities/cat.cpp b/Userland/Utilities/cat.cpp index 89f984b7991..4c0d5be9e86 100644 --- a/Userland/Utilities/cat.cpp +++ b/Userland/Utilities/cat.cpp @@ -45,15 +45,18 @@ ErrorOr serenity_main(Main::Arguments arguments) TRY(Core::System::pledge("stdio", nullptr)); + Array buffer; for (auto& fd : fds) { for (;;) { - char buffer[32768]; - auto nread = TRY(Core::System::read(fd, buffer, sizeof(buffer))); + auto buffer_span = buffer.span(); + auto nread = TRY(Core::System::read(fd, buffer_span)); if (nread == 0) break; - ssize_t already_written = 0; - while (already_written < nread) - already_written += TRY(Core::System::write(STDOUT_FILENO, buffer + already_written, nread - already_written)); + buffer_span = buffer_span.trim(nread); + while (!buffer_span.is_empty()) { + auto already_written = TRY(Core::System::write(STDOUT_FILENO, buffer_span)); + buffer_span = buffer_span.slice(already_written); + } } TRY(Core::System::close(fd)); }