hrr_rb_ssh 0.1.1 → 0.1.2

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

@@ -1,37 +1,20 @@
 # coding: utf-8
 # vim: et ts=2 sw=2
 
-require 'zlib'
-
-require 'hrr_rb_ssh/logger'
+require 'hrr_rb_ssh/transport/compression_algorithm/compression_algorithm'
+require 'hrr_rb_ssh/transport/compression_algorithm/functionable'
 
 module HrrRbSsh
   class Transport
     class CompressionAlgorithm
-      name_list = [
-        'zlib'
-      ]
-
-      class Zlib
-        def initialize
-          @logger = HrrRbSsh::Logger.new self.class.name
-
-          @deflator = ::Zlib::Deflate.new
-          @inflator = ::Zlib::Inflate.new
-        end
+      class Zlib < CompressionAlgorithm
+        NAME = 'zlib'
 
-        def deflate data
-          @deflator.deflate(data, ::Zlib::SYNC_FLUSH)
+        def initialize direction
+          super
         end
 
-        def inflate data
-          @inflator.inflate(data)
-        end
-      end
-
-      @@list ||= Hash.new
-      name_list.each do |name|
-        @@list[name] = Zlib
+        include Functionable
       end
     end
   end

data/lib/hrr_rb_ssh/transport/compression_algorithm.rb

@@ -1,22 +1,24 @@
 # coding: utf-8
 # vim: et ts=2 sw=2
 
-require 'hrr_rb_ssh/logger'
-require 'hrr_rb_ssh/transport/compression_algorithm/none'
-require 'hrr_rb_ssh/transport/compression_algorithm/zlib'
-
 module HrrRbSsh
   class Transport
     class CompressionAlgorithm
-      @@list ||= Hash.new
-
-      def self.[] key
-        @@list[key]
+      def self.list
+        CompressionAlgorithm.list
       end
 
       def self.name_list
-        @@list.keys
+        CompressionAlgorithm.name_list
+      end
+
+      def self.[] key
+        CompressionAlgorithm[key]
       end
     end
   end
 end
+
+require 'hrr_rb_ssh/transport/compression_algorithm/compression_algorithm'
+require 'hrr_rb_ssh/transport/compression_algorithm/none'
+require 'hrr_rb_ssh/transport/compression_algorithm/zlib'

data/lib/hrr_rb_ssh/transport/direction.rb

@@ -0,0 +1,11 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+module HrrRbSsh
+  class Transport
+    module Direction
+      OUTGOING = :outgoing
+      INCOMING = :incoming
+    end
+  end
+end

data/lib/hrr_rb_ssh/transport/encryption_algorithm/aes128_cbc.rb

@@ -0,0 +1,19 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/transport/encryption_algorithm/encryption_algorithm'
+require 'hrr_rb_ssh/transport/encryption_algorithm/functionable'
+
+module HrrRbSsh
+  class Transport
+    class EncryptionAlgorithm
+      class Aes128Cbc < EncryptionAlgorithm
+        NAME        = 'aes128-cbc'
+        CIPHER_NAME = "AES-128-CBC"
+        BLOCK_SIZE  = 16
+
+        include Functionable
+      end
+    end
+  end
+end

data/lib/hrr_rb_ssh/transport/encryption_algorithm/aes128_ctr.rb

@@ -0,0 +1,19 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/transport/encryption_algorithm/encryption_algorithm'
+require 'hrr_rb_ssh/transport/encryption_algorithm/functionable'
+
+module HrrRbSsh
+  class Transport
+    class EncryptionAlgorithm
+      class Aes128Ctr < EncryptionAlgorithm
+        NAME        = 'aes128-ctr'
+        CIPHER_NAME = "AES-128-CTR"
+        BLOCK_SIZE  = 16
+
+        include Functionable
+      end
+    end
+  end
+end

data/lib/hrr_rb_ssh/transport/encryption_algorithm/aes192_cbc.rb

@@ -0,0 +1,19 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/transport/encryption_algorithm/encryption_algorithm'
+require 'hrr_rb_ssh/transport/encryption_algorithm/functionable'
+
+module HrrRbSsh
+  class Transport
+    class EncryptionAlgorithm
+      class Aes192Cbc < EncryptionAlgorithm
+        NAME        = 'aes192-cbc'
+        CIPHER_NAME = "AES-192-CBC"
+        BLOCK_SIZE  = 16
+
+        include Functionable
+      end
+    end
+  end
+end

data/lib/hrr_rb_ssh/transport/encryption_algorithm/aes192_ctr.rb

