From e602cff985169e792ca2f5fe047c4cc8ab5ad040 Mon Sep 17 00:00:00 2001
From: Bond_009 <bond.009@outlook.com>
Date: Fri, 4 Dec 2020 18:37:07 +0100
Subject: Support benchmarking

---
 2/part2.c | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

(limited to '2/part2.c')

diff --git a/2/part2.c b/2/part2.c
index b167fcf..c6219a1 100644
--- a/2/part2.c
+++ b/2/part2.c
@@ -13,9 +13,13 @@ bool is_valid_password(char *pass, char *policy)
     return (pass[pos1] == c) != (pass[pos2] == c);
 }
 
-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[128] = { 0 };
     int valid = 0;
     while (fgets(buffer, 128, file)) {
-- 
cgit v1.2.3