cow 0.1.1

data/.gitignore

@@ -0,0 +1,9 @@
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in cow.gemspec
+gemspec

data/README.md

@@ -0,0 +1,43 @@
+# Cow
+
+Cow - console server wrapper. 
+
+## Installation
+
+`gem install cow`
+
+`mkdir /var/cache/cow`
+
+`sudo cow add [hostname] [type] [community]`
+
+## Usage
+
+Type `cow`. Then you can see usage.
+
+`cow update` should be run periodically.
+
+## Example
+
+```
+$ cow list
+console.local/1        -       tokyo-core01     |       ssh -l kazubu:7001 console.local
+console.local/2        -       tokyo-core02     |       ssh -l kazubu:7002 console.local
+console.local/3        -       tokyo-sw01       |       ssh -l kazubu:7003 console.local
+console.local/4        -       tokyo-sw02       |       ssh -l kazubu:7004 console.local
+console.local/5        -       tokyo-serv01     |       ssh -l kazubu:7005 console.local
+console.local/6        -       tokyo-serv02     |       ssh -l kazubu:7006 console.local
+console.local/7        -       tokyo-serv03     |       ssh -l kazubu:7007 console.local
+console.local/8        -       0F-B3-10P08      |       ssh -l kazubu:7008 console.local
+
+$ cow find core
+console.local/1        -       tokyo-core01     |       ssh -l kazubu:7001 console.local
+console.local/2        -       tokyo-core02     |       ssh -l kazubu:7002 console.local
+
+$ cow connect tokyo-core01
+Found on console.local/1. Connecting...
+Password:
+
+tokyo-core01 (ttyd0)
+
+login: 
+```

data/Rakefile

@@ -0,0 +1,2 @@
+require "bundler/gem_tasks"
+task :default => :spec

data/bin/console

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require "cow"
+
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+# (If you use this, don't forget to add pry to your Gemfile!)
+# require "pry"
+# Pry.start
+
+require "irb"
+IRB.start

data/bin/setup

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/cow.gemspec

@@ -0,0 +1,32 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'cow/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "cow"
+  spec.version       = Cow::VERSION
+  spec.authors       = ["Kazuki Shimizu"]
+  spec.email         = ["kazubu@kazubu.jp"]
+
+  spec.summary       = %q{Console server Wrapper}
+  spec.description   = %q{Ease to connect console server ports}
+  spec.homepage      = "https://www.kazubu.jp/"
+
+  # Prevent pushing this gem to RubyGems.org. To allow pushes either set the 'allowed_push_host'
+  # to allow pushing to a single host or delete this section to allow pushing to any host.
+  #if spec.respond_to?(:metadata)
+  #  spec.metadata['allowed_push_host'] = "TODO: Set to 'http://mygemserver.com'"
+  #else
+  #  raise "RubyGems 2.0 or newer is required to protect against public gem pushes."
+  #end
+
+  spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.12"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_dependency 'snmp'
+end

data/exe/cow

@@ -0,0 +1,10 @@
+#!/usr/bin/env ruby
+
+begin
+  require 'rubygems'
+rescue LoadError
+end
+
+require 'cow/application'
+
+Cow::Application.new.run

data/lib/cow.rb

@@ -0,0 +1,17 @@
+require "cow/version"
+require "cow/plugins/acs"
+require "cow/plugins/acs5000"
+require "cow/plugins/acs6000"
+require "cow/plugins/digi_ts"
+require "cow/plugins/cisco"
+
+module Cow
+  SERVER_TYPES = {
+    acs: Cow::ACS::Server,
+    acs5000: Cow::ACS5000::Server,
+    acs6000: Cow::ACS6000::Server,
+    digi_ts_telnet: Cow::DIGI_TS::Telnet::Server,
+    cisco_telnet: Cow::Cisco::Telnet::Server,
+    cisco_ssh: Cow::Cisco::SSH::Server
+  }
+end

data/lib/cow/application.rb

