Class: EncryptDecrypt::EncryptedServer

Inherits:
Object
  • Object
show all
Defined in:
lib/accu-encrypt.rb

Overview

A class which allows for the creation of an encrypted session over a server, with the client acting as the server.

Codes: 0 - Connection refused, 1 - Connection accepted, 2 - Ready, 3 - Sending encrypted message, 4 - End communications, 5 - Verification message inbound

Instance Method Summary collapse

Constructor Details

#initialize(server) ⇒ EncryptedServer

Initializes EncryptedServer object.



53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
 
# File 'lib/accu-encrypt.rb', line 53

def initialize( server )
  @server = server
  print "Connection attempt from |" + @server.peeraddr[3] + "|.  Would you like to connect? "
  response = gets.chomp.downcase
  if response == "y" then
    puts "Connection accepted."
    send_code(1)
    @password = EncryptDecrypt.get_password "Please enter password for communications: "
    @listen_thread, @interactive_thread = nil, nil
    @mode = EncryptDecrypt.get_mode
    if listen_code() == 2 then
      send_code(2)
      puts "Ready code received.\nSuccessfully connected!"
      listen_encrypted()
      interactive()
      @listen_thread.join
      @interactive_thread.join
      puts "Ending connection."
    else
      puts "Connection refused."
    end
  else
    send_code 0
    puts "Connection refused."
  end
  server.close
end

Instance Method Details

#interactive ⇒ Object

Opens an interactive session.



162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
 
# File 'lib/accu-encrypt.rb', line 162

def interactive()
  @interactive_thread = Thread.new {
    loop {
      print "What would you like to do? |S Q| "
      response = gets.chomp.downcase
      while response != "q" and response != "s" do
        puts "Incorrect input: |" + response + "|"
        print "What would you like to do? |S Q| "
        response = gets.chomp.downcase
      end
      if response == "q" then
        puts "Sending close message to client."
        send_code(4)
        @listen_thread.exit
        @interactive_thread.exit
      else
        print "Type message and end with END:\n\n"
        lines = ""
        while line = gets and line.chomp != "END" do
          #puts "|" + line + "|"
          lines << line
        end
        lines = lines.chomp
        puts "Message read.\nEncrypting now."
        send_encrypted(lines)
        puts "Message sent."
      end
    }
  }
end

#listen {|msg| ... } ⇒ Object

Listens for a generic network communication and yields to a passed block (if any).

Yields:

  • (msg) 


104
105
106
107
 
# File 'lib/accu-encrypt.rb', line 104

def listen()
  msg = @server.gets.chomp
  yield msg
end

#listen_code ⇒ Object

Listens for a network code.



121
122
123
124
125
126
127
128
129
130
131
 
# File 'lib/accu-encrypt.rb', line 121

def listen_code()
  listen { |msg|
    if msg.slice(0..1) == "C:" then
      #puts "Code present."
      return msg.slice(2..msg.length).to_i
    else
      #puts "Code not present."
      return msg.slice(0..1)
    end
  }
end

#listen_encrypted ⇒ Object

Listens for encrypted messages.



82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
 
# File 'lib/accu-encrypt.rb', line 82

def listen_encrypted()
  @listen_thread = Thread.new {
    loop {
      code = listen_code()
      if code == 3 then
        lines = ""
        while line = @server.gets and line != "END\n" do
          lines << line
        end
        secret,text = EncryptDecrypt.decrypt(@password,lines,@mode)
        puts "Message received:\n\n" + text.chomp + "\n\nMessage end."
      elsif code == 4 then
        puts "Client closed connection."
        @interactive_thread.exit
        @listen_thread.exit
      end
    }
  }
end

#listen_lines ⇒ Object

Listens for multiple lines, appends them together and stops at END.



111
112
113
114
115
116
117
118
 
# File 'lib/accu-encrypt.rb', line 111

def listen_lines()
  lines = []
  while line = @server.gets.chomp and line != "END" do
    #puts "|" + line + "|"
    lines << line
  end
  lines
end

#send(text) ⇒ Object

Sends a plaintext message.



134
135
136
 
# File 'lib/accu-encrypt.rb', line 134

def send(text)
  @server.puts text
end

#send_code(code) ⇒ Object

Sends a code.



157
158
159
 
# File 'lib/accu-encrypt.rb', line 157

def send_code(code)
  send("C:#{code}")
end

#send_encrypted(lines) ⇒ Object

Send over an encrypted multiline message.



149
150
151
152
153
154
 
# File 'lib/accu-encrypt.rb', line 149

def send_encrypted(lines)
  secret,text = EncryptDecrypt.encrypt(@password,lines,@mode)
  send_code(3)
  send(text)
  send("END")
end

#send_lines(lines) ⇒ Object

Sends multiple lines and appends END to end the send.



140
141
142
143
144
145
146
 
# File 'lib/accu-encrypt.rb', line 140

def send_lines(lines)
  array = lines.split("\n")
  array.each do |line|
    send(line)
  end
  send("END")
end