hrr_rb_ssh 0.1.0

data/lib/hrr_rb_ssh/authentication/method/password/context.rb

@@ -0,0 +1,29 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/logger'
+
+module HrrRbSsh
+  class Authentication
+    module Method
+      class Password
+        class Context
+          attr_reader :username, :password
+
+          def initialize username, password
+            @username = username
+            @password = password
+
+            @logger = HrrRbSsh::Logger.new self.class.name
+          end
+
+          def verify username, password
+            @logger.info("verify username and password")
+            @logger.debug("username is #{username}, @username is #{@username}, and password is #{password}, @password is #{@password}")
+            username == @username and password == @password
+          end
+        end
+      end
+    end
+  end
+end

data/lib/hrr_rb_ssh/authentication/method/password.rb

@@ -0,0 +1,37 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/logger'
+require 'hrr_rb_ssh/authentication/method/password/context'
+
+module HrrRbSsh
+  class Authentication
+    module Method
+      name_list = [
+        'password'
+      ]
+
+      class Password
+        def initialize options
+          @logger = HrrRbSsh::Logger.new self.class.name
+
+          @authenticator = options.fetch( 'authentication_password_authenticator', Authenticator.new { false } )
+        end
+
+        def authenticate userauth_request_message
+          @logger.info("authenticate")
+          @logger.debug("userauth request: " + userauth_request_message.inspect)
+          username = userauth_request_message['user name']
+          password = userauth_request_message['plaintext password']
+          context = Context.new(username, password)
+          @authenticator.authenticate context
+        end
+      end
+
+      @@list ||= Hash.new
+      name_list.each do |name|
+        @@list[name] = Password
+      end
+    end
+  end
+end

data/lib/hrr_rb_ssh/authentication/method.rb

@@ -0,0 +1,21 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/authentication/method/none'
+require 'hrr_rb_ssh/authentication/method/password'
+
+module HrrRbSsh
+  class Authentication
+    module Method
+      @@list ||= Hash.new
+
+      def self.[] key
+        @@list[key]
+      end
+
+      def self.name_list
+        @@list.keys
+      end
+    end
+  end
+end

data/lib/hrr_rb_ssh/authentication.rb

@@ -0,0 +1,107 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/logger'
+require 'hrr_rb_ssh/message'
+require 'hrr_rb_ssh/closed_authentication_error'
+require 'hrr_rb_ssh/authentication/authenticator'
+require 'hrr_rb_ssh/authentication/method'
+
+module HrrRbSsh
+  class Authentication
+    SERVICE_NAME = 'ssh-userauth'
+
+    def initialize transport, options={}
+      @transport = transport
+      @options = options
+
+      @logger = HrrRbSsh::Logger.new self.class.name
+
+      @transport.register_acceptable_service SERVICE_NAME
+
+      @closed = nil
+
+      @username = nil
+    end
+
+    def send payload
+      raise HrrRbSsh::ClosedAuthenticationError if @closed
+      begin
+        @transport.send payload
+      rescue HrrRbSsh::ClosedTransportError
+        raise HrrRbSsh::ClosedAuthenticationError
+      end
+    end
+
+    def receive
+      raise HrrRbSsh::ClosedAuthenticationError if @closed
+      begin
+        @transport.receive
+      rescue HrrRbSsh::ClosedTransportError
+        raise HrrRbSsh::ClosedAuthenticationError
+      end
+    end
+
+    def start
+      @transport.start
+      authenticate
+    end
+
+    def close
+      return if @closed
+      @closed = true
+      @transport.close
+    end
+
+    def closed?
+      @closed
+    end
+
+    def username
+      raise HrrRbSsh::ClosedAuthenticationError if @closed
+      @username
+    end
+
+    def authenticate
+      loop do
+        payload = @transport.receive
+        case payload[0,1].unpack("C")[0]
+        when HrrRbSsh::Message::SSH_MSG_USERAUTH_REQUEST::VALUE
+          userauth_request_message = HrrRbSsh::Message::SSH_MSG_USERAUTH_REQUEST.decode payload
+          method_name = userauth_request_message['method name']
+          method = Method[method_name].new(@options)
+          if method.authenticate(userauth_request_message)
+            send_userauth_success
+            @username = userauth_request_message['user name']
+            @closed = false
+            break
+          else
+            send_userauth_failure
+            @closed = true
+          end
+        else
+          @closed = true
+          raise
+        end
+      end
+    end
+
+    def send_userauth_failure
+      message = {
+        'SSH_MSG_USERAUTH_FAILURE'          => HrrRbSsh::Message::SSH_MSG_USERAUTH_FAILURE::VALUE,
+        'authentications that can continue' => Method.name_list,
+        'partial success'                   => false,
+      }
+      payload = HrrRbSsh::Message::SSH_MSG_USERAUTH_FAILURE.encode message
+      @transport.send payload
+    end
+
+    def send_userauth_success
+      message = {
+        'SSH_MSG_USERAUTH_SUCCESS' => HrrRbSsh::Message::SSH_MSG_USERAUTH_SUCCESS::VALUE,
+      }
+      payload = HrrRbSsh::Message::SSH_MSG_USERAUTH_SUCCESS.encode message
+      @transport.send payload
+    end
+  end
+end

