diff options
author | SeekingBlues <seekingblues@gmail.com> | 2021-10-13 20:05:28 -0400 |
---|---|---|
committer | Brian Gianforcaro <b.gianfo@gmail.com> | 2021-10-17 12:07:09 -0700 |
commit | 9705580c53b481278a7c30f84483bb8fa0ad0c0c (patch) | |
tree | f01239731ab65746150166ec1317506435e96780 /Userland | |
parent | 6d4e58efeab6776c5ec05b5fd94449497276ced7 (diff) | |
download | serenity-9705580c53b481278a7c30f84483bb8fa0ad0c0c.zip |
rev: Unbreak reading from standard input
Since Core::File does not handle streaming input properly (see #5093
and #4198), we use the LibC APIs instead.
Diffstat (limited to 'Userland')
-rw-r--r-- | Userland/Utilities/rev.cpp | 51 |
1 files changed, 35 insertions, 16 deletions
diff --git a/Userland/Utilities/rev.cpp b/Userland/Utilities/rev.cpp index af361f442a..5b42754203 100644 --- a/Userland/Utilities/rev.cpp +++ b/Userland/Utilities/rev.cpp @@ -4,10 +4,9 @@ * SPDX-License-Identifier: BSD-2-Clause */ -#include <AK/String.h> -#include <AK/Vector.h> #include <LibCore/ArgsParser.h> -#include <LibCore/File.h> +#include <errno.h> +#include <string.h> #include <unistd.h> int main(int argc, char** argv) @@ -24,30 +23,50 @@ int main(int argc, char** argv) args_parser.add_positional_argument(paths, "File path", "path", Core::ArgsParser::Required::No); args_parser.parse(argc, argv); - Vector<RefPtr<Core::File>> files; - if (paths.is_empty()) { - files.append(Core::File::standard_input()); - } else { + Vector<FILE*> streams; + auto num_paths = paths.size(); + streams.ensure_capacity(num_paths ? num_paths : 1); + + if (!paths.is_empty()) { for (auto const& path : paths) { - auto file_or_error = Core::File::open(path, Core::OpenMode::ReadOnly); - if (file_or_error.is_error()) { - warnln("Failed to open {}: {}", path, file_or_error.error()); + FILE* stream = fopen(path.characters(), "r"); + if (!stream) { + warnln("Failed to open {}: {}", path, strerror(errno)); continue; } - - files.append(file_or_error.value()); + streams.append(stream); } + } else { + streams.append(stdin); } + char* buffer = nullptr; + ScopeGuard guard = [&] { + free(buffer); + for (auto* stream : streams) { + if (fclose(stream)) + perror("fclose"); + } + }; + if (pledge("stdio", nullptr) < 0) { perror("pledge"); return 1; } - for (auto& file : files) { - while (file->can_read_line()) { - auto line = file->read_line(); - outln("{}", line.reverse()); + for (auto* stream : streams) { + for (;;) { + size_t n = 0; + errno = 0; + ssize_t buflen = getline(&buffer, &n, stream); + if (buflen == -1) { + if (errno != 0) { + perror("getline"); + return 1; + } + break; + } + outln("{}", String { buffer, Chomp }.reverse()); } } |