hooray 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: e7ef2ad4641d2f2c855dbd7269067c578e0810c4
+  data.tar.gz: 63abe0efb004ead77086adf20558e42d788b1ca6
+SHA512:
+  metadata.gz: 6feb8a504adfb80d0310f63d9805f2e0572d58f7d498ca642c7b10702ba4c59e4b2485d9163afe9e02e2bd8fd1215055ed3fbdb18bc27a489d1941f6a21f8a6e
+  data.tar.gz: f59f4a37cc59cd917d5ccd0efb843a98d294b48de8eee7d63e91aebe1aba3fecd915cb0429b36d0c59da2aa81e1f91bb69572abcf5f0e0722ccd6801ec0da9c5

data/README.md

@@ -0,0 +1,81 @@
+Hooray
+======
+
+Find devices around you.
+
+```
+hoo list
+```
+```
+Motorola Mobility           | 192.168.1.70  | 34:bb:XX:XX:XX:XX
+Giga-byte Technology Co.    | 192.168.1.71  | 94:de:XX:XX:XX:XX
+Veih's Android              | 192.168.1.72  | 04:46:XX:XX:XX:XX
+Nery's LG                   | 192.168.1.73  | cc:fa:XX:XX:XX:XX
+nofxx's iPhone              | 192.168.1.77  | 64:a3:XX:XX:XX:XX
+LG Electronics              | 192.168.1.79  | cc:fa:XX:XX:XX:XX
+TP-Link                     | 192.168.1.253 | f8:d1:XX:XX:XX:XX
+Siemens Subscriber Networks | 192.168.1.254 | 00:0b:XX:XX:XX:XX
+---
+8 devices @ 2014-12-11 13:32:04 -0200 2.85s
+```
+
+Find running services:
+
+```
+hoo list web
+```
+```
+nofxx desktop               | 192.168.1.77  | 64:a3:XX:XX:XX:XX
+```
+
+Or simply by ports:
+
+
+```
+hoo list 80
+hoo list 6777 udp
+```
+
+Monitor:
+
+```
+hoo watch
+```
+```
+New nodes @ 2014-12-11 13:53:06 -0200
+NAME         | IP           | MAC
+-------------|--------------|------------------
+iPhone nofxx | 192.168.1.76 | 64:a3:XX:XX:XX:XX
+```
+
+Use as a lib:
+```
+require 'hooray'
+Hooray::Seek.lan(port, protocol).devices
+Hooray::Seek.new(network, port, protocol).devices
+```
+
+## Why?
+
+Bind macs or fix IP's is also boring and a per device work.
+As more we use wifi/ethernet devices in our company, nmap gets boring.
+Not to mention slow, there's no easy way to assign names to devices I know,
+which makes OS scan/port scan a needed option most times (thus making it slower).
+Also (please open a issue if you know how) even in fast mode nmap won't run in under 2s.
+And, on recent updates, you need sudo for all of that.
+
+## How?
+
+```
+hoo init
+```
+
+A ~/.horray folder is prepopulated:
+devices.yml regex list you mac addresses with names you recognize.
+services.yml regex list ports to names you recognize.
+
+
+## Credits
+
+* Ruby net-ping gem.
+* Nmap`s mac precompiled prefixes.

data/Rakefile

@@ -0,0 +1,9 @@
+require 'bundler/gem_tasks'
+
+require 'rspec/core/rake_task'
+require 'rubocop/rake_task'
+
+RSpec::Core::RakeTask.new
+RuboCop::RakeTask.new
+
+task default: [:spec, :rubocop]

data/bin/hoo

@@ -0,0 +1,9 @@
+#!/usr/bin/env ruby
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+
+require 'hooray'
+
+puts
+puts '                             Hooray!'
+
+Hooray::CLI.start(ARGV)

data/lib/hooray.rb

@@ -0,0 +1,20 @@
+# require 'pry'
+require 'yaml'
+require 'open3'
+require 'socket'
+require 'ipaddr'
+require 'macaddr'
+require 'net/ping'
+require 'paint/pa'
+require 'table_print'
+
+require 'hooray/settings'
+require 'hooray/node'
+require 'hooray/port'
+require 'hooray/seek'
+
+require 'hooray/cli'
+
+# Hooray!
+module Hooray
+end

data/lib/hooray/cli.rb

@@ -0,0 +1,114 @@
+require 'thor'
+
+module Hooray
+  # Nice cli
+  class CLI < Thor
+    class_option :verbose, type: :boolean, aliases: :v
+    class_option :network, type: :string, aliases: :n
+
+    def initialize(*args)
+      super
+      return if ARGV.first =~ /init/
+      Settings.load!
+      @start = Time.now
+    end
+
+    desc 'init', 'creates settings on ~'
+    long_desc <<-LONG
+
+    Creates local settings on your home folder.
+
+    LONG
+    def init
+      return if Dir.exist?(Settings::CONFIG_DIR)
+      pa 'Creating ~/.hooray directory', :red
+      Dir.mkdir(Settings::CONFIG_DIR)
+      settings_dir = File.join(File.dirname(__FILE__), 'settings')
+      %w(settings devices services).each do |file|
+        pa "Creating ~/.hooray/#{file}.yml", :red
+        FileUtils.cp "#{settings_dir}/#{file}.yml", Settings::CONFIG_DIR
+      end
+    end
+
+    desc 'list FILTER', 'list and apply FILTER'
+    long_desc <<-LONG
+
+    Lists out all devices connected to the current network.
+
+    LONG
+    def list(*filter)
+      pa "Listing devices * #{filter}", :red
+      print_table Seek.new(options[:network], *filter).nodes
+    end
+
+    desc 'watch FILTER', 'watch in realtime  FILTER'
+    long_desc <<-LONG
+
+    Keeps listing out all devices connected to the current network.
+
+    LONG
+    def watch
+      print_table old_seek = Seek.new(options[:network]).nodes
+      pa "Starting watch...", :red
+      loop do
+        sleep 5
+        new_seek = Seek.new(options[:network]).nodes
+        print_change :new, (new_seek - old_seek)
+        print_change :old, (old_seek - new_seek), :red
+        old_seek = new_seek
+        # binding.pry
+      end
+    end
+
+    desc 'update', 'updates ssh config files'
+    long_desc <<-LONG
+
+    Updates your config files based on devices.
+
+    LONG
+    def update
+    end
+
+    desc 'local', 'local port status'
+    long_desc <<-LONG
+
+    This is a helper for those who can never remember netstat options.
+
+    LONG
+    def local
+      Kernel.system 'netstat -na | grep "tcp.*LISTEN"'
+    end
+
+    private
+
+    def debug(message)
+      return unless options[:verbose]
+      say message
+    end
+
+    def print_change(txt, changes, color = :blue)
+      return if changes.empty?
+      pa "#{txt.to_s.capitalize} nodes @ #{Time.now}", color
+      tp changes, :name, :ip, :mac
+    end
+
+    def print_table(nodes)
+      tp nodes, :name, :ip, :mac
+      puts "---"
+      took = (Time.now - @start).round(2)
+      pa "#{nodes.count} devices @ #{Time.now} #{took}s", '#777', :bold
+    end
+
+    def method_missing(*params)
+      case params.size
+      when 1 then
+        puts "Do you want to `#{params.first}` to a device?"
+        puts "Use #{ARGV.first} #{params.first} <device name>"
+      when 2 then
+        command, device = *params
+        system "#{command} "
+      else super
+      end
+    end
+  end
+end

