Allow IPRange#include? to accept String arguments

rapid7 · Apr 2, 2015 · 45118a2 · 45118a2
1 parent 11563fb
commit 45118a2
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/lib/nexpose/site.rb b/lib/nexpose/site.rb
@@ -723,11 +723,14 @@ def eql?(other)
       @from == other.from && @to == other.to
     end
 
-    def include?(single_ip)
-      return false unless single_ip.respond_to? :from
+    def include?(subset)
       from = IPAddr.new(@from)
       to = @to.nil? ? from : IPAddr.new(@to)
-      other = IPAddr.new(single_ip)
+      if subset.respond_to?(:from)
+        other = IPAddr.new(subset.from)
+      else
+        other = IPAddr.new(subset)
+      end
 
       if other < from
         false