Added timeout to cli options

1 files changed, 7 insertions, 2 deletions

diff --git a/src/main/java/com/it_jaros/networkScanner/CliParser.java b/src/main/java/com/it_jaros/networkScanner/CliParser.java
index eb452ac..4fb91ef 100644
--- a/src/main/java/com/it_jaros/networkScanner/CliParser.java
+++ b/src/main/java/com/it_jaros/networkScanner/CliParser.java
@@ -13,6 +13,7 @@ public class CliParser {
         Set<String> targets = new HashSet<>();
         int socketLimit = 1000;
         int delayInMillis = 0;
+        int timeoutInMillis = 1000;
         for (int i = 0; i < args.length; i++) {
             final String currentArg = args[i];
             if (!currentArg.startsWith("-")) {
@@ -30,6 +31,9 @@ public class CliParser {
                     case "--delay", "-d":
                         delayInMillis = Integer.parseInt(args[++i]);
                         break;
+                    case "--timeout", "-t":
+                        timeoutInMillis = Integer.parseInt(args[++i]);
+                        break;
                     default:
                         throw new IllegalArgumentException("No such param " + currentArg);
                 }
@@ -40,9 +44,9 @@ public class CliParser {
         }
 
         System.out.printf(
-                "Scanning targets '%s' with socketLimit(%s), delayInMillis(%s) \n", targets, socketLimit, delayInMillis);
+                "Scanning targets '%s' with socketLimit(%s), delayInMillis(%s), timeoutInMillis(%s)\n", targets, socketLimit, delayInMillis, timeoutInMillis);
         
-        return new ScanOptions(targets.stream().toList(), socketLimit, delayInMillis);
+        return new ScanOptions(targets.stream().toList(), socketLimit, delayInMillis, timeoutInMillis);
     }
 
     public void usageAndExit() {
@@ -53,6 +57,7 @@ public class CliParser {
                 Options:
                     --delay, -d: Waiting period in millis for connection attempts for a host
                     --socketLimit, -sl: Socket Connection limit for a host
+                    --timeout, -t: Timeout per connection attempt
 
                 Examples:
                     java -jar scanner.jar localhost