From f24c7aafc14a662d52a9622522742a99ce8d88fa Mon Sep 17 00:00:00 2001
From: "Alexander \"Arav\" Andreev" <me@arav.su>
Date: Fri, 26 Apr 2024 04:29:26 +0400
Subject: [PATCH] Let's put some more safety to parse_arguments().

---
 main.c | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/main.c b/main.c
index e6c59e2..dc8aebc 100644
--- a/main.c
+++ b/main.c
@@ -39,9 +39,12 @@ int parse_arguments(const int argc, const char *const *const argv,
 
     for (int i = 1, pos = 0; i < argc; ++i)
         if (argv[i][0] == '-') {
-            if (argv[i][1] == 'h' || argv[i][2] == 'h')
+            size_t l = strlen(argv[i]);
+            if (l < 2)
+                return 99;
+            if (argv[i][1] == 'h' || (argv[i][1] == '-' && l > 2 && argv[i][2] == 'h'))
                 return 1;
-            else if (argv[i][1] == 'v' || argv[i][2] == 'v')
+            else if (argv[i][1] == 'v' || (argv[i][1] == '-' && l > 2 && argv[i][2] == 'v'))
                 return 2;
             else
                 return 99;