Module: Crypt::CBC
Constant Summary collapse
- ULONG =
0x100000000
Instance Method Summary collapse
- #carefully_open_file(filename, mode) ⇒ Object
- #decrypt_file(crypt_filename, plain_filename) ⇒ Object
- #decrypt_stream(crypt_stream, plain_stream) ⇒ Object
- #decrypt_string(crypt_text) ⇒ Object
- #encrypt_file(plain_filename, crypt_filename) ⇒ Object
- #encrypt_stream(plain_stream, crypt_stream) ⇒ Object
- #encrypt_string(plain_text) ⇒ Object
- #generate_initialization_vector(words) ⇒ Object
Instance Method Details
#carefully_open_file(filename, mode) ⇒ Object
71 72 73 74 75 76 77 78 79 80 |
# File 'lib/crypt/cbc.rb', line 71 def carefully_open_file(filename, mode) begin a_file = File.new(filename, mode) rescue puts "Sorry. There was a problem opening the file <#{filename}>." a_file.close() unless a_file.nil? raise end return(a_file) end |
#decrypt_file(crypt_filename, plain_filename) ⇒ Object
92 93 94 95 96 97 98 |
# File 'lib/crypt/cbc.rb', line 92 def decrypt_file(crypt_filename, plain_filename) crypt_file = carefully_open_file(crypt_filename, 'rb') plain_file = carefully_open_file(plain_filename, 'wb+') decrypt_stream(crypt_file, plain_file) crypt_file.close unless crypt_file.closed? plain_file.close unless plain_file.closed? end |
#decrypt_stream(crypt_stream, plain_stream) ⇒ Object
53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
# File 'lib/crypt/cbc.rb', line 53 def decrypt_stream(crypt_stream, plain_stream) # Cypher-block-chain mode chain = crypt_stream.read(block_size()) while (block = crypt_stream.read(block_size())) decrypted = decrypt_block(block) plain_text = decrypted ^ chain plain_stream.write(plain_text) unless crypt_stream.eof? chain = block end # write the final block, omitting the padding buffer = plain_text.split('') = buffer.last.unpack('C').first .times { plain_stream.write(buffer.shift) } end |
#decrypt_string(crypt_text) ⇒ Object
110 111 112 113 114 115 116 |
# File 'lib/crypt/cbc.rb', line 110 def decrypt_string(crypt_text) crypt_stream = StringIO.new(crypt_text) plain_stream = StringIO.new('') decrypt_stream(crypt_stream, plain_stream) plain_text = plain_stream.string return(plain_text) end |
#encrypt_file(plain_filename, crypt_filename) ⇒ Object
83 84 85 86 87 88 89 |
# File 'lib/crypt/cbc.rb', line 83 def encrypt_file(plain_filename, crypt_filename) plain_file = carefully_open_file(plain_filename, 'rb') crypt_file = carefully_open_file(crypt_filename, 'wb+') encrypt_stream(plain_file, crypt_file) plain_file.close unless plain_file.closed? crypt_file.close unless crypt_file.closed? end |
#encrypt_stream(plain_stream, crypt_stream) ⇒ Object
24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
# File 'lib/crypt/cbc.rb', line 24 def encrypt_stream(plain_stream, crypt_stream) # Cypher-block-chain mode init_vector = generate_initialization_vector(block_size() / 4) chain = encrypt_block(init_vector) crypt_stream.write(chain) while ((block = plain_stream.read(block_size())) && (block.length == block_size())) block = block ^ chain encrypted = encrypt_block(block) crypt_stream.write(encrypted) chain = encrypted end # write the final block # At most block_size()-1 bytes can be part of the message. # That means the final byte can be used to store the number of meaningful # bytes in the final block block = '' if block.nil? buffer = block.split('') = buffer.length buffer << .chr * (block_size() - ) block = buffer.join('') block = block ^ chain encrypted = encrypt_block(block) crypt_stream.write(encrypted) end |
#encrypt_string(plain_text) ⇒ Object
101 102 103 104 105 106 107 |
# File 'lib/crypt/cbc.rb', line 101 def encrypt_string(plain_text) plain_stream = StringIO.new(plain_text) crypt_stream = StringIO.new('') encrypt_stream(plain_stream, crypt_stream) crypt_text = crypt_stream.string return(crypt_text) end |
#generate_initialization_vector(words) ⇒ Object
14 15 16 17 18 19 20 21 |
# File 'lib/crypt/cbc.rb', line 14 def generate_initialization_vector(words) srand(Time.now.to_i) vector = "" words.times { vector << [rand(ULONG)].pack('N') } return(vector) end |