hrr_rb_ssh 0.1.1 → 0.1.2

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

@@ -0,0 +1,34 @@
+# 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/channel_type/session/request_type/request_type'
+
+module HrrRbSsh
+  class Connection
+    class Channel
+      module ChannelType
+        class Session
+          module RequestType
+            class Env < RequestType
+              NAME = '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
+          end
+        end
+      end
+    end
+  end
+end
+
+require 'hrr_rb_ssh/connection/channel/channel_type/session/request_type/env/context'

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

@@ -0,0 +1,44 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/logger'
+
+module HrrRbSsh
+  class Connection
+    class Channel
+      module ChannelType
+        class Session
+          module RequestType
+            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
+  end
+end

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

@@ -0,0 +1,34 @@
+# 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/channel_type/session/request_type/request_type'
+
+module HrrRbSsh
+  class Connection
+    class Channel
+      module ChannelType
+        class Session
+          module RequestType
+            class Exec < RequestType
+              NAME = '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
+          end
+        end
+      end
+    end
+  end
+end
+
+require 'hrr_rb_ssh/connection/channel/channel_type/session/request_type/exec/context'

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

@@ -0,0 +1,54 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/logger'
+
+module HrrRbSsh
+  class Connection
+    class Channel
+      module ChannelType
+        class Session
+          module RequestType
+            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
+  end
+end

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

@@ -0,0 +1,34 @@
+# 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/channel_type/session/request_type/request_type'
+
+module HrrRbSsh
+  class Connection
+    class Channel
+      module ChannelType
+        class Session
+          module RequestType
+            class PtyReq < RequestType
+              NAME = 'pty-req'
+
+              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
+          end
+        end
+      end
+    end
+  end
+end
+
+require 'hrr_rb_ssh/connection/channel/channel_type/session/request_type/pty_req/context'

data/lib/hrr_rb_ssh/connection/channel/channel_type/session/request_type/request_type.rb

@@ -0,0 +1,34 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+module HrrRbSsh
+  class Connection
+    class Channel
+      module ChannelType
+        class Session
+          module RequestType
+            class RequestType
+              @@list = Array.new
+
+              def self.inherited klass
+                @@list.push klass
+              end
+
+              def self.list
+                @@list
+              end
+
+              def self.name_list
+                @@list.map{ |klass| klass::NAME }
+              end
+
+              def self.[] key
+                @@list.find{ |klass| key == klass::NAME }
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/hrr_rb_ssh/connection/channel/channel_type/session/request_type/shell/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 ChannelType
+        class Session
+          module RequestType
+            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
+  end
+end

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

@@ -0,0 +1,34 @@
+# 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/channel_type/session/request_type/request_type'
+
+module HrrRbSsh
+  class Connection
+    class Channel
+      module ChannelType
+        class Session
+          module RequestType
+            class Shell < RequestType
+              NAME = '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
+          end
+        end
+      end
+    end
+  end
+end
+
+require 'hrr_rb_ssh/connection/channel/channel_type/session/request_type/shell/context'

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

@@ -0,0 +1,44 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/logger'
+
+module HrrRbSsh
+  class Connection
+    class Channel
+      module ChannelType
+        class Session
+          module RequestType
+            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
+  end
+end

data/lib/hrr_rb_ssh/connection/channel/channel_type/session/request_type/subsystem.rb

@@ -0,0 +1,34 @@
+# 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/channel_type/session/request_type/request_type'
+
+module HrrRbSsh
+  class Connection
+    class Channel
+      module ChannelType
+        class Session
+          module RequestType
+            class Subsystem < RequestType
+              NAME = 'subsystem'
+
+              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_subsystem', RequestHandler.new {})
+                handler.run context
+
+                proc_chain.connect context.chain_proc
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end
+
+require 'hrr_rb_ssh/connection/channel/channel_type/session/request_type/subsystem/context'

data/lib/hrr_rb_ssh/connection/channel/channel_type/session/request_type.rb

@@ -0,0 +1,33 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+module HrrRbSsh
+  class Connection
+    class Channel
+      module ChannelType
+        class Session
+          module RequestType
+            def self.list
+              RequestType.list
+            end
+
+            def self.name_list
+              RequestType.name_list
+            end
+
+            def self.[] key
+              RequestType[key]
+            end
+          end
+        end
+      end
+    end
+  end
+end
+
+require 'hrr_rb_ssh/connection/channel/channel_type/session/request_type/request_type'
+require 'hrr_rb_ssh/connection/channel/channel_type/session/request_type/pty_req'
+require 'hrr_rb_ssh/connection/channel/channel_type/session/request_type/env'
+require 'hrr_rb_ssh/connection/channel/channel_type/session/request_type/shell'
+require 'hrr_rb_ssh/connection/channel/channel_type/session/request_type/exec'
+require 'hrr_rb_ssh/connection/channel/channel_type/session/request_type/subsystem'

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

