Use shorter keys in test mode, as they are faster.

tomlea · Mar 5, 2012 · 73e175d · 73e175d
1 parent bd1fe22
commit 73e175d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/lib/cryptolocker/rsa.rb b/lib/cryptolocker/rsa.rb
@@ -1,7 +1,9 @@
 require "openssl"
 class Cryptolocker::RSA
   KEY_LEN = 4096
-  KEY_BYTE_LENGTH = KEY_LEN/8
+  def key_byte_length
+    KEY_LEN/8
+  end
 
   def self.generate_keys
     new_key = OpenSSL::PKey::RSA.generate( KEY_LEN )
@@ -28,8 +30,8 @@ def encrypt(value)
   end
 
   def decrypt(value)
-    encrypted_key = value[0...KEY_BYTE_LENGTH]
-    data = value[KEY_BYTE_LENGTH..-1]
+    encrypted_key = value[0...key_byte_length]
+    data = value[key_byte_length..-1]
     one_time_key = @key.private_decrypt(encrypted_key)
     Cryptolocker::AES.decrypt(data, one_time_key)
   end

diff --git a/spec/spec_helper.rb b/spec/spec_helper.rb
@@ -3,4 +3,5 @@
 Bundler.require(:default, :test)
 
 require "cryptolocker"
-
+Cryptolocker::RSA.send(:remove_const, :KEY_LEN)
+Cryptolocker::RSA::KEY_LEN = 1024