@@ -0,0 +1,237 @@
+require 'cow'
+require 'yaml'
+
+module Cow
+  class Application
+    class Cache
+      attr_accessor :servers
+
+      def initialize
+        @servers = []
+      end
+
+      def add_server(_server)
+        @servers.each_with_index{|server, idx|
+          if server.hostname == _server.hostname
+            @servers[idx] = _server
+            return _server
+          end
+        }
+
+        @servers << _server
+      end
+
+    end
+
+    CACHE_FILE = '/var/cache/cow/cow.cache'
+
+    @cache = nil
+
+    def initialize
+      @name = 'cow'
+    end
+
+    def create_cache(_cache = CACHE_FILE)
+      File.write(_cache, Cow::Application::Cache.new.to_yaml)
+    end
+
+    def load_cache(_cache = CACHE_FILE)
+      unless File.exist?(_cache)
+        puts 'Cache file is not found. Creating...'
+        create_cache(_cache)
+      end
+
+      if File.size(_cache) < 5
+        puts 'Cache file looks empty. Creating...'
+        create_cache(_cache)
+      end
+
+      cache = YAML.load(File.read(_cache))
+
+      if cache.class == Cow::Application::Cache
+        @cache = cache
+      else
+        raise 'Cache file is corrupt?'
+      end
+    end
+
+    def edit_cache(_cache = CACHE_FILE)
+      unless @cache
+        load_cache
+      end
+
+      cache = @cache
+
+      File.open(_cache, 'w') do |file|
+        puts 'Updating cache file.'
+        if file.flock(File::LOCK_EX|File::LOCK_NB)
+
+          begin
+            yield cache
+          rescue
+            cache = @cache
+          end
+
+          if cache.class == Cow::Application::Cache
+            file.write(cache.to_yaml)
+          else
+            file.write(@cache.to_yaml)
+          end
+        else
+          raise 'File is already locked.'
+        end
+      end
+      load_cache(_cache)
+    end
+
+    def run
+      command = ARGV[0]
+      option1 = ARGV[1]
+      option2 = ARGV[2]
+      option3 = ARGV[3]
+
+      case command
+      when 'add', 'a'
+        cmd_add(option1, option2, option3)
+      when 'connect', 'c'
+        cmd_connect(option1)
+      when 'find', 'f'
+        cmd_find(option1)
+      when 'list', 'l'
+        cmd_list(option1)
+      when 'server-list', 's'
+        cmd_server_list
+      when 'update', 'u'
+        cmd_update(option1)
+      else
+        show_help
+      end
+    end
+
+    def cmd_add(_hostname, _type, _snmp_community)
+      raise ArgumentError unless _hostname.class == String && _type.class == String && _snmp_community.class == String
+      raise ArgumentError unless Cow::SERVER_TYPES.keys.include?(_type.to_sym)
+
+      puts "#{_hostname}(#{_type}) will be added"
+
+      server = Cow::SERVER_TYPES[_type.to_sym].new(_hostname, _snmp_community)
+
+      server.ports.each do |port|
+        puts "%s %s" % ["#{server.hostname.to_s}/#{port.port.to_s}".ljust(30), port.name.to_s.ljust(20) ]
+      end
+
+      load_cache
+
+      edit_cache do |cache|
+        cache.add_server server
+      end if server
+    end
+
+    def cmd_connect(_portname)
+      raise ArgumentError unless _portname.class == String
+
+      load_cache
+
+      @cache.servers.each do |server|
+        server.find(_portname).each do |port|
+          if port.name == _portname
+            puts "Found on #{server.hostname}/#{port.port}. Connecting..."
+            server.connect(port)
+          end
+          return
+        end
+      end
+    end
+
+    def cmd_find(_keyword)
+      raise ArgumentError unless _keyword.class == String
+
+      load_cache
+      puts "%s %s %s" % ["SERVER/PORT".ljust(30), "PORTNAME".ljust(20), "COMMAND"]
+
+      @cache.servers.each do |server|
+        server.find(_keyword).each do |port|
+          puts "%s %s %s" % ["#{server.hostname.to_s}/#{port.port.to_s}".ljust(30), port.name.to_s.ljust(20), server.connect_command(port)]
+        end
+      end
+    end
+
+    def cmd_list(_hostname)
+      load_cache
+      puts "%s %s %s" % ["SERVER/PORT".ljust(30), "PORTNAME".ljust(20), "COMMAND"]
+
+      @cache.servers.each do |server|
+        if _hostname.nil? || server.hostname == _hostname
+          server.ports.each do |port|
+            puts "%s %s %s" % ["#{server.hostname.to_s}/#{port.port.to_s}".ljust(30), port.name.to_s.ljust(20), server.connect_command(port)]
+          end
+        end
+      end
+    end
+
+    def cmd_server_list
+      load_cache
+      puts "SERVER(TYPE)"
+
+      @cache.servers.each do |server|
+        type = nil
+        Cow::SERVER_TYPES.each_key{|t| type = t if Cow::SERVER_TYPES[t] == server.class}
+
+        puts "#{server.hostname}(#{type})"
+      end
+    end
+
+    def cmd_update(_hostname)
+      load_cache
+
+      puts "Updating local cache..."
+      puts "%s %s" % ["SERVER/PORT".ljust(30), "PORTNAME".ljust(20)]
+
+      edit_cache do |cache|
+        cache.servers.each do |server|
+          if _hostname.nil? || server.hostname == _hostname
+            puts "Updating server #{server.hostname}"
+            server.update_ports
+
+            server.ports.each do |port|
+              puts "%s %s" % ["#{server.hostname.to_s}/#{port.port.to_s}".ljust(30), port.name.to_s.ljust(20) ]
+            end
+
+            cache.add_server server
+          end
+        end
+      end
+    end
+
+    def show_help
+      puts <<EOF
+cow - Console Server Wrapper
+
+Usage:
+  cow [command] [options]
+
+Commands:
+  add [hostname] [type] [snmp_community]
+    Add new console server.
+    available types: #{Cow::SERVER_TYPES.keys.join(' ')}
+
+  connect [portname]
+    Connect to console port.
+
+  find [keyword]
+    Find console port which includes given name.
+
+  list [hostname(optional)]
+    Show console server port list. If hostname is not specified,
+    all console ports will be shown.
+
+  server-list
+    Show all console servers.
+
+  update [hostname(optional)]
+    Update console server port list. If hostname is not specified,
+    all port lists will be updated.
+EOF
+    end
+  end
+end

