Contents

# TODO that file isn't refactored yet!
module Pubnub
  class Crypto
    def initialize(cipher_key)
      Pubnub.logger.debug(:pubnub){'Initializing Crypto'}
      @alg = 'AES-256-CBC'
      sha256_key = Digest::SHA256.hexdigest(cipher_key)
      @key = sha256_key.slice(0,32)

      @iv = '0123456789012345'
      Pubnub.logger.debug(:pubnub){'Initialized Crypto'}
    end


    def encrypt(message)

      aes = OpenSSL::Cipher::Cipher.new(@alg)
      aes.encrypt
      aes.key = @key
      aes.iv = @iv

      json_message = message.to_json
      cipher = aes.update(json_message)
      cipher << aes.final

      Base64.strict_encode64(cipher)

    end

    def decrypt(cipher_text)
      Pubnub.logger.debug(:pubnub){"Decrypting: #{cipher_text}"}
      decode_cipher = OpenSSL::Cipher::Cipher.new(@alg)
      decode_cipher.decrypt
      decode_cipher.key = @key
      decode_cipher.iv = @iv

      plain_text = ''

      begin
        undecoded_text = Base64.decode64(cipher_text)
        plain_text = decode_cipher.update(undecoded_text)
        plain_text << decode_cipher.final
      rescue => e
        Pubnub.logger.error(:pubnub){'DECRYPTION ERROR'}
        plain_text = '"DECRYPTION ERROR"'
      end

      begin
        JSON.load(plain_text)
      rescue JSON::ParserError
        JSON.load("[#{plain_text}]")[0] # srsly ruby? srsly?
      end

      Pubnub.logger.debug(:pubnub){'Finished decrypting'}

      plain_text

    end
  end
end

Version data entries

7 entries across 7 versions & 1 rubygems

Version	Path
pubnub-3.7.12	lib/pubnub/crypto.rb
pubnub-3.7.11	lib/pubnub/crypto.rb
pubnub-3.7.10	lib/pubnub/crypto.rb
pubnub-3.7.9	lib/pubnub/crypto.rb
pubnub-3.7.7	lib/pubnub/crypto.rb
pubnub-3.7.5	lib/pubnub/crypto.rb
pubnub-3.7.1	lib/pubnub/crypto.rb