python - 服务器/客户端应用程序和 JSONDecodeError : Unterminated string python

Question

我有异步套接字服务器文件和客户端文件。
当我向客户端发送类似“下载文件名.ex”的内容时，该客户端的代码会处理我的请求:

try:
    content = read(sp_data[-1]).decode('utf-8')
    print(content)
    msg = json.dumps({'file': sp_data[-1], 'command': data, 'content': content,
                                  'msg': f'[+] File {sp_data[-1]} has been successfully downloaded.'}).encode('utf-8')
except FileNotFoundError:
    msg = json.dumps({'msg': f'[-] File {sp_data[-1]} not found', 'command': data}).encode('utf-8')
    s.send(msg)

当客户端向 socketserver 发送一些数据时，该服务器的代码句柄接收到消息:

def recv_message(client_socket):
    global messages
    data = json.loads(client_socket.recv(4096).decode('utf-8').strip()) ##Important here i got this error json.decoder.JSONDecodeError: Unterminated string starting at: line 1 column 67 (char 66)
    raddr = get_raddr(str(client_socket))
    raddr = f'{raddr[0]}:{raddr[1]}'
    message = f'From: {raddr}\nCommand: {data["command"]}\nOutput: \n\n{data["msg"]}'
    try:
        d = messages[raddr]
        d.append(message)
        messages[raddr] = d
    except KeyError:
        messages[raddr] = [message]
    except AttributeError:
        print(message, messages)
    if 'content' in data.keys(): ##Important
        print(data['content'])
        threading.Thread(target=create_file, args=(data['file'], data['content'],), daemon=False).start()

错误:

data = json.loads(client_socket.recv(4096).decode('utf-8').strip())
json.decoder.JSONDecodeError: Unterminated string starting at: line 1 column 67 (char 66)

但是上面的服务器代码在收到来自第一个代码的消息时给了我这个错误(当我向客户端发送类似“download file.ex”的内容时，客户端将我的命令检测为它的特殊命令，执行第一个代码，发送 json 文件到服务器。但是如果我向客户端发送“dir”命令，它会像 shell 命令一样检测我的命令，将通过子进程运行命令，将结果发送回服务器，我不会收到任何错误。)
注意:我还减少了 socketserver 的代码。因此，我的代码中的某些内容可能会更糟。这篇文章的主要目标 - 使下载功能有效。我也明白我的代码很大。我离开了“ ##Important” 我的文件中的评论。您只能观看由这些注释定位的代码。
服务器 :

import selectors
import socket
import threading
import json
import base64
import shlex

selector = selectors.DefaultSelector()

connections = {}


def accept_conn(server_socket):
    sock, addr = server_socket.accept()
    connections[len(connections) + 1] = [sock, f'{addr[0]}:{addr[-1]}']
    selector.register(fileobj=sock, events=selectors.EVENT_READ, data=recv_message)


s = socket.socket()
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
s.bind(('localhost', 4444))
s.listen()
selector.register(fileobj=s, events=selectors.EVENT_READ, data=accept_conn)

messages = {}

##Important 
def create_file(file, content): #content - base64 string
    print(content)
    with open(file, 'wb') as f:
        f.write(base64.b64decode(content.encode('utf-8')))


def recv_message(client_socket):
    global messages
    data = json.loads(client_socket.recv(4096).decode('utf-8').strip()) ##Important here i got this error json.decoder.JSONDecodeError: Unterminated string starting at: line 1 column 67 (char 66)
    raddr = get_raddr(str(client_socket))
    raddr = f'{raddr[0]}:{raddr[1]}'
    message = f'From: {raddr}\nCommand: {data["command"]}\nOutput: \n\n{data["msg"]}'
    try:
        d = messages[raddr]
        d.append(message)
        messages[raddr] = d
    except KeyError:
        messages[raddr] = [message]
    except AttributeError:
        print(message, messages)
    if 'content' in data.keys(): ##Important
        print(data['content'])
        threading.Thread(target=create_file, args=(data['file'], data['content'],), daemon=False).start()


def get_raddr(string):
    '''Get raddr parameter from client socket'''
    raddr = string.replace('>', '')
    return eval(raddr[raddr.find('raddr')::].replace('raddr=', ''))


def is_manage_string(sub, string):
    tokens = shlex.split(string)
    try:
        if len(tokens) == 2 and tokens[0] == sub and str(int(tokens[-1])):
            return True, int(tokens[-1])
    except Exception as e:
        print(e)
    return False


