public/Socket_Programming
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Update app_secure_client.py and lib_secure_server.py minor updates

Browse Source
This commit is contained in:
George 2020-04-01 21:32:31 -07:00
parent 32d5945805
commit 98e28fb5f4
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
app_secure_client.py
View File

@ -106,8 +106,8 @@ class Message:
self.read_to_buffer(content_len) self.read_to_buffer(content_len)
data = self._recv_buffer[:content_len] data = self._recv_buffer[:content_len]
# decrypt data # decrypt data
iv = bytes(data[:16]) iv = bytes(data[:AES.block_size])
encrypted = bytes(data[16:]) encrypted = bytes(data[AES.block_size:])
key = self.sessionkey + self.sessionkey[::-1] key = self.sessionkey + self.sessionkey[::-1]
cipher = AES.new(key, AES.MODE_CFB, iv) cipher = AES.new(key, AES.MODE_CFB, iv)
data = cipher.decrypt(encrypted) data = cipher.decrypt(encrypted)

4
lib_secure_server.py
View File

@ -282,8 +282,8 @@ class Message:
self._set_selector_events_mask("r") self._set_selector_events_mask("r")
else: else:
print(f"Receiving request (encrypted)") print(f"Receiving request (encrypted)")
iv = data[:16] iv = data[:AES.block_size]
encrypted = data[16:] encrypted = data[AES.block_size:]
key = self.sessionkey + self.sessionkey[::-1] key = self.sessionkey + self.sessionkey[::-1]
cipher = AES.new(key, AES.MODE_CFB, iv) cipher = AES.new(key, AES.MODE_CFB, iv)
data = cipher.decrypt(encrypted) data = cipher.decrypt(encrypted)