socket FTP-1

基于socket实现文件的传输以及md5验证

server:

 import socket
 import os
 import hashlib
 server=socket.socket()
 server.bind(('localhost',9090))
 server.listen()

 while True:
     print('ready to receive:')
     conn,addr=server.accept()

     while True:
         data1=conn.recv(1024)
         file_name=data1.decode()

         if os.path.isfile(file_name):
             file_size=os.path.getsize(file_name)
             conn.send(str(file_size).encode())
             f=open(file_name,'rb')
             md=hashlib.md5()

             data2=conn.recv(1024)

             for line in f:
                 conn.send(line)
                 md.update(line)

             f.close()
             print('发送完成...')
             conn.send(md.hexdigest().encode())

         else:
             print('无文件')

 server.close()

 import socket
 import  os
 import hashlib

 client=socket.socket()
 client.connect(('localhost',9090))

 while True:
     choice=input('>>')
     if len(choice)==0:continue

     client.send(choice.encode())
     data1=client.recv(1024)
     data_size=int(data1.decode())
     print('data_size:',data_size)

     count_size=0
     recv_data=b''
     # if os.path.isfile(choice):
     f=open(choice+'.new','wb')
     client.send(b'recving')
     md=b''
     ma_data=hashlib.md5(md)

     while count_size < data_size:
         data=client.recv(1024)
         recv_data+=data
         recv_data_size=len(data.decode())
         count_size+=recv_data_size
         ma_data.update(data)

     else:
         f.write(recv_data)
         print('complete')
         f.close()

     md1=client.recv(1024).decode()
     print(ma_data.hexdigest(),md1)
     if ma_data.hexdigest()==md1:
         print('无异常')

 client.close()