Повреждение данных при передаче файлов aes python tkinter - PullRequest
Новая
       15

Повреждение данных при передаче файлов aes python tkinter

0 голосов
/ 16 февраля 2019

Я работаю над проектом, в котором вы можете отправить файл через сокет, и он зашифрован.И он работает, он шифрует данные и отправляет их через сокет, и он прекрасно его расшифровывает, но изображение становится испорченным ниже, это пример изображения before и после того, как я могу исправить это?

Before: enter image description here

После: enter image description here

Шифрование и дешифрование находятся в отдельном модуле, который я создал, поэтому они ссылаются так, как ониниже

Код шифрования и дешифрования: 

def encrypt(key, data, iv):
    encryptor = AES.new(key, AES.MODE_CBC, iv)
    return encryptor.encrypt(data)

def decrypt(key, data, iv):
    decryptor = AES.new(key, AES.MODE_CBC, iv)
    return decryptor.decrypt(data)

Код сервера: 

s = socket.socket()
s.bind(("127.0.0.1", 5000))
s.listen(1)
c, addr = self.s.accept

name = c.recv(1024)
name = name.split("/")[-1]

newFile = open(self.curpath + name, "wb")
stop = False

while True:
    data = c.recv(24*1024)

    print(data[-4:])
    if data[-4:] == "DONE":
        break

    data = decrypt(self.key, data, self.iv)
    newFile.write(data)

    print("STOP")
    newFile.close()

Код клиента: 

s. = socket.socket()
s.connect(("127.0.0.1", 5000)

item = "Splash.png"
s.send(item)

with open(item, "rb") as fp:
        while True:
            chunk = fp.read(64*1024)

            if len(chunk) == 0:
                break
            elif len(chunk) % 16 != 0:
                chunk += " " * (16 - len(chunk) % 16)
            s.send(encrypt(self.key, chunk, self.iv))

s.send("DONE")
print("Done")
...