data/lib/hrr_rb_ssh/closed_authentication_error.rb

@@ -0,0 +1,7 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+module HrrRbSsh
+  class ClosedAuthenticationError < StandardError
+  end
+end

data/lib/hrr_rb_ssh/closed_connection_error.rb

@@ -0,0 +1,7 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+module HrrRbSsh
+  class ClosedConnectionError < StandardError
+  end
+end

data/lib/hrr_rb_ssh/closed_transport_error.rb

@@ -0,0 +1,7 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+module HrrRbSsh
+  class ClosedTransportError < StandardError
+  end
+end

data/lib/hrr_rb_ssh/compat.rb

@@ -0,0 +1,65 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+if RUBY_VERSION < "2.1"
+  class Array
+    def to_h
+      h = Hash.new
+      self.each do |k, v|
+        h[k] = v
+      end
+      h
+    end
+  end
+
+  require 'openssl'
+  class OpenSSL::BN
+    alias_method :__initialize__, :initialize
+
+    def initialize *args
+      args[0] = case args[0]
+                when OpenSSL::BN, Fixnum, Bignum
+                  args[0].to_s
+                else
+                  args[0]
+                end
+      __initialize__ *args
+    end
+  end
+end
+
+if RUBY_VERSION < "2.3"
+  class ClosedQueueError < StandardError
+  end
+
+  class Queue
+    require 'timeout'
+
+    alias_method :__enq__, :enq
+    alias_method :__deq__, :deq
+
+    def close
+      @closed = true
+    end
+
+    def closed?
+      @closed == true
+    end
+
+    def enq arg
+      raise ClosedQueueError if @closed == true
+      __enq__ arg
+    end
+
+    def deq
+      begin
+        Timeout.timeout(0.1) do
+          __deq__
+        end
+      rescue Timeout::Error
+        return nil if @closed == true
+        retry
+      end
+    end
+  end
+end

data/lib/hrr_rb_ssh/connection/channel/proc_chain/chain_context.rb

@@ -0,0 +1,22 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/logger'
+
+module HrrRbSsh
+  class Connection
+    class Channel
+      class ProcChain
+        class ChainContext
+          def initialize proc_chain
+            @logger = HrrRbSsh::Logger.new self.class.name
+            @proc_chain = proc_chain
+          end
+          def call_next *args
+            @proc_chain.call_next *args
+          end
+        end
+      end
+    end
+  end
+end

data/lib/hrr_rb_ssh/connection/channel/proc_chain.rb

@@ -0,0 +1,25 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/logger'
+require 'hrr_rb_ssh/connection/channel/proc_chain/chain_context'
+
+module HrrRbSsh
+  class Connection
+    class Channel
+      class ProcChain
+        def initialize
+          @logger = HrrRbSsh::Logger.new self.class.name
+          @q = Queue.new
+        end
+        def connect next_proc
+          @q.enq next_proc if next_proc
+        end
+        def call_next *args
+          next_proc = @q.deq
+          next_proc.call ChainContext.new(self), *args
+        end
+      end
+    end
+  end
+end

