Browse Source

Utilities: Make syscall(1) use Core::ArgsParser

Note that this should not change the behavior at all.
Ben Wiederhake 4 years ago
parent
commit
244c81bcf2
1 changed files with 18 additions and 47 deletions
  1. 18 47
      Userland/Utilities/syscall.cpp

+ 18 - 47
Userland/Utilities/syscall.cpp

@@ -24,71 +24,42 @@
  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
-#include <errno.h>
-#include <getopt.h>
-#include <stdint.h>
+#include <LibCore/ArgsParser.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <syscall.h>
-#include <unistd.h>
-
-#if !defined __ENUMERATE_SYSCALL
-#    define __ENUMERATE_SYSCALL(x) SC_##x,
-#endif
 
 #define SC_NARG 4
 
-Syscall::Function syscall_table[] = {
-    ENUMERATE_SYSCALLS(__ENUMERATE_SYSCALL)
-};
-
 FlatPtr arg[SC_NARG];
 char buf[BUFSIZ];
 
-FlatPtr parse(char* s);
+static FlatPtr parse(const char* s);
 
 int main(int argc, char** argv)
 {
-    int oflag;
-    int opt;
-    while ((opt = getopt(argc, argv, "olh")) != -1) {
-        switch (opt) {
-        case 'o':
-            oflag = 1;
-            break;
-        case 'l':
-            for (auto sc : syscall_table) {
-                fprintf(stdout, "%s ", Syscall::to_string(sc));
-            }
-            return EXIT_SUCCESS;
-        case 'h':
-            fprintf(stderr, "usage: \tsyscall [-o] [-l] [-h] <syscall-name> <args...> [buf==BUFSIZ buffer]\n");
-            fprintf(stderr, "\tsyscall write 1 hello 5\n");
-            fprintf(stderr, "\tsyscall -o read 0 buf 5\n");
-            fprintf(stderr, "\tsyscall sleep 3\n");
-            break;
-        default:
-            exit(EXIT_FAILURE);
-        }
-    }
-
-    if (optind >= argc) {
-        fprintf(stderr, "No entry specified\n");
-        return -1;
-    }
-
-    for (int i = 0; i < argc - optind; i++) {
-        arg[i] = parse(argv[i + optind]);
+    bool output_buffer = false;
+    bool list_syscalls = false;
+    Vector<const char*> arguments;
+
+    Core::ArgsParser args_parser;
+    args_parser.add_option(output_buffer, "Output the contents of the buffer (beware of stray zero bytes!)", "output-buffer", 'o');
+    args_parser.add_option(list_syscalls, "List all existing syscalls", "list-syscalls", 'l');
+    args_parser.add_positional_argument(arguments, "Syscall arguments (can be strings, 'buf' for the output buffer, or numbers like 1234 or 0xffffffff)", "syscall-arguments");
+    args_parser.parse(argc, argv);
+
+    for (size_t i = 0; i < arguments.size(); i++) {
+        arg[i] = parse(arguments[i]);
     }
 
-    for (auto sc : syscall_table) {
-        if (strcmp(Syscall::to_string(sc), (char*)arg[0]) == 0) {
+    for (int sc = 0; sc < Syscall::Function::__Count; ++sc) {
+        if (strcmp(Syscall::to_string((Syscall::Function)sc), (char*)arg[0]) == 0) {
             int rc = syscall(sc, arg[1], arg[2], arg[3]);
             if (rc == -1) {
                 perror("syscall");
             } else {
-                if (oflag)
+                if (output_buffer)
                     fwrite(buf, 1, sizeof(buf), stdout);
             }
 
@@ -101,7 +72,7 @@ int main(int argc, char** argv)
     return -1;
 }
 
-FlatPtr parse(char* s)
+FlatPtr parse(const char* s)
 {
     char* t;
     FlatPtr l;