python實現ftp文件傳輸功能
本文實例為大家分享了python實現ftp文件傳輸的具體代碼,供大家參考,具體內容如下
主要步驟可以分為以下幾步:
1.讀取文件名2.檢測文件是否存在3.打開文件4.檢測文件大小5.發送文件大小和 md5值給客戶端6.等客戶端確認7.開始邊讀邊發數據
服務器端代碼:
import socket,os,timeimport 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: zjtimport 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,os,timeimport hashlib server =socket.socket()server.bind((’0.0.0.0’,8888))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.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 socketimport hashlibclient = 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:',file_total_size,' present: ',received_size) else: new_file_md5 = m.hexdigest() print('client file md5:',new_file_md5) print('file has received done!') print('total:',file_total_size,' present: ',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 socketimport hashlibclient = 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: #添加一次判斷,使最后一次剩多少就接收多少,避免發生粘包 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:',new_file_md5) print('file has received done!') print('total:',file_total_size,' present: ',received_size) print('下一句關閉文件') f.close() print('開始接收md5 ') sever_file_md5 = client.recv(1024) print('client file md5:',new_file_md5) print('server file md5:',sever_file_md5)client.close()
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持好吧啦網。
相關文章:
