ipscan 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 61800893649005026086244f7edc773894947869
-  data.tar.gz: fdf8e90e53edfe30ad48198a79fbf76afb843e93
+  metadata.gz: 28f0852ab8dc39016f37d4e781441b5c74e27d98
+  data.tar.gz: 25e3f199daaa1504e888e49ba78c5479c02c2382
 SHA512:
-  metadata.gz: d18d2d37e7ff94760c4c98d3422c2459201176aea8044f2cc542a4c7bff1259038f5f29a84ad2e8b35bea4ade2b744a5e3b7d634d60d30268eb101a4f0fb650f
-  data.tar.gz: 14acc5b6cc5ac6df30015384ae0dfd23c1d0ef00c912c3224889dd3440fcc1a4fb6f363aa94e278e7af6a2cffa6b9b80015d532e9cf666f7455b290ba701d847
+  metadata.gz: 985db2876f5d07380465e5b2403e0d8dcc6a9613dbe54507748b9e3bbada9ddf0e878fe69b6927436e07bb101165b0271f27a061e0df2dd1c9cc3549c6af2ca1
+  data.tar.gz: b8bdef4f08be552a0a601da6867a1eafb8a9152243448c6c8ca3cf2d75cc1a2dde6bbf3ef4df9fb042f5d1d2d7aec6eb29ca3fdd21ed565eae139a5b73255625

data/lib/ipscan.rb

@@ -10,40 +10,97 @@ require 'ruby-progressbar'
 
 module IPScan
 
-  def self.scan(network_addr = nil)
-    selfip = IPSocket::getaddress(Socket::gethostname)
-    unless network_addr
-      network = selfip.split(".")[0..2].join(".")
+  def self.arp(ip)
+    case RbConfig::CONFIG['host_os']
+    when /linux/i
+      line = `arp #{ip}`.split("\n").last.split(" ")
+      return {name: line[0],  mac: line[2]}
+    when /bsd|osx|mach|darwin/i
+      line = `arp #{ip}`.split("\n").last.split(" ")
+      return {name: line[0],  mac: line[3]}
+    else
+      raise "host_os=#{RbConfig::CONFIG['host_os']} are not supported"
     end
-    puts "check #{network}.0 ~ #{network}.254"
+  end
 
+  def self.get_range_addresses(network_addr, start_host, last_host)
     addrs = []
-    0.upto(254) do |i|
-      addrs << "#{network}.#{i}"
+    start_host.upto(last_host) do |i|
+      addrs << "#{network_addr}.#{i}"
     end
+    addrs
+  end
+
+  def self.get_netbios_info(range)
+    nbtinfo = {}
+    body_started = false
+    result = `nbtscan #{range}`
+    result.each_line do |line|
+      if line.index("-----------")
+        body_started = true
+        next
+      end
+      next unless body_started
+      h            = line.split(" ")
+      ip           = h[0]
+      netbios_name = h[1]
+      nbtinfo[ip] = {netbios_name: netbios_name}
+    end
+    nbtinfo
+  end
+
+  def self.scan(renge)
+    unless renge.index("-")
+      renge = renge + "-254"
+    end
+    scan_result  = renge.scan(/(.*)-(.*)/)[0]
+    network_addr = scan_result[0].split(".")[0..2].join(".")
+    start_host   = scan_result[0].split(".")[3].to_i
+    last_host    = scan_result[1].to_i
+    host_total   = (start_host..last_host).size
+    puts "check #{network_addr}.#{start_host} ~ #{network_addr}.#{last_host}"
+
+    netbios_info = {}
+    if `which nbtscan` != ""
+      netbios_info = get_netbios_info(range)
+    end
+
+    addrs = get_range_addresses(network_addr, start_host, last_host)
 
     results = []
-    progressbar = ProgressBar.create(:total => 255)
+    progressbar = ProgressBar.create(:total => host_total)
     Parallel.each(addrs, in_threads: 100) do |addr|
       data = {:ip => addr}
-      if addr == selfip
-        data[:name] = "(localhost)"
-        data[:mac]  = "(localhost)"
-        results << data
-      else
-        pinger = Net::Ping::External.new(addr)
-        if pinger.ping?
-          arp_resp = `arp #{addr}`.split(" ")
-          data[:name] = arp_resp[0]
-          data[:mac]  = arp_resp[3]
-          results << data
+      data.merge!(arp(addr))
+      pinger = Net::Ping::External.new(addr)
+      if pinger.ping?
+        arp_line = `arp #{addr}`
+        arp_hash = arp_line.split(" ")
+        if arp_hash[0] == "?"
+          data[:name] = "<unknown>"
+        else
+          data[:name] = arp_hash[0]
+        end
+
+        if arp_line.index("no entry")
+          data[:mac]  = "<no entry>     "
+        else
+          data[:mac]  = arp_hash[3]
+        end
+
+        if netbios_info[addr]
+          data[:netbios_name] = netbios_info[addr][:netbios_name]
+        else
+          data[:netbios_name] = "<unknown>"
         end
+        results << data
       end
       progressbar.increment
     end
 
+    puts "IP Address \tMac Address     \tName         \tNetBIOS Name"
     results.each do |res|
-      puts "#{res[:ip]}\t#{res[:mac]}\t#{res[:name]}"
+      puts "#{res[:ip]} \t#{res[:mac]} \t#{res[:name]} \t#{res[:netbios_name]}"
     end
   end
 end

data/lib/ipscan/cli.rb

@@ -4,9 +4,9 @@ require 'thor'
 
 module IPScan
   class CLI < Thor
-    desc "scan [NETWORK_ADDR]", "scan LAN devices. ex: scan 192.168.1"
-    def scan(network_addr=nil)
-      IPScan.scan(network_addr)
+    desc "range [SCAN_RANGE]", "scan 192.168.1.0-254"
+    def range(range)
+      IPScan.scan(range)
     end
 
     default_task :scan

data/lib/ipscan/version.rb

@@ -1,3 +1,3 @@
 module IPScan
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ipscan
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - k-yamada
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-05-18 00:00:00.000000000 Z
+date: 2014-05-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler