【小程序】使用socket实现文件的收发

功能：使用socket实现文件的收发，客户端发送[get 文件名]，在客户端接收服务端的文件，接受后校验MD5值。

功能：使用socket实现文件的收发，客户端发送[get 文件名]，在客户端接收服务端的文件，接受后校验MD5值。

client端：

import socket,hashlib

client=socket.socket()

client.connect((\'localhost\',9999))

while True:
    msg=input(\'>>>\').strip()
    if len(msg)==0:
        continue
    if msg.startswith(\'get\'):
        client.send(msg.encode(\'utf-8\'))
        server_response=client.recv(1024)
        print(\'file size:\',server_response.decode())
        client.send(\'...\'.encode(\'utf-8\'))
        file_total_size=int(server_response.decode())
        recv_size=0
        filename=msg.split()[1]
        f=open(filename+\'_new\',\'wb\')
        m=hashlib.md5()

        while recv_size<file_total_size:
            data=client.recv(1024)
            recv_size+=len(data)
            f.write(data)
            m.update(data)
            print(\'recv:%d/%d\'%(recv_size,file_total_size))
        else:
            print(\'recv done...\')
            f.close()
            new_file_md5=m.hexdigest()
            client.send(\'recv done\'.encode(\'utf-8\'))
            server_files_md5=client.recv(1024).decode()
            print(\'server_files_md5:\',server_files_md5)
            print(\'new_file_md5:\',new_file_md5)

client.close()

server端：

import socket,os,hashlib

server=socket.socket()
server.bind((\'localhost\',9999))
print(\'waiting...\')
server.listen()
while True:
    conn,addr=server.accept()
    print(\'new conn:\',addr)
    while True:
        data=conn.recv(1024).decode()
        if not data:
            print(\'客户端断开...\')
            break
        print(\'data\',data)
        cmd,filename=data.split()
        print(\'cmd:\',cmd)
        print(\'filename:\',filename)
        if os.path.isfile(filename):
            file_size=os.stat(filename).st_size
            conn.send(str(file_size).encode(\'utf-8\'))
            conn.recv(1024).decode()#防粘包
            f = open(filename,\'rb\')
            m=hashlib.md5()
            for line in f:
                m.update(line)
                conn.send(line)#按行发送
                res_md5=m.hexdigest()
            print(\'MD5:\',res_md5)
            f.close()
            conn.recv(1024).decode()#防粘包
            conn.send(res_md5.encode(\'utf-8\'))#发送MD5


        print(\'send done\')

server.close()