data/lib/hooray/node.rb

@@ -0,0 +1,38 @@
+module Hooray
+  # Node representing a device
+  class Node
+    attr_accessor :host, :name, :nick, :mac, :ip, :ports
+
+    def initialize(params = {})
+      @ip = params[:ip]
+      @mac = params[:mac]
+      @mac ||= Mac.addr if @ip == Seek.my_lan_ip
+      set_name
+    end
+
+    def set_name
+      return unless mac
+      if [Mac.addr].flatten.include?(mac)
+        @name = Socket.gethostname
+      else
+        @name = Settings.device(mac) || Settings.family(mac)
+      end
+    end
+
+    def to_ip
+      Addrinfo.ip(ip)
+    end
+
+    def <=>(other)
+      ip <=> other.ip
+    end
+
+    def eql?(other)
+      ip == other.ip || mac == other.mac
+    end
+
+    def hash
+      [ip, mac].hash
+    end
+  end
+end

data/lib/hooray/port.rb

@@ -0,0 +1,7 @@
+module Hooray
+  class Port < Struct.new(:number, :protocol, :name)
+    def name
+      @name || Settings.services[number][protocol]
+    end
+  end
+end

data/lib/hooray/scan.rb

@@ -0,0 +1,19 @@
+module Hooray
+  #
+  # Main runner
+  #
+  class Scan < Struct.new(:target, :ports, :opts)
+    def run
+      scan = []
+      bots = []
+      pa "Starting #{Settings.list}  on '#{target}' #{ports}"
+      [ports].flatten.each do |port|
+        bots << Thread.new do
+          scan << ip if Net::Ping::TCP.new(ip.to_s, port, 1).ping?
+        end
+      end
+      bots.each(&:join)
+      scan
+    end
+  end
+end

