Fix UDP scan (#115)

FCG-LLC · Sep 26, 2017 · 7d95d8b · 7d95d8b
1 parent 6911f12
commit 7d95d8b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scans/scanner.py b/scans/scanner.py
@@ -94,7 +94,7 @@ async def run_scan(self, nodes, scanners, scan, protocol=PROTOCOL, scan_only=Fal
             for scanner in scanners[ip_protocol]:
                 log.info("Scanning %i %s %s nodes for open ports.", len(dict_nodes[ip_protocol]), protocol.name,
                          ip_protocol)
-                if protocol.name == TransportProtocol.UDP:
+                if protocol == TransportProtocol.UDP:
                     ports.extend(await scanner.scan_ports(dict_nodes[ip_protocol]))
                 else:
                     for node in dict_nodes[ip_protocol]: