loyal3-sentry 0.4.4 → 0.5.0

data/VERSION

@@ -1 +1 @@
-0.4.4
+0.5.0

data/lib/sentry/asymmetric_sentry.rb

@@ -21,7 +21,41 @@ module Sentry
     def encrypt(data)
       raise NoPublicKeyError unless public?
       rsa = public_rsa
-      rsa.public_encrypt(data)
+      return rsa.public_encrypt(data)
+    end
+
+    def decrypt_large_from_base64(data, key=nil)
+      padding_length = data[0]
+      chunk_length = public_rsa.max_encryptable_length + 11 # 11 is magic padding for RSA encoding
+      data = Base64.decode64(data[1,data.length])
+      return (0...data.length).step(chunk_length).inject("") { |accum, idx| accum + decrypt_with_padding(data.slice(idx, chunk_length), padding_length, key)}
+    end
+
+    def chunk_size(padding_length)
+      return public_rsa.max_encryptable_length - padding_length
+    end
+    
+    def encrypt_large_to_base64(data)
+      padding_length = 8
+      chunk_length = chunk_size(padding_length)
+      return padding_length.chr + Base64.encode64( (0...data.length).step(chunk_length).inject("") {|accum, idx| accum + encrypt_with_padding( data.slice(idx, chunk_length), padding_length)} )
+    end
+
+    def decrypt_with_padding(data, padding_length, key=nil)
+      decrypted = decrypt(data, key)
+      return decrypted[0, decrypted.length - padding_length]
+    end
+
+    def encrypt_with_padding(data, padding_length)
+      encrypt(data + rand_string(padding_length))
+    end
+
+    @@CHARS = ('a'..'z').to_a + ('A'..'Z').to_a + ('0'..'9').to_a
+
+    def rand_string(length=8)
+      s=''
+      length.times{ s << @@CHARS[rand(@@CHARS.length)] }
+      s
     end
   
     def encrypt_to_base64(data)
@@ -31,7 +65,7 @@ module Sentry
     def decrypt(data, key = nil)
       raise NoPrivateKeyError unless private?
       rsa = private_rsa(key)
-      rsa.private_decrypt(data)
+      return rsa.private_decrypt(data)
     end
   
     def decrypt_from_base64(data, key = nil)

data/lib/sentry.rb

@@ -45,3 +45,27 @@ begin
 rescue NameError
   nil
 end
+
+class OpenSSL::PKey::RSA
+  def max_encryptable_length
+    @max_encryption_length ||= calc_max_encrypted_length
+  end
+
+  private
+
+  def calc_max_encrypted_length
+    upper_bound = 4*1024
+    test_length = upper_bound / 2
+    while test_length != (upper_bound - 1)
+      probe = "a" * test_length
+      begin
+        self.public_encrypt(probe)
+        test_length = test_length + ((upper_bound - test_length) / 2)
+      rescue Exception => e
+        upper_bound = test_length
+        test_length = test_length / 2
+      end
+    end
+    return test_length
+  end
+end

data/sentry.gemspec

@@ -5,7 +5,7 @@
 
 Gem::Specification.new do |s|
   s.name = %q{sentry}
-  s.version = "0.4.4"
+  s.version = "0.5.0"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["John Pelly", "David Stevenson"]
@@ -43,6 +43,7 @@ Gem::Specification.new do |s|
      "test/keys/encrypted_public",
      "test/keys/private",
      "test/keys/public",
+     "test/rsa_key_test.rb",
      "test/schema.rb",
      "test/sha_sentry_test.rb",
      "test/symmetric_sentry_callback_test.rb",
@@ -59,6 +60,7 @@ Gem::Specification.new do |s|
      "test/asymmetric_sentry_callback_test.rb",
      "test/asymmetric_sentry_test.rb",
      "test/fixtures/user.rb",
+     "test/rsa_key_test.rb",
      "test/schema.rb",
      "test/sha_sentry_test.rb",
      "test/symmetric_sentry_callback_test.rb",

data/test/asymmetric_sentry_test.rb

@@ -9,19 +9,19 @@ class AsymmetricSentryTest < Test::Unit::TestCase
     @encrypted_public_key_file = File.dirname(__FILE__) + '/keys/encrypted_public'
     @encrypted_private_key_file = File.dirname(__FILE__) + '/keys/encrypted_private'
     @sentry = Sentry::AsymmetricSentry.new
-    
+
     @orig = 'sentry'
     @data = "vYfMxtVB8ezXmQKSNqTC9sPgi8TbsYRxWd7DVbpprzyuEdZ7gftJ/0IXsbXm\nXCU08bTAl0uEFm7dau+eJMXEJg==\n"
     @encrypted_data = "q2obYAITmK93ylzVS01mJx1jSlnmylMX15nFpb4uKesVgnqvtzBRHZ/SK+Nm\nEzceIoAcJc3DHosVa4VUE/aK/A==\n"
     Sentry::AsymmetricSentry.default_public_key_file = nil
     Sentry::AsymmetricSentry.default_private_key_file = nil
   end
-  
+
   def test_should_decrypt_files
     set_key_files @public_key_file, @private_key_file
     assert_equal @orig, @sentry.decrypt_from_base64(@data)
   end
-  
+
   def test_should_decrypt_files_with_encrypted_key
     set_key_files @encrypted_public_key_file, @encrypted_private_key_file
     assert_equal @orig, @sentry.decrypt_from_base64(@encrypted_data, @key)
@@ -32,7 +32,7 @@ class AsymmetricSentryTest < Test::Unit::TestCase
     assert !@sentry.private?
     set_key_files @public_key_file, @private_key_file
   end
-  
+
   def test_should_read_encrypted_key_files
     assert !@sentry.public?
     assert !@sentry.private?
@@ -43,17 +43,32 @@ class AsymmetricSentryTest < Test::Unit::TestCase
     set_default_key_files @public_key_file, @private_key_file
     assert_equal @orig, @sentry.decrypt_from_base64(@data)
   end
-  
+
   def test_should_decrypt_files_with_default_encrypted_key
     set_default_key_files @encrypted_public_key_file, @encrypted_private_key_file
     assert_equal @orig, @sentry.decrypt_from_base64(@encrypted_data, @key)
   end
 
+  def test_should_decrypt_block_by_block_for_large_data
+    set_default_key_files @encrypted_public_key_file, @encrypted_private_key_file
+    large_data = "asdf" * 2048
+    encrypted = @sentry.encrypt_large_to_base64(large_data)
+    assert_not_equal large_data, encrypted
+    assert_equal large_data, @sentry.decrypt_large_from_base64(encrypted, @key)
+  end
+
   def test_should_decrypt_files_with_default_key_using_class_method
     set_default_key_files @public_key_file, @private_key_file
     assert_equal @orig, Sentry::AsymmetricSentry.decrypt_from_base64(@data)
   end
-  
+
+  def test_foo
+    puts "^^^^^^^"
+    puts `ruby -v`
+    puts (1...20).step(3).inspect
+
+  end
+
   def test_should_decrypt_files_with_default_encrypted_key_using_class_method
     set_default_key_files @encrypted_public_key_file, @encrypted_private_key_file
     assert_equal @orig, Sentry::AsymmetricSentry.decrypt_from_base64(@encrypted_data, @key)
@@ -64,25 +79,26 @@ class AsymmetricSentryTest < Test::Unit::TestCase
     assert !@sentry.private?
     set_default_key_files @public_key_file, @private_key_file
   end
-  
+
   def test_should_read_encrypted_key_files_with_default_key
     assert !@sentry.public?
     assert !@sentry.private?
     set_default_key_files @encrypted_public_key_file, @encrypted_private_key_file
   end
 
-  private  
+  private
+
   def set_key_files(public_key, private_key)
     @sentry.public_key_file = public_key
     @sentry.private_key_file = private_key
     assert @sentry.private?
     assert @sentry.public?
   end
-  
+
   def set_default_key_files(public_key, private_key)
     Sentry::AsymmetricSentry.default_public_key_file = public_key
     Sentry::AsymmetricSentry.default_private_key_file = private_key
     assert @sentry.private?
     assert @sentry.public?
   end
-end
+end

data/test/rsa_key_test.rb

@@ -0,0 +1,11 @@
+require 'abstract_unit'
+
+class RsaKeyTest < Test::Unit::TestCase
+  def setup
+    @public_key = OpenSSL::PKey::RSA.new(File.open(File.dirname(__FILE__) + '/keys/encrypted_public') { |f| f.read })
+  end
+
+  def test_can_find_max_encoded_length_for_key
+    assert_equal 53, @public_key.max_encryptable_length
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: loyal3-sentry
 version: !ruby/object:Gem::Version 
-  version: 0.4.4
+  version: 0.5.0
 platform: ruby
 authors: 
 - John Pelly
@@ -50,6 +50,7 @@ files:
 - test/keys/encrypted_public
 - test/keys/private
 - test/keys/public
+- test/rsa_key_test.rb
 - test/schema.rb
 - test/sha_sentry_test.rb
 - test/symmetric_sentry_callback_test.rb
@@ -87,6 +88,7 @@ test_files:
 - test/asymmetric_sentry_callback_test.rb
 - test/asymmetric_sentry_test.rb
 - test/fixtures/user.rb
+- test/rsa_key_test.rb
 - test/schema.rb
 - test/sha_sentry_test.rb
 - test/symmetric_sentry_callback_test.rb