需要將多個文件從客戶端傳輸到服務器

Question

我最近正在開展一個項目，我基本上都在制作一個Dropbox克隆。 服務器和客戶端工作正常，但我有一個小問題。 我能夠將單個文件從客戶端傳輸到服務器但是當我嘗試將所有文​​件一起傳輸時，它會在傳輸第一個文件后給出錯誤，所以基本上我的代碼只適用於單個文件。 我需要讓它適用於多個文件。 任何幫助將不勝感激。 這是我的代碼

服務器代碼

import socket
import thread
import hashlib

serversock = socket.socket()
host = socket.gethostname();
port = 9000;
serversock.bind((host,port));
filename = ""
serversock.listen(10);
print "Waiting for a connection....."

clientsocket,addr = serversock.accept() 
print("Got a connection from %s" % str(addr))
while True:
    size = clientsocket.recv(1)
    filesz = clientsocket.recv(1)
    if filesz.isdigit():
        size += filesz
        filesize = int(size)
    else:
        filesize = int(size)    
    print filesize
    for i in range(0,filesize):
        if filesz.isdigit():
            filename += clientsocket.recv(1)
        else:
            filename += filesz
            filesz = "0"
    print filename      
    file_to_write = open(filename, 'wb')
    while True:
        data = clientsocket.recv(1024)
        #print data
        if not data:
            break
        file_to_write.write(data)
    file_to_write.close()
    print 'File received successfully'
serversock.close()

客戶代碼

import socket
import os
import thread

s = socket.socket() 
host = socket.gethostname()                           
port = 9000
s.connect((host, port))
path = "C:\Users\Fahad\Desktop"
directory = os.listdir(path)
for files in directory:
    print files  
    filename = files
    size = bytes(len(filename))
    #print size
    s.send(size)
    s.send(filename)
    file_to_send = open(filename, 'rb')
    l = file_to_send.read()
    s.sendall(l)
    file_to_send.close()       
    print 'File Sent'                                              
s.close()                                                                           

這是我得到的錯誤

Waiting for a connection.....
Got a connection from ('192.168.0.100', 58339)
13
Betternet.lnk
File received successfully
Traceback (most recent call last):
  File "server.py", line 22, in <module>
    filesize = int(size)
ValueError: invalid literal for int() with base 10: ''

Answer 1

您的代碼段中有幾個小問題。 也許你可以做這樣的事情？

import socket
import thread
import hashlib

serversock = socket.socket()
host = socket.gethostname();
port = 9000;
serversock.bind((host,port));
filename = ""
serversock.listen(10);
print "Waiting for a connection....."

clientsocket,addr = serversock.accept()
print("Got a connection from %s" % str(addr))
while True:
    size = clientsocket.recv(16) # Note that you limit your filename length to 255 bytes.
    if not size:
        break
    size = int(size, 2)
    filename = clientsocket.recv(size)
    filesize = clientsocket.recv(32)
    filesize = int(filesize, 2)
    file_to_write = open(filename, 'wb')
    chunksize = 4096
    while filesize > 0:
        if filesize < chunksize:
            chunksize = filesize
        data = clientsocket.recv(chunksize)
        file_to_write.write(data)
        filesize -= len(data)

    file_to_write.close()
    print 'File received successfully'
serversock.close()

客戶：

import socket
import os
import thread

s = socket.socket()
host = socket.gethostname()
port = 9000
s.connect((host, port))
path = "blah"
directory = os.listdir(path)
for files in directory:
    print files
    filename = files
    size = len(filename)
    size = bin(size)[2:].zfill(16) # encode filename size as 16 bit binary
    s.send(size)
    s.send(filename)

    filename = os.path.join(path,filename)
    filesize = os.path.getsize(filename)
    filesize = bin(filesize)[2:].zfill(32) # encode filesize as 32 bit binary
    s.send(filesize)

    file_to_send = open(filename, 'rb')

    l = file_to_send.read()
    s.sendall(l)
    file_to_send.close()
    print 'File Sent'

s.close()

這里客戶端也發送文件的大小。 size和filesize都編碼為二進制字符串（您可以使用另一種方法）。 文件名長度（大小）最多可以取2 ^ 16，發送文件最多可以包含2 ^ 32字節（即2 ^文件大小）。

Answer 2

您可以創建多個套接字並使用makefile編寫：

服務器：

import socket
import threading
import time

serversock = socket.socket()
host = socket.gethostname()
port = 9000
serversock.bind((host, port))

serversock.listen(10)
print("Waiting for a connection.....")


def reader(client):
    fle = client.makefile('r')
    filename = fle.readline()
    client.send("Got file {}\n".format(filename))
    file_to_write = open(filename.rstrip(), 'wb')
    client.send("Starting writing {}\n".format(filename))
    file_to_write.write(fle.read())
    file_to_write.close()
    client.send("Finished writing {}\n".format(filename))


while True:
    client, addr = serversock.accept()
    print("Got a connection from %s" % str(addr))
    client_serve_thread = threading.Thread(target=reader, args=tuple((client,)))
    client_serve_thread.start()
    time.sleep(0.001)

serversock.close()

客戶：

import socket
import os
import thread
import os

host = socket.gethostname()
port = 9000

path = "/home/padraic/t"
directory = os.listdir(path)
for file in directory:
    s = socket.socket()
    s.connect((host, port))
    filename = os.path.join(path, file)
    s.send(file+"\n")
    print(s.recv(1024))
    file_to_send = open(os.path.join(path, file), 'rb')
    s.send(file_to_send.read())
    print('File Sent\n')
    file_to_send.close()
    rec = s.recv(1024)
    print(rec)
    s.close()

Answer 3

我相信你實際上是在接收所有文件的內容，然后將它們全部寫入一個文件。

您的服務器只接受單個連接，並將收到的任何數據寫入文件，直到它不再接收數據為止。 直到客戶端在最后關閉其套接字時才會發生這種情況。

有幾種方法可以解決這個問題。

1. 將accept調用移動到服務器循環中，將connect調用移動到客戶端循環中。 讓您的客戶端連接，發送文件名，傳輸單個文件的全部內容，然后關閉連接。 在下一次迭代中，重新執行一遍。
2. 或者，在每個文件的開頭，讓客戶端向服務器發送要傳輸的文件名和文件大小（因此服務器知道如何查找文件內容的結尾）。 然后准確地將多個字節寫入服務器。 （但另請參閱下面的傳輸文件大小。）

我建議（1）更健壯，更容易實現。

第二個主題：發送文件名的機制有缺陷。 如果我正確地遵循它，如果正在傳輸的文件名以數字開頭，則程序將無法正常工作，因為服務器將無法確定用於發送文件名長度的字節數。 有兩種常用的發送方式：

1. 使用struct模塊以明確定義的方式格式化二進制整數。 您實際上發送“打包”格式，服務器將解壓縮它。 然后它會准確知道文件名接收的字節長度。
2. 只需發送一個包含文件名的標題行，后跟空字節或換行符（或其他一些明確定義的終結符字節）。 然后，服務器可以一次讀取一個字節，直到它看到終結符。