Support benchmarking

3 files changed, 15 insertions, 6 deletions

diff --git a/1/part1.c b/1/part1.c
index 62ef852..07f1251 100644
--- a/1/part1.c
+++ b/1/part1.c
@@ -18,9 +18,12 @@ int repair(const int * arr)
     return 0;
 }
 
-int main()
+int main(int argc, char *argv[])
 {
-    FILE *file = fopen("input", "r");
+    FILE *file = fopen(argv[1], "r");
+    if (!file) {
+        return 1;
+    }
 
     char buffer[8] = { 0 };
     int input[INPUT_LEN] = { 0 };
diff --git a/1/part2.c b/1/part2.c
index 6efb387..adf487c 100644
--- a/1/part2.c
+++ b/1/part2.c
@@ -20,9 +20,12 @@ int repair(const int *arr)
     return 0;
 }
 
-int main()
+int main(int argc, char *argv[])
 {
-    FILE *file = fopen("input", "r");
+    FILE *file = fopen(argv[1], "r");
+    if (!file) {
+        return 1;
+    }
 
     char buffer[8] = { 0 };
     int input[INPUT_LEN] = { 0 };
diff --git a/1/part2_fast.c b/1/part2_fast.c
index 1d9afb1..b38894b 100644
--- a/1/part2_fast.c
+++ b/1/part2_fast.c
@@ -121,9 +121,12 @@ int repair_avx(const int *arr)
     return 0;
 }
 
-int main()
+int main(int argc, char *argv[])
 {
-    FILE *file = fopen("input", "r");
+    FILE *file = fopen(argv[1], "r");
+    if (!file) {
+        return 1;
+    }
 
     char buffer[8] = { 0 };
     int input[INPUT_LEN] = { 0 };