将多个文件从本地服务器下载到客户端

代码的主要问题是您正在发送/接收任意数量的数据。如果您的缓冲区（1024）小于文件大小，则客户端的文件将包含较少的信息，而如果文件缓冲区较大，则文件可能包含更多的信息（来自下一个文件的数据）。

您可以通过发送一个表示文件结尾的值来解决此问题。该方法的问题在于，该值不能包含在任何文件中，并且客户端必须在接收到的数据中扫描该值。

另一种可能的解决方案是计算文件大小，然后在文件数据之前发送该信息。这样，管理员将知道每个文件需要多少数据。

使用struct.pack我们可以创建一个最小的四个字节的文件头。

def send_file(soc, path):
    with open(path, 'rb') as f:
        data = f.read()
    size = struct.pack('!I', len(data))
    soc.send(size + data)

然后，客户端可以通过读取四个字节并将其指定unpack为int来获取文件大小。

def recv_file(soc, path):
    size_header = soc.recv(4)
    size = struct.unpack('!I', size_header)[0]
    data = soc.recv(size)
    with open(path, 'wb') as f:
        f.write(data)

请注意，如果文件大小大于套接字缓冲区，则一次调用发送/接收文件可能会引起套接字错误。在这种情况下，您必须循环读取较小的数据块，或使用增大缓冲区的大小socket.setsockopt。

这是上述功能的修改后的版本，可以处理大文件：

import struct
import os.path

def send_file(soc, path):
    file_size = os.path.getsize(path)
    size_header = struct.pack('!Q', file_size)
    soc.send(size_header)
    with open(path, 'rb') as f:
        while True:
            data = f.read(1024)
            if not data:
                break
            soc.send(data)

def recv_file(soc, path):
    size_header = soc.recv(8)
    file_size = struct.unpack('!Q', size_header)[0]
    chunks = [1024 for i in range(file_size / 1024)]
    with open(path, 'wb') as f:
        for chunk in chunks:
            f.write(soc.recv(chunk))
        f.write(soc.recv(file_size % 1024))

我尚未彻底测试过此代码，但它适用于任何大小的文件。

send_file在服务器中使用该功能的示例：

host = ''
skServer = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
skServer.bind((host,2525))
skServer.listen(10)
print "Server currently active"

Content,Address = skServer.accept()
print Address

files = ['tmp.bsp', 'tmp.seq', 'tmp.dms']
for file in files:
    send_file(Content, file)

Content.close()
print "Sending is over"
skServer.close()

recv_file在客户端中使用：

skClient = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
skClient.connect(("127.0.0.1",2525))

files = ['tmp.bsp', 'tmp.seq', 'tmp.dms']
for file in files:
    recv_file(skClient, file)

print "Download over"
skClient.close()