Python使用socket实现简单的文件传输并校验

24132eqw

因为工作需要，要在两台设备之间进行压力测试。即A设备不断往B设备发送文件，B设备接收文件后校验文件是否正确接收。       用Python的socket模块写了简单的Server和Client脚本。Server负责监听端口，接收Client发来的文件并做MD5校验，确认接收的文件是否准确无误。Client端负责向Server发送文件及文件的MD5值。
       文件名通过Client发送特定字符串提取，文件结束符合通过检测'EOF'字符实现。实际运行过程中，发现总是会有误判，把结束符'EOF'错误写入文件了。希望有大神不吝赐教！！

Socket_Server.py

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67

#!/usr/bin/python # -*- coding: utf-8 -*- # Python:          2.7 # Platform:        Linux # Author:          wucl(wucl-20@163.com) # Program:         Socket_Server监听网络端口，接收Client发来的文件 # Version:         2.0 # History:         2016.1.28 #                  2016.2.1   接收Client发来的文件并做MD5校验 import socket, time, hashlib, os, base64, re def CalcMD5(fineName, block_size=64 * 1024):   with open(fineName, 'rb') as f:     md5 = hashlib.md5()     while True:       data = f.read(block_size)       if not data:         break       md5.update(data)     retmd5 = md5.hexdigest()     print(retmd5)     return retmd5 def main(IP, Port, log, error_log):     s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)     s.bind((IP, Port))     s.listen(12)     print 'Listened on ', socket.gethostname(), Port     reg = re.compile('filename:(.*?):(.*)::::')     while True:         conn, addr = s.accept()         print(time.ctime(), 'Connected from ', addr)         recvd = conn.recv(1024)         if  recvd:             ss = reg.findall(recvd)             filename = ss[0][0]             md5 = ss[0][1]             f = open(filename, 'wb')             while True:                 data = conn.recv(1024)                 if not data:                     break                 if data == 'EOF':                     f.close()                     log.write('%s Received %s\n'%(time.ctime(), filename))                     log.flush()                     if CalcMD5(filename) == md5:                         print('%s Received Successfully!\n' %filename)                     else:                         error_log.write('%s %s Recevied failed\n'%(time.ctime(), filename))                         error_log.flush()                         print(CalcMD5(filename))                 else:                     f.write(data)                     f.flush()             conn.close()     s.close() if __name__ == "__main__":     IP = ''     Port = 12345     log = open('log.txt','w+')     error_log = open('error_log.txt','w+')     main(IP, Port, log, error_log)

Socket_Client.py

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56

#!/usr/bin/python # -*- coding: utf-8 -*- # Python:          2.7 # Platform:        Linux # Author:          wucl(wucl-20@163.com) # Program:         Socket_Client连接网络端，并向网络端发送文件。 # Version:         2.0 # History:         2016.1.28 #                  2016.2.1    Client向Server端发送文件及文件的MD5值 import socket, hashlib import time, os, base64, os.path def CalcMD5(fineName, block_size=64 * 1024):   with open(fineName, 'rb') as f:     md5 = hashlib.md5()     while True:       data = f.read(block_size)       if not data:         break       md5.update(data)     retmd5 = md5.hexdigest()     print(retmd5)     return retmd5 def SendFile(s, filename):     print('Sending %s' % filename)     s.sendall('filename:%s:%s::::'%(filename, CalcMD5(filename)))     f = open(filename, 'rb')     while True:         chunk = f.read(chunksize)         if not chunk:             f.close()             print('%s Successfully Sent %s\n' %(time.ctime(), filename))             break         s.sendall(chunk)     time.sleep(1)     s.sendall('EOF')     s.close() def main(dirname, IP, Port, MB = 1024):     for filename in os.listdir(dirname):         if os.path.isfile(filename):             s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)             s.connect((IP, Port))             SendFile(s, filename) if __name__ == '__main__':     dirname = '/root/test/'     IP = '10.88.106.200'     Port = 12345     MB = 1024     chunksize = int(MB*1024*1024)     main(dirname, IP, Port, MB)