data/lib/cow/plugins/acs.rb

@@ -0,0 +1,40 @@
+require 'cow/plugins/tmpl'
+require 'snmp'
+
+module Cow
+  module ACS
+    OID_DESCRIPTION_LIST = '1.3.6.1.4.1.2925.4.2.6.2.1.1.3'
+    OID_TCPPORT_LIST = '1.3.6.1.4.1.2925.4.2.6.2.1.1.26'
+
+    class Port < Cow::Tmpl::Port
+    end
+
+    class Server < Cow::Tmpl::Server
+      def port(_port)
+        return _port if _port.class == Cow::ACS::Port
+        super
+      end
+
+      def connect_command(_port, _user = ENV['USER'])
+        return "ssh -l #{_user}:#{port(_port).tcp_port} #{@hostname}"
+      end
+
+      private
+
+      def get_ports
+        ret = []
+        snmp{|s|
+          s.walk(OID_DESCRIPTION_LIST){|x|
+            next if x.name.last == 0
+            port = x.name.last
+            name = x.value.to_s
+            tcp_port = snmp{|s| break s.get_value(OID_TCPPORT_LIST + ".#{x.name.last.to_s}")}
+
+            ret <<  Cow::ACS::Port.new(port, name, tcp_port)
+          }
+        }
+        return ret
+      end
+    end
+  end
+end

data/lib/cow/plugins/acs5000.rb

