network_scanner 0.0.2 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 523886c0cc9e935ee6629722850bf2aac37a2a94
-  data.tar.gz: a249a6aba70f6c99e6ac6ebb67e6a905f84c9afb
+  metadata.gz: b9bdf3586d28cb2c6ba41ce2337d054d1ebcb041
+  data.tar.gz: 23164e74226b74b5f02f50dbe5c212778d3db68d
 SHA512:
-  metadata.gz: 26091157cbe2b73b7faf985f049f8fe96d3dbe712a4ee487d59eb9a933d60459b6afa12b252c5c30ae8d56c868e6d73cd6d5a815be86f3fd093aeb16cb524af1
-  data.tar.gz: 618b76d95a7e2558ab2c8a80cc7fccc6457410b5670753320d8c495c703d5e306a7531233f135089cc40bbea8b4bf50a637885412e9ad1283aeb313f655f8b78
+  metadata.gz: f23d8cb010bd8f316a86aaa50d2f16b07725344a7c55e387c91e3b0678e29e89a977a194cf3d848a42262d3284c74c625587d879f9779f676f068ef906a0c8ee
+  data.tar.gz: d54b2b76b0019752e3a9ef42d4a767d15cd8b6a3b423b3e81f49ce868ab35caf52ac7f0ed45084145c712e12ee177842c28ed65c9fcd28c8224bd36a50d064e8

data/bin/network_scanner

@@ -5,42 +5,63 @@ require 'optparse'
 
 options = {}
 OptionParser.new do |opts|
-  opts.on('--cache FILE', String, 'Caches the IPs up on the network in FILE for quicker scanning') do |file|
-    options[:cache] = file
+  opts.on('-f', '--format FORMAT', String, 'Format to read/write (text/json). Defaults to text') do |format|
+    options[:format] = format
   end
 
-  opts.on('--interface INTERFACE', String, 'Specifys interface to scan') do |interface|
+  opts.on('-s', '--poolsize POOLSIZE', Integer, 'Specify thread pool size (default 100)') do |pool|
+    options[:pool] = pool
+  end
+
+  opts.on('-i', '--interface INTERFACE', String, 'Specifys interface to scan') do |interface|
     options[:interface] = interface
   end
 
-  opts.on('--range RANGE', String, 'Specify a range to scan(192.168.1.100-192.168.1.200)') do |range|
+  opts.on('-r', '--range RANGE', String, 'Specify a range to scan(192.168.1.100-192.168.1.200)') do |range|
     options[:range] = range
   end
 
-  opts.on('--portscan PORT', Integer, 'Port to scan for') do |port|
-    options[:port] = port
+  opts.on('-c', '--cache FILE', String, 'Cache to read ips from') do |file|
+    options[:cache] = file
   end
 
-  opts.on('--cacheread FILE', String, 'Cache to read ips from') do |file|
-    options[:cacheread] = file
+  opts.on('-P', '--ping', 'Checks for all machines that respond to ping') do
+    options[:ping] = true
   end
 
-  opts.on('--poolsize POOLSIZE', Integer, 'Specify thread pool size (default 100)') do |pool|
-    options[:pool] = pool
+  opts.on('-p', '--portscan PORT', Integer, 'Port to scan for') do |port|
+    options[:port] = port
   end
 
-  opts.on_tail('--help', 'Show help message') do
+  opts.on('-o', '--output OUTPUTFILE', String, 'File to output to') do |file|
+    options[:output] = file
+  end
+
+  opts.on_tail('-h', '--help', 'Show help message') do
     puts opts
     exit
   end
 end.parse!
 
-
 network_scanner = NetworkScanner.new
 if options[:pool]
+  unless options[:pool] > 0
+    raise ArgumentError.new('Specify a pool size greater than 0')
+  end
   network_scanner.pool_size = options[:pool]
 end
 
+if options[:format]
+  unless ['text','json'].include?(options[:format])
+    raise ArgumentError.new('Specify either text or json as a format')
+  end
+  network_scanner.format = options[:format]
+end
+
+if options[:output]
+  network_scanner.output_name = options[:output]
+end
+
 if options[:interface]
   network_scanner.get_interface_ips(options[:interface])
 end
@@ -50,11 +71,11 @@ if options[:range]
 end
 
 if options[:cache]