data/lib/hrr_rb_ssh/connection/channel/session/env/context.rb

@@ -0,0 +1,43 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/logger'
+
+module HrrRbSsh
+  class Connection
+    class Channel
+      module Session
+        class Env
+          class Context
+            attr_reader \
+              :logger,
+              :username,
+              :io,
+              :variables,
+              :vars,
+              :variable_name,
+              :variable_value
+
+            def initialize proc_chain, username, io, variables, message
+              @logger = HrrRbSsh::Logger.new self.class.name
+
+              @proc_chain = proc_chain
+              @username   = username
+              @io         = io
+              @variables  = variables
+              @vars       = variables
+
+              @variable_name  = message['variable name']
+              @variable_value = message['variable value']
+            end
+
+            def chain_proc &block
+              @proc = block || @proc
+            end
+          end
+        end
+      end
+    end
+  end
+end
+

data/lib/hrr_rb_ssh/connection/channel/session/env.rb

@@ -0,0 +1,31 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/logger'
+require 'hrr_rb_ssh/connection/request_handler'
+require 'hrr_rb_ssh/connection/channel/session/env/context'
+
+module HrrRbSsh
+  class Connection
+    class Channel
+      module Session
+        request_type = 'env'
+
+        class Env
+          def self.run proc_chain, username, io, variables, message, options
+            logger = HrrRbSsh::Logger.new self.class.name
+
+            context = Context.new proc_chain, username, io, variables, message
+            handler = options.fetch('connection_channel_request_env', RequestHandler.new {})
+            handler.run context
+
+            proc_chain.connect context.chain_proc
+          end
+        end
+
+        @@request_type_list ||= Hash.new
+        @@request_type_list[request_type] = Env
+      end
+    end
+  end
+end

data/lib/hrr_rb_ssh/connection/channel/session/exec/context.rb

@@ -0,0 +1,41 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/logger'
+
+module HrrRbSsh
+  class Connection
+    class Channel
+      module Session
+        class Exec
+          class Context
+            attr_reader \
+              :logger,
+              :username,
+              :io,
+              :variables,
+              :vars,
+              :command
+
+            def initialize proc_chain, username, io, variables, message
+              @logger = HrrRbSsh::Logger.new self.class.name
+
+              @proc_chain = proc_chain
+              @username   = username
+              @io         = io
+              @variables  = variables
+              @vars       = variables
+
+              @command = message['command']
+            end
+
+            def chain_proc &block
+              @proc = block || @proc
+            end
+          end
+        end
+      end
+    end
+  end
+end
+

data/lib/hrr_rb_ssh/connection/channel/session/exec.rb

@@ -0,0 +1,31 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/logger'
+require 'hrr_rb_ssh/connection/request_handler'
+require 'hrr_rb_ssh/connection/channel/session/exec/context'
+
+module HrrRbSsh
+  class Connection
+    class Channel
+      module Session
+        request_type = 'exec'
+
+        class Exec
+          def self.run proc_chain, username, io, variables, message, options
+            logger = HrrRbSsh::Logger.new self.class.name
+
+            context = Context.new proc_chain, username, io, variables, message
+            handler = options.fetch('connection_channel_request_exec', RequestHandler.new {})
+            handler.run context
+
+            proc_chain.connect context.chain_proc
+          end
+        end
+
+        @@request_type_list ||= Hash.new
+        @@request_type_list[request_type] = Exec
+      end
+    end
+  end
+end

data/lib/hrr_rb_ssh/connection/channel/session/pty_req/context.rb