@@ -0,0 +1,30 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/connection/channel/channel_type/channel_type'
+
+module HrrRbSsh
+  class Connection
+    class Channel
+      module ChannelType
+        class Session < ChannelType
+          NAME = 'session'
+
+          def self.list
+            RequestType.list
+          end
+
+          def self.name_list
+            RequestType.name_list
+          end
+
+          def self.[] key
+            RequestType[key]
+          end
+        end
+      end
+    end
+  end
+end
+
+require 'hrr_rb_ssh/connection/channel/channel_type/session/request_type'

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

@@ -0,0 +1,25 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+module HrrRbSsh
+  class Connection
+    class Channel
+      module ChannelType
+        def self.list
+          ChannelType.list
+        end
+
+        def self.name_list
+          ChannelType.name_list
+        end
+
+        def self.[] key
+          ChannelType[key]
+        end
+      end
+    end
+  end
+end
+
+require 'hrr_rb_ssh/connection/channel/channel_type/channel_type'
+require 'hrr_rb_ssh/connection/channel/channel_type/session'

data/lib/hrr_rb_ssh/connection/channel.rb

@@ -4,21 +4,11 @@
 require 'socket'
 require 'hrr_rb_ssh/logger'
 require 'hrr_rb_ssh/connection/channel/proc_chain'
-require 'hrr_rb_ssh/connection/channel/session'
+require 'hrr_rb_ssh/connection/channel/channel_type'
 
 module HrrRbSsh
   class Connection
     class Channel
-      @@type_list ||= Hash.new
-
-      def self.[] key
-        @@type_list[key]
-      end
-
-      def self.type_list
-        @@type_list.keys
-      end
-
       INITIAL_WINDOW_SIZE = 100000
       MAXIMUM_PACKET_SIZE = 100000
 
@@ -219,7 +209,7 @@ module HrrRbSsh
 
       def request message, variables
         request_type = message['request type']
-        @@type_list[@channel_type][request_type].run @proc_chain, @connection.username, @request_handler_io, variables, message, @connection.options
+        ChannelType[@channel_type][request_type].run @proc_chain, @connection.username, @request_handler_io, variables, message, @connection.options
       end
 
       def send_channel_success

data/lib/hrr_rb_ssh/transport/compression_algorithm/compression_algorithm.rb

@@ -0,0 +1,34 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/logger'
+
+module HrrRbSsh
+  class Transport
+    class CompressionAlgorithm
+      class CompressionAlgorithm
+        @@list = Array.new
+
+        def self.inherited klass
+          @@list.push klass
+        end
+
+        def self.list
+          @@list
+        end
+
+        def self.name_list
+          @@list.map{ |klass| klass::NAME }
+        end
+
+        def self.[] key
+          @@list.find{ |klass| key == klass::NAME }
+        end
+
+        def initialize direction
+          @logger = HrrRbSsh::Logger.new self.class.name
+        end
+      end
+    end
+  end
+end

data/lib/hrr_rb_ssh/transport/compression_algorithm/functionable.rb

@@ -0,0 +1,31 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'zlib'
+
+module HrrRbSsh
+  class Transport
+    class CompressionAlgorithm
+      module Functionable
+        def initialize direction
+          super
+
+          case direction
+          when HrrRbSsh::Transport::Direction::OUTGOING
+            @deflator = ::Zlib::Deflate.new
+          when HrrRbSsh::Transport::Direction::INCOMING
+            @inflator = ::Zlib::Inflate.new
+          end
+        end
+
+        def deflate data
+          @deflator.deflate(data, ::Zlib::SYNC_FLUSH)
+        end
+
+        def inflate data
+          @inflator.inflate(data)
+        end
+      end
+    end
+  end
+end

data/lib/hrr_rb_ssh/transport/compression_algorithm/none.rb

@@ -1,32 +1,20 @@
 # coding: utf-8
 # vim: et ts=2 sw=2
 
-require 'hrr_rb_ssh/logger'
+require 'hrr_rb_ssh/transport/compression_algorithm/compression_algorithm'
+require 'hrr_rb_ssh/transport/compression_algorithm/unfunctionable'
 
 module HrrRbSsh
   class Transport
     class CompressionAlgorithm
-      name_list = [
-        'none'
-      ]
+      class None < CompressionAlgorithm
+        NAME = 'none'
 
-      class None
-        def initialize
-          @logger = HrrRbSsh::Logger.new self.class.name
+        def initialize direction=nil
+          super
         end
 
-        def deflate data
-          data
-        end
-
-        def inflate data
-          data
-        end
-      end
-
-      @@list ||= Hash.new
-      name_list.each do |name|
-        @@list[name] = None
+        include Unfunctionable
       end
     end
   end

data/lib/hrr_rb_ssh/transport/compression_algorithm/unfunctionable.rb

@@ -0,0 +1,20 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/logger'
+
+module HrrRbSsh
+  class Transport
+    class CompressionAlgorithm
+      module Unfunctionable
+        def deflate data
+          data
+        end
+
+        def inflate data
+          data
+        end
+      end
+    end
+  end
+end