class NetworkClipboard::AESConnection

Attributes

remote_client_id[R]
socket[R]

Public Class Methods

new(config,socket) click to toggle source
# File lib/network_clipboard/connection.rb, line 19
def initialize(config,socket)
  @socket = socket

  @encryptor = OpenSSL::Cipher::AES.new(128, :CBC)
  @encryptor.encrypt
  @encryptor.key = config.secret

  @decryptor = OpenSSL::Cipher::AES.new(128, :CBC)
  @decryptor.decrypt
  @decryptor.key = config.secret

  handshake(config.client_id)
end

Public Instance Methods

close_read() click to toggle source
# File lib/network_clipboard/connection.rb, line 68
def close_read
  @socket.close_read
end
close_write() click to toggle source
# File lib/network_clipboard/connection.rb, line 72
def close_write
  @socket.close_write
end
handshake(client_id) click to toggle source
# File lib/network_clipboard/connection.rb, line 33
def handshake(client_id)
  iv = @encryptor.random_iv
  @socket.send([iv.size].pack('N'),0)
  @socket.send(iv,0)
  iv_size = @socket.read(4).unpack('N')[0]
  @decryptor.iv = @socket.read(iv_size)

  # Verify it all worked.
  send(HANDSHAKE_STRING)
  raise HandshakeException unless receive == HANDSHAKE_STRING

  send(client_id)
  @remote_client_id = receive
end
receive() click to toggle source
# File lib/network_clipboard/connection.rb, line 55
def receive()
  size_bits = @socket.read(4)
  if size_bits.nil? and @socket.eof?
    raise DisconnectedError
  end
  ciphertext_size = size_bits.unpack('N')[0]
  ciphertext = @socket.read(ciphertext_size)
  plaintext = @decryptor.update(ciphertext) + @decryptor.final
  @decryptor.reset

  return plaintext
end
send(new_content) click to toggle source
# File lib/network_clipboard/connection.rb, line 48
def send(new_content)
  ciphertext = @encryptor.update(new_content) + @encryptor.final
  @socket.send([ciphertext.size].pack('N'),0)
  @socket.send(ciphertext,0)
  @encryptor.reset
end