net-ops 0.0.4.pre → 0.0.5.pre

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f73921d344261f93ed38678f0be1ed30e058c786
-  data.tar.gz: 2dc6ca24c0ff3ef0fb849b96b9ed3f8967e1a9f8
+  metadata.gz: 6ddd5aa75208c9e4c9cca7e3b132134453117747
+  data.tar.gz: 7f7cb228b4234a18e901c3d0bf77d19a0c637c90
 SHA512:
-  metadata.gz: 6571aa3b754121c2218abfde7d7f08586022ac6b84944aa290a8faae0deb642a5e21ec757ba621c05aee747b807d66120f943abf07b8edcf1c8bc6678aa89ff3
-  data.tar.gz: c3f8fbab57e7c2642363b9f4ae492c66b3ed12cf87a5343b8a237e9d0f78e1bb4b97bec4e0c8f2b2dbd9cb44fd1d468fc200dac32d73ae3b54a474c6714c5009
+  metadata.gz: 024cbbd43efc3e9381db9fa65446fee4f08626cb424cebeefefe04ec039f9679f23cd85d9d6fcbde32f95cc956af0f0212b5f1a089164ecb107c47f106ff8ad0
+  data.tar.gz: b74d3628a96ae7753f37268c5031be99251ee92321b3985255bdb2a209274211fd189e9ec4a0b9cd1088cd73aab24c52e105829af685321de3434d1ac5afcfb8

data/lib/net/ops.rb

@@ -2,17 +2,19 @@ require 'yaml'
 require 'logger'
 require 'thread/pool'
 
-Dir['../lib/net/transport/*.rb'].each { |file| require file }
+#Dir['../lib/net/transport/*.rb'].each { |file| require file }
+require 'net/transport/ssh'
+require 'net/transport/telnet'
 
 module Net
   module Ops
-  
+
     # Provides a DSL for interacting with Cisco switches and routers.
     class Session
-      
+
       attr_reader :transport
       attr_reader :transports
-      
+
       # Initialize a new session.
       #
       # @param [String] host the target host.
@@ -24,14 +26,14 @@ module Net
       # @param [Logger] logger the logger to use.
       #
       # @return [Session] the new session.