@@ -0,0 +1,19 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/transport/encryption_algorithm/encryption_algorithm'
+require 'hrr_rb_ssh/transport/encryption_algorithm/functionable'
+
+module HrrRbSsh
+  class Transport
+    class EncryptionAlgorithm
+      class Aes192Ctr < EncryptionAlgorithm
+        NAME        = 'aes192-ctr'
+        CIPHER_NAME = "AES-192-CTR"
+        BLOCK_SIZE  = 16
+
+        include Functionable
+      end
+    end
+  end
+end

data/lib/hrr_rb_ssh/transport/encryption_algorithm/aes256_cbc.rb

@@ -0,0 +1,19 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/transport/encryption_algorithm/encryption_algorithm'
+require 'hrr_rb_ssh/transport/encryption_algorithm/functionable'
+
+module HrrRbSsh
+  class Transport
+    class EncryptionAlgorithm
+      class Aes256Cbc < EncryptionAlgorithm
+        NAME        = 'aes256-cbc'
+        CIPHER_NAME = "AES-256-CBC"
+        BLOCK_SIZE  = 16
+
+        include Functionable
+      end
+    end
+  end
+end

data/lib/hrr_rb_ssh/transport/encryption_algorithm/aes256_ctr.rb

@@ -0,0 +1,19 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/transport/encryption_algorithm/encryption_algorithm'
+require 'hrr_rb_ssh/transport/encryption_algorithm/functionable'
+
+module HrrRbSsh
+  class Transport
+    class EncryptionAlgorithm
+      class Aes256Ctr < EncryptionAlgorithm
+        NAME        = 'aes256-ctr'
+        CIPHER_NAME = "AES-256-CTR"
+        BLOCK_SIZE  = 16
+
+        include Functionable
+      end
+    end
+  end
+end

data/lib/hrr_rb_ssh/transport/encryption_algorithm/arcfour.rb

@@ -0,0 +1,19 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/transport/encryption_algorithm/encryption_algorithm'
+require 'hrr_rb_ssh/transport/encryption_algorithm/functionable'
+
+module HrrRbSsh
+  class Transport
+    class EncryptionAlgorithm
+      class Arcfour < EncryptionAlgorithm
+        NAME        = 'arcfour'
+        CIPHER_NAME = "RC4"
+        BLOCK_SIZE  = 8
+
+        include Functionable
+      end
+    end
+  end
+end

data/lib/hrr_rb_ssh/transport/encryption_algorithm/blowfish_cbc.rb

@@ -0,0 +1,19 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/transport/encryption_algorithm/encryption_algorithm'
+require 'hrr_rb_ssh/transport/encryption_algorithm/functionable'
+
+module HrrRbSsh
+  class Transport
+    class EncryptionAlgorithm
+      class BlowfishCbc < EncryptionAlgorithm
+        NAME        = 'blowfish-cbc'
+        CIPHER_NAME = "BF-CBC"
+        BLOCK_SIZE  = 8
+
+        include Functionable
+      end
+    end
+  end
+end

data/lib/hrr_rb_ssh/transport/encryption_algorithm/cast128_cbc.rb

@@ -0,0 +1,19 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/transport/encryption_algorithm/encryption_algorithm'
+require 'hrr_rb_ssh/transport/encryption_algorithm/functionable'
+
+module HrrRbSsh
+  class Transport
+    class EncryptionAlgorithm
+      class Cast128Cbc < EncryptionAlgorithm
+        NAME        = 'cast128-cbc'
+        CIPHER_NAME = "CAST5-CBC"
+        BLOCK_SIZE  = 8
+
+        include Functionable
+      end
+    end
+  end
+end

