Added feature to distinguish between blocked by firewall and open ports

1 files changed, 15 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 bde3f3b..dcb73dd 100644
--- a/src/main/java/com/it_jaros/network_scanner/App.java
+++ b/src/main/java/com/it_jaros/network_scanner/App.java
@@ -1,5 +1,6 @@
 package com.it_jaros.network_scanner;
 
+import java.util.BitSet;
 import java.util.List;
 import java.util.stream.Collectors;
 
@@ -24,13 +25,21 @@ public class App {
     }
 
     private static void printResult(ScanResult result) {
-        String ports = result.openPorts().stream().mapToObj(String::valueOf).collect(Collectors.joining(","));
-        String output = "[" + result.target() + "]:";
-        if (ports.isEmpty()) {
+        if (result.openPorts().isEmpty() && result.blockedPorts().isEmpty()) {
             return;
-        } else {
-            output = output + ports;
         }
-        System.out.print(output + "\n");
+
+        System.out.println(result.target());
+        if (!result.openPorts().isEmpty()) {
+            System.out.println("\topen:\t" + map(result.openPorts()));
+        }
+        if (!result.blockedPorts().isEmpty()) {
+            System.out.println("\tblocked:\t" + map(result.blockedPorts()));
+        }
+        System.out.println();
+    }
+
+    private static String map(BitSet ports) {
+        return ports.stream().mapToObj(String::valueOf).collect(Collectors.joining(","));
     }
 }