python實現ftp檔案傳輸功能
阿新 • • 發佈:2020-03-22
本文例項為大家分享了python實現ftp檔案傳輸的具體程式碼,供大家參考,具體內容如下
主要步驟可以分為以下幾步:
1.讀取檔名
2.檢測檔案是否存在
3.開啟檔案
4.檢測檔案大小
5.傳送檔案大小和 md5值給客戶端
6.等客戶端確認
7.開始邊讀邊發資料
伺服器端程式碼:
import socket,os,time import hashlib server =socket.socket() server.bind(('0.0.0.0',6666)) server.listen() print("等待....") while True: conn,addr = server.accept() print("new conn:",conn) while True: data = conn.recv(1024) if not data: print("client is disconnection") break cmd,filename = data.decode().split() #記錄指令和檔名 print(filename) #判斷當前目錄是否存在該檔案,而且必須是檔案,而不是目錄 if os.path.isfile(filename): f = open(filename,'rb') #m = hashlib.md5() # 建立md5 file_size = os.stat(filename).st_size #stat() 可以返回檔案的大小值 conn.send((str(file_size)).encode()) # 傳送檔案大小 conn.recv(1024) #等待返回資訊 for line in f: # m.updata(line) conn.send(line) #print("file md5",m.hexdigest()) #列印md5值 f.close()
客戶端程式碼:
# Author: zjt import socket client = socket.socket() client.connect(("0.0.0.0",6666)) while True: cmd = input(">>>:").strip() if len(cmd)==0 :continue if cmd.startswith("get"): client.send(cmd.encode()) server_response = client.recv(1024) print("server response: ",server_response) client.send(b"ready to recv file") # 開始接收檔案 file_total_size = int(server_response.decode()) received_size = 0 # 記錄接收檔案的大小 filename = cmd.split()[1] # 因為兩個目錄一致,接收的檔名不能與原檔案相同 f = open(filename+".new","wb") while received_size < file_total_size: data = client.recv(1024) received_size += len(data) f.write(data) print("total:",file_total_size," present: ",received_size) else: print("file has received done!") f.close() client.close()
用80M的檔案傳輸測試,效果如下:
程序升級:
前面的程式碼還沒新增md5進行驗證,現在對程式碼進行升級
伺服器端程式碼:
import socket,8888)) server.listen() print("等待....") while True: conn,'rb') m = hashlib.md5() # 建立md5 file_size = os.stat(filename).st_size #stat() 可以返回檔案的大小值 conn.send((str(file_size)).encode()) # 傳送檔案大小 conn.recv(1024) #等待返回資訊 for line in f: m.update(line) conn.send(line) print("file md5",m.hexdigest()) #列印md5值 f.close() conn.send(m.hexdigest().encode()) # 傳送md5 print("我真的已經發過去了",m.hexdigest().encode()) print("send done") server.close()
客戶端程式碼:
import socket import hashlib client = socket.socket() client.connect(("0.0.0.0",8888)) while True: cmd = input(">>>:").strip() if len(cmd)==0 :continue if cmd.startswith("get"): client.send(cmd.encode()) server_response = client.recv(1024) print("server response: ",server_response) client.send(b"ready to recv file") # 開始接收檔案 file_total_size = int(server_response.decode()) received_size = 0 # 記錄接收檔案的大小 filename = cmd.split()[1] # 因為兩個目錄一致,接收的檔名不能與原檔案相同 f = open(filename+".new","wb") m = hashlib.md5() while received_size < file_total_size: data = client.recv(1024) received_size += len(data) m.update(data) f.write(data) #print("total:",received_size) else: new_file_md5 = m.hexdigest() print("client file md5:",new_file_md5) print("file has received done!") print("total:",received_size) f.close() sever_file_md5 = client.recv(1024) print("client file md5:",new_file_md5) print("server file md5:",sever_file_md5) client.close()
兩個程式在linux 環境下執行,結果如下:
可以看到傳輸後文件大小變大了一點點,而且md5前後值也不同,說明檔案傳輸發生了改變。
現在講程式在windows環境下執行,結果如下:
此時可以看到windows上沒有問題,檔案大小相同,且md5值也一致。
原因分析:
之所以會發生這種情況,是因為在linux上執行時,最後一次傳輸檔案與傳送md5值的時候,發生可粘包,導致最後一次接收檔案的時候,連同md5的資料一併傳送了。而客戶端也當作一條接收資訊,全部接收了。所以客戶端出現沒有收到來自伺服器端的md5值,多出來的那一點點,就是md5值的大小。
解決方法:
在接收檔案的時候,判斷當前剩餘多少檔案需要接收,如果大於1024,就接收1024大小的檔案,否則就只接收剩下全部的檔案,防止最後一次接收多餘的資料。
只需要對客戶端程式碼進行修改,修改後程式碼如下:
import socket import hashlib client = socket.socket() client.connect(("0.0.0.0",server_response) client.send(b"ready to recv file") # 開始接收檔案 file_total_size = int(server_response.decode()) received_size = 0 # 記錄接收檔案的大小 filename = cmd.split()[1] f = open(filename+".new","wb") m = hashlib.md5() while received_size < file_total_size: #新增一次判斷,使最後一次剩多少就接收多少,避免發生粘包 if file_total_size - received_size > 1024: size = 1024 else: # 最後一次,剩多少收多少 size = file_total_size - received_size data = client.recv(size) received_size += len(data) m.update(data) f.write(data) else: new_file_md5 = m.hexdigest() print("client file md5:",received_size) print("下一句關閉檔案") f.close() print("開始接收md5 ") sever_file_md5 = client.recv(1024) print("client file md5:",sever_file_md5) client.close()
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。