@@ -0,0 +1,40 @@
+require 'cow/plugins/tmpl'
+require 'snmp'
+
+module Cow
+  module ACS5000
+    OID_DESCRIPTION_LIST = '1.3.6.1.4.1.10418.15.2.2.6.2.1.1.3'
+    OID_TCPPORT_LIST = '1.3.6.1.4.1.10418.15.2.2.6.2.1.1.26'
+
+    class Port < Cow::Tmpl::Port
+    end
+
+    class Server < Cow::Tmpl::Server
+      def port(_port)
+        return _port if _port.class == Cow::ACS5000::Port
+        super
+      end
+
+      def connect_command(_port, _user = ENV['USER'])
+        return "ssh -l #{_user}:#{port(_port).tcp_port} #{@hostname}"
+      end
+
+      private
+
+      def get_ports
+        ret = []
+        snmp{|s|
+          s.walk(OID_DESCRIPTION_LIST){|x|
+            next if x.name.last == 0
+            port = x.name.last
+            name = x.value.to_s
+            tcp_port = snmp{|s| break s.get_value(OID_TCPPORT_LIST + ".#{x.name.last.to_s}")}
+
+            ret <<  Cow::ACS5000::Port.new(port, name, tcp_port)
+          }
+        }
+        return ret
+      end
+    end
+  end
+end

data/lib/cow/plugins/acs6000.rb

@@ -0,0 +1,39 @@
+require 'cow/plugins/tmpl'
+require 'snmp'
+
+module Cow
+  module ACS6000
+    OID_DESCRIPTION_LIST = '1.3.6.1.4.1.10418.16.2.3.2.1.4'
+
+    class Port < Cow::Tmpl::Port
+    end
+
+    class Server < Cow::Tmpl::Server
+      def port(_port)
+        return _port if _port.class == Cow::ACS6000::Port
+        super
+      end
+
+      def connect_command(_port, _user = ENV['USER'])
+        return "ssh -l #{_user}:#{port(_port).tcp_port} #{@hostname}"
+      end
+
+      private
+
+      def get_ports
+        ret = []
+        snmp{|s|
+          s.walk(OID_DESCRIPTION_LIST){|x|
+            next if x.name.last == 0
+            port = x.name.last
+            name = x.value.to_s
+            tcp_port = port.to_i + 7000
+
+            ret <<  Cow::ACS6000::Port.new(port, name, tcp_port)
+          }
+        }
+        return ret
+      end
+    end
+  end
+end

data/lib/cow/plugins/cisco.rb

@@ -0,0 +1,54 @@
+require 'cow/plugins/tmpl'
+require 'snmp'
+
+module Cow
+  module Cisco
+    OID_DESCRIPTION_LIST = '1.3.6.1.4.1.9.2.9.2.1.8'
+
+    class Port < Cow::Tmpl::Port
+    end
+
+    module Server
+      def port(_port)
+        return _port if _port.class == Cow::Cisco::Port
+        super
+      end
+
+      def get_ports
+        ret = []
+        snmp{|s|
+          s.walk(OID_DESCRIPTION_LIST){|x|
+            next if x.name.last == 0
+            next if x.value.to_s.length == 0
+            port = x.name.last
+            name = x.value.to_s
+            tcp_port = 2000 + port.to_i
+
+            ret <<  Cow::Cisco::Port.new(port, name, tcp_port)
+          }
+        }
+        return ret
+      end
+    end
+
+    module Telnet
+      class Server < Cow::Tmpl::Server
+        include Cow::Cisco::Server
+
+        def connect_command(_port, _user = ENV['USER'])
+          return "telnet #{@hostname} #{port(_port).tcp_port}"
+        end
+      end
+    end
+
+    module SSH
+      class Server < Cow::Tmpl::Server
+        include Cow::Cisco::Server
+
+        def connect_command(_port, _user = ENV['USER'])
+          return "ssh -l #{_user}:#{port(_port).port} #{@hostname}"
+        end
+      end
+    end
+  end
+end

data/lib/cow/plugins/digi_ts.rb