data/lib/hrr_rb_ssh/transport/encryption_algorithm/encryption_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 EncryptionAlgorithm
+      class EncryptionAlgorithm
+        @@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, iv, key
+          @logger = HrrRbSsh::Logger.new self.class.name
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,61 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/logger'
+
+module HrrRbSsh
+  class Transport
+    class EncryptionAlgorithm
+      module Functionable
+        def self.included klass
+          cipher = OpenSSL::Cipher.new(klass::CIPHER_NAME)
+          klass.const_set(:IV_LENGTH,  cipher.iv_len)
+          klass.const_set(:KEY_LENGTH, cipher.key_len)
+        end
+
+        def initialize direction, iv, key
+          super
+
+          @cipher = OpenSSL::Cipher.new(self.class::CIPHER_NAME)
+          case direction
+          when HrrRbSsh::Transport::Direction::OUTGOING
+            @cipher.encrypt
+          when HrrRbSsh::Transport::Direction::INCOMING
+            @cipher.decrypt
+          end
+          @cipher.padding = 0
+          @cipher.iv  = iv
+          @cipher.key = key
+        end
+
+        def block_size
+          self.class::BLOCK_SIZE
+        end
+
+        def iv_length
+          self.class::IV_LENGTH
+        end
+
+        def key_length
+          self.class::KEY_LENGTH
+        end
+
+        def encrypt data
+          if data.empty?
+            data
+          else
+            @cipher.update(data) + @cipher.final
+          end
+        end
+
+        def decrypt data
+          if data.empty?
+            data
+          else
+            @cipher.update(data) + @cipher.final
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -2,47 +2,20 @@
 # vim: et ts=2 sw=2
 
 require 'hrr_rb_ssh/logger'
+require 'hrr_rb_ssh/transport/encryption_algorithm/unfunctionable'
 
 module HrrRbSsh
   class Transport
     class EncryptionAlgorithm
-      name_list = [
-        'none'
-      ]
-
-      class None
+      class None < EncryptionAlgorithm
+        NAME        = 'none'
         BLOCK_SIZE  = 0
-        IV_LENGTH   = 0
-        KEY_LENGTH  = 0
-
-        def initialize iv=nil, key=nil
-          @logger = HrrRbSsh::Logger.new self.class.name
-        end
-
-        def block_size
-          BLOCK_SIZE
-        end
-
-        def iv_length
-          IV_LENGTH
-        end
-
-        def key_length
-          KEY_LENGTH
-        end
 
-        def encrypt data
-          data
+        def initialize direction=nil, iv=nil, key=nil
+          super
         end
 
-        def decrypt 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/encryption_algorithm/three_des_cbc.rb

@@ -0,0 +1,19 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+require 'hrr_rb_ssh/transport/encryption_algorithm/encryption_algorithm'
+require 'hrr_rb_ssh/transport/encryption_algorithm/functionable'
+
+module HrrRbSsh
+  class Transport
+    class EncryptionAlgorithm
+      class ThreeDesCbc < EncryptionAlgorithm
+        NAME        = '3des-cbc'
+        CIPHER_NAME = "DES-EDE3-CBC"
+        BLOCK_SIZE  = 8
+
+        include Functionable
+      end
+    end
+  end
+end

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

@@ -0,0 +1,35 @@
+# coding: utf-8
+# vim: et ts=2 sw=2
+
+module HrrRbSsh
+  class Transport
+    class EncryptionAlgorithm
+      module Unfunctionable
+        def self.included klass
+          klass.const_set(:IV_LENGTH,  0)
+          klass.const_set(:KEY_LENGTH, 0)
+        end
+
+        def block_size
+          self.class::BLOCK_SIZE
+        end
+
+        def iv_length
+          self.class::IV_LENGTH
+        end
+
+        def key_length
+          self.class::KEY_LENGTH
+        end
+
+        def encrypt data
+          data
+        end
+
+        def decrypt data
+          data
+        end
+      end
+    end
+  end
+end

data/lib/hrr_rb_ssh/transport/encryption_algorithm.rb

@@ -1,22 +1,33 @@
 # coding: utf-8
 # vim: et ts=2 sw=2
 
-require 'hrr_rb_ssh/logger'
-require 'hrr_rb_ssh/transport/encryption_algorithm/none'
-require 'hrr_rb_ssh/transport/encryption_algorithm/aes_128_cbc'
-
 module HrrRbSsh
   class Transport
     class EncryptionAlgorithm
-      @@list ||= Hash.new
-
-      def self.[] key
-        @@list[key]
+      def self.list
+        EncryptionAlgorithm.list
       end
 
       def self.name_list
-        @@list.keys
+        EncryptionAlgorithm.name_list
+      end
+
+      def self.[] key
+        EncryptionAlgorithm[key]
       end
     end
   end
 end
+
+require 'hrr_rb_ssh/transport/encryption_algorithm/encryption_algorithm'
+require 'hrr_rb_ssh/transport/encryption_algorithm/none'
+require 'hrr_rb_ssh/transport/encryption_algorithm/three_des_cbc'
+require 'hrr_rb_ssh/transport/encryption_algorithm/blowfish_cbc'
+require 'hrr_rb_ssh/transport/encryption_algorithm/aes128_cbc'
+require 'hrr_rb_ssh/transport/encryption_algorithm/aes192_cbc'
+require 'hrr_rb_ssh/transport/encryption_algorithm/aes256_cbc'
+require 'hrr_rb_ssh/transport/encryption_algorithm/arcfour'
+require 'hrr_rb_ssh/transport/encryption_algorithm/cast128_cbc'
+require 'hrr_rb_ssh/transport/encryption_algorithm/aes128_ctr'
+require 'hrr_rb_ssh/transport/encryption_algorithm/aes192_ctr'
+require 'hrr_rb_ssh/transport/encryption_algorithm/aes256_ctr'

