block-tea 1.3.1 → 1.3.2

data/lib/crypt/block_tea.rb

@@ -7,13 +7,21 @@ module Crypt
     end
 
 
-  	# encrypt: Use Corrected Block TEA to encrypt plaintext using password
-  	# Return encrypted text as string
-  	def encrypt(plaintext)
-  		if plaintext.length == 0
-  			return('')  # nothing to encrypt
-  		end
+    def encrypt(plaintext)
+      raise StandardError, "Nothing to encrypt" if plaintext.length == 0
+      encryption_algorithm(plaintext)
+    end
+
+
+    def decrypt(ciphertext)
+      raise StandardError, "Nothing to decrypt" if ciphertext.length == 0
+      decryption_algorithm(ciphertext)
+    end
+
 
+    private
+
+    def encryption_algorithm(plaintext)
   		# 'escape' plaintext so chars outside ISO-8859-1 work in single-byte packing, but  
   		# keep spaces as spaces (not '%20') so encrypted text doesn't grow too long, and 
   		# convert result to longs
@@ -44,18 +52,12 @@ module Crypt
   			z = v[n-1]
   		}
 
-  		ciphertext = longs_to_str(v)
-  		return ciphertext.unpack('a*').pack('m').delete("\n") # base64 encode it without newlines
-  	end
-
+      return base64_encode(v)
+    end
 
-  	# decrypt: Use Corrected Block TEA to decrypt ciphertext using password
-  	def decrypt(ciphertext)
-  		if ciphertext.length == 0
-  			return('')
-  		end
 
-  		v = str_to_longs(ciphertext.unpack('m').pack("a*"))	# base64 decode and convert to array of 'longs'
+    def decryption_algorithm(ciphertext)
+  		v = base64_decode(ciphertext)
   		k = str_to_longs(@password.ljust(16).slice(0,16))
   		n = v.length	
 
@@ -81,29 +83,43 @@ module Crypt
   		plaintext = longs_to_str(v)
   		# strip trailing null chars resulting from filling 4-char blocks:
   		plaintext = plaintext.gsub(/\0+$/,'')
+
   		return plaintext
-  	end
+    end
 
 
-    private
+    def base64_encode(v)
+      ciphertext = longs_to_str(v)
+      ciphertext.unpack('a*').pack('m').delete("\n") # base64 encode it without newlines
+    end
+
+
+    def base64_decode(ciphertext)
+      v = ciphertext.unpack('m').pack("a*")         # base64 decode and convert to array of 'longs'
+      str_to_longs(v)
+    end
 
 
     def mx(z, y, sum, p)
       ((z>>5 ^ ((y<<2)&0xffffffff)) + (y>>3 ^ ((z<<4)&0xffffffff)) ^ (sum^y) + (p ^ z)) & 0xffffffff
     end
 
-  	def self.str_to_longs(s, include_count = false)
-  		s << [0,0,0].pack('c*') # Pad with at most three nulls
-  		return s.unpack('L*')
-  	end
+
+    def self.str_to_longs(s, include_count = false)
+      s << [0,0,0].pack('c*') # Pad with at most three nulls
+      return s.unpack('L*')
+    end
+
 
     def str_to_longs(s, include_count = false) # :nodoc:
       self.class.str_to_longs s, include_count
     end
 
-  	def self.longs_to_str(l, include_count = false)   # convert array of longs back to string
-  		l.pack('L*')
-  	end
+
+    def self.longs_to_str(l, include_count = false)   # convert array of longs back to string
+      l.pack('L*')
+    end
+
 
     def longs_to_str(l, include_count = false) # :nodoc:
       self.class.longs_to_str l, include_count

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: block-tea
 version: !ruby/object:Gem::Version 
-  hash: 25
+  hash: 31
   prerelease: 
   segments: 
   - 1
   - 3
-  - 1
-  version: 1.3.1
+  - 2
+  version: 1.3.2
 platform: ruby
 authors: 
 - Mikhailov Anatoly
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-03-09 00:00:00 +06:00
+date: 2011-03-10 00:00:00 +06:00
 default_executable: block_tea
 dependencies: []