day36 作业

客户端

import struct

import json

from socket import *

client=socket(AF_INET,SOCK_STREAM)

# client.connect(('121.36.98.49',8822))

client.connect(('127.0.0.1',8811))

while True:

    cmd=input('请输入命令（dw or up +文件）>>：').strip()

    if len(cmd) == 0:continue

    client.send(cmd.encode('utf-8'))

    # 接收端

    # 1、先手4个字节，从中提取接下来要收的头的长度

    x=client.recv(4)

    print(x)

    header_len=struct.unpack('i',x)[0]

    # 2、接收头，并解析

    json_str_bytes=client.recv(header_len)

    json_str=json_str_bytes.decode('utf-8')

    header_dic=json.loads(json_str)

    print(header_dic)

    total_size=header_dic["file_size"]

    # 3、接收真实的数据

    recv_size = 0

    while recv_size < total_size:

        recv_data=client.recv(1024)

        recv_size+=len(recv_data)

        with open(f"my{header_dic['file_name']}",'wb') as f :

            f.write(recv_data)

    else:

        print()

服务端

import struct

import json

from socket import *

server = socket(AF_INET, SOCK_STREAM)

server.setsockopt(SOL_SOCKET, SO_REUSEADDR, 1)

server.bind(('127.0.0.1',8811))

# server.bind(('0.0.0.0',8812))

server.listen(5)

while True:

    conn, client_addr = server.accept()

    while True:

        try:

            cmd = conn.recv(1024)

            if len(cmd) == 0: break

            cmd = cmd.decode('utf-8')

            cmd = cmd.split(',')

            print(cmd[0],cmd[1])

            if cmd[0] == 'up':

                pass # 执行上传

            elif cmd[0] == 'dw': # 执行下载

                print('开始传送')

                with open(f'/hz/{cmd[1]}','rb') as f :

                    file_data = f.read()

                    print(file_data)

                    print('传送完毕')

                file_size = len(file_data)

                header_dic = {

                    "mode": cmd[0],

                    'file_name':cmd[1],

                    "file_size": file_size,

                }

                print(header_dic)

                json_str = json.dumps(header_dic)

                json_str_bytes = json_str.encode('utf-8')

                x = struct.pack('i', len(json_str_bytes))

                print(x)

                conn.send(x)

                conn.send(json_str_bytes)

                conn.send(file_data)

        except Exception:

            break

    conn.close()

巴特西

day36 作业

客户端

服务端

最新文章

热门文章