data/lib/hrr_rb_ssh/transport/kex_algorithm/diffie_hellman.rb

@@ -1,13 +1,13 @@
 # coding: utf-8
 # vim: et ts=2 sw=2
 
-require 'hrr_rb_ssh/logger'
+require 'openssl'
 require 'hrr_rb_ssh/transport/data_type'
 
 module HrrRbSsh
   class Transport
     class KexAlgorithm
-      class DiffieHellman
+      module DiffieHellman
         H0_DEFINITION = [
           ['string', 'V_C'],
           ['string', 'V_S'],
@@ -20,7 +20,7 @@ module HrrRbSsh
         ]
 
         def initialize
-          @logger = HrrRbSsh::Logger.new self.class.name
+          super
 
           @dh = OpenSSL::PKey::DH.new
           if @dh.respond_to?(:set_pqg)
@@ -122,7 +122,6 @@ module HrrRbSsh
           key_length = HrrRbSsh::Transport::MacAlgorithm[mac_algorithm_s_to_c_name]::KEY_LENGTH
           build_key(shared_secret, hash(transport), 'F'.ord, transport.session_id, key_length)
         end
-
       end
     end
   end

data/lib/hrr_rb_ssh/transport/kex_algorithm/diffie_hellman_group14_sha1.rb

@@ -1,16 +1,15 @@
 # coding: utf-8
 # vim: et ts=2 sw=2
 
+require 'hrr_rb_ssh/transport/kex_algorithm/kex_algorithm'
 require 'hrr_rb_ssh/transport/kex_algorithm/diffie_hellman'
 
 module HrrRbSsh
   class Transport
     class KexAlgorithm
-      name_list = [
-        'diffie-hellman-group14-sha1'
-      ]
+      class DiffieHellmanGroup14Sha1 < KexAlgorithm
+        NAME = 'diffie-hellman-group14-sha1'
 
-      class DiffieHellmanGroup14Sha1 < DiffieHellman
         P = \
           "FFFFFFFF" "FFFFFFFF" "C90FDAA2" "2168C234" \
           "C4C6628B" "80DC1CD1" "29024E08" "8A67CC74" \
@@ -31,11 +30,8 @@ module HrrRbSsh
         G = 2
 
         DIGEST = 'sha1'
-      end
 
-      @@list ||= Hash.new
-      name_list.each do |name|
-        @@list[name] = DiffieHellmanGroup14Sha1
+        include DiffieHellman
       end
     end
   end

data/lib/hrr_rb_ssh/transport/kex_algorithm/diffie_hellman_group1_sha1.rb

@@ -1,16 +1,15 @@
 # coding: utf-8
 # vim: et ts=2 sw=2
 
+require 'hrr_rb_ssh/transport/kex_algorithm/kex_algorithm'
 require 'hrr_rb_ssh/transport/kex_algorithm/diffie_hellman'
 
 module HrrRbSsh
   class Transport
     class KexAlgorithm
-      name_list = [
-        'diffie-hellman-group1-sha1'
-      ]
+      class DiffieHellmanGroup1Sha1 < KexAlgorithm
+        NAME = 'diffie-hellman-group1-sha1'
 
-      class DiffieHellmanGroup1Sha1 < DiffieHellman
         P = \
           "FFFFFFFF" "FFFFFFFF" "C90FDAA2" "2168C234" \
           "C4C6628B" "80DC1CD1" "29024E08" "8A67CC74" \
@@ -23,11 +22,8 @@ module HrrRbSsh
         G = 2
 
         DIGEST = 'sha1'
-      end
 
-      @@list ||= Hash.new
-      name_list.each do |name|
-        @@list[name] = DiffieHellmanGroup1Sha1
+        include DiffieHellman
       end
     end
   end

data/lib/hrr_rb_ssh/transport/kex_algorithm/kex_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 KexAlgorithm
+      class KexAlgorithm
+        @@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
+          @logger = HrrRbSsh::Logger.new self.class.name
+        end
+      end
+    end
+  end
+end