Fixed issue #5

0xf104a · Dec 31, 2017 · c03979d · c03979d
1 parent 332473b
commit c03979d
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/util.c b/util.c
@@ -7,6 +7,7 @@
 //
 
 #include "util.h"
+#include "message.h"
 #include <stdio.h>
 int randport()
 {
@@ -38,8 +39,10 @@ const char* getarg(const char arg[2], const char* argv[], int argc)
 #endif
         if (!strcmp(arg, argv[i])) {
 
-            if (i < argc) {
+            if (i < argc-1) {
                 return argv[i + 1];
+            }else{
+                die("Argument required for key:%s",arg);
             }
         }
     }
@@ -54,8 +57,10 @@ const char* getlarg(const char* arg, const char* argv[], int argc)
 #endif
         if (!strcmp(arg, argv[i])) {
 
-            if (i < argc) {
+            if (i < argc-1) {
                 return argv[i + 1];
+            }else{
+                die("Argument required for key:%s",arg);
             }
         }
     }