manage_process = False


def manage():
    global manage_process
    while True:
        manage_process = False
        command = input('>>> ').strip()
        if command == 'list':
            try:
                for i in range(1, len(connections) + 1):
                    print(f'{i}\t{connections[i][-1]}')
            except KeyError:
                pass
            if len(connections) == 0:
                print('[-] There are not any connections')
        elif 'manage' in command:
            index = is_manage_string('manage', command)
            if index:
                index = index[-1]
            else:
                print('[-] Invalid command\nUse manage "number_of_connection"\nEx: manage 1')
                continue
            if index >= 1 and index <= len(connections):
                sock, addr = connections[index]
                print(addr)
                print(f'{addr} is used')
                while True: ##Important here i launch loop which send data to socket
                    manage_process = True
                    command = input('>>> ').strip()
                    if command == 'messages':
                        try:
                            if messages[addr] == list():
                                print()
                                continue
                        except KeyError:
                            pass
                        try:
                            print('\n\n'.join(messages[addr]))
                        except KeyError:
                            print()
                    elif command == 'message':
                        try:
                            print(messages[addr][-1])
                        except:
                            print()
                    elif command == 'clear_messages':
                        try:
                            if messages[addr]:
                                messages[addr] = []
                        except KeyError:
                            print('[-] There are not any messages for cleaning up')
                    elif command == 'leave':
                        print(f'Leaving connection {addr}')
                        break
                    elif command: ##Important if command hasn't been detected as my special command(leave, messages), it will be executed like shell command
                        try:
                            sock.send(command.encode('utf-8'))
                            print(
                                'Your input has not been detected as special command and will execute like shell command or like client special command(ex: download; see client file)')
                        except ConnectionResetError:
                            print("Connection has been lost, therefore shell commands can't be used")
                    else:
                        continue

            else:
                print('[-] Invalid number of connection')

        elif command:
            print('[-] Invalid command\nType "help" to see avalible commands')

##Important
def event_loop():
    while True:
        data = selector.select()
        for key, _ in data:
            try:
                key.data(key.fileobj)
            except ConnectionResetError:
                selector.unregister(key.fileobj)


##Important
threading.Thread(target=manage, daemon=True).start()
event_loop()

客户 :

import socket
import subprocess
import shlex
import threading
import json
import base64


s = socket.socket()
s.connect(('localhost', 4444))


##Important
def read(file):
    with open(file, 'rb') as f:
        return base64.b64encode(f.read())


def runner(data):
    sp_data = shlex.split(data)
    try:
        print(sp_data)
        if len(sp_data) == 2 and sp_data[0] == 'download': ###Important here we create json object which will be send to socketserver
            try:
                content = read(sp_data[-1]).decode('utf-8')
                print(content)
                msg = json.dumps({'file': sp_data[-1], 'command': data, 'content': content,
                                  'msg': f'[+] File {sp_data[-1]} has been successfully downloaded.'}).encode('utf-8')
            except FileNotFoundError:
                msg = json.dumps({'msg': f'[-] File {sp_data[-1]} not found', 'command': data}).encode('utf-8')
            s.send(msg)
            return ''
    except Exception as e:
        print(e)
    command = subprocess.run(data, shell=True, encoding='cp866', text=True, capture_output=True)
    command = command.stderr if command.stderr else command.stdout
    command = json.dumps({'msg': command, 'command': data})
    s.send(command.encode('utf-8'))


while True:##Important
    data = s.recv(4096).decode('utf-8').strip()
    threading.Thread(target=runner, args=(data,)).start()

Answer 1

import socket
import struct


class Socket(socket.socket):
    def __init__(self):
        self.sock = socket.socket()
        super().__init__(socket.AF_INET, socket.SOCK_STREAM)

    def send_msg(self, msg):
        # Prefix each message with a 4-byte length (network byte order)
        msg = struct.pack('>I', len(msg)) + msg
        self.sock.sendall(msg)

    def recv_msg(self):
        # Read message length and unpack it into an integer
        raw_msglen = self.recv_all(4)
        if not raw_msglen:
            return None
        msglen = struct.unpack('>I', raw_msglen)[0]
        # Read the message data
        return self.recv_all(msglen)

    def recv_all(self, n):
        data = bytearray()
        while len(data) < n:
            packet = self.sock.recv(n - len(data))
            if not packet:
                return None
            data.extend(packet)
        return data

我将您的代码改造成 Socket 类。