Browse Source

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.
SeekingBlues 3 years ago
parent
commit
9705580c53
1 changed files with 35 additions and 16 deletions
  1. 35 16
      Userland/Utilities/rev.cpp

+ 35 - 16
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());
         }
     }