从客户端接收字节后制作文件

Question

我想执行一个简单的文件传输程序，该程序将文件从客户端传输到服务器。服务器将路径发送到客户端，客户端将文件发送到服务器。但是我不知道如何在服务器部分中创建文件。实际上，我以二进制模式发送所有文件，但是之后，我在服务器模式下接收每个字节，我不知道如何从收到的字节中制作文件。

客户零件

sock = 0
path = " "

def open_socket():
    global sock
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    sock.connect(("localhost", 9000))

def transfer_file(path):
    f = open(path, "rb").read()
    sock.send(f)

def main():
    open_socket()
    path = sock.recv(1024)

if __name__ == "__main__":
    main()

服务器部分

def open_connection():
    global client
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    sock.bind(("0.0.0.0", 9000))
    sock.listen(1)
    client, address = sock.accept()

def file_transfer():
    path = raw_input("Enter path of file: ")
    client.send(path)
    while True:
        file_from_client = client.recv(1024)

def main():
    open_connection()
    file_transfer()

if __name__ == "__main__":
    main()

谢谢！

Answer 1

您需要在插座上进行更多工作，包括使用sendall而不是send（可能会发送的少于完整的缓冲区），并找出一种方法，让阅读方了解数据何时具有数据完成的发送。由于您没有消息边界可以让您要求多个文件，因此我的示例关闭了套接字，让另一侧知道传输已完成。我也摆脱了全球变量。这将传输一个文件：

client.py

import socket
def open_socket():
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    sock.connect(("localhost", 9000))
    return sock
def transfer_file(sock, path):
    f = open(path, "rb").read()
    sock.sendall(f)
def main():
    sock = open_socket()
    path = sock.recv(1024)
    transfer_file(sock, path)
    sock.shutdown(socket.SHUT_RDWR)
    sock.close()
if __name__ == "__main__":
    main()

server.py

import socket
import os
def open_connection():
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
    sock.bind(("0.0.0.0", 9000))
    sock.listen(1)
    client, address = sock.accept()
    sock.close()
    return client
def file_transfer(client):
    path = raw_input("Enter path of file on client: ")
    local_path = os.path.join(os.getcwd(), os.path.basename(path))
    print 'will write to local file', local_path
    client.send(path)
    with open(local_path + '.tmp', 'wb') as outfile:
        while True:
            block = client.recv(1024)
            if not block:
                break
            outfile.write(block)
    os.rename(local_path + '.tmp', local_path)
    print 'wrote', os.stat(local_path).st_size, 'bytes'
def main():
    client = open_connection()
    file_transfer(client)
    client.close()

if __name__ == "__main__":
    main()