data/lib/hooray/seek.rb

@@ -0,0 +1,79 @@
+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 :nodes, :devices
+
+    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
+
+    def arp_table
+      return @arp_table if @arp_table
+      @arp_table ||= {}
+      `arp -n`.split(/\n/).each do |l|
+        ip, _hw, mac, _flag, iface = l.split(/\s+/)
+        # p "#{ip}  #{mac} #{iface}"
+        @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

data/lib/hooray/settings.rb

@@ -0,0 +1,59 @@
+module Hooray
+  # App settings
+  class Settings
+    CONFIG_DIR = ENV['HOME'] + '/.hooray/'
+
+    class << self
+      attr_accessor :all, :services, :devices, :macs
+
+      def no_config_folder
+        puts "No config folder, run `#{$PROGRAM_NAME} init`"
+        puts
+        exit 1
+      end
+
+      def load!
+        no_config_folder unless Dir.exist?(CONFIG_DIR)
+        @all = YAML.load_file(CONFIG_DIR + 'settings.yml')
+        @services = YAML.load_file(CONFIG_DIR + 'services.yml')
+        @devices  = YAML.load_file(CONFIG_DIR + 'devices.yml')
+        @macs = {}
+        File.read(CONFIG_DIR + 'nmap-mac-prefixes').each_line do |line|
+          next if line =~ /^\s*#/
+          prefix, *name = *line.split(/\s/)
+          @macs.store(prefix, name.join(' '))
+        end
+      end
+
+      def list
+        out = 'SCAN: '
+        out += ' SYN' if syn_scan
+        out += ' SERVICE' if service_scan
+        out += ' OS' if os_fingerprint
+        out
+      end
+
+      def device(mac)
+        devices[mac.to_sym] || devices[mac.to_s]
+      end
+
+      def family(mac)
+        prefix = mac.to_s.gsub(':', '')[0,6].upcase
+        macs[prefix]
+      end
+
+      def all
+        @all ||= {}
+      end
+
+      def method_missing(meth, *params)
+        if meth =~ /=/
+          all[meth.to_s.gsub('=', '')] = params.first
+        else
+          arg = meth.to_s.gsub('?', '')
+          all[arg] || all[arg.to_sym]
+        end
+      end
+    end # class << self
+  end # Settings
+end # Hooray