wc: Use pledge()

1 files changed, 10 insertions, 0 deletions

diff --git a/Userland/wc.cpp b/Userland/wc.cpp
index 60a4c66e13..e2d56d802d 100644
--- a/Userland/wc.cpp
+++ b/Userland/wc.cpp
@@ -122,6 +122,11 @@ Count get_total_count(Vector<Count>& counts)
 
 int main(int argc, char** argv)
 {
+    if (pledge("stdio rpath", nullptr) < 0) {
+        perror("pledge");
+        return 1;
+    }
+
     Vector<const char*> files;
 
     Core::ArgsParser args_parser;
@@ -140,6 +145,11 @@ int main(int argc, char** argv)
         counts.append(count);
     }
 
+    if (pledge("stdio", nullptr) < 0) {
+        perror("pledge");
+        return 1;
+    }
+
     if (files.size() > 1) {
         Count total_count = get_total_count(counts);
         counts.append(total_count);