Added cli option to be able to select whether to display blocked ports

or not

1 files changed, 6 insertions, 6 deletions

diff --git a/src/main/java/com/it_jaros/network_scanner/App.java b/src/main/java/com/it_jaros/network_scanner/App.java
index dcb73dd..1ae45ba 100644
--- a/src/main/java/com/it_jaros/network_scanner/App.java
+++ b/src/main/java/com/it_jaros/network_scanner/App.java
@@ -11,7 +11,7 @@ public class App {
             ScanOptions options = cliParser.parseArgs(args);
             Scanner scanner = new Scanner(options.openSocketLimit(), options.timeoutInMillis(),
                     options.delayInMillis(), options.maxWorkersPerHost(), options.maxTargetsLimit());
-            printResults(scanner.scan(options));
+            printResults(scanner.scan(options), options.showBlockedPorts());
         } catch (IllegalArgumentException e) {
             System.out.printf("Error while parsing arguments: %s", e.getMessage());
             cliParser.usageAndExit();
@@ -20,12 +20,12 @@ public class App {
         }
     }
 
-    public static void printResults(List<ScanResult> results) {
-        results.forEach(App::printResult);
+    public static void printResults(List<ScanResult> results, boolean showBlockedPorts) {
+        results.forEach((ScanResult r) -> printResult(r, showBlockedPorts));
     }
 
-    private static void printResult(ScanResult result) {
-        if (result.openPorts().isEmpty() && result.blockedPorts().isEmpty()) {
+    private static void printResult(ScanResult result, boolean showBlockedPorts) {
+        if (result.openPorts().isEmpty() && (!showBlockedPorts || result.blockedPorts().isEmpty())) {
             return;
         }
 
@@ -33,7 +33,7 @@ public class App {
         if (!result.openPorts().isEmpty()) {
             System.out.println("\topen:\t" + map(result.openPorts()));
         }
-        if (!result.blockedPorts().isEmpty()) {
+        if (showBlockedPorts && !result.blockedPorts().isEmpty()) {
             System.out.println("\tblocked:\t" + map(result.blockedPorts()));
         }
         System.out.println();