网络编程基础【day09】：socket实现文件发送（六）

本节内容

1、概述

2、文件下载实现

3、MD5值校验

一、概述

　　我们如何利用socket去下载一个文件，整体思路是这样的：

1. 读取文件名
2. 检测文件是否存在
3. 打开文件
4. 检测文件大小
5. 发送文件大小给客户端
6. 等客户确认
7. 开始边读边发数据
8. 发送md5值给客户端校验

友情提示：以下代码都是在Linux系统，并且是python3换将下实验的。

二、文件下载实现

2.1、服务端代码

逻辑：获取命令和文件名->判断文件是否存在->打开文件->获取文件大小->发送文件大小给客户端->等待客户端确认->边读边发

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

import  hashlib import  socket,os   server  =  socket.socket() server.bind(( "localhost" , 9999 )) server.listen() while  True :      conn,addr  =  server.accept()      print ( "new conn:" ,addr)      while  True :          print ( "等待新指令" )          data  =  conn.recv( 1024 )          if  not  data:              print ( "客户端已断开" )              break          cmd,filename  =  data.decode().split()    #接收客户端发过来的命令和文件名          print (filename)          if  os.path.isfile(filename):     #判断文件是否存在              # m = hashlib.md5()              with  open (filename, "rb" ) as f:                  file_size  =  os.stat(filename).st_size    #获取文件大小                  conn.send(  str (file_size).encode() )  #发送文件大小                  conn.recv( 1024 )                  for  line  in  f:                      #m.update(line)                      conn.send(line)      #边读边发给客户端                  #print("file md5",m.hexdigest())          print ( "send done" )   server.close()

2.2、客户端代码

逻辑：判断是否是下载命令(get) ->发送下载命令和文件名 ->获取文件大小->发送确认信息->判断时候已经全部接收

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24

import   socket   client  =  socket.socket() client.connect(( "localhost" , 9999 )) while  True :      cmd  =  input ( ">>>:" ).strip()      if  len (cmd)  = =  0 : continue      if  cmd.startswith( "get" ):          client.send(cmd.encode())     #发送下载命令和文件名          server_respose  =  client.recv( 1024 )     #接收文件大小          print ( "server response:" ,server_respose)          client.send( "ready to recv file" .encode())     #发送确认信息          file_total_size  =  int (server_respose.decode())          revived_size  =  0    #初始化接收大小          filename  =  cmd.split()[ 1 ]    #获取文件名          with  open (filename  +  ".new" , "wb" ) as f:              while  revived_size < file_total_size:     #判断接收大小和文件大小比较                  data  =  client.recv( 1024 )                  revived_size  + =  len (data)    #接收大小                  f.write(data)    #写入文件              else :                  print (file_total_size,revived_size)                  print ( "file recv done" ) client.close()

三、MD5值校验

说明：上面只提到了怎么传输，我们在概述里面也提到了最后要用MD5值做校验，但是很显然没看到，下面我们就来说说怎么用MD5值做比较

3.1、服务端

说明：生成md5的对象->计算MD5值->生成16进制的形式->编码后发送给客户端

①代码

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

import  hashlib import  socket,os   server  =  socket.socket() server.bind(( "localhost" , 9999 )) server.listen() while  True :      conn,addr  =  server.accept()      print ( "new conn:" ,addr)      while  True :          print ( "等待新指令" )          data  =  conn.recv( 1024 )          if  not  data:              print ( "客户端已断开" )              break          cmd,filename  =  data.decode().split()          print (filename)          if  os.path.isfile(filename):              m  =  hashlib.md5()   #生成MD5的对象              with  open (filename, "rb" ) as f:                  file_size  =  os.stat(filename).st_size                  conn.send(  str (file_size).encode() )  #send file size                  conn.recv( 1024 )                  for  line  in  f:                      m.update(line)   #计算md5值                      conn.send(line)                  print ( "file md5" ,m.hexdigest())              conn.recv( 1024 )    #等待客户确认发送MD5值              conn.send(m.hexdigest().encode())   #生成MD5值并且发送给客户端          print ( "send done" )   server.close()

②改动的地方

 

3.2、客户端

说明：生成MD5值->计算接收的数据的MD5值->生成接收数据的MD5值16进制的形式->发送接收MD5值确认信息->接收客户端的MD5值->客户端和服务端的MD5值做比较

①代码

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

# __auther__ == zhangqigao import   socket,hashlib   client  =  socket.socket() client.connect(( "localhost" , 9999 )) while  True :      cmd  =  input ( ">>>:" ).strip()      if  len (cmd)  = =  0 : continue      if  cmd.startswith( "get" ):          client.send(cmd.encode())          server_respose  =  client.recv( 1024 )          print ( "server response:" ,server_respose)          client.send( "ready to recv file" .encode())          file_total_size  =  int (server_respose.decode())          revived_size  =  0          filename  =  cmd.split()[ 1 ]          m  =  hashlib.md5()   #生成MD5对象          with  open (filename  +  ".new" , "wb" ) as f:              while  revived_size < file_total_size:                  data  =  client.recv( 1024 )                  revived_size  + =  len (data)                  m.update(data)    #计算数据接收的MD5值                  f.write(data)              else :                  print (file_total_size,revived_size)                  client_md5_vaule  =  m.hexdigest()   #生成接收数据的MD5值16进制形式                  client.send( "ready to recv file md5 value" .encode())                  server_md5_value  =  client.recv( 1024 )   #接收客户端的MD5值                  if  client_md5_vaule  = =  server_md5_value.decode():   #客户端和服务端的MD5值做比较                      print ( "file recv done" )                  else :                      print (client_md5_vaule,server_md5_value.decode()) client.close()

 ②改动地方

总结：

1. 上传和下载都是以客户端或者服务端加载文件，然后另外一边接收再写入文件。
2. 解决粘包问题，在接收数据大小后需要等待确认信息。
3. 看文件能都上传和下载，应该用MD5值去校验。
4. 创建一个无限大的文件，请用：dd if=/dev/sda1 of=文件名，比如创建test.txt，则：dd if=/dev/sda1 of=test.txt
5. 获取一个文件的大小：os.stat(文件名).st_size

转载于:https://www.cnblogs.com/luoahong/p/9896130.html