-      def initialize(host, options = { timeout: 10, prompt: /.+(#|>)/ }, logger = nil)
+      def initialize(host, options = { timeout: 10, prompt: /.+(#|>|\])/ }, logger = nil)
         @host    = host
         @options = options
-        @transports = []        
-        
+        @transports = []
+
         setup_logger(logger)
-        
-        Net::Ops::Transport.constants.each do |c| 
+
+        Net::Ops::Transport.constants.each do |c|
           register_transport(c) if Class === Net::Ops::Transport.const_get(c)
         end
       end
@@ -48,7 +50,7 @@ module Net
         @credentials = credentials
 
         @logger.debug(@host) { "Opening session as #{ credentials[:username] }" }
-        
+
         @transports.each do |transport|
           @transport ||= transport.open(@host, @options, credentials)
         end
@@ -86,13 +88,13 @@ module Net
         output = ''
         @transport.cmd(command) { |c| output += c }
 
-        @logger.debug("#{ @host } (#{ get_mode })") { output }
+        # @logger.debug("#{ @host } (#{ get_mode })") { output }
         # @logger.warn(@host) { 'Net::Ops::IOSInvalidInput'; puts output } if /nvalid input detected/.match(output)
         fail Net::Ops::IOSInvalidInput if /nvalid input detected/.match(output)
-        
+
         output
       end
-      
+
       # Get the specified item on the device.
       # Equivalent to the Cisco show command.
       #
@@ -110,7 +112,7 @@ module Net
       def set(item, value)
         run("#{ item } #{ value }")
       end
-      
+
       # Enable the specified item on the device.
       #
       # @param item [String] the item to enable.
@@ -127,14 +129,14 @@ module Net
       def disable(item)
         run("no #{ item }")
       end
-      
+
       def zeroize(item)
         @logger.debug(@host) { "Executing #{ item } zeroize" }
-      
+
         @transport.cmd('String' => "#{ item } zeroize", 'Match' => /.+/)
         @transport.cmd('yes')
       end
-      
+
       def generate(item, options)
         run("#{ item } generate #{ options }")
       end
@@ -184,37 +186,37 @@ module Net
 
         write! if options == :enforce_save
       end
-      
+
       def interface(interface, &block)
         ensure_mode(:configuration)
-        
+
         run("interface #{ interface }")
         instance_eval(&block)
       end
-      
+
       def interfaces(interfaces = /.+/, &block)
         ints = privileged do
-          get('interfaces status').select do |int| 
+          get('interfaces status').select do |int|
             interfaces.match("#{ int['short_type'] }#{ int['port_number'] }")
           end
         end
-        
+
         ints.each do |int|
           interface("#{ int['short_type'] }#{ int['port_number'] }") do
             instance_eval(&block)
           end
         end
       end
-      
+
       def lines(lines, &block)
         ensure_mode(:configuration)
-        
+
         run("line #{ lines }")
         instance_eval(&block)
       end
 
       private
-      
+
       def register_transport(klass)
         @logger.debug(@host) { "Registering transport #{ klass }" }
         @transports << Net::Ops::Transport.const_get(klass)
@@ -229,8 +231,7 @@ module Net
         @logger = logger
 
         # Otherwise we create a new one.
-        # logger = Logger.new(STDOUT)
-        logger = Logger.new('log.log')
+        logger = Logger.new(STDOUT)
         logger.level = Logger::DEBUG
         @logger ||= logger
       end
@@ -240,9 +241,9 @@ module Net
       # @return [Symbol] the current command mode.
       def get_mode
         prompt = ''
-        @transport.cmd('') { |c| prompt += c }        
+        @transport.cmd('') { |c| prompt += c }
         match = /(?<hostname>[^\(-\)]+)(\((?<text>[\w\-]+)\))?(?<char>#|>)/.match(prompt)
-        
+
         mode  = nil
 
         if match && match['char']
@@ -252,12 +253,12 @@ module Net
                  when '#' then :privileged
                  end
 
-        end         
-        
+        end
+
         if match && match['text']
           mode = match['text'].to_sym
         end
-        
+
         mode
       end
 
@@ -273,7 +274,7 @@ module Net
 
         when :privileged
           run('end') if configuration?
-          enable_privileged(@credentials['password']) if user?
+          enable_privileged(@credentials[:password]) if user?
 
         when :configuration
           run('configure terminal') unless configuration?
@@ -312,7 +313,7 @@ module Net
       end
 
     end
-    
+
     class Parser
 
       def initialize(file)
@@ -320,16 +321,16 @@ module Net
       end
 
       def parse(command, output)
-        results = []        
+        results = []
         path    = explore_tree(command.split(/ /))
-        
+
         if path.has_key?('regex')
           regex = Regexp.new(path.fetch('regex').delete(' '))
-          
+
           output.each_line do |line|
             results << regex.match(line) if regex.match(line)
           end
-          
+
         else results = output
         end
 
@@ -342,40 +343,40 @@ module Net
         level = @regexs['cisco']
 
         path.each { |p| level[p] ? level = level[p] : break }
-        
+
         level
       end
 
     end
-    
+
     class Task
       include Net::Ops
 
       def initialize(id)
         @id = id
-        
+
         @logger = Logger.new(STDOUT)
         @logger.level = Logger::INFO
       end
-      
+
       def log(severity, message)
         @logger.add(severity, message, @id)
       end
-      
+
       def info(message)
         log(Logger::INFO, message)
       end
-      
+
       def warn(message)
         log(Logger::WARN, message)
       end
-      
+
       def error(message)
         log(Logger::ERROR, message)
       end
 
     end
-    
+
     #
     class TransportUnavailable < Exception; end
     class IOSInvalidInput < Exception; end

data/lib/net/transport/ssh.rb

@@ -0,0 +1,39 @@
+require 'net/ssh/telnet'
+
+module Net
+  module Ops
+    module Transport
+    
+      #
+      class SSH
+        
+        # Open an SSH session to the specified host using net/ssh/telnet.
+        #
+        # @param host [String] the destination host.
+        # @param options [Hash]
+        # @param credentials [Hash] credentials to use to connect.
+        def self.open(host, options, credentials)
+          session = nil
+
+          ssh = Net::SSH.start(host, credentials[:username], :password => credentials[:password])
+          session = Net::SSH::Telnet.new('Session' => ssh,
+                                         'Timeout' => options[:timeout],
+                                         'Prompt'  => options[:prompt])
+
+        rescue Errno::ECONNREFUSED => e
+          session = nil
+
+        rescue Net::SSH::AuthenticationFailed => e
+          session = nil
+
+        rescue Exception => e
+          session = nil
+
+        return session
+        end
+        
+      end
+    
+    end
+  end
+end

data/lib/net/transport/telnet.rb

@@ -0,0 +1,53 @@
+require 'net/telnet'
+
+module Net
+  module Ops
+    module Transport
+
+      #
+      class Telnet
+
+        # Open a Telnet session to the specified host using net/ssh.
+        #
+        # @param host [String] the destination host.
+        # @param options [Hash]
+        # @param credentials [Hash] credentials to use to connect.
+        def self.open(host, options, credentials)
+          session = nil
+
+          session = Net::Telnet.new('Host' => host,
+                                    'Timeout' => options[:timeout],
+                                    'Prompt'  => options[:prompt])
+
+          output = ''
+          session.cmd('String' => '', 'Match'  => /.+/) { |c| output += c }
+
+          if /[Uu]sername:/.match(output) then
+            session.cmd('String' => credentials[:username],
+                      'Match'  => /.+/)
+            session.cmd(credentials[:password])
+          end
+
+          if /[Pp]assword:/.match(output) then
+            session.cmd(credentials[:password])
+          end
+
+          return session
+
+        rescue Errno::ECONNREFUSED => e
+          session = nil
+
+        rescue Net::OpenTimeout => e
+          session = nil
+
+        rescue Exception => e
+          session = nil
+
+        return session
+        end
+
+      end
+
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: net-ops
 version: !ruby/object:Gem::Version
-  version: 0.0.4.pre
+  version: 0.0.5.pre
 platform: ruby
 authors:
 - Maxime Mouchet
@@ -38,7 +38,7 @@ dependencies:
     - - ~>
       - !ruby/object:Gem::Version
         version: 0.0.2
-description: Framework for interacting with network devices.
+description: Framework to automate daily operations on network devices.
 email:
 - max@maxmouchet.com
 executables: []
@@ -46,6 +46,8 @@ extensions: []
 extra_rdoc_files: []
 files:
 - lib/net/ops.rb
+- lib/net/transport/ssh.rb
+- lib/net/transport/telnet.rb
 homepage: http://github.com/maxmouchet/qscripts
 licenses:
 - MIT