-  network_scanner.cache(options[:cache])
+  network_scanner.get_cache_ips(options[:cache])
 end
 
-if options[:cacheread]
-  network_scanner.cacheread(options[:cacheread])
+if options[:ping]
+  network_scanner.check_pings
 end
 
 if options[:port]

data/lib/network_scanner.rb

@@ -24,10 +24,19 @@ module NetworkScanner
   end
 
   class Scanner
-    attr_accessor :pool_size
+    attr_accessor :pool_size, :format, :output_name
 
     def initialize(opts = {})
       @pool_size = 100
+      @format = 'text'
+    end
+    
+    def text?
+      @format == 'text'
+    end
+
+    def json?
+      @format == 'json'
     end
 
     def get_interface_ips(interface)
@@ -59,7 +68,6 @@ module NetworkScanner
       end
 
       puts "Checking ips in (#{first_range}).(#{second_range}).(#{third_range}).(#{fourth_range})"
-      @ips = @ips_to_check
       return @ips_to_check
     end
 
@@ -83,26 +91,57 @@ module NetworkScanner
       end
 
       puts "Checking ips in (#{first_range}).(#{second_range}).(#{third_range}).(#{fourth_range})"
-      @ips = @ips_to_check
       return @ips_to_check
     end
 
-    def get_ips
-      raise Exception.new("Must specify an interface to get ips to check") unless @ips_to_check
+    def get_cache_ips(file)
+      @ips_to_check = []
+
+      if text?
+        File.open(file).each_line do |ip|
+          @ips_to_check << ip
+        end
+      elsif json?
+        JSON.parse(File.read(file)).each do |ip|
+          @ips_to_check << ip
+        end
+      end
+      return @ips_to_check
+    end
+
+    def check_pings
+      raise Exception.new("Must specify ips to check(interface/range/cache)") unless @ips_to_check
 
       @ips = []
 
       pool = Thread.pool(@pool_size)
 
+      if self.output_name
+        out = File.open(self.output_name, 'w')
+      else
+        out = STDOUT
+      end
+
       @ips_to_check.each do |ip|
         pool.process do
           # For some reason &> isn't working, so pipe stdout and then stderr
-          @ips << ip if system("ping -c 1 #{ip} >> /dev/null 2> /dev/null")
+          if system("ping -c 1 #{ip} >> /dev/null 2> /dev/null")
+            @ips << ip
+            if text?
+              out.print "#{ip}\n"
+            end
+          end
         end
       end
 
       pool.shutdown
 
+      if json?
+        out.puts(JSON.pretty_generate(@ips))
+      end
+
+      out.close unless out == STDOUT
+
       return @ips
     end
 
@@ -117,30 +156,40 @@ module NetworkScanner
     end
 
     def check_ports(port)
-      raise Exception.new("Must scan ips first (Specify an interface or cacheread)") unless @ips
+      raise Exception.new("Must scan ips first (Specify an interface or cacheread)") unless @ips_to_check
 
-      puts "Checking for ports out of a total of #{@ips.length} ips"
+      puts "Checking for ports out of a total of #{@ips_to_check.length} ips"
+
+      if self.output_name
+        out = File.open(self.output_name, 'w')
+      else
+        out = STDOUT
+      end
 
       pool = Thread.pool(@pool_size)
 
-      @ips.each do |ip|
+      @ips = []
+
+      @ips_to_check.each do |ip|
         pool.process do
-          puts ip if port_open?(ip, port)
+          if port_open?(ip, port)
+            @ips << ip
+            if text?
+              out.print "#{ip}\n"
+            end
+          end
         end
       end
 
       pool.shutdown
-    end
 
-    def cache(file)
-      raise Exception.new("Must scan ips first (Specify an interface or cacheread)") unless @ips
-      get_ips
+      if json?
+        out.puts(JSON.pretty_generate(@ips))
+      end
 
-      File.open(file, 'w'){|f| f.puts(JSON.pretty_generate(@ips))}
-    end
+      out.close unless out == STDOUT
 
-    def cacheread(file)
-      @ips = JSON.parse(File.read(file))
+      return @ips
     end
   end
 end

data/lib/network_scanner/version.rb

@@ -1,3 +1,3 @@
 module NetworkScanner
-  VERSION = "0.0.2"
+  VERSION = "0.1.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: network_scanner
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.1.0
 platform: ruby
 authors:
 - Aaron Neyer