@@ -0,0 +1,50 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/logger'
+
+module HrrRbSsh
+  class Connection
+    class Channel
+      module Session
+        class PtyReq
+          class Context
+            attr_reader \
+              :logger,
+              :username,
+              :io,
+              :variables,
+              :vars,
+              :term_environment_variable_value,
+              :terminal_width_characters,
+              :terminal_height_rows,
+              :terminal_width_pixels,
+              :terminal_height_pixels,
+              :encoded_terminal_modes
+
+            def initialize proc_chain, username, io, variables, message
+              @logger = HrrRbSsh::Logger.new self.class.name
+
+              @proc_chain = proc_chain
+              @username   = username
+              @io         = io
+              @variables  = variables
+              @vars       = variables
+
+              @term_environment_variable_value = message['TERM environment variable value']
+              @terminal_width_characters       = message['terminal width, characters']
+              @terminal_height_rows            = message['terminal height, rows']
+              @terminal_width_pixels           = message['terminal width, pixels']
+              @terminal_height_pixels          = message['terminal height, pixels']
+              @encoded_terminal_modes          = message['encoded terminal modes']
+            end
+
+            def chain_proc &block
+              @proc = block || @proc
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/hrr_rb_ssh/connection/channel/session/pty_req.rb

@@ -0,0 +1,31 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/logger'
+require 'hrr_rb_ssh/connection/request_handler'
+require 'hrr_rb_ssh/connection/channel/session/pty_req/context'
+
+module HrrRbSsh
+  class Connection
+    class Channel
+      module Session
+        request_type = 'pty-req'
+
+        class PtyReq
+          def self.run proc_chain, username, io, variables, message, options
+            logger = HrrRbSsh::Logger.new self.class.name
+
+            context = Context.new proc_chain, username, io, variables, message
+            handler = options.fetch('connection_channel_request_pty_req', RequestHandler.new {})
+            handler.run context
+
+            proc_chain.connect context.chain_proc
+          end
+        end
+
+        @@request_type_list ||= Hash.new
+        @@request_type_list[request_type] = PtyReq
+      end
+    end
+  end
+end

data/lib/hrr_rb_ssh/connection/channel/session/shell/context.rb

@@ -0,0 +1,37 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/logger'
+
+module HrrRbSsh
+  class Connection
+    class Channel
+      module Session
+        class Shell
+          class Context
+            attr_reader \
+              :logger,
+              :username,
+              :io,
+              :variables,
+              :vars
+
+            def initialize proc_chain, username, io, variables, message
+              @logger = HrrRbSsh::Logger.new self.class.name
+
+              @proc_chain = proc_chain
+              @username   = username
+              @io         = io
+              @variables  = variables
+              @vars       = variables
+            end
+
+            def chain_proc &block
+              @proc = block || @proc
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/hrr_rb_ssh/connection/channel/session/shell.rb

@@ -0,0 +1,31 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/logger'
+require 'hrr_rb_ssh/connection/request_handler'
+require 'hrr_rb_ssh/connection/channel/session/shell/context'
+
+module HrrRbSsh
+  class Connection
+    class Channel
+      module Session
+        request_type = 'shell'
+
+        class Shell
+          def self.run proc_chain, username, io, variables, message, options
+            logger = HrrRbSsh::Logger.new self.class.name
+
+            context = Context.new proc_chain, username, io, variables, message
+            handler = options.fetch('connection_channel_request_shell', RequestHandler.new {})
+            handler.run context
+
+            proc_chain.connect context.chain_proc
+          end
+        end
+
+        @@request_type_list ||= Hash.new
+        @@request_type_list[request_type] = Shell
+      end
+    end
+  end
+end

data/lib/hrr_rb_ssh/connection/channel/session/subsystem/context.rb

@@ -0,0 +1,40 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/logger'
+
+module HrrRbSsh
+  class Connection
+    class Channel
+      module Session
+        class Subsystem
+          class Context
+            attr_reader \
+              :logger,
+              :username,
+              :io,
+              :variables,
+              :vars,
+              :subsystem_name
+
+            def initialize proc_chain, username, io, variables, message
+              @logger = HrrRbSsh::Logger.new self.class.name
+
+              @proc_chain = proc_chain
+              @username   = username
+              @io         = io
+              @variables  = variables
+              @vars       = variables
+
+              @subsystem_name = message['subsystem name']
+            end
+
+            def chain_proc &block
+              @proc = block || @proc
+            end
+          end
+        end
+      end
+    end
+  end
+end