Contents

module Hooray
  #
  # Main runner
  #
  class Seek
    attr_accessor :network, :opts, :nodes

    NET_MASK = 24
    TIMEOUT  = 1

    def initialize(network = nil, *params)
      @network = network || Seek.local_mask
      unless params.empty?
        ports, words = params.flatten.partition { |s| s =~ /\d+/ }
        @ports = ports.map(&:to_i).first # TODO
        @protocol = words.select { |w| w =~ /udp|tcp/ }.join
      end

      @nodes = ping.sort.map do |n|
        Node.new(ip: n, mac: arp_table[n.to_s]) # , name: find_name(n))
      end
    end
    alias_method :devices, :nodes

    def ping_class
      return Net::Ping::External unless @protocol
      @protocol =~ /udp/ ? Net::Ping::UDP : Net::Ping::TCP
    end

    #
    # fast -> -sn -PA
    #
    def ping
      scan = []
      bots = []
      # pa "Starting #{Settings.list} on '#{network}' #{@ports} #{@protocol}"
      @network.to_range.each do |ip|
        # next if ip == my_lan_ip
        bots << Thread.new do
          if ping_class.new(ip.to_s, @ports, TIMEOUT).ping?
            scan << ip
            print '.'
          end
        end
      end
      bots.each(&:join)
      puts
      scan
    end

    #
    # ARP Table reader
    #
    def arp_table
      return @arp_table if @arp_table
      @arp_table ||= {}
      `arp -na`.split(/\n/).each do |l|
        _q, ip, _at, mac, *iface = l.split(/\s+/)
        next unless mac =~ /:\w{2}:/
        ip = ip[1..-2] # (ip) -> ip
        @arp_table.merge!(ip => mac) if iface
      end
      @arp_table
    end

    class << self
      def my_ips
        Socket.ip_address_list.select do |ip|
          ip.ipv4_private? && !ip.ipv4_loopback?
        end
      end

      def my_lan_ip
        IPAddr.new(my_ips.first.ip_address)
      end

      def local_mask
        my_lan_ip.mask(NET_MASK)
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
hooray-0.0.7	lib/hooray/seek.rb