Initial commit

p1-check/main.c

@@ -0,0 +1,66 @@
+/*
+ * CS 261: Main driver
+ *
+ * Name: Nicholas Tamassia
+ */
+
+#include "p1-check.h"
+
+/*
+ * helper function for printing help text
+ */
+void usage (char **argv)
+{
+    printf("Usage: %s <option(s)> mini-elf-file\n", argv[0]);
+    printf(" Options are:\n");
+    printf("  -h      Display usage\n");
+    printf("  -H      Show the Mini-ELF header\n");
+}
+
+int main (int argc, char **argv)
+{
+    int c;
+    bool display_header = false;
+
+    while ((c = getopt(argc, argv, "hH")) != -1) {
+        switch (c) {
+            case 'h':
+                usage(argv);
+                return EXIT_SUCCESS;
+            case 'H':
+                display_header = true;
+                break;
+            default:
+                usage(argv);
+                return EXIT_FAILURE;
+        }
+    }
+
+    /* Check if file path was the last argument */
+    if (optind != argc - 1) {
+        usage(argv);
+        return EXIT_FAILURE;
+    }
+
+    elf_hdr_t header = { 0 };
+    FILE *header_file = fopen(argv[optind], "r");
+
+    /* Invalid file path or failed to read correct data from file */
+    if (!read_header(header_file, &header)) {
+        printf("Failed to read file\n");
+
+        if (header_file != NULL) {
+            fclose(header_file);
+        }
+
+        return EXIT_FAILURE;
+    }
+
+    if (display_header) {
+        dump_header(&header);
+    }
+
+    fclose(header_file);
+
+    return EXIT_SUCCESS;
+}