@@ -0,0 +1,46 @@
+require 'cow/plugins/tmpl'
+require 'snmp'
+
+module Cow
+  module DIGI_TS
+    OID_DESCRIPTION_LIST = '1.3.6.1.4.1.332.11.5.3.3.25.13.1.29'
+    OID_TCPPORT_BASE = '1.3.6.1.4.1.332.11.5.3.3.20.22.0'
+
+    class Port < Cow::Tmpl::Port
+    end
+
+    module Telnet
+      class Server < Cow::Tmpl::Server
+        def port(_port)
+          return _port if _port.class == Cow::DIGI_TS::Port
+          super
+        end
+
+        def connect_command(_port, _user = ENV['USER'])
+          #return "ssh -l #{_user}:#{port(_port).tcp_port+500} #{@hostname}"
+          return "telnet #{@hostname} #{port(_port).tcp_port}"
+        end
+
+        private
+
+        def get_ports
+          ret = []
+
+          base_port = snmp{|s| break s.get_value(OID_TCPPORT_BASE)}
+
+          snmp{|s|
+            s.walk(OID_DESCRIPTION_LIST){|x|
+              next if x.value.to_s.chomp.length == 0
+              port = x.name.last
+              name = x.value.to_s
+              tcp_port = base_port + port.to_i
+
+              ret <<  Cow::DIGI_TS::Port.new(port, name, tcp_port)
+            }
+          }
+          return ret
+        end
+      end
+    end
+  end
+end

data/lib/cow/plugins/tmpl.rb

@@ -0,0 +1,99 @@
+require 'snmp'
+
+module Cow
+  module Tmpl
+    class Port
+      attr_accessor :port, :name, :tcp_port
+
+      @port = nil
+      @name = nil
+      @tcp_port = nil
+
+      def initialize
+      end
+
+      def initialize(_port, _name, _tcp_port)
+        raise ArgumentError if _port.nil? or _name.nil? or _tcp_port.nil?
+
+        @port = _port.to_s
+        @name = _name.to_s
+        @tcp_port = _tcp_port.to_s
+      end
+    end
+
+    class Server
+      attr_accessor :hostname
+
+      @hostname = nil
+      @snmp_community = nil
+      @ports = nil
+
+      def initialize
+      end
+
+      def initialize(_hostname, _snmp_community)
+        @hostname = _hostname
+        @snmp_community = _snmp_community
+
+        @ports = get_ports
+      end
+
+      def ports
+        update_ports unless @ports
+        return @ports
+      end
+
+      def update_ports
+        @ports = get_ports
+      end
+
+      def port(_port)
+        case _port
+        when Fixnum
+          ports.each{|p|
+            return p if p.port == _port
+          }
+        when String
+          ports.each{|p|
+            return p if p.name == _port
+          }
+        else
+        end
+
+        nil
+      end
+
+      def find(_portname)
+        _ports = []
+        ports.each{|p|
+          _ports << p if p.name.include?(_portname)
+        }
+        return _ports
+      end
+
+      def connect_command(_port, _user = ENV['USER'])
+        raise NotImplementedError
+      end
+
+      def connect(_port, _user = nil)
+        if _user
+          system(connect_command(_port, _user))
+        else
+          system(connect_command(_port))
+        end
+      end
+
+      private
+
+      def get_ports
+        raise NotImplementedError
+      end
+
+      def snmp
+        SNMP::Manager.open(:host => @hostname, :community => @snmp_community) do|snmp|
+          yield(snmp)
+        end
+      end
+    end
+  end
+end

data/lib/cow/version.rb

@@ -0,0 +1,3 @@
+module Cow
+  VERSION = "0.1.1"
+end

metadata

@@ -0,0 +1,117 @@
+--- !ruby/object:Gem::Specification
+name: cow
+version: !ruby/object:Gem::Version
+  version: 0.1.1
+  prerelease: 
+platform: ruby
+authors:
+- Kazuki Shimizu
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2016-09-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.12'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.12'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: snmp
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Ease to connect console server ports
+email:
+- kazubu@kazubu.jp
+executables:
+- cow
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- cow.gemspec
+- exe/cow
+- lib/cow.rb
+- lib/cow/application.rb
+- lib/cow/plugins/acs.rb
+- lib/cow/plugins/acs5000.rb
+- lib/cow/plugins/acs6000.rb
+- lib/cow/plugins/cisco.rb
+- lib/cow/plugins/digi_ts.rb
+- lib/cow/plugins/tmpl.rb
+- lib/cow/version.rb
+homepage: https://www.kazubu.jp/
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: -2846258190445444436
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: -2846258190445444436
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Console server Wrapper
+test_files: []