Socket聊天中的解密问题

在Socket聊天程序中，当使用加密算法对发送的数据进行加密后，接收方需要对接收到的数据进行解密。如果在解密过程中出现问题，可能导致接收方无法正确地解密消息，从而出现解密错误的情况。

以下是一个解决Socket聊天中解密问题的示例代码：

import socket
import hashlib
import base64
import json

# 客户端
class ChatClient:
    def __init__(self, host, port, password):
        self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.socket.connect((host, port))
        self.password = password
        self.key = None

    def send(self, data):
        if self.key:
            data = self.encrypt(data)
        self.socket.sendall(data.encode())

    def receive(self):
        data = self.socket.recv(1024).decode()
        if self.key:
            data = self.decrypt(data)
        return data

    def encrypt(self, data):
        cipher = self.get_cipher()
        encrypted_data = cipher.encrypt(data.encode())
        encrypted_data = base64.b64encode(encrypted_data).decode()
        return encrypted_data

    def decrypt(self, data):
        encrypted_data = base64.b64decode(data)
        cipher = self.get_cipher()
        decrypted_data = cipher.decrypt(encrypted_data).decode()
        return decrypted_data

    def get_cipher(self):
        md5 = hashlib.md5()
        md5.update(self.password.encode())
        key = md5.digest()
        cipher = AES.new(key, AES.MODE_ECB)
        return cipher

    def login(self):
        data = {
            'type': 'login',
            'username': 'user',
            'password': self.password
        }
        self.send(json.dumps(data))
        login_data = json.loads(self.receive())
        if login_data['status'] == 'ok':
            self.key = login_data['key']
        
# 服务端
class ChatServer:
    def __init__(self, host, port, password):
        self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.socket.bind((host, port))
        self.socket.listen(1)