Merge pull request #387 from mozilla/add_exception_handling_targets

Fix exception feedback
mozilla · Jun 14, 2017 · ef3e0d0 · ef3e0d0
2 parents 08c6579 + 830454d
commit ef3e0d0
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lib/ssh_scan/target_parser.rb b/lib/ssh_scan/target_parser.rb
@@ -28,7 +28,11 @@ def enumerateIPRange(ip,port)
           end
           return ip_array
         elsif ip.include? "/"
-          cidr = NetAddr::CIDR.create(ip)
+          begin
+            cidr = NetAddr::CIDR.create(ip)
+          rescue
+            raise ArgumentError, "Invalid target: #{ip}"
+          end
           ip_array = cidr.enumerate
           ip_array.delete(cidr.network)
           ip_array.delete(cidr.last)