1 files changed, 119 insertions, 0 deletions
diff --git a/emulator/qtools/parse_options.cpp b/emulator/qtools/parse_options.cpp
new file mode 100644
index 0000000..395e9a1
--- /dev/null
+++ b/emulator/qtools/parse_options.cpp
@@ -0,0 +1,119 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <ctype.h>
+#include "bitvector.h"
+#include "parse_options.h"
+#include "hash_table.h"
+
+const char *root = "";
+bool lump_kernel;
+bool lump_libraries;
+Bitvector pid_include_vector(32768);
+Bitvector pid_exclude_vector(32768);
+bool include_some_pids;
+bool exclude_some_pids;
+
+HashTable<int> excluded_procedures(2000);
+HashTable<int> included_procedures(2000);
+bool exclude_some_procedures;
+bool include_some_procedures;
+
+bool exclude_kernel_syms;
+bool exclude_library_syms;
+bool include_kernel_syms;
+bool include_library_syms;
+bool demangle = true;
+
+static const char *OptionsUsageStr =
+    "  -e :kernel exclude all kernel symbols\n"
+    "  -e :libs   exclude all library symbols\n"
+    "  -e <func>  exclude function <func>\n"
+    "  -e <pid>   exclude process <pid>\n"
+    "  -i :kernel include all kernel symbols\n"
+    "  -i :libs   include all library symbols\n"
+    "  -i <func>  include function <func>\n"
+    "  -i <pid>   include process <pid>\n"
+    "  -l :kernel lump all the kernel symbols together\n"
+    "  -l :libs   lump all the library symbols together\n"
+    "  -m         do not demangle C++ symbols (m for 'mangle')\n"
+    "  -r <root>  use <root> as the path for finding ELF executables\n"
+    ;
+
+void OptionsUsage()
+{
+    fprintf(stderr, "%s", OptionsUsageStr);
+}
+
+void ParseOptions(int argc, char **argv)
+{
+    bool err = false;
+    while (!err) {
+        int opt = getopt(argc, argv, "+e:i:l:mr:");
+        if (opt == -1)
+            break;
+        switch (opt) {
+        case 'e':
+            if (*optarg == ':') {
+                if (strcmp(optarg, ":kernel") == 0)
+                    exclude_kernel_syms = true;
+                else if (strcmp(optarg, ":libs") == 0)
+                    exclude_library_syms = true;
+                else
+                    err = true;
+                excluded_procedures.Update(optarg, 1);
+                exclude_some_procedures = true;
+            } else if (isdigit(*optarg)) {
+                int bitnum = atoi(optarg);
+                pid_exclude_vector.SetBit(bitnum);
+                exclude_some_pids = true;
+            } else {
+                excluded_procedures.Update(optarg, 1);
+                exclude_some_procedures = true;
+            }
+            break;
+        case 'i':
+            if (*optarg == ':') {
+                if (strcmp(optarg, ":kernel") == 0)
+                    include_kernel_syms = true;
+                else if (strcmp(optarg, ":libs") == 0)
+                    include_library_syms = true;
+                else
+                    err = true;
+                included_procedures.Update(optarg, 1);
+                include_some_procedures = true;
+            } else if (isdigit(*optarg)) {
+                int bitnum = atoi(optarg);
+                pid_include_vector.SetBit(bitnum);
+                include_some_pids = true;
+            } else {
+                included_procedures.Update(optarg, 1);
+                include_some_procedures = true;
+            }
+            break;
+        case 'l':
+            if (strcmp(optarg, ":kernel") == 0)
+                lump_kernel = true;
+            else if (strcmp(optarg, ":libs") == 0)
+                lump_libraries = true;
+            else
+                err = true;
+            break;
+        case 'm':
+            demangle = false;
+            break;
+        case 'r':
+            root = optarg;
+            break;
+        default:
+            err = true;
+            break;
+        }
+    }
+
+    if (err) {
+        Usage(argv[0]);
+        